Support overlay modes for "codeql database init"

Author: cklin

lib/codeql.js

@@ -1,3 +1,4 @@
+import assert from "assert";
 import * as fs from "fs";
 import * as path from "path";
 
@@ -585,6 +586,15 @@ export async function getCodeQLForCmd(
         extraArgs.push(`--qlconfig-file=${qlconfigFile}`);
       }
 
+      const overlayDatabaseEnv = process.env["CODEQL_OVERLAY_DATABASE_INIT"];
+      if (overlayDatabaseEnv === "overlay") {
+        assert(config.buildMode === BuildMode.None);
+        extraArgs.push("--overlay");
+      } else if (overlayDatabaseEnv === "overlay-base") {
+        assert(config.buildMode === BuildMode.None);
+        extraArgs.push("--overlay-base");
+      }
+
       const overwriteFlag = isSupportedToolsFeature(
         await this.getVersion(),
         ToolsFeature.ForceOverwrite,

lib/codeql.js.map

@@ -163,6 +163,11 @@ export function cleanupDatabaseClusterDirectory(
   // for testing.
   rmSync = fs.rmSync,
 ): void {
+  const overlayDatabaseEnv = process.env["CODEQL_OVERLAY_DATABASE_INIT"];
+  if (overlayDatabaseEnv === "overlay") {
+    return;
+  }
+
   if (
     fs.existsSync(config.dbLocation) &&
     (fs.statSync(config.dbLocation).isFile() ||