#!/bin/bash

# 이미지 소스 리스트
registries=("none" "docker.io" "registry.k8s.io" "quay.io" "ghcr.io")

for registry in "${registries[@]}"; do
  count=$(podman images | grep "$registry" | wc -l)

  if [ "$count" -eq 0 ]; then
    echo -e "\033[33mNo search ${registry} container image\033[0m"
  else
    echo -e "\033[32mDelete ${registry} container images: \033[31m$count \033[0m"

    # none은 이름 대신 ID로 삭제, 그 외에는 태그로 삭제
    if [ "$registry" == "none" ]; then
      podman images | grep "$registry" | awk '{print $3}' | xargs -r -I {} podman rmi {} > /dev/null 2>&1
    else
      podman images | grep "$registry" | awk '{print $1":"$2}' | xargs -r -I {} podman rmi {} > /dev/null 2>&1
    fi
  fi
done