From b9ed7e8532c3bfe8fdad51db149e39be7307aa1a Mon Sep 17 00:00:00 2001
From: Skyler Malinowski <smalinowski@nvidia.com>
Date: Mon, 26 Oct 2026 11:08:11 -0400
Subject: [PATCH] Add arm64 build and image release support

Signed-off-by: Skyler Malinowski <smalinowski@nvidia.com>
---
 .github/workflows/release.yml |  2 ++
 .goreleaser.yaml              | 17 +++++++++++------
 Dockerfile                    |  3 ++-
 3 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 0fcde58..35de698 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -19,6 +19,8 @@ jobs:
         uses: actions/setup-go@v6
         with:
           go-version: '1.26.10'
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
       - name: Login to Docker Hub
         if: startsWith(github.ref, 'refs/tags/')
         uses: docker/login-action@v4
diff --git a/.goreleaser.yaml b/.goreleaser.yaml
index 08ec22b..053029c 100644
--- a/.goreleaser.yaml
+++ b/.goreleaser.yaml
@@ -11,6 +11,7 @@ builds:
       - windows
     goarch:
       - amd64
+      - arm64
     env:
       - CGO_ENABLED=0
 archives:
@@ -57,13 +58,17 @@ nfpms:
     formats:
       - apk
     bindir: /usr/bin
-dockers:
-  - image_templates:
-      - "containerssh/agent:latest"
-      - "containerssh/agent:{{ .Tag }}"
-      - "quay.io/containerssh/agent:latest"
-      - "quay.io/containerssh/agent:{{ .Tag }}"
+dockers_v2:
+  - images:
+      - "containerssh/agent"
+      - "quay.io/containerssh/agent"
+    tags:
+      - "latest"
+      - "{{ .Tag }}"
     dockerfile: Dockerfile
+    platforms:
+      - linux/amd64
+      - linux/arm64
 source:
   enabled: true
   name_template: 'containerssh-agent-{{ .Version }}-source'
diff --git a/Dockerfile b/Dockerfile
index 1d51991..40b6d55 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,2 +1,3 @@
 FROM scratch
-COPY containerssh-agent /usr/bin/containerssh-agent
+ARG TARGETPLATFORM
+COPY $TARGETPLATFORM/containerssh-agent /usr/bin/containerssh-agent