fix : parsing-for-port-url

Signed-off-by: aniket866 <iamaniketkumarmaner@gmail.com>

Author: aniket866

cmd/importURL.go

@@ -97,28 +97,10 @@ func NewImportURLCommand(globalClientOpts *connectors.ClientOptions) *cobra.Comm
 			}
 			sepSpecificationFiles := strings.Split(specificationFiles, ",")
 			for _, f := range sepSpecificationFiles {
-				mainArtifact := true
-				secret := ""
-
-				// Check if URL starts with https or http
-				if strings.HasPrefix(f, "https://") || strings.HasPrefix(f, "http://") {
-					urlAndMainAtrifactAndSecretName := strings.Split(f, ":")
-					n := len(urlAndMainAtrifactAndSecretName)
-					f = urlAndMainAtrifactAndSecretName[0] + ":" + urlAndMainAtrifactAndSecretName[1]
-					if n > 2 {
-						val, err := strconv.ParseBool(urlAndMainAtrifactAndSecretName[2])
-						if err != nil {
-							fmt.Println(err)
-						}
-						mainArtifact = val
-					}
-					if n > 3 {
-						secret = urlAndMainAtrifactAndSecretName[3]
-					}
-				}
+				parsedURL, mainArtifact, secret := parseImportURLArg(f)
 
 				// Try downloading the artifcat
-				msg, err := mc.DownloadArtifact(f, mainArtifact, secret)
+				msg, err := mc.DownloadArtifact(parsedURL, mainArtifact, secret)
 				if err != nil {
 					fmt.Printf("Got error when invoking Microcks client importing Artifact: %s", err)
 					os.Exit(1)
@@ -130,3 +112,28 @@ func NewImportURLCommand(globalClientOpts *connectors.ClientOptions) *cobra.Comm
 
 	return importURLCmd
 }
+
+func parseImportURLArg(f string) (string, bool, string) {
+	mainArtifact := true
+	secret := ""
+
+	// Check if URL starts with https or http and safely extract flags from the end
+	if strings.HasPrefix(f, "https://") || strings.HasPrefix(f, "http://") {
+		parts := strings.Split(f, ":")
+		n := len(parts)
+		// If the last parts represent mainArtifact or secret name
+		if n > 2 {
+			if val, err := strconv.ParseBool(parts[n-1]); err == nil {
+				mainArtifact = val
+				f = strings.Join(parts[:n-1], ":")
+			} else if n > 3 {
+				if val, err := strconv.ParseBool(parts[n-2]); err == nil {
+					mainArtifact = val
+					secret = parts[n-1]
+					f = strings.Join(parts[:n-2], ":")
+				}
+			}
+		}
+	}
+	return f, mainArtifact, secret
+}

cmd/importURL_test.go

@@ -0,0 +1,91 @@
+/*
+ * Copyright The Microcks Authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestParseImportURLArg(t *testing.T) {
+	tests := []struct {
+		name                 string
+		input                string
+		expectedURL          string
+		expectedMainArtifact bool
+		expectedSecret       string
+	}{
+		{
+			name:                 "Simple URL, defaults",
+			input:                "http://localhost/spec.yaml",
+			expectedURL:          "http://localhost/spec.yaml",
+			expectedMainArtifact: true,
+			expectedSecret:       "",
+		},
+		{
+			name:                 "URL with port, defaults",
+			input:                "http://localhost:8585/spec.yaml",
+			expectedURL:          "http://localhost:8585/spec.yaml",
+			expectedMainArtifact: true,
+			expectedSecret:       "",
+		},
+		{
+			name:                 "URL with port and mainArtifact true",
+			input:                "http://localhost:8585/spec.yaml:true",
+			expectedURL:          "http://localhost:8585/spec.yaml",
+			expectedMainArtifact: true,
+			expectedSecret:       "",
+		},
+		{
+			name:                 "URL with port and mainArtifact false",
+			input:                "http://localhost:8585/spec.yaml:false",
+			expectedURL:          "http://localhost:8585/spec.yaml",
+			expectedMainArtifact: false,
+			expectedSecret:       "",
+		},
+		{
+			name:                 "URL with port, mainArtifact false, and secret",
+			input:                "http://localhost:8585/spec.yaml:false:mysecret",
+			expectedURL:          "http://localhost:8585/spec.yaml",
+			expectedMainArtifact: false,
+			expectedSecret:       "mysecret",
+		},
+		{
+			name:                 "URL with port, mainArtifact true, and secret",
+			input:                "http://localhost:8585/spec.yaml:true:mysecret",
+			expectedURL:          "http://localhost:8585/spec.yaml",
+			expectedMainArtifact: true,
+			expectedSecret:       "mysecret",
+		},
+		{
+			name:                 "Simple URL, mainArtifact false, and secret",
+			input:                "http://localhost/spec.yaml:false:mysecret",
+			expectedURL:          "http://localhost/spec.yaml",
+			expectedMainArtifact: false,
+			expectedSecret:       "mysecret",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			url, mainArtifact, secret := parseImportURLArg(tt.input)
+			assert.Equal(t, tt.expectedURL, url)
+			assert.Equal(t, tt.expectedMainArtifact, mainArtifact)
+			assert.Equal(t, tt.expectedSecret, secret)
+		})
+	}
+}