Limit OverlayAnalysis to internal repos

Author: cklin

src/config-utils.test.ts

@@ -815,6 +815,7 @@ const calculateAugmentationMacro = test.macro({
     const actualAugmentationProperties =
       await configUtils.calculateAugmentation(
         getCachedCodeQL(),
+        { owner: "github", repo: "repo" },
         createFeatures([]),
         rawPacksInput,
         rawQueriesInput,
@@ -946,6 +947,7 @@ const calculateAugmentationErrorMacro = test.macro({
       () =>
         configUtils.calculateAugmentation(
           getCachedCodeQL(),
+          { owner: "github", repo: "repo" },
           createFeatures([]),
           rawPacksInput,
           rawQueriesInput,

src/config-utils.ts

@@ -507,6 +507,7 @@ export async function getDefaultConfig({
 
   const augmentationProperties = await calculateAugmentation(
     codeql,
+    repository,
     features,
     packsInput,
     queriesInput,
@@ -621,6 +622,7 @@ async function loadConfig({
 
   const augmentationProperties = await calculateAugmentation(
     codeql,
+    repository,
     features,
     packsInput,
     queriesInput,
@@ -664,6 +666,7 @@ async function loadConfig({
  * the config file sent to the CLI.
  *
  * @param codeql The CodeQL object.
+ * @param repository The repository to analyze.
  * @param features The feature enablement object.
  * @param rawPacksInput The packs input from the action configuration.
  * @param rawQueriesInput The queries input from the action configuration.
@@ -682,6 +685,7 @@ async function loadConfig({
 // exported for testing.
 export async function calculateAugmentation(
   codeql: CodeQL,
+  repository: RepositoryNwo,
   features: FeatureEnablement,
   rawPacksInput: string | undefined,
   rawQueriesInput: string | undefined,
@@ -705,6 +709,7 @@ export async function calculateAugmentation(
   const { overlayDatabaseMode, useOverlayDatabaseCaching } =
     await getOverlayDatabaseMode(
       codeql,
+      repository,
       features,
       sourceRoot,
       buildMode,
@@ -785,6 +790,7 @@ function parseQueriesFromInput(
  */
 async function getOverlayDatabaseMode(
   codeql: CodeQL,
+  repository: RepositoryNwo,
   features: FeatureEnablement,
   sourceRoot: string,
   buildMode: BuildMode | undefined,
@@ -809,7 +815,10 @@ async function getOverlayDatabaseMode(
       `Setting overlay database mode to ${overlayDatabaseMode} ` +
         "from the CODEQL_OVERLAY_DATABASE_MODE environment variable.",
     );
-  } else if (await features.getValue(Feature.OverlayAnalysis, codeql)) {
+  } else if (
+    repository.owner in ["github", "dsp-testing"] &&
+    (await features.getValue(Feature.OverlayAnalysis, codeql))
+  ) {
     if (isAnalyzingPullRequest()) {
       overlayDatabaseMode = OverlayDatabaseMode.Overlay;
       useOverlayDatabaseCaching = true;