fix: expose arch var for image (#14)

* fix: expose arch var for image

* format

Author: jamilbk

main.tf

@@ -21,9 +21,9 @@ locals {
   compute_region_zones = length(var.compute_instance_availability_zones) == 0 ? data.google_compute_zones.in_region.names : var.compute_instance_availability_zones
 }
 
-# Fetch most recent COS image
+# Fetch most recent Ubuntu image based on architecture
 data "google_compute_image" "ubuntu" {
-  family  = "ubuntu-2404-lts"
+  family  = "ubuntu-2404-lts-${var.compute_instance_architecture}"
   project = "ubuntu-os-cloud"
 }
 

variables.tf

@@ -33,6 +33,17 @@ variable "compute_instance_type" {
   type = string
 }
 
+variable "compute_instance_architecture" {
+  type        = string
+  default     = "amd64"
+  description = "The architecture of the compute instance (amd64 or arm64)"
+
+  validation {
+    condition     = contains(["amd64", "arm64"], var.compute_instance_architecture)
+    error_message = "Architecture must be either 'amd64' or 'arm64'."
+  }
+}
+
 variable "compute_provision_public_ipv4_address" {
   type        = bool
   default     = true