From 354dd156209c336a876de3e2fe1c2a61007a85a5 Mon Sep 17 00:00:00 2001
From: Judson James <judsonjames96@gmail.com>
Date: Fri, 12 Jan 2024 23:52:39 -0500
Subject: [PATCH] Rewrite ARM builds to use arm-runner-action to resolve
 OrangePi5 images (#1143)

---
 .github/workflows/build.yml | 26 ++++++++++++++++++++++----
 scripts/armrunner.sh        | 12 ++++++++++++
 2 files changed, 34 insertions(+), 4 deletions(-)
 create mode 100644 scripts/armrunner.sh

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 4f8606189..b4f17e5ad 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -273,14 +273,20 @@ jobs:
             artifact-name: LinuxArm64
             image_suffix: RaspberryPi
             image_url: https://github.com/PhotonVision/photon-image-modifier/releases/download/v2024.0.4/photonvision_raspi.img.xz
+            cpu: cortex-a7
+            image_additional_mb: 0
           - os: ubuntu-latest
             artifact-name: LinuxArm64
             image_suffix: limelight2
             image_url: https://github.com/PhotonVision/photon-image-modifier/releases/download/v2024.0.4/photonvision_limelight.img.xz
+            cpu: cortex-a7
+            image_additional_mb: 0
           - os: ubuntu-latest
             artifact-name: LinuxArm64
             image_suffix: orangepi5
             image_url: https://github.com/PhotonVision/photon-image-modifier/releases/download/v2024.0.4/photonvision_opi5.img.xz
+            cpu: cortex-a8
+            image_additional_mb: 4096
 
     runs-on: ${{ matrix.os }}
     name: "Build image - ${{ matrix.image_url }}"
@@ -293,11 +299,23 @@ jobs:
       - uses: actions/download-artifact@v4
         with:
           name: jar-${{ matrix.artifact-name }}
-      # TODO- replace with the arm-runner action and run this inside of the chroot. but this works for now.
-      - name: Generate image
+      - uses: pguyot/arm-runner-action@v2
+        name: Generate image
+        id: generate_image
+        with:
+          base_image: ${{ matrix.image_url }}
+          image_additional_mb: ${{ matrix.image_additional_mb }}
+          optimize_image: yes
+          cpu: ${{ matrix.cpu }}
+          commands: |
+            chmod +x scripts/arm_runner_installer.sh
+            ./scripts/arm_runner_installer.sh
+      - name: Compress image
         run: |
-          chmod +x scripts/generatePiImage.sh
-          ./scripts/generatePiImage.sh ${{ matrix.image_url }} ${{ matrix.image_suffix }}
+          new_jar=$(realpath $(find . -name photonvision\*-linuxarm64.jar))
+          new_image_name=$(basename "${new_jar/.jar/_${{ matrix.image_suffix }}.img}")
+          mv ${{ steps.generate_image.outputs.image }} $new_image_name
+          sudo xz -T 0 -v $new_image_name
       - uses: actions/upload-artifact@v4
         name: Upload image
         with:
diff --git a/scripts/armrunner.sh b/scripts/armrunner.sh
new file mode 100644
index 000000000..4e59e345a
--- /dev/null
+++ b/scripts/armrunner.sh
@@ -0,0 +1,12 @@
+###
+# Alternative ARM Runner installer to setup PhotonVision JAR
+# for ARM based builds such as Raspberry Pi, Orange Pi, etc.
+# This assumes that the image provided to arm-runner-action contains
+# the servicefile needed to auto-launch PhotonVision.
+###
+NEW_JAR=$(realpath $(find . -name photonvision\*-linuxarm64.jar))
+echo "Using jar: " $(basename $NEW_JAR)
+
+DEST_PV_LOCATION=/opt/photonvision
+sudo mkdir -p $DEST_PV_LOCATION
+sudo cp $NEW_JAR ${DEST_PV_LOCATION}/photonvision.jar