valknarandClaude Sonnet 5
9dc45f751a
Our upgrade handler on the shared httpServer was destroying every socket that wasn't for /ws/runs, which silently killed Next's own dev-mode HMR websocket (/_next/hmr) too - breaking hot reload with no useful error, just a failed WebSocket connection in the browser console. Delegate to app.getUpgradeHandler() instead, which must be called after prepare(). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
74 lines
2.3 KiB
TypeScript
74 lines
2.3 KiB
TypeScript
import "./src/bootstrap/async-local-storage-polyfill";
|
|
import { createServer } from "node:http";
|
|
import next from "next";
|
|
import { WebSocketServer } from "ws";
|
|
import { getConfig } from "./src/lib/config/load";
|
|
import { migrateOnBoot } from "./src/lib/db/client";
|
|
import { syncAuthFromConfig } from "./src/lib/auth/sync";
|
|
import { reconcileOrphanedRuns } from "./src/lib/runner/engine";
|
|
import { killAllRuns } from "./src/lib/runner/registry";
|
|
import { attachWsServer, authenticateUpgrade } from "./src/lib/ws/server";
|
|
|
|
const dev = process.env.NODE_ENV !== "production";
|
|
const { config } = getConfig();
|
|
|
|
const port = Number(process.env.PORT ?? config.server.port);
|
|
const hostname = process.env.HOST ?? config.server.host;
|
|
|
|
migrateOnBoot();
|
|
syncAuthFromConfig();
|
|
reconcileOrphanedRuns();
|
|
|
|
const app = next({ dev, hostname, port });
|
|
const handle = app.getRequestHandler();
|
|
|
|
app.prepare().then(() => {
|
|
const nextUpgradeHandler = app.getUpgradeHandler();
|
|
|
|
const httpServer = createServer((req, res) => {
|
|
handle(req, res);
|
|
});
|
|
|
|
const wss = new WebSocketServer({ noServer: true });
|
|
attachWsServer(wss);
|
|
|
|
httpServer.on("upgrade", (req, socket, head) => {
|
|
const { pathname } = new URL(req.url ?? "/", "http://internal");
|
|
if (pathname !== "/ws/runs") {
|
|
// Anything else (e.g. Next's own dev-mode HMR websocket at /_next/hmr) is Next's to handle.
|
|
nextUpgradeHandler(req, socket, head).catch(() => socket.destroy());
|
|
return;
|
|
}
|
|
|
|
authenticateUpgrade(req)
|
|
.then((ok) => {
|
|
if (!ok) {
|
|
socket.write("HTTP/1.1 401 Unauthorized\r\n\r\n");
|
|
socket.destroy();
|
|
return;
|
|
}
|
|
wss.handleUpgrade(req, socket, head, (ws) => {
|
|
wss.emit("connection", ws, req);
|
|
});
|
|
})
|
|
.catch(() => socket.destroy());
|
|
});
|
|
|
|
httpServer.listen(port, hostname, () => {
|
|
console.log(
|
|
`> triggershell ready on http://${hostname}:${port} (${dev ? "development" : "production"})`,
|
|
);
|
|
});
|
|
|
|
const shutdown = (signal: string) => {
|
|
console.log(`> received ${signal}, shutting down...`);
|
|
killAllRuns();
|
|
httpServer.close(() => process.exit(0));
|
|
// Force-exit if graceful shutdown hangs (e.g. a stuck WS connection).
|
|
setTimeout(() => process.exit(1), 5000).unref();
|
|
};
|
|
|
|
process.on("SIGTERM", () => shutdown("SIGTERM"));
|
|
process.on("SIGINT", () => shutdown("SIGINT"));
|
|
});
|