Add the capability to the latest GCE images from GCE projects

Add a function to get the latest GCE image based on the project name,
the family and the architecture.

Signed-off-by: Arnaud Meukam <[email protected]>

Author: ameukam

config/jobs/kubernetes/kops/build_jobs.py

@@ -43,6 +43,7 @@
     skip_jobs,
     image,
     networking_options,
+    gce_distro_options,
     distro_options,
     k8s_versions,
     kops_versions,
@@ -485,7 +486,7 @@ def generate_grid():
     # TODO(justinsb): merge into above block when we can
     # pylint: disable=too-many-nested-blocks
     for networking in ['kubenet', 'calico', 'cilium', 'gce']: # TODO: all networking_options:
-        for distro in ['u2204']: # TODO: all distro_options:
+        for distro in gce_distro_options: # TODO: all distro_options:
             for k8s_version in k8s_versions:
                 for kops_version in [None]: # TODO: all kops_versions:
                     results.append(

config/jobs/kubernetes/kops/build_vars.py

@@ -29,6 +29,11 @@
     "kopeio",
 ]
 
+# GCE distributions
+gce_distro_options = [
+    "u2204",
+]
+
 # AWS distributions
 distro_options = [
     "al2023",

config/jobs/kubernetes/kops/helpers.py

@@ -174,6 +174,51 @@ def latest_aws_image(owner, name, arch='x86_64'):
     set_pinned(pin, image)
     return image
 
+# latest_gce_image returns the latest GCE image for the given project, family, and arch
+# If the image is pinned, it returns the pinned image
+# Otherwise, it fetches the latest image from the specified family
+def latest_gce_image(project, family, arch="X86_64"):
+    pin = f"gce://images/{project}/{family}:{arch}"
+    image = get_pinned(pin)
+    if image:
+        return image
+
+    import googleapiclient.discovery  # pylint: disable=import-error, import-outside-toplevel
+
+    compute = googleapiclient.discovery.build("compute", "v1", cache_discovery=False)
+    try:
+        # Get the latest image from the specified family
+        # pylint: disable=no-member
+        image_response = (
+            compute.images().getFromFamily(project=project, family=family).execute()
+        )
+        # pylint: enable=no-member
+    except Exception as e:
+        raise RuntimeError(
+            f"Failed to get image from family {family} in project {project}: {str(e)}"
+        )
+
+    # Verify architecture matches
+    if "architecture" not in image_response:
+        raise RuntimeError(
+            f"Image {image_response['name']} has no architecture specified"
+        )
+
+    if arch not in image_response["architecture"]:
+        raise RuntimeError(
+            f"Image family {family} has architecture {image_response['architecture']} "
+            f"which doesn't match requested {arch}"
+        )
+
+    image_name = image_response["name"]
+    set_pinned(pin, image_name)
+    return f"{project}/{image_name}"
+
+# Get latest images from some public images families
+gce_distro_images = {
+    "u2204": latest_gce_image("ubuntu-os-cloud", "ubuntu-2204-lts"),
+}
+
 distro_images = {
     'al2023': latest_aws_image('137112412989', 'al2023-ami-2*-kernel-6.1-x86_64'),
     'amzn2': latest_aws_image('137112412989', 'amzn2-ami-kernel-5.10-hvm-*-x86_64-gp2'),

config/jobs/kubernetes/kops/pinned.list

@@ -1,3 +1,4 @@
+gce://images/ubuntu-os-cloud/ubuntu-2204-lts:X86_64=ubuntu-2204-jammy-v20250826
 aws://images/137112412989/al2023-ami-2*-kernel-6.1-x86_64:x86_64=137112412989/al2023-ami-2023.8.20250818.0-kernel-6.1-x86_64
 aws://images/137112412989/amzn2-ami-kernel-5.10-hvm-*-x86_64-gp2:x86_64=137112412989/amzn2-ami-kernel-5.10-hvm-2.0.20250818.2-x86_64-gp2
 aws://images/136693071363/debian-11-amd64-*:x86_64=136693071363/debian-11-amd64-20250801-2191