feat: support arm64

minchao · Jun 5, 2024 · 4bb2bf1 · 4bb2bf1
1 parent 6cc2c90
commit 4bb2bf1
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 14 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/util.ts b/src/util.ts
@@ -30,20 +30,31 @@ export async function download(
   return cachedPath
 }
 
+function getArchitecture(): string {
+  const arch = os.arch()
+  switch (arch) {
+    case 'arm64': {
+      return arch
+    }
+    case 'x64': {
+      return 'amd64'
+    }
+    default: {
+      throw new Error(`Unsupported architecture: ${arch}`)
+    }
+  }
+}
+
 function getDownloadURL(toolName: string, version: string): string {
   const baseURL =
     'https://github.com/open-policy-agent/gatekeeper/releases/download/'
-
-  const arch = os.arch()
-  if (arch !== 'x64') {
-    throw new Error(`Unsupported architecture: ${arch}`)
-  }
+  const arch = getArchitecture()
 
   switch (os.type()) {
     case 'Darwin':
-      return `${baseURL}${version}/${toolName}-${version}-darwin-amd64.tar.gz`
+      return `${baseURL}${version}/${toolName}-${version}-darwin-${arch}.tar.gz`
     case 'Linux':
-      return `${baseURL}${version}/${toolName}-${version}-linux-amd64.tar.gz`
+      return `${baseURL}${version}/${toolName}-${version}-linux-${arch}.tar.gz`
     default:
       throw new Error(`Unsupported OS: ${os.type()}`)
   }