From a471dabe46a305fa47d821fa91417ab93357b047 Mon Sep 17 00:00:00 2001
From: zeaslity <zeaslity@qq.com>
Date: Fri, 26 Jan 2024 10:45:36 +0800
Subject: [PATCH] [ Cmii ] [ Operator ] - clean cmii image

---
 cmii_operator/CmiiImageSync.go      | 22 ++++++++++++++++++++++
 cmii_operator/CmiiImageSync_test.go |  7 +++++++
 cmii_operator/CmiiOperator_test.go  |  2 ++
 3 files changed, 31 insertions(+)

diff --git a/cmii_operator/CmiiImageSync.go b/cmii_operator/CmiiImageSync.go
index eaf768a..7a731a9 100644
--- a/cmii_operator/CmiiImageSync.go
+++ b/cmii_operator/CmiiImageSync.go
@@ -102,6 +102,28 @@ func ImageDelete(imageName string) []types.ImageDeleteResponseItem {
 	return remove
 }
 
+func ImagePruneAllCmiiImages() (errorRemoveImageNameList []string) {
+
+	imageGetAll := ImageGetAll()
+
+	for _, imageSummary := range imageGetAll {
+		if strings.Contains(imageSummary.RepoTags[0], CmiiHarborPrefix) {
+			_, err := apiClient.ImageRemove(context.TODO(), imageSummary.ID, types.ImageRemoveOptions{
+				Force:         true,
+				PruneChildren: false,
+			})
+			if err != nil {
+				log.ErrorF("[ImageDelete] -- ImageRemove error  ! %s", err.Error())
+				errorRemoveImageNameList = append(errorRemoveImageNameList, imageSummary.RepoTags[0])
+			}
+
+			log.InfoF("[ImageDelete] - image remove of [%s] success!", imageSummary.RepoTags[0])
+		}
+	}
+
+	return errorRemoveImageNameList
+}
+
 func ImageTagFromSourceToTarget(sourceImageName, targetImageName string) bool {
 
 	getByName := ImageGetByName(sourceImageName)
diff --git a/cmii_operator/CmiiImageSync_test.go b/cmii_operator/CmiiImageSync_test.go
index 20f47ac..b1f9179 100644
--- a/cmii_operator/CmiiImageSync_test.go
+++ b/cmii_operator/CmiiImageSync_test.go
@@ -140,3 +140,10 @@ func TestConvertImageGzipFileName(t *testing.T) {
 		t.Logf(" %s to %s", s, gzipFileName)
 	}
 }
+
+func TestImagePruneAllCmiiImages(t *testing.T) {
+
+	errorRemoveImageNameList := ImagePruneAllCmiiImages()
+
+	utils.BeautifulPrintListWithTitle(errorRemoveImageNameList, "CMII Image Prune Error")
+}
diff --git a/cmii_operator/CmiiOperator_test.go b/cmii_operator/CmiiOperator_test.go
index 1409bc8..faad9e4 100644
--- a/cmii_operator/CmiiOperator_test.go
+++ b/cmii_operator/CmiiOperator_test.go
@@ -6,10 +6,12 @@ import (
 )
 
 func TestFetchDemoImages(t *testing.T) {
+
 	errorPullImageList, errorGzipImageList := FetchDemoImages("cqga", true)
 
 	utils.BeautifulPrintListWithTitle(errorPullImageList, "cmii errorPullImageList")
 	utils.BeautifulPrintListWithTitle(errorGzipImageList, "cmii errorGzipImageList")
+
 }
 
 func TestFetchDependencyRepos(t *testing.T) {