1
This commit is contained in:
414
scripts/smoke-same-origin-stack.ts
Normal file
414
scripts/smoke-same-origin-stack.ts
Normal file
@@ -0,0 +1,414 @@
|
||||
import assert from 'node:assert/strict';
|
||||
import { spawn, type ChildProcess } from 'node:child_process';
|
||||
import fs from 'node:fs';
|
||||
import http from 'node:http';
|
||||
import path from 'node:path';
|
||||
import { fileURLToPath } from 'node:url';
|
||||
|
||||
import { httpRequest } from '../server-node/src/testHttp.ts';
|
||||
|
||||
const scriptPath = fileURLToPath(import.meta.url);
|
||||
const repoRoot = path.resolve(path.dirname(scriptPath), '..');
|
||||
const bundledNodePath = path.join(
|
||||
repoRoot,
|
||||
'.tools',
|
||||
'node-v22.22.2-win-x64',
|
||||
process.platform === 'win32' ? 'node.exe' : 'bin/node',
|
||||
);
|
||||
const runtimeNodePath = fs.existsSync(bundledNodePath)
|
||||
? bundledNodePath
|
||||
: process.execPath;
|
||||
const serverBuildPath = path.join(repoRoot, 'server-node', 'dist', 'server.js');
|
||||
const webBuildPath = path.join(repoRoot, 'dist', 'index.html');
|
||||
const proxyPort = 18080;
|
||||
const nodePort = 18081;
|
||||
const proxyBaseUrl = `http://127.0.0.1:${proxyPort}`;
|
||||
const nodeBaseUrl = `http://127.0.0.1:${nodePort}`;
|
||||
|
||||
type ManagedChild = {
|
||||
name: string;
|
||||
process: ChildProcess;
|
||||
};
|
||||
|
||||
function assertBuildArtifacts() {
|
||||
if (!fs.existsSync(serverBuildPath)) {
|
||||
throw new Error(
|
||||
'server-node/dist/server.js 不存在，请先运行 npm run server-node:build',
|
||||
);
|
||||
}
|
||||
|
||||
if (!fs.existsSync(webBuildPath)) {
|
||||
throw new Error('dist/index.html 不存在，请先运行 npm run build');
|
||||
}
|
||||
}
|
||||
|
||||
function sleep(ms: number) {
|
||||
return new Promise((resolve) => setTimeout(resolve, ms));
|
||||
}
|
||||
|
||||
async function waitForReady(
|
||||
label: string,
|
||||
url: string,
|
||||
validate: (bodyText: string, status: number) => void,
|
||||
timeoutMs = 20000,
|
||||
) {
|
||||
const startedAt = Date.now();
|
||||
let lastError: unknown = null;
|
||||
|
||||
while (Date.now() - startedAt < timeoutMs) {
|
||||
try {
|
||||
const response = await httpRequest(url);
|
||||
const bodyText = await response.text();
|
||||
validate(bodyText, response.status);
|
||||
return;
|
||||
} catch (error) {
|
||||
lastError = error;
|
||||
await sleep(250);
|
||||
}
|
||||
}
|
||||
|
||||
throw new Error(
|
||||
`[smoke:proxy] ${label} 未在 ${timeoutMs}ms 内就绪: ${lastError instanceof Error ? lastError.message : String(lastError)}`,
|
||||
);
|
||||
}
|
||||
|
||||
function spawnManagedChild(
|
||||
name: string,
|
||||
command: string,
|
||||
args: string[],
|
||||
env: NodeJS.ProcessEnv,
|
||||
): ManagedChild {
|
||||
const child = spawn(command, args, {
|
||||
cwd: repoRoot,
|
||||
env,
|
||||
stdio: 'inherit',
|
||||
shell: false,
|
||||
});
|
||||
|
||||
child.on('error', (error) => {
|
||||
console.error(`[smoke:proxy] ${name} 启动失败`, error);
|
||||
});
|
||||
|
||||
return {
|
||||
name,
|
||||
process: child,
|
||||
};
|
||||
}
|
||||
|
||||
async function stopChild(child: ManagedChild | null) {
|
||||
if (!child || child.process.exitCode !== null) {
|
||||
return;
|
||||
}
|
||||
|
||||
child.process.kill('SIGTERM');
|
||||
|
||||
await Promise.race([
|
||||
new Promise<void>((resolve) => {
|
||||
child.process.once('exit', () => resolve());
|
||||
}),
|
||||
sleep(2000),
|
||||
]);
|
||||
|
||||
if (child.process.exitCode === null) {
|
||||
child.process.kill('SIGKILL');
|
||||
await new Promise<void>((resolve) => {
|
||||
child.process.once('exit', () => resolve());
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
function contentTypeFor(filePath: string) {
|
||||
if (filePath.endsWith('.html')) {
|
||||
return 'text/html; charset=utf-8';
|
||||
}
|
||||
if (filePath.endsWith('.js')) {
|
||||
return 'text/javascript; charset=utf-8';
|
||||
}
|
||||
if (filePath.endsWith('.css')) {
|
||||
return 'text/css; charset=utf-8';
|
||||
}
|
||||
if (filePath.endsWith('.json')) {
|
||||
return 'application/json; charset=utf-8';
|
||||
}
|
||||
|
||||
return 'application/octet-stream';
|
||||
}
|
||||
|
||||
function resolveStaticFile(urlPath: string) {
|
||||
const cleanPath = decodeURIComponent(urlPath.split('?')[0] || '/');
|
||||
const normalizedPath = cleanPath === '/' ? '/index.html' : cleanPath;
|
||||
const trimmedRelativePath = normalizedPath.replace(/^\/+/u, '');
|
||||
const candidatePath = path.resolve(repoRoot, 'dist', trimmedRelativePath);
|
||||
const distRoot = path.resolve(repoRoot, 'dist');
|
||||
|
||||
if (
|
||||
candidatePath.startsWith(distRoot) &&
|
||||
fs.existsSync(candidatePath) &&
|
||||
fs.statSync(candidatePath).isFile()
|
||||
) {
|
||||
return candidatePath;
|
||||
}
|
||||
|
||||
return webBuildPath;
|
||||
}
|
||||
|
||||
async function startSameOriginProxy() {
|
||||
const server = http.createServer((request, response) => {
|
||||
const requestUrl = request.url || '/';
|
||||
|
||||
if (requestUrl === '/healthz') {
|
||||
response.writeHead(200, {
|
||||
'Content-Type': 'text/plain; charset=utf-8',
|
||||
});
|
||||
response.end('ok');
|
||||
return;
|
||||
}
|
||||
|
||||
if (requestUrl.startsWith('/api/')) {
|
||||
const upstream = http.request(
|
||||
{
|
||||
hostname: '127.0.0.1',
|
||||
port: nodePort,
|
||||
path: requestUrl,
|
||||
method: request.method,
|
||||
headers: {
|
||||
...request.headers,
|
||||
host: `127.0.0.1:${nodePort}`,
|
||||
},
|
||||
},
|
||||
(upstreamResponse) => {
|
||||
response.writeHead(
|
||||
upstreamResponse.statusCode ?? 502,
|
||||
upstreamResponse.headers,
|
||||
);
|
||||
upstreamResponse.pipe(response);
|
||||
},
|
||||
);
|
||||
|
||||
upstream.on('error', (error) => {
|
||||
response.writeHead(502, {
|
||||
'Content-Type': 'application/json; charset=utf-8',
|
||||
});
|
||||
response.end(
|
||||
JSON.stringify({
|
||||
error: {
|
||||
message:
|
||||
error instanceof Error ? error.message : 'proxy upstream failed',
|
||||
},
|
||||
}),
|
||||
);
|
||||
});
|
||||
|
||||
request.pipe(upstream);
|
||||
return;
|
||||
}
|
||||
|
||||
const filePath = resolveStaticFile(requestUrl);
|
||||
response.writeHead(200, {
|
||||
'Content-Type': contentTypeFor(filePath),
|
||||
});
|
||||
fs.createReadStream(filePath).pipe(response);
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve, reject) => {
|
||||
server.once('error', reject);
|
||||
server.listen(proxyPort, '127.0.0.1', () => resolve());
|
||||
});
|
||||
|
||||
return server;
|
||||
}
|
||||
|
||||
async function stopProxyServer(server: http.Server | null) {
|
||||
if (!server) {
|
||||
return;
|
||||
}
|
||||
|
||||
await new Promise<void>((resolve, reject) => {
|
||||
server.close((error) => {
|
||||
if (error) {
|
||||
reject(error);
|
||||
return;
|
||||
}
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
async function authEntry(baseUrl: string) {
|
||||
const requestId = 'proxy-smoke-auth-entry';
|
||||
const username = `proxy_${Date.now().toString(36)}`;
|
||||
const response = await httpRequest(`${baseUrl}/api/auth/entry`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'X-Request-Id': requestId,
|
||||
},
|
||||
body: JSON.stringify({
|
||||
username,
|
||||
password: 'proxy-secret-123',
|
||||
}),
|
||||
});
|
||||
const payload = (await response.json()) as {
|
||||
token: string;
|
||||
user: {
|
||||
id: string;
|
||||
username: string;
|
||||
};
|
||||
};
|
||||
|
||||
assert.equal(response.status, 200);
|
||||
assert.equal(response.headers.get('x-request-id'), requestId);
|
||||
assert.equal(payload.user.username, username);
|
||||
assert.ok(payload.token);
|
||||
|
||||
return payload;
|
||||
}
|
||||
|
||||
async function main() {
|
||||
assertBuildArtifacts();
|
||||
|
||||
let serverChild: ManagedChild | null = null;
|
||||
let proxyServer: http.Server | null = null;
|
||||
|
||||
try {
|
||||
console.log('[smoke:proxy] starting built node server');
|
||||
serverChild = spawnManagedChild(
|
||||
'server-node',
|
||||
runtimeNodePath,
|
||||
[serverBuildPath],
|
||||
{
|
||||
...process.env,
|
||||
PROJECT_ROOT: repoRoot,
|
||||
NODE_ENV: 'test',
|
||||
NODE_SERVER_ADDR: `:${nodePort}`,
|
||||
DATABASE_URL: 'pg-mem://genarrative-proxy-smoke',
|
||||
LOG_LEVEL: 'silent',
|
||||
JWT_SECRET: 'proxy-smoke-secret',
|
||||
JWT_ISSUER: 'genarrative-proxy-smoke',
|
||||
LLM_API_KEY: '',
|
||||
DASHSCOPE_API_KEY: '',
|
||||
},
|
||||
);
|
||||
|
||||
await waitForReady(
|
||||
'node server',
|
||||
`${nodeBaseUrl}/healthz`,
|
||||
(bodyText, status) => {
|
||||
assert.equal(status, 200);
|
||||
const payload = JSON.parse(bodyText) as {
|
||||
ok: boolean;
|
||||
service: string;
|
||||
};
|
||||
assert.equal(payload.ok, true);
|
||||
assert.equal(payload.service, 'genarrative-node-server');
|
||||
},
|
||||
);
|
||||
console.log('[smoke:proxy] node server ready');
|
||||
|
||||
console.log('[smoke:proxy] starting same-origin reverse proxy harness');
|
||||
proxyServer = await startSameOriginProxy();
|
||||
|
||||
await waitForReady(
|
||||
'reverse proxy',
|
||||
`${proxyBaseUrl}/healthz`,
|
||||
(bodyText, status) => {
|
||||
assert.equal(status, 200);
|
||||
assert.equal(bodyText.trim(), 'ok');
|
||||
},
|
||||
);
|
||||
console.log('[smoke:proxy] reverse proxy ready');
|
||||
|
||||
const homeResponse = await httpRequest(`${proxyBaseUrl}/`);
|
||||
const homeHtml = await homeResponse.text();
|
||||
assert.equal(homeResponse.status, 200);
|
||||
assert.match(homeHtml, /<div id="root"><\/div>/u);
|
||||
console.log('[smoke:proxy] static web entry ok');
|
||||
|
||||
const entry = await authEntry(proxyBaseUrl);
|
||||
console.log('[smoke:proxy] proxied auth entry ok');
|
||||
|
||||
const meResponse = await httpRequest(`${proxyBaseUrl}/api/auth/me`, {
|
||||
headers: {
|
||||
Authorization: `Bearer ${entry.token}`,
|
||||
},
|
||||
});
|
||||
const mePayload = (await meResponse.json()) as {
|
||||
user: {
|
||||
username: string;
|
||||
};
|
||||
};
|
||||
assert.equal(meResponse.status, 200);
|
||||
assert.equal(mePayload.user.username, entry.user.username);
|
||||
console.log('[smoke:proxy] proxied auth me ok');
|
||||
|
||||
const saveResponse = await httpRequest(
|
||||
`${proxyBaseUrl}/api/runtime/save/snapshot`,
|
||||
{
|
||||
method: 'PUT',
|
||||
headers: {
|
||||
Authorization: `Bearer ${entry.token}`,
|
||||
'Content-Type': 'application/json',
|
||||
'X-Genarrative-Response-Envelope': 'v1',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
gameState: {
|
||||
worldType: 'WUXIA',
|
||||
chapter: 2,
|
||||
},
|
||||
bottomTab: 'adventure',
|
||||
currentStory: {
|
||||
text: 'proxy smoke story',
|
||||
},
|
||||
}),
|
||||
},
|
||||
);
|
||||
const savePayload = (await saveResponse.json()) as {
|
||||
ok: true;
|
||||
data: {
|
||||
gameState: {
|
||||
chapter: number;
|
||||
};
|
||||
};
|
||||
meta: {
|
||||
requestId: string;
|
||||
operation: string;
|
||||
};
|
||||
};
|
||||
assert.equal(saveResponse.status, 200);
|
||||
assert.equal(savePayload.ok, true);
|
||||
assert.equal(savePayload.data.gameState.chapter, 2);
|
||||
assert.equal(savePayload.meta.operation, 'PUT /api/runtime/save/snapshot');
|
||||
assert.ok(savePayload.meta.requestId);
|
||||
console.log('[smoke:proxy] proxied runtime save ok');
|
||||
|
||||
const getResponse = await httpRequest(
|
||||
`${proxyBaseUrl}/api/runtime/save/snapshot`,
|
||||
{
|
||||
headers: {
|
||||
Authorization: `Bearer ${entry.token}`,
|
||||
},
|
||||
},
|
||||
);
|
||||
const getPayload = (await getResponse.json()) as {
|
||||
gameState: {
|
||||
chapter: number;
|
||||
};
|
||||
bottomTab: string;
|
||||
};
|
||||
assert.equal(getResponse.status, 200);
|
||||
assert.equal(getPayload.gameState.chapter, 2);
|
||||
assert.equal(getPayload.bottomTab, 'adventure');
|
||||
console.log('[smoke:proxy] proxied runtime snapshot read ok');
|
||||
|
||||
console.log('[smoke:proxy] all checks passed');
|
||||
} finally {
|
||||
await stopProxyServer(proxyServer);
|
||||
await stopChild(serverChild);
|
||||
}
|
||||
}
|
||||
|
||||
void main().catch((error) => {
|
||||
console.error('[smoke:proxy] failed');
|
||||
console.error(error);
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user