Select tmate binaries based on CPU architecture (#91)

* Select tmate binaries based on CPU architecture

Now arm64 and amd64 are both supported.

* Replace switch statement with an Object map

Based on feedback in the PR

Author: joeyparrish

lib/index.js

@@ -10291,6 +10291,16 @@ const execShellCommand = (cmd) => {
 
 const TMATE_LINUX_VERSION = "2.4.0"
 
+// Map os.arch() values to the architectures in tmate release binary filenames.
+// Possible os.arch() values documented here:
+// https://nodejs.org/api/os.html#os_os_arch
+// Available tmate binaries listed here:
+// https://github.com/tmate-io/tmate/releases/
+const TMATE_ARCH_MAP = {
+  arm64: 'arm64v8',
+  x64: 'amd64',
+};
+
 /** @param {number} ms */
 const sleep = (ms) => new Promise(resolve => setTimeout(resolve, ms));
 
@@ -10308,7 +10318,11 @@ async function run() {
       await execShellCommand(optionalSudoPrefix + 'apt-get update');
       await execShellCommand(optionalSudoPrefix + 'apt-get install -y openssh-client xz-utils');
 
-      const tmateReleaseTar = await tool_cache.downloadTool(`https://github.com/tmate-io/tmate/releases/download/${TMATE_LINUX_VERSION}/tmate-${TMATE_LINUX_VERSION}-static-linux-amd64.tar.xz`);
+      const tmateArch = TMATE_ARCH_MAP[external_os_default().arch()];
+      if (!tmateArch) {
+        throw new Error(`Unsupported architecture: ${external_os_default().arch()}`)
+      }
+      const tmateReleaseTar = await tool_cache.downloadTool(`https://github.com/tmate-io/tmate/releases/download/${TMATE_LINUX_VERSION}/tmate-${TMATE_LINUX_VERSION}-static-linux-${tmateArch}.tar.xz`);
       const tmateDir = external_path_default().join(external_os_default().tmpdir(), "tmate")
       tmateExecutable = external_path_default().join(tmateDir, "tmate")
 

src/index.js

@@ -11,6 +11,16 @@ import { execShellCommand } from "./helpers"
 
 const TMATE_LINUX_VERSION = "2.4.0"
 
+// Map os.arch() values to the architectures in tmate release binary filenames.
+// Possible os.arch() values documented here:
+// https://nodejs.org/api/os.html#os_os_arch
+// Available tmate binaries listed here:
+// https://github.com/tmate-io/tmate/releases/
+const TMATE_ARCH_MAP = {
+  arm64: 'arm64v8',
+  x64: 'amd64',
+};
+
 /** @param {number} ms */
 const sleep = (ms) => new Promise(resolve => setTimeout(resolve, ms));
 
@@ -28,7 +38,11 @@ export async function run() {
       await execShellCommand(optionalSudoPrefix + 'apt-get update');
       await execShellCommand(optionalSudoPrefix + 'apt-get install -y openssh-client xz-utils');
 
-      const tmateReleaseTar = await tc.downloadTool(`https://github.com/tmate-io/tmate/releases/download/${TMATE_LINUX_VERSION}/tmate-${TMATE_LINUX_VERSION}-static-linux-amd64.tar.xz`);
+      const tmateArch = TMATE_ARCH_MAP[os.arch()];
+      if (!tmateArch) {
+        throw new Error(`Unsupported architecture: ${os.arch()}`)
+      }
+      const tmateReleaseTar = await tc.downloadTool(`https://github.com/tmate-io/tmate/releases/download/${TMATE_LINUX_VERSION}/tmate-${TMATE_LINUX_VERSION}-static-linux-${tmateArch}.tar.xz`);
       const tmateDir = path.join(os.tmpdir(), "tmate")
       tmateExecutable = path.join(tmateDir, "tmate")