91 lines
2.1 KiB
JavaScript
91 lines
2.1 KiB
JavaScript
const { spawn } = require("child_process");
|
|
const net = require("net");
|
|
const path = require("path");
|
|
|
|
function waitForPort(host, port, timeoutMs = 30_000) {
|
|
const start = Date.now();
|
|
return new Promise((resolve, reject) => {
|
|
const tick = () => {
|
|
const socket = net.createConnection({ host, port });
|
|
const timer = setTimeout(() => {
|
|
socket.destroy();
|
|
if (Date.now() - start > timeoutMs) {
|
|
reject(new Error(`等待端口超时：${host}:${port}`));
|
|
return;
|
|
}
|
|
setTimeout(tick, 200);
|
|
}, 500);
|
|
|
|
socket.once("connect", () => {
|
|
clearTimeout(timer);
|
|
socket.end();
|
|
resolve();
|
|
});
|
|
socket.once("error", () => {
|
|
clearTimeout(timer);
|
|
if (Date.now() - start > timeoutMs) {
|
|
reject(new Error(`等待端口超时：${host}:${port}`));
|
|
return;
|
|
}
|
|
setTimeout(tick, 200);
|
|
});
|
|
};
|
|
tick();
|
|
});
|
|
}
|
|
|
|
function spawnTask(name, command, cwd, extraEnv) {
|
|
console.log(`[${name}] ${command}`);
|
|
const child = spawn(command, {
|
|
cwd,
|
|
stdio: "inherit",
|
|
shell: true,
|
|
env: { ...process.env, ...extraEnv },
|
|
});
|
|
return child;
|
|
}
|
|
|
|
async function main() {
|
|
const rootDir = process.cwd();
|
|
const frontendDir = path.join(rootDir, "wolai-frontend");
|
|
|
|
const frontendPort = Number(process.env.FRONTEND_PORT || 3000);
|
|
|
|
const frontend = spawnTask(
|
|
"frontend",
|
|
`pnpm -C wolai-frontend dev -p ${frontendPort}`,
|
|
rootDir,
|
|
{},
|
|
);
|
|
|
|
await waitForPort("127.0.0.1", frontendPort);
|
|
|
|
const electron = spawnTask(
|
|
"electron",
|
|
"electron desktop-electron/main.js",
|
|
rootDir,
|
|
{ MNOTE_LOAD_URL: `http://127.0.0.1:${frontendPort}` },
|
|
);
|
|
|
|
const shutdown = () => {
|
|
try {
|
|
if (!electron.killed) electron.kill("SIGINT");
|
|
} catch {}
|
|
try {
|
|
if (!frontend.killed) frontend.kill("SIGINT");
|
|
} catch {}
|
|
};
|
|
|
|
process.on("SIGINT", shutdown);
|
|
process.on("SIGTERM", shutdown);
|
|
|
|
electron.on("exit", () => shutdown());
|
|
frontend.on("exit", () => shutdown());
|
|
}
|
|
|
|
main().catch((err) => {
|
|
console.error(err instanceof Error ? err.stack : String(err));
|
|
process.exit(1);
|
|
});
|
|
|