code-server-2/packages/server/src/vscode/bootstrapFork.ts

import * as cp from "child_process";
import * as fs from "fs";
import * as net from "net";
import * as path from "path";
import { logger, field } from "@coder/logger/src";

declare var __non_webpack_require__: typeof require;

export const requireModule = (modulePath: string): void => {
	process.env.AMD_ENTRYPOINT = modulePath;
	process.env.VSCODE_ALLOW_IO = "true";

	if (!process.send) {
		const socket = new net.Socket({ fd: 3 });
		socket.on("data", (data) => {
			process.emit("message", JSON.parse(data.toString()), undefined);
		});

		process.send = (message: any): void => {
			socket.write(JSON.stringify(message));
		};
	}

	const content = fs.readFileSync(path.join(process.env.BUILD_DIR as string || path.join(__dirname, "../.."), "./build/bootstrap-fork.js"));
	eval(content.toString());
};

/**
 * Uses the internal bootstrap-fork.js to load a module
 * @example
 * const cp = forkModule("vs/code/electron-browser/sharedProcess/sharedProcessMain", true);
 * cp.stdout.on("data", (data) => console.log(data.toString("utf8")));
 * cp.stderr.on("data", (data) => console.log(data.toString("utf8")));
 * @param modulePath Path of the VS Code module to load.
 * @param stdio Whether to use stdio (spawn) or send/onMessage (fork).
 */
export const forkModule = (modulePath: string, stdio?: boolean): cp.ChildProcess => {
	const basename = modulePath.split("/").pop()!;
	let i = 0;
	for (; i < basename.length; i++) {
		const character = basename.charAt(i);
		if (character === character.toUpperCase()) {
			break;
		}
	}
	const childLogger = logger.named(basename.substring(0, i));
	childLogger.debug("Forking...", field("module", modulePath));

	let proc: cp.ChildProcess | undefined;

	const args = ["--bootstrap-fork", modulePath];
	const options: cp.SpawnOptions = {
		stdio: [null, null, null, "pipe"],
	};
	if (process.env.CLI === "true") {
		proc = stdio ? cp.spawn(process.execPath, args, options) : cp.fork(process.execPath, args, options);
	} else {
		proc = cp.spawn(process.execArgv[0], ["-r", "tsconfig-paths/register", process.argv[1], ...args], options);
	}

	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("message", (message) => {
		childLogger.debug("message", field("message", message.toString().trim()));
	});

	proc.on("exit", (exitCode) => {
		childLogger.debug(`Exited with ${exitCode}`);
	});

	return proc;
};