From 96923a5a7f2017f9dd7eecd753d173e47e47cf9a Mon Sep 17 00:00:00 2001
From: Code Hz <codehz@users.noreply.github.com>
Date: Wed, 11 Nov 2020 19:15:17 +0800
Subject: [PATCH] allow use exact commit hash (#13) (#14)

---
 dist/index.js | 61 ++++++++++++++++++++++++++++++++++-----------------
 index.js      | 61 ++++++++++++++++++++++++++++++++++-----------------
 2 files changed, 82 insertions(+), 40 deletions(-)

diff --git a/dist/index.js b/dist/index.js
index cb01b2a..3f77d6a 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -36,29 +36,50 @@ async function downloadZig (version) {
     win32: 'zip'
   }[os.platform()]
 
-  const index = await getJSON({ url: 'https://ziglang.org/download/index.json' })
+  if (version.includes('+')) {
+    // use exact commit hash
+    const addrhost = {
+      linux: 'linux-x86_64',
+      darwin: 'macos-x86_64',
+      win32: 'windows-x86_64'
+    }[os.platform()]
+    const downloadUrl = `https://ziglang.org/builds/zig-${addrhost}-${version}.${ext}`
+    const variantName = `zig-${addrhost}-${version}`
 
-  const availableVersions = Object.keys(index)
-  const useVersion = semver.valid(version)
-    ? semver.maxSatisfying(availableVersions.filter((v) => semver.valid(v)), version)
-    : null
+    const downloadPath = await cache.downloadTool(downloadUrl)
+    const zigPath = ext === 'zip'
+      ? await cache.extractZip(downloadPath)
+      : await cache.extractTar(downloadPath, undefined, 'x')
 
-  const meta = index[useVersion || version]
-  if (!meta || !meta[host]) {
-    throw new Error(`Could not find version ${version} for platform ${host}`)
+    const binPath = path.join(zigPath, variantName)
+    const cachePath = await cache.cacheDir(binPath, 'zig', variantName)
+
+    return cachePath
+  } else {
+    const index = await getJSON({ url: 'https://ziglang.org/download/index.json' })
+
+    const availableVersions = Object.keys(index)
+    const useVersion = semver.valid(version)
+      ? semver.maxSatisfying(availableVersions.filter((v) => semver.valid(v)), version)
+      : null
+
+    const meta = index[useVersion || version]
+    if (!meta || !meta[host]) {
+      throw new Error(`Could not find version ${version} for platform ${host}`)
+    }
+
+    const variantName = path.basename(meta[host].tarball).replace(`.${ext}`, '')
+
+    const downloadPath = await cache.downloadTool(meta[host].tarball)
+    const zigPath = ext === 'zip'
+      ? await cache.extractZip(downloadPath)
+      : await cache.extractTar(downloadPath, undefined, 'x')
+
+    const binPath = path.join(zigPath, variantName)
+    const cachePath = await cache.cacheDir(binPath, 'zig', variantName)
+
+    return cachePath
   }
-
-  const variantName = path.basename(meta[host].tarball).replace(`.${ext}`, '')
-
-  const downloadPath = await cache.downloadTool(meta[host].tarball)
-  const zigPath = ext === 'zip'
-    ? await cache.extractZip(downloadPath)
-    : await cache.extractTar(downloadPath, undefined, 'x')
-
-  const binPath = path.join(zigPath, variantName)
-  const cachePath = await cache.cacheDir(binPath, 'zig', variantName)
-
-  return cachePath
 }
 
 async function main () {
diff --git a/index.js b/index.js
index e2fe21f..4ee7d61 100644
--- a/index.js
+++ b/index.js
@@ -29,29 +29,50 @@ async function downloadZig (version) {
     win32: 'zip'
   }[os.platform()]
 
-  const index = await getJSON({ url: 'https://ziglang.org/download/index.json' })
+  if (version.includes('+')) {
+    // use exact commit hash
+    const addrhost = {
+      linux: 'linux-x86_64',
+      darwin: 'macos-x86_64',
+      win32: 'windows-x86_64'
+    }[os.platform()]
+    const downloadUrl = `https://ziglang.org/builds/zig-${addrhost}-${version}.${ext}`
+    const variantName = `zig-${addrhost}-${version}`
 
-  const availableVersions = Object.keys(index)
-  const useVersion = semver.valid(version)
-    ? semver.maxSatisfying(availableVersions.filter((v) => semver.valid(v)), version)
-    : null
+    const downloadPath = await cache.downloadTool(downloadUrl)
+    const zigPath = ext === 'zip'
+      ? await cache.extractZip(downloadPath)
+      : await cache.extractTar(downloadPath, undefined, 'x')
 
-  const meta = index[useVersion || version]
-  if (!meta || !meta[host]) {
-    throw new Error(`Could not find version ${version} for platform ${host}`)
+    const binPath = path.join(zigPath, variantName)
+    const cachePath = await cache.cacheDir(binPath, 'zig', variantName)
+
+    return cachePath
+  } else {
+    const index = await getJSON({ url: 'https://ziglang.org/download/index.json' })
+
+    const availableVersions = Object.keys(index)
+    const useVersion = semver.valid(version)
+      ? semver.maxSatisfying(availableVersions.filter((v) => semver.valid(v)), version)
+      : null
+
+    const meta = index[useVersion || version]
+    if (!meta || !meta[host]) {
+      throw new Error(`Could not find version ${version} for platform ${host}`)
+    }
+
+    const variantName = path.basename(meta[host].tarball).replace(`.${ext}`, '')
+
+    const downloadPath = await cache.downloadTool(meta[host].tarball)
+    const zigPath = ext === 'zip'
+      ? await cache.extractZip(downloadPath)
+      : await cache.extractTar(downloadPath, undefined, 'x')
+
+    const binPath = path.join(zigPath, variantName)
+    const cachePath = await cache.cacheDir(binPath, 'zig', variantName)
+
+    return cachePath
   }
-
-  const variantName = path.basename(meta[host].tarball).replace(`.${ext}`, '')
-
-  const downloadPath = await cache.downloadTool(meta[host].tarball)
-  const zigPath = ext === 'zip'
-    ? await cache.extractZip(downloadPath)
-    : await cache.extractTar(downloadPath, undefined, 'x')
-
-  const binPath = path.join(zigPath, variantName)
-  const cachePath = await cache.cacheDir(binPath, 'zig', variantName)
-
-  return cachePath
 }
 
 async function main () {