Add support for overriding images of target containers

examples/mounts-template.yaml

@@ -4,3 +4,5 @@
     labels: ${LABELS}
     entrypoints:
       ${ENTRYPOINTS}
+    images:
+      ${IMAGES}

operator/crd.yaml

@@ -112,6 +112,9 @@ spec:
                       description: Override the entrypoint of specific containers in the target Deployment or Statefulset when mounting the code volume.
                       additionalProperties:
                         type: string
+                    images:
+                      type: object
+                      description: Override the image of specific containers in the target Deployment or Statefulset when mounting the code volume.
                     mounted:
                       type: boolean
                       description: Whether the volume should actually be mounted.

operator/op.py

@@ -311,3 +311,19 @@ def restore_entrypoints(*, manifest: dict, entrypoints: dict) -> None:
         if container.get("command"):
             del container["command"]
             del container["args"]
+
+
+def update_images(*, manifest: dict, images: dict) -> None:
+    for container in manifest["spec"]["template"]["spec"]["containers"]:
+        image = images.get(container["name"])
+        if image is None:
+            continue
+        container["x-original-image"] = container["image"]
+        container["image"] = image
+
+
+def restore_images(*, manifest: dict) -> None:
+    for container in manifest["spec"]["template"]["spec"]["containers"]:
+        if container.get("x-original-image"):
+            container["image"] = container["x-original-image"]
+            del container["x-original-image"]