rewrite func getImageNameTag for supporting registries with port

Signed-off-by: artem198315 <> Signed-off-by: artem198315 <[email protected]>
metallb · Aug 2, 2023 · 43ce79d · 43ce79d
1 parent 82e861c
commit 43ce79d
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/pkg/helm/config.go b/pkg/helm/config.go
@@ -330,9 +330,19 @@ func valueWithDefault(name string, def int) (int, error) {
 }
 
 func getImageNameTag(envValue string) (string, string) {
-	img := strings.Split(envValue, ":")
+	var img []string
+	var repoPath string
+
+	pos := strings.LastIndex(envValue, "/")
+	if pos > 0 {
+		img = strings.Split(envValue[pos+1:], ":")
+		repoPath = envValue[:pos] + "/"
+	} else {
+		img = strings.Split(envValue, ":")
+	}
+
 	if len(img) == 1 {
-		return img[0], ""
+		return repoPath + img[0], ""
 	}
-	return img[0], img[1]
+	return repoPath + img[0], img[1]
 }