import { field, logger } from "@coder/logger"; import { ReadWriteConnection } from "@coder/protocol"; import { Server, ServerOptions } from "@coder/protocol/src/node/server"; import { NewSessionMessage } from '@coder/protocol/src/proto'; import { ChildProcess } from "child_process"; import * as express from "express"; import * as http from "http"; import * as ws from "ws"; import { forkModule } from "./vscode/bootstrapFork"; export const createApp = (registerMiddleware?: (app: express.Application) => void, options?: ServerOptions): { readonly express: express.Application; readonly server: http.Server; readonly wss: ws.Server; } => { const app = express(); if (registerMiddleware) { registerMiddleware(app); } const server = http.createServer(app); const wss = new ws.Server({ server }); wss.shouldHandle = (req): boolean => { // Should handle auth here return true; }; wss.on("connection", (ws) => { const connection: ReadWriteConnection = { onMessage: (cb): void => { ws.addEventListener("message", (event) => cb(event.data)); }, close: (): void => ws.close(), send: (data): void => { try { ws.send(data); } catch (error) { logger.error(error.message); } }, onClose: (cb): void => ws.addEventListener("close", () => cb()), }; const server = new Server(connection, options ? { ...options, forkProvider: (message: NewSessionMessage): ChildProcess => { const command = message.getCommand(); const childLogger = logger.named(command.split("/").pop()!); childLogger.debug("Forking...", field("module", command)); let proc: ChildProcess; if (message.getIsBootstrapFork()) { proc = forkModule(command); } else { throw new Error("No support for non bootstrap-forking yet"); } proc.stdout.on("data", (message) => { childLogger.debug("stdout", field("message", message.toString().trim())); }); proc.stderr.on("data", (message) => { childLogger.debug("stderr", field("message", message.toString().trim())); }); proc.stdin.on("data", (message) => { childLogger.debug("stdin", field("message", message.toString().trim())); }); proc.on("exit", (exitCode) => { childLogger.debug(`Exited with ${exitCode}`); }); return proc; }, } : undefined); }); /** * We should static-serve the `web` package at this point. */ app.get("/", (req, res, next) => { res.write("Example! :)"); res.status(200); res.end(); }); return { express: app, server, wss, }; };