72bf4547d4
* Clean up workbench and integrate initialization data * Uncomment Electron fill * Run server & client together * Clean up Electron fill & patch * Bind fs methods This makes them usable with the promise form: `promisify(access)(...)`. * Add space between tag and title to browser logger * Add typescript dep to server and default __dirname for path * Serve web files from server * Adjust some dev options * Rework workbench a bit to use a class and catch unexpected errors * No mkdirs for now, fix util fill, use bash with exec * More fills, make general client abstract * More fills * Fix cp.exec * Fix require calls in fs fill being aliased * Create data and storage dir * Implement fs.watch Using exec for now. * Implement storage database fill * Fix os export and homedir * Add comment to use navigator.sendBeacon * Fix fs callbacks (some args are optional) * Make sure data directory exists when passing it back * Update patch * Target es5 * More fills * Add APIs required for bootstrap-fork to function (#15) * Add bootstrap-fork execution * Add createConnection * Bundle bootstrap-fork into cli * Remove .node directory created from spdlog * Fix npm start * Remove unnecessary comment * Add webpack-hot-middleware if CLI env is not set * Add restarting to shared process * Fix starting with yarn
80 lines
2.0 KiB
TypeScript
80 lines
2.0 KiB
TypeScript
import { ReadWriteConnection } from "@coder/protocol";
|
|
import { Server, ServerOptions } from "@coder/protocol/src/node/server";
|
|
import * as express from "express";
|
|
import * as http from "http";
|
|
import * as ws from "ws";
|
|
import * as url from "url";
|
|
import { ClientMessage, SharedProcessInitMessage } from '@coder/protocol/src/proto';
|
|
|
|
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 => {
|
|
if (typeof req.url === "undefined") {
|
|
return false;
|
|
}
|
|
|
|
const parsedUrl = url.parse(req.url, true);
|
|
const sharedProcessInit = parsedUrl.query["shared_process_init"];
|
|
if (typeof sharedProcessInit === "undefined" || Array.isArray(sharedProcessInit)) {
|
|
return false;
|
|
}
|
|
|
|
try {
|
|
const msg = ClientMessage.deserializeBinary(Buffer.from(sharedProcessInit, "base64"));
|
|
if (!msg.hasSharedProcessInit()) {
|
|
return false;
|
|
}
|
|
const spm = msg.getSharedProcessInit()!;
|
|
(<any>req).sharedProcessInit = spm;
|
|
} catch (ex) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
};
|
|
|
|
wss.on("connection", (ws: WebSocket, req) => {
|
|
const spm = (<any>req).sharedProcessInit as SharedProcessInitMessage;
|
|
if (!spm) {
|
|
ws.close();
|
|
return;
|
|
}
|
|
|
|
const connection: ReadWriteConnection = {
|
|
onMessage: (cb): void => {
|
|
ws.addEventListener("message", (event) => cb(event.data));
|
|
},
|
|
close: (): void => ws.close(),
|
|
send: (data): void => ws.send(data),
|
|
onClose: (cb): void => ws.addEventListener("close", () => cb()),
|
|
};
|
|
|
|
const server = new Server(connection, options);
|
|
});
|
|
|
|
/**
|
|
* 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,
|
|
};
|
|
};
|