Branch tool to create GCE image with Docker.

tools/baseimage/cmd/create_gce_debian_container/main.go

@@ -0,0 +1,102 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"maps"
+	"slices"
+
+	"github.com/google/android-cuttlefish/tools/baseimage/pkg/cli"
+	"github.com/google/android-cuttlefish/tools/baseimage/pkg/gce"
+	"github.com/google/android-cuttlefish/tools/baseimage/pkg/gce/scripts"
+	"google.golang.org/api/googleapi"
+)
+
+// Cuttlefish base images are based on debian images.
+const debianSourceImageProject = "debian-cloud"
+
+var sourceImageMap = map[gce.Arch]map[int]string{
+	gce.ArchX86: {
+		12: "debian-12-bookworm-v20260114",
+		13: "debian-13-trixie-v20260114",
+	},
+	gce.ArchArm: {
+		12: "debian-12-bookworm-arm64-v20260114",
+		13: "debian-13-trixie-arm64-v20260114",
+	},
+}
+
+// Flags
+var (
+	project       string
+	zone          string
+	arch          cli.Arch
+	debianVersion int
+	imageName     string
+)
+
+func init() {
+	flag.StringVar(&project, "project", "", "GCP project whose resources will be used for creating the amended image")
+	flag.StringVar(&zone, "zone", "us-central1-a", "GCP zone used for creating relevant resources")
+	flag.Var(&arch, "arch", "architecture of GCE image. Supports either x86_64 or arm64")
+	flag.IntVar(&debianVersion, "debian-version", 13, "Debian version: https://www.debian.org/releases")
+	flag.StringVar(&imageName, "image-name", "", "output GCE image name")
+}
+
+func main() {
+	flag.Parse()
+
+	if project == "" {
+		log.Fatal("usage: `-project` must not be empty")
+	}
+	if zone == "" {
+		log.Fatal("usage: `-zone` must not be empty")
+	}
+	if imageName == "" {
+		log.Fatal("usage: `-image-name` must not be empty")
+	}
+
+	if _, ok := sourceImageMap[arch.GceArch()]; !ok {
+		log.Fatalf("no source image found for arch %s: supported archs: %v",
+			arch,
+			slices.Collect(maps.Keys(sourceImageMap)))
+	}
+
+	if _, ok := sourceImageMap[arch.GceArch()][debianVersion]; !ok {
+		log.Fatalf("no source image found for debian %d: supported versions: %v",
+			debianVersion,
+			slices.Collect(maps.Keys(sourceImageMap[arch.GceArch()])))
+	}
+
+	buildImageOpts := gce.BuildImageOpts{
+		Arch:               arch.GceArch(),
+		SourceImageProject: debianSourceImageProject,
+		SourceImage:        sourceImageMap[arch.GceArch()][debianVersion],
+		ImageName:          imageName,
+		ModifyFunc: func(project, zone, insName string) error {
+			if err := gce.UploadBashScript(project, zone, insName, "install_docker.sh", scripts.InstallDocker); err != nil {
+				return fmt.Errorf("error uploading script: %v", err)
+			}
+			return gce.RunCmd(project, zone, insName, "./install_docker.sh")
+		},
+	}
+
+	h, err := gce.NewGceHelper(project, zone)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	_, err = h.Service.Images.Get(project, imageName).Do()
+	if err == nil {
+		log.Fatalf("image %q already exists", imageName)
+	}
+	gerr, ok := err.(*googleapi.Error)
+	if !ok || gerr.Code != 404 {
+		log.Fatalf("failed to check if image exists: %v", err)
+	}
+
+	if err := h.BuildImage(project, zone, buildImageOpts); err != nil {
+		log.Fatal(err)
+	}
+}

tools/baseimage/pkg/gce/scripts/install_docker.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+set -e
+
+# Install Docker on Debian 12
+
+sudo apt-get update
+sudo apt-get install -y ca-certificates curl
+sudo install -m 0755 -d /etc/apt/keyrings
+sudo curl -fsSL https://download.docker.com/linux/debian/gpg -o /etc/apt/keyrings/docker.asc
+sudo chmod a+r /etc/apt/keyrings/docker.asc
+
+# Add the repository to Apt sources:
+echo \
+  "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.asc] https://download.docker.com/linux/debian \
+  $(. /etc/os-release && echo "$VERSION_CODENAME") stable" | \
+  sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
+sudo apt-get update
+
+sudo apt-get install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin

tools/baseimage/pkg/gce/scripts/scripts.go

@@ -39,3 +39,6 @@ var InstallKernelMain string
 
 //go:embed validate_cuttlefish_image.sh
 var ValidateCuttlefishImage string
+
+//go:embed install_docker.sh
+var InstallDocker string