Add ImageRegistrySpec helpers

Signed-off-by: Tamal Saha <[email protected]>
kmodules · Oct 29, 2023 · bcb43a7 · bcb43a7
1 parent 4a3745f
commit bcb43a7
Showing 1 changed file with 63 additions and 0 deletions.
diff --git a/apis/shared/helpers.go b/apis/shared/helpers.go
@@ -91,3 +91,66 @@ func (r ResourceLocator) GraphQuery(oid kmapi.OID) (string, map[string]interface
 	}
 	return "", nil, fmt.Errorf("unknown query type %+v, oid %s", r, oid)
 }
+
+func (r ImageRegistrySpec) DockerHubProxy() string {
+	addr := r.Proxies.DockerHub
+	if addr == "" {
+		addr = r.RegistryFQDN
+	}
+	addr = strings.TrimSpace(addr)
+	addr = strings.TrimSuffix(addr, "/")
+	return addr
+}
+
+func (r ImageRegistrySpec) DockerLibraryProxy() string {
+	addr := r.Proxies.DockerLibrary
+	if addr == "" {
+		addr = r.Proxies.DockerHub
+	}
+	if addr == "" {
+		addr = r.RegistryFQDN
+	}
+	addr = strings.TrimSpace(addr)
+	addr = strings.TrimSuffix(addr, "/")
+	return addr
+}
+
+func (r ImageRegistrySpec) GHCRProxy() string {
+	addr := r.Proxies.GHCR
+	if addr == "" {
+		addr = r.RegistryFQDN
+	}
+	addr = strings.TrimSpace(addr)
+	addr = strings.TrimSuffix(addr, "/")
+	return addr
+}
+
+func (r ImageRegistrySpec) QuayProxy() string {
+	addr := r.Proxies.Quay
+	if addr == "" {
+		addr = r.RegistryFQDN
+	}
+	addr = strings.TrimSpace(addr)
+	addr = strings.TrimSuffix(addr, "/")
+	return addr
+}
+
+func (r ImageRegistrySpec) KubernetesRegistryProxy() string {
+	addr := r.Proxies.Kubernetes
+	if addr == "" {
+		addr = r.RegistryFQDN
+	}
+	addr = strings.TrimSpace(addr)
+	addr = strings.TrimSuffix(addr, "/")
+	return addr
+}
+
+func (r ImageRegistrySpec) AppsCodeRegistryProxy() string {
+	addr := r.Proxies.AppsCode
+	if addr == "" {
+		addr = r.RegistryFQDN
+	}
+	addr = strings.TrimSpace(addr)
+	addr = strings.TrimSuffix(addr, "/")
+	return addr
+}