Refactor project structure

cmd/rm.go

@@ -0,0 +1,101 @@
+package cmd
+
+import (
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"gitea.stuzer.link/stuzer05/docker-registry-manager/internal/app"
+	"net/http"
+	"strings"
+)
+
+func RegistryRm(app *app.App, imageArg string) {
+	parts := strings.SplitN(imageArg, ":", 2)
+	imageName := parts[0]
+	tag := ""
+	if len(parts) > 1 {
+		tag = parts[1]
+	}
+
+	if tag == "" {
+		deleteImageByName(app, imageName)
+	} else {
+		deleteImageByTag(app, imageName, tag)
+	}
+}
+
+func deleteImageByName(app *app.App, imageName string) {
+	type tagList struct {
+		Tags []string `json:"tags"`
+	}
+
+	resp, err := http.Get(fmt.Sprintf("%s/v2/%s/tags/list", app.RegistryURL, imageName))
+	if err != nil {
+		fmt.Println("Error fetching tags:", err)
+		return
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode == http.StatusNotFound {
+		fmt.Printf("Error: Image '%s' not found in the registry.\n", imageName)
+		return
+	}
+
+	var tl tagList
+	if err := json.NewDecoder(resp.Body).Decode(&tl); err != nil {
+		fmt.Println("Error decoding response:", err)
+		return
+	}
+
+	for _, tag := range tl.Tags {
+		deleteImageByTag(app, imageName, tag)
+	}
+}
+
+func deleteImageByTag(app *app.App, imageName, tag string) {
+	client := &http.Client{}
+
+	req, err := http.NewRequest("HEAD", fmt.Sprintf("%s/v2/%s/manifests/%s", app.RegistryURL, imageName, tag), nil)
+	if err != nil {
+		fmt.Println("Error creating request:", err)
+		return
+	}
+	req.Header.Set("Accept", "application/vnd.docker.distribution.manifest.v2+json")
+
+	resp, err := client.Do(req)
+	if err != nil {
+		fmt.Println("Error fetching manifest:", err)
+		return
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode == http.StatusNotFound {
+		fmt.Printf("Error: Image '%s:%s' not found in the registry.\n", imageName, tag)
+		return
+	}
+
+	digest := resp.Header.Get("Docker-Content-Digest")
+
+	req, err = http.NewRequest("DELETE", fmt.Sprintf("%s/v2/%s/manifests/%s", app.RegistryURL, imageName, digest), nil)
+	if err != nil {
+		fmt.Println("Error creating delete request:", err)
+		return
+	}
+
+	// Join username and password for Basic Auth
+	registryCredentials := app.RegistryUsername + ":" + app.RegistryPassword
+	req.Header.Set("Authorization", "Basic "+base64.StdEncoding.EncodeToString([]byte(registryCredentials)))
+
+	resp, err = client.Do(req)
+	if err != nil {
+		fmt.Println("Error deleting image:", err)
+		return
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode == http.StatusAccepted {
+		fmt.Printf("Deleted image: %s:%s\n", imageName, tag)
+	} else {
+		fmt.Printf("Failed to delete image: %s:%s (status code: %d)\n", imageName, tag, resp.StatusCode)
+	}
+}