Centralize fork logging

packages/server/src/vscode/bootstrapFork.ts

@@ -1,6 +1,7 @@
 import * as cp from "child_process";
 import * as fs from "fs";
 import * as path from "path";
+import { logger, field } from "@coder/logger/src";
 
 declare var __non_webpack_require__: typeof require;
 
@@ -14,16 +15,55 @@ export const requireModule = (modulePath: string): void => {
 /**
  * Uses the internal bootstrap-fork.js to load a module
  * @example
- * const cp = forkModule("vs/code/electron-browser/sharedProcess/sharedProcessMain");
+ * 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
+ * @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): cp.ChildProcess => {
+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];
 	if (process.env.CLI === "true") {
-		return cp.spawn(process.execPath, args);
+		proc = stdio ? cp.spawn(process.execPath, args) : cp.fork(process.execPath, args);
+	} else if (stdio) {
+		proc = cp.spawn("npm", ["start", "--scripts-prepend-node-path", "--", ...args]);
 	} else {
-		return cp.spawn("npm", ["start", "--scripts-prepend-node-path", "--", ...args]);
+		// TODO: need to fork somehow so we get send/onMessage.
+		proc = cp.spawn("npm", ["start", "--scripts-prepend-node-path", "--", ...args]);
 	}
+
+	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;
 };