Add linting steps and improve testing steps

src/node/entry.ts

@@ -49,7 +49,7 @@ const main = async (args: Args = {}): Promise<void> => {
     new VscodeHttpProvider([], { base: "/vscode-embed", auth, password })
   )
 
-  ipcMain.onDispose(() => httpServer.dispose())
+  ipcMain().onDispose(() => httpServer.dispose())
 
   const serverAddress = await httpServer.listen()
   logger.info(`Server listening on ${serverAddress}`)
@@ -84,4 +84,18 @@ const main = async (args: Args = {}): Promise<void> => {
   }
 }
 
-wrap(main)
+// TODO: Implement CLI parser.
+if (process.argv.includes("--version")) {
+  const version = require("../../package.json").version
+  if (process.argv.includes("--json")) {
+    console.log({
+      codeServer: version,
+      vscode: require("../../lib/vscode/package.json").version,
+    })
+  } else {
+    console.log(version)
+  }
+  process.exit(0)
+} else {
+  wrap(main)
+}

src/node/wrapper.ts

@@ -21,14 +21,6 @@ export class ProcessError extends Error {
   }
 }
 
-/**
- * Ensure we control when the process exits.
- */
-const exit = process.exit
-process.exit = function(code?: number) {
-  logger.warn(`process.exit() was prevented: ${code || "unknown code"}.`)
-} as (code?: number) => never
-
 /**
  * Allows the wrapper and inner processes to communicate.
  */
@@ -37,19 +29,26 @@ export class IpcMain {
   public readonly onMessage = this._onMessage.event
   private readonly _onDispose = new Emitter<NodeJS.Signals | undefined>()
   public readonly onDispose = this._onDispose.event
+  public readonly exit: (code?: number) => never
 
   public constructor(public readonly parentPid?: number) {
     process.on("SIGINT", () => this._onDispose.emit("SIGINT"))
     process.on("SIGTERM", () => this._onDispose.emit("SIGTERM"))
     process.on("exit", () => this._onDispose.emit(undefined))
 
+    // Ensure we control when the process exits.
+    this.exit = process.exit
+    process.exit = function(code?: number) {
+      logger.warn(`process.exit() was prevented: ${code || "unknown code"}.`)
+    } as (code?: number) => never
+
     this.onDispose((signal) => {
       // Remove listeners to avoid possibly triggering disposal again.
       process.removeAllListeners()
 
       // Let any other handlers run first then exit.
       logger.debug(`${parentPid ? "inner process" : "wrapper"} ${process.pid} disposing`, field("code", signal))
-      setTimeout(() => exit(0), 0)
+      setTimeout(() => this.exit(0), 0)
     })
 
     // Kill the inner process if the parent dies. This is for the case where the
@@ -117,9 +116,17 @@ export class IpcMain {
   }
 }
 
-export const ipcMain = new IpcMain(
-  typeof process.env.CODE_SERVER_PARENT_PID !== "undefined" ? parseInt(process.env.CODE_SERVER_PARENT_PID) : undefined
-)
+let _ipcMain: IpcMain
+export const ipcMain = (): IpcMain => {
+  if (!_ipcMain) {
+    _ipcMain = new IpcMain(
+      typeof process.env.CODE_SERVER_PARENT_PID !== "undefined"
+        ? parseInt(process.env.CODE_SERVER_PARENT_PID)
+        : undefined
+    )
+  }
+  return _ipcMain
+}
 
 export interface WrapperOptions {
   maxMemory?: number
@@ -135,14 +142,14 @@ export class WrapperProcess {
   private started?: Promise<void>
 
   public constructor(private currentVersion: string, private readonly options?: WrapperOptions) {
-    ipcMain.onDispose(() => {
+    ipcMain().onDispose(() => {
       if (this.process) {
         this.process.removeAllListeners()
         this.process.kill()
       }
     })
 
-    ipcMain.onMessage(async (message) => {
+    ipcMain().onMessage(async (message) => {
       switch (message.type) {
         case "relaunch":
           logger.info(`Relaunching: ${this.currentVersion} -> ${message.version}`)
@@ -156,7 +163,7 @@ export class WrapperProcess {
             await this.start()
           } catch (error) {
             logger.error(error.message)
-            exit(typeof error.code === "number" ? error.code : 1)
+            ipcMain().exit(typeof error.code === "number" ? error.code : 1)
           }
           break
         default:
@@ -170,12 +177,14 @@ export class WrapperProcess {
     if (!this.started) {
       const child = this.spawn()
       logger.debug(`spawned inner process ${child.pid}`)
-      this.started = ipcMain.handshake(child).then(() => {
-        child.once("exit", (code) => {
-          logger.debug(`inner process ${child.pid} exited unexpectedly`)
-          exit(code || 0)
+      this.started = ipcMain()
+        .handshake(child)
+        .then(() => {
+          child.once("exit", (code) => {
+            logger.debug(`inner process ${child.pid} exited unexpectedly`)
+            ipcMain().exit(code || 0)
+          })
         })
-      })
       this.process = child
     }
     return this.started
@@ -201,23 +210,23 @@ export class WrapperProcess {
 // // It's possible that the pipe has closed (for example if you run code-server
 // // --version | head -1). Assume that means we're done.
 if (!process.stdout.isTTY) {
-  process.stdout.on("error", () => exit())
+  process.stdout.on("error", () => ipcMain().exit())
 }
 
 export const wrap = (fn: () => Promise<void>): void => {
-  if (ipcMain.parentPid) {
-    ipcMain
+  if (ipcMain().parentPid) {
+    ipcMain()
       .handshake()
       .then(() => fn())
       .catch((error: ProcessError): void => {
         logger.error(error.message)
-        exit(typeof error.code === "number" ? error.code : 1)
+        ipcMain().exit(typeof error.code === "number" ? error.code : 1)
       })
   } else {
     const wrapper = new WrapperProcess(require("../../package.json").version)
     wrapper.start().catch((error) => {
       logger.error(error.message)
-      exit(typeof error.code === "number" ? error.code : 1)
+      ipcMain().exit(typeof error.code === "number" ? error.code : 1)
     })
   }
 }