feat: add option to removed loaded images

Author: aldbr

run_demo.sh

@@ -216,6 +216,11 @@ while [ -n "${1:-}" ]; do case $1 in
     shift
     continue ;;
 
+  --prune-loaded-images)
+    prune_loaded_images=1
+    shift
+    continue ;;
+
   --ci-values)
     shift
     if [[ -z "${1:-}" ]]; then
@@ -573,6 +578,17 @@ printf "%b Loading docker images...\n" ${UNICORN_EMOJI}
 for img_name in "${docker_images_to_load[@]}"
 do
   "${demo_dir}/kind" --name diracx-demo load docker-image "${img_name}"
+
+  if [[ ${prune_loaded_images:-0} -eq 1 ]]; then
+    printf "%b Pruning ${img_name} locally\n" ${UNICORN_EMOJI}
+    # delete the tag
+    docker image rm -f "${img_name}"
+    # delete the digest that kind added so layers really disappear
+    digest="$(docker image inspect --format='{{index .RepoDigests 0}}' "${img_name}" 2>/dev/null)"
+    if [[ -n "$digest" ]]; then
+      docker image rm -f "$digest"
+    fi
+  fi
 done