Support cli-version-x.y.z-pre.txt marker files

lib/codeql.test.js

@@ -172,24 +172,36 @@ function mockApiDetails(apiDetails) {
         t.deepEqual(result.toolsVersion, "0.0.0-20200610");
     });
 });
-(0, ava_1.default)("tries to cache an explicitly requested bundle with its CLI version number", async (t) => {
-    await util.withTmpDir(async (tmpDir) => {
-        (0, testing_utils_1.setupActionsVars)(tmpDir, tmpDir);
-        mockApiDetails(sampleApiDetails);
-        sinon.stub(actionsUtil, "isRunningLocalAction").returns(true);
-        const releaseApiMock = mockReleaseApi({
-            assetNames: ["cli-version-2.10.0.txt"],
-            tagName: "codeql-bundle-20200610",
+const EXPLICITLY_REQUESTED_BUNDLE_TEST_CASES = [
+    {
+        cliVersion: "2.10.0",
+        expectedToolcacheVersion: "2.10.0-20200610",
+    },
+    {
+        cliVersion: "2.10.0-pre",
+        expectedToolcacheVersion: "0.0.0-20200610",
+    },
+];
+for (const { cliVersion, expectedToolcacheVersion, } of EXPLICITLY_REQUESTED_BUNDLE_TEST_CASES) {
+    (0, ava_1.default)(`caches an explicitly requested bundle containing CLI ${cliVersion} as ${expectedToolcacheVersion}`, async (t) => {
+        await util.withTmpDir(async (tmpDir) => {
+            (0, testing_utils_1.setupActionsVars)(tmpDir, tmpDir);
+            mockApiDetails(sampleApiDetails);
+            sinon.stub(actionsUtil, "isRunningLocalAction").returns(true);
+            const releaseApiMock = mockReleaseApi({
+                assetNames: [`cli-version-${cliVersion}.txt`],
+                tagName: "codeql-bundle-20200610",
+            });
+            const url = mockDownloadApi({
+                tagName: "codeql-bundle-20200610",
+            });
+            const result = await codeql.setupCodeQL(url, sampleApiDetails, tmpDir, util.GitHubVariant.DOTCOM, false, SAMPLE_DEFAULT_CLI_VERSION, (0, logging_1.getRunnerLogger)(true), false);
+            t.assert(releaseApiMock.isDone(), "Releases API should have been called");
+            t.assert(toolcache.find("CodeQL", expectedToolcacheVersion));
+            t.deepEqual(result.toolsVersion, cliVersion);
         });
-        const url = mockDownloadApi({
-            tagName: "codeql-bundle-20200610",
-        });
-        const result = await codeql.setupCodeQL(url, sampleApiDetails, tmpDir, util.GitHubVariant.DOTCOM, false, SAMPLE_DEFAULT_CLI_VERSION, (0, logging_1.getRunnerLogger)(true), false);
-        t.assert(releaseApiMock.isDone(), "Releases API should have been called");
-        t.assert(toolcache.find("CodeQL", "2.10.0-20200610"));
-        t.deepEqual(result.toolsVersion, "2.10.0");
     });
-});
+}
 for (const { isCached, tagName, toolcacheCliVersion } of [
     {
         isCached: true,

lib/setup-codeql.js

@@ -88,21 +88,12 @@ async function getCodeQLBundleReleasesDotcomOnly(logger) {
         repo: codeQLActionRepository.split("/")[1],
     });
     logger.debug(`Found ${releases.length} releases.`);
-    return releases.flatMap((release) => {
-        const cliVersionFileVersions = release.assets
-            .map((asset) => { var _a; return (_a = asset.name.match(/cli-version-(.*)\.txt/)) === null || _a === void 0 ? void 0 : _a[1]; })
-            .filter((v) => v)
-            .map((v) => v);
-        if (cliVersionFileVersions.length > 1) {
-            logger.warning(`Ignoring release ${release.tag_name} with multiple CLI version marker files.`);
-            return [];
-        }
-        return [
-            { cliVersion: cliVersionFileVersions[0], tagName: release.tag_name },
-        ];
-    });
+    return releases.map((release) => ({
+        cliVersion: tryGetCodeQLCliVersionForRelease(release, logger),
+        tagName: release.tag_name,
+    }));
 }
-async function tryGetCodeQLCliVersionForRelease(release, logger) {
+function tryGetCodeQLCliVersionForRelease(release, logger) {
     const cliVersionsFromMarkerFiles = release.assets
         .map((asset) => { var _a; return (_a = asset.name.match(/cli-version-(.*)\.txt/)) === null || _a === void 0 ? void 0 : _a[1]; })
         .filter((v) => v)
@@ -429,9 +420,14 @@ async function downloadCodeQL(codeqlURL, maybeCliVersion, apiDetails, variant, t
         (variant === util.GitHubVariant.DOTCOM &&
             (await tryFindCliVersionDotcomOnly(`codeql-bundle-${bundleVersion}`, logger))) ||
         undefined;
-    // Include the bundle version in the toolcache version number so that if the user requests the
-    // same URL again, we can get it from the cache without having to call any of the Releases API.
-    const toolcacheVersion = cliVersion
+    // Include both the CLI version and the bundle version in the toolcache version number. That way
+    // if the user requests the same URL again, we can get it from the cache without having to call
+    // any of the Releases API.
+    //
+    // Special case: If the CLI version is a pre-release, then cache the bundle as
+    // `0.0.0-<bundleVersion>` to avoid the bundle being interpreted as containing a stable CLI
+    // release.
+    const toolcacheVersion = cliVersion && !cliVersion.includes("-")
         ? `${cliVersion}-${bundleVersion}`
         : convertToSemVer(bundleVersion, logger);
     return {

src/codeql.test.ts

@@ -239,36 +239,52 @@ test("downloads an explicitly requested bundle even if a different version is ca
   });
 });
 
-test("tries to cache an explicitly requested bundle with its CLI version number", async (t) => {
-  await util.withTmpDir(async (tmpDir) => {
-    setupActionsVars(tmpDir, tmpDir);
+const EXPLICITLY_REQUESTED_BUNDLE_TEST_CASES = [
+  {
+    cliVersion: "2.10.0",
+    expectedToolcacheVersion: "2.10.0-20200610",
+  },
+  {
+    cliVersion: "2.10.0-pre",
+    expectedToolcacheVersion: "0.0.0-20200610",
+  },
+];
 
-    mockApiDetails(sampleApiDetails);
-    sinon.stub(actionsUtil, "isRunningLocalAction").returns(true);
+for (const {
+  cliVersion,
+  expectedToolcacheVersion,
+} of EXPLICITLY_REQUESTED_BUNDLE_TEST_CASES) {
+  test(`caches an explicitly requested bundle containing CLI ${cliVersion} as ${expectedToolcacheVersion}`, async (t) => {
+    await util.withTmpDir(async (tmpDir) => {
+      setupActionsVars(tmpDir, tmpDir);
 
-    const releaseApiMock = mockReleaseApi({
-      assetNames: ["cli-version-2.10.0.txt"],
-      tagName: "codeql-bundle-20200610",
+      mockApiDetails(sampleApiDetails);
+      sinon.stub(actionsUtil, "isRunningLocalAction").returns(true);
+
+      const releaseApiMock = mockReleaseApi({
+        assetNames: [`cli-version-${cliVersion}.txt`],
+        tagName: "codeql-bundle-20200610",
+      });
+      const url = mockDownloadApi({
+        tagName: "codeql-bundle-20200610",
+      });
+
+      const result = await codeql.setupCodeQL(
+        url,
+        sampleApiDetails,
+        tmpDir,
+        util.GitHubVariant.DOTCOM,
+        false,
+        SAMPLE_DEFAULT_CLI_VERSION,
+        getRunnerLogger(true),
+        false
+      );
+      t.assert(releaseApiMock.isDone(), "Releases API should have been called");
+      t.assert(toolcache.find("CodeQL", expectedToolcacheVersion));
+      t.deepEqual(result.toolsVersion, cliVersion);
     });
-    const url = mockDownloadApi({
-      tagName: "codeql-bundle-20200610",
-    });
-
-    const result = await codeql.setupCodeQL(
-      url,
-      sampleApiDetails,
-      tmpDir,
-      util.GitHubVariant.DOTCOM,
-      false,
-      SAMPLE_DEFAULT_CLI_VERSION,
-      getRunnerLogger(true),
-      false
-    );
-    t.assert(releaseApiMock.isDone(), "Releases API should have been called");
-    t.assert(toolcache.find("CodeQL", "2.10.0-20200610"));
-    t.deepEqual(result.toolsVersion, "2.10.0");
   });
-});
+}
 
 for (const { isCached, tagName, toolcacheCliVersion } of [
   {

src/setup-codeql.ts

@@ -76,28 +76,16 @@ async function getCodeQLBundleReleasesDotcomOnly(
   });
   logger.debug(`Found ${releases.length} releases.`);
 
-  return releases.flatMap((release) => {
-    const cliVersionFileVersions = release.assets
-      .map((asset) => asset.name.match(/cli-version-(.*)\.txt/)?.[1])
-      .filter((v) => v)
-      .map((v) => v as string);
-
-    if (cliVersionFileVersions.length > 1) {
-      logger.warning(
-        `Ignoring release ${release.tag_name} with multiple CLI version marker files.`
-      );
-      return [];
-    }
-    return [
-      { cliVersion: cliVersionFileVersions[0], tagName: release.tag_name },
-    ];
-  });
+  return releases.map((release) => ({
+    cliVersion: tryGetCodeQLCliVersionForRelease(release, logger),
+    tagName: release.tag_name,
+  }));
 }
 
-async function tryGetCodeQLCliVersionForRelease(
+function tryGetCodeQLCliVersionForRelease(
   release,
   logger: Logger
-): Promise<string | undefined> {
+): string | undefined {
   const cliVersionsFromMarkerFiles = release.assets
     .map((asset) => asset.name.match(/cli-version-(.*)\.txt/)?.[1])
     .filter((v) => v)
@@ -573,11 +561,17 @@ export async function downloadCodeQL(
         logger
       ))) ||
     undefined;
-  // Include the bundle version in the toolcache version number so that if the user requests the
-  // same URL again, we can get it from the cache without having to call any of the Releases API.
-  const toolcacheVersion = cliVersion
-    ? `${cliVersion}-${bundleVersion}`
-    : convertToSemVer(bundleVersion, logger);
+  // Include both the CLI version and the bundle version in the toolcache version number. That way
+  // if the user requests the same URL again, we can get it from the cache without having to call
+  // any of the Releases API.
+  //
+  // Special case: If the CLI version is a pre-release, then cache the bundle as
+  // `0.0.0-<bundleVersion>` to avoid the bundle being interpreted as containing a stable CLI
+  // release.
+  const toolcacheVersion =
+    cliVersion && !cliVersion.includes("-")
+      ? `${cliVersion}-${bundleVersion}`
+      : convertToSemVer(bundleVersion, logger);
   return {
     toolsVersion: cliVersion || toolcacheVersion,
     codeqlFolder: await toolcache.cacheDir(