code-server-2/ci/dev/watch.ts

import { spawn, fork, ChildProcess } from "child_process"
import del from "del"
import { promises as fs } from "fs"
import * as path from "path"
import { CompilationStats, onLine, OnLineCallback, VSCodeCompileStatus } from "../../src/node/util"

interface DevelopmentCompilers {
  [key: string]: ChildProcess | undefined
  vscode: ChildProcess
  vscodeWebExtensions: ChildProcess
  codeServer: ChildProcess
  plugins: ChildProcess | undefined
}

class Watcher {
  private rootPath = path.resolve(process.cwd())
  private readonly paths = {
    /** Path to uncompiled VS Code source. */
    vscodeDir: path.join(this.rootPath, "vendor", "modules", "code-oss-dev"),
    compilationStatsFile: path.join(this.rootPath, "out", "watcher.json"),
    pluginDir: process.env.PLUGIN_DIR,
  }

  //#region Web Server

  /** Development web server. */
  private webServer: ChildProcess | undefined

  private reloadWebServer = (): void => {
    if (this.webServer) {
      this.webServer.kill()
    }

    // Pass CLI args, save for `node` and the initial script name.
    const args = process.argv.slice(2)
    this.webServer = fork(path.join(this.rootPath, "out/node/entry.js"), args)
    const { pid } = this.webServer

    this.webServer.on("exit", () => console.log("[Code Server]", `Web process ${pid} exited`))

    console.log("\n[Code Server]", `Spawned web server process ${pid}`)
  }

  //#endregion

  //#region Compilers

  private readonly compilers: DevelopmentCompilers = {
    codeServer: spawn("tsc", ["--watch", "--pretty", "--preserveWatchOutput"], { cwd: this.rootPath }),
    vscode: spawn("yarn", ["watch"], { cwd: this.paths.vscodeDir }),
    vscodeWebExtensions: spawn("yarn", ["watch-web"], { cwd: this.paths.vscodeDir }),
    plugins: this.paths.pluginDir ? spawn("yarn", ["build", "--watch"], { cwd: this.paths.pluginDir }) : undefined,
  }

  private vscodeCompileStatus = VSCodeCompileStatus.Loading

  public async initialize(): Promise<void> {
    for (const event of ["SIGINT", "SIGTERM"]) {
      process.on(event, () => this.dispose(0))
    }

    if (!this.hasVerboseLogging) {
      console.log("\n[Watcher]", "Compiler logs will be minimal. Pass --log to show all output.")
    }

    this.cleanFiles()

    for (const [processName, devProcess] of Object.entries(this.compilers)) {
      if (!devProcess) continue

      devProcess.on("exit", (code) => {
        this.log(`[${processName}]`, "Terminated unexpectedly")
        this.dispose(code)
      })

      if (devProcess.stderr) {
        devProcess.stderr.on("data", (d: string | Uint8Array) => process.stderr.write(d))
      }
    }

    onLine(this.compilers.vscode, this.parseVSCodeLine)
    onLine(this.compilers.codeServer, this.parseCodeServerLine)

    if (this.compilers.plugins) {
      onLine(this.compilers.plugins, this.parsePluginLine)
    }
  }

  //#endregion

  //#region Line Parsers

  private parseVSCodeLine: OnLineCallback = (strippedLine, originalLine) => {
    if (!strippedLine.includes("watch-extensions") || this.hasVerboseLogging) {
      console.log("[VS Code]", originalLine)
    }

    switch (this.vscodeCompileStatus) {
      case VSCodeCompileStatus.Loading:
        // Wait for watch-client since "Finished compilation" will appear multiple
        // times before the client starts building.
        if (strippedLine.includes("Starting 'watch-client'")) {
          console.log("[VS Code] 🚧 Compiling 🚧", "(This may take a moment!)")
          this.vscodeCompileStatus = VSCodeCompileStatus.Compiling
        }
        break
      case VSCodeCompileStatus.Compiling:
        if (strippedLine.includes("Finished compilation")) {
          console.log("[VS Code] ✨ Finished compiling! ✨", "(Refresh your web browser ♻️)")
          this.vscodeCompileStatus = VSCodeCompileStatus.Compiled

          this.emitCompilationStats()
          this.reloadWebServer()
        }
        break
      case VSCodeCompileStatus.Compiled:
        console.log("[VS Code] 🔔 Finished recompiling! 🔔", "(Refresh your web browser ♻️)")
        this.emitCompilationStats()
        this.reloadWebServer()
        break
    }
  }

  private parseCodeServerLine: OnLineCallback = (strippedLine, originalLine) => {
    if (!strippedLine.length) return

    console.log("[Compiler][Code Server]", originalLine)

    if (strippedLine.includes("Watching for file changes")) {
      console.log("[Compiler][Code Server]", "Finished compiling!", "(Refresh your web browser ♻️)")

      this.reloadWebServer()
    }
  }

  private parsePluginLine: OnLineCallback = (strippedLine, originalLine) => {
    if (!strippedLine.length) return

    console.log("[Compiler][Plugin]", originalLine)

    if (strippedLine.includes("Watching for file changes...")) {
      this.reloadWebServer()
    }
  }

  //#endregion

  //#region Utilities

  /**
   * Cleans files from previous builds.
   */
  private cleanFiles(): Promise<string[]> {
    console.log("[Watcher]", "Cleaning files from previous builds...")

    return del([
      "out/**/*",
      // Included because the cache can sometimes enter bad state when debugging compiled files.
      ".cache/**/*",
    ])
  }

  /**
   * Emits a file containing compilation data.
   * This is especially useful when Express needs to determine if VS Code is still compiling.
   */
  private emitCompilationStats(): Promise<void> {
    const stats: CompilationStats = {
      status: this.vscodeCompileStatus,
      lastCompiledAt: new Date(),
    }

    this.log("Writing watcher stats...")
    return fs.writeFile(this.paths.compilationStatsFile, JSON.stringify(stats, null, 2))
  }

  private log(...entries: string[]) {
    process.stdout.write(entries.join(" "))
  }

  private dispose(code: number | null): void {
    for (const [processName, devProcess] of Object.entries(this.compilers)) {
      this.log(`[${processName}]`, "Killing...\n")
      devProcess?.removeAllListeners()
      devProcess?.kill()
    }
    process.exit(typeof code === "number" ? code : 0)
  }

  private get hasVerboseLogging() {
    return process.argv.includes("--log")
  }

  //#endregion
}

async function main(): Promise<void> {
  try {
    const watcher = new Watcher()
    await watcher.initialize()
  } catch (error: any) {
    console.error(error.message)
    process.exit(1)
  }
}

main()