From 4d59e1a1f490576bc6dd9aca6c27ea1a12c5b634 Mon Sep 17 00:00:00 2001
From: Rishav Dhar <19497993+rdhar@users.noreply.github.com>
Date: Tue, 1 Oct 2024 13:53:16 +0100
Subject: [PATCH] fix: tenv version and arch selection

Signed-off-by: Rishav Dhar <19497993+rdhar@users.noreply.github.com>
---
 README.md  |  1 +
 action.yml | 10 ++++++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 06fc2afe..55beaa87 100644
--- a/README.md
+++ b/README.md
@@ -121,6 +121,7 @@ In order to locally decrypt the TF plan file, use the following command (noting
 | `fmt_enable`</br>Default: `true`                       | Boolean flag to enable TF fmt command and display diff of changes.                                     |
 | `label_pr`</br>Default: `true`                         | Boolean flag to add PR label of TF command to run.                                                     |
 | `plan_parity`</br>Default: `false`                     | Boolean flag to compare the TF plan file with a newly-generated one to prevent stale apply.            |
+| `tenv_arch`</br>Default: `arm64`                       | String architecture of the tenv tool to install and use.                                               |
 | `tenv_version`</br>Example: `v3.1.0`                   | String version tag of the tenv tool to install and use.                                                |
 | `tf_tool`</br>Default: `terraform`                     | String name of the TF tool to use and override default assumption from wrapper environment variable.   |
 | `tf_version`</br>Example: `~> 1.8.0`                   | String version constraint of the TF tool to install and use.                                           |
diff --git a/action.yml b/action.yml
index c9c9d7dd..768b0ad8 100644
--- a/action.yml
+++ b/action.yml
@@ -32,6 +32,10 @@ inputs:
     description: Boolean flag to compare the TF plan file with a newly-generated one to prevent stale apply.
     required: false
     default: "false"
+  tenv_arch:
+    description: String architecture of the tenv tool to install and use.
+    required: false
+    default: "arm64"
   tenv_version:
     description: String version tag of the tenv tool to install and use.
     required: false
@@ -278,16 +282,18 @@ runs:
     - name: Install TF via tenv
       if: inputs.tf_version != ''
       env:
+        TENV_ARCH: ${{ inputs.tenv_arch }}
         TENV_VERSION: ${{ inputs.tenv_version }}
         TF_TOOL: ${{ inputs.tf_tool }}
         TF_VERSION: ${{ inputs.tf_version }}
       shell: bash
       run: |
+        # If $TENV_VERSION is not set, then retrieve the latest version.
         if [ -z "$TENV_VERSION" ]; then
           TENV_VERSION=$(curl --no-progress-meter --location https://api.github.com/repos/tofuutils/tenv/releases/latest | jq -r .tag_name)
         fi
-        curl --remote-name --no-progress-meter --location "https://github.com/tofuutils/tenv/releases/latest/download/tenv_${TENV_VERSION}_amd64.deb"
-        sudo dpkg --install "tenv_${TENV_VERSION}_amd64.deb"
+        curl --remote-name --no-progress-meter --location "https://github.com/tofuutils/tenv/releases/download/${TENV_VERSION}/tenv_${TENV_VERSION}_${TENV_ARCH}.deb"
+        sudo dpkg --install "tenv_${TENV_VERSION}_${TENV_ARCH}.deb"
         tenv "$TF_TOOL" install "$TF_VERSION"
         tenv update-path