ref(repo): move validate repo url to pkg/repo

cmd/helm/repository.go

@@ -19,10 +19,7 @@ package main
 import (
 	"encoding/json"
 	"errors"
-	"net/url"
 	"path/filepath"
-	"regexp"
-	"strings"
 
 	"github.com/codegangsta/cli"
 	"github.com/kubernetes/helm/pkg/format"
@@ -35,9 +32,6 @@ func init() {
 
 const chartRepoPath = "repositories"
 
-// URL is the url pattern used to check if a given repo url is valid
-const URL string = `^((http|gs|https?):\/\/)?(\S+(:\S*)?@)?((([1-9]\d?|1\d\d|2[01]\d|22[0-3])(\.(1?\d{1,2}|2[0-4]\d|25[0-5])){2}(?:\.([0-9]\d?|1\d\d|2[0-4]\d|25[0-4]))|(([a-zA-Z0-9]+([-\.][a-zA-Z0-9]+)*)|((www\.)?))?(([a-zA-Z\x{00a1}-\x{ffff}0-9]+-?-?)*[a-zA-Z\x{00a1}-\x{ffff}0-9]+)(?:\.([a-zA-Z\x{00a1}-\x{ffff}]{2,}))?))(:(\d{1,5}))?((\/|\?|#)[^\s]*)?$`
-
 const repoDesc = `Helm repositories store Helm charts.
 
    The repository commands are used to manage which Helm repositories Helm may
@@ -95,10 +89,6 @@ func addRepo(c *cli.Context) error {
 	}
 	name := args[0]
 	repoURL := args[1]
-	valid := IsValidURL(repoURL)
-	if !valid {
-		return errors.New(repoURL + " is not a valid REPOSITOTY_URL argument \n" + addRepoDesc)
-	}
 	payload, _ := json.Marshal(repo.Repo{URL: repoURL, Name: name})
 	msg := ""
 	if _, err := NewClient(c).Post(chartRepoPath, payload, &msg); err != nil {
@@ -138,22 +128,3 @@ func removeRepo(c *cli.Context) error {
 	format.Msg(name + " has been removed.\n")
 	return nil
 }
-
-// IsValidURL checks if the string is a valid URL.
-// This was inspired by the IsURL function in govalidator https://github.com/asaskevich/govalidator
-func IsValidURL(str string) bool {
-	if str == "" || len(str) >= 2083 || len(str) <= 3 || strings.HasPrefix(str, ".") {
-		return false
-	}
-	u, err := url.Parse(str)
-	if err != nil {
-		return false
-	}
-	if strings.HasPrefix(u.Host, ".") {
-		return false
-	}
-	if u.Host == "" && (u.Path != "" && !strings.Contains(u.Path, ".")) {
-		return false
-	}
-	return regexp.MustCompile(URL).MatchString(str)
-}

cmd/helm/repository_test.go

@@ -1,37 +0,0 @@
-/*
-Copyright 2016 The Kubernetes Authors All rights reserved.
-
-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 main
-
-import (
-	"testing"
-)
-
-func TestHasValidPrefix(t *testing.T) {
-	tests := map[string]bool{
-		"https://host/bucket": true,
-		"http://host/bucket":  true,
-		"gs://bucket-name":    true,
-		"charts":              false,
-	}
-
-	for url, expectedResult := range tests {
-		result := IsValidURL(url)
-		if result != expectedResult {
-			t.Errorf("Expected: %v, Got: %v for url: %v", expectedResult, result, url)
-		}
-	}
-}

cmd/manager/chartrepos.go

@@ -57,6 +57,9 @@ func addChartRepoHandlerFunc(w http.ResponseWriter, r *http.Request, c *router.C
 		httputil.BadRequest(w, r, err)
 		return nil
 	}
+	if err := repo.ValidateRepoURL(cr.URL); err != nil {
+		return err
+	}
 
 	if err := c.Manager.AddRepo(cr); err != nil {
 		httputil.BadRequest(w, r, err)

pkg/repo/repo.go

@@ -17,10 +17,16 @@ limitations under the License.
 package repo
 
 import (
+	"errors"
 	"fmt"
 	"net/url"
+	"regexp"
+	"strings"
 )
 
+// URL is the url pattern used to check if a given repo url is valid
+const URL string = `^((http|gs|https?):\/\/)?(\S+(:\S*)?@)?((([1-9]\d?|1\d\d|2[01]\d|22[0-3])(\.(1?\d{1,2}|2[0-4]\d|25[0-5])){2}(?:\.([0-9]\d?|1\d\d|2[0-4]\d|25[0-4]))|(([a-zA-Z0-9]+([-\.][a-zA-Z0-9]+)*)|((www\.)?))?(([a-zA-Z\x{00a1}-\x{ffff}0-9]+-?-?)*[a-zA-Z\x{00a1}-\x{ffff}0-9]+)(?:\.([a-zA-Z\x{00a1}-\x{ffff}]{2,}))?))(:(\d{1,5}))?((\/|\?|#)[^\s]*)?$`
+
 // NewRepo takes params and returns a IRepo
 func NewRepo(URL, credentialName, repoName, repoFormat, repoType string) (IRepo, error) {
 	return newRepo(URL, credentialName, repoName, ERepoFormat(repoFormat), ERepoType(repoType))
@@ -113,3 +119,26 @@ func validateRepo(tr IRepo, wantURL, wantCredentialName string, wantFormat ERepo
 
 	return nil
 }
+
+// ValidateRepoURL checks if the string is a valid URL.
+// This was inspired by the IsURL function in govalidator https://github.com/asaskevich/govalidator
+func ValidateRepoURL(str string) error {
+	err := errors.New("Invalid Repository URL")
+	if str == "" || len(str) >= 2083 || len(str) <= 3 || strings.HasPrefix(str, ".") {
+		return err
+	}
+	u, err := url.Parse(str)
+	if err != nil {
+		return err
+	}
+	if strings.HasPrefix(u.Host, ".") {
+		return err
+	}
+	if u.Host == "" && (u.Path != "" && !strings.Contains(u.Path, ".")) {
+		return err
+	}
+	if !regexp.MustCompile(URL).MatchString(str) {
+		return err
+	}
+	return nil
+}

pkg/repo/repo_test.go

@@ -65,3 +65,26 @@ func TestInvalidRepoFormat(t *testing.T) {
 		t.Fatalf("expected error did not occur for invalid format")
 	}
 }
+
+func TestValidateRepoURL(t *testing.T) {
+	validURLs := []string{
+		"https://host/bucket",
+		"http://host/bucket",
+		"gs://bucket-name",
+	}
+	invalidURLs := []string{"charts"}
+
+	for _, url := range validURLs {
+		err := ValidateRepoURL(url)
+		if err != nil {
+			t.Fatalf("Expected repo url: %v to be valid but threw an error")
+		}
+	}
+
+	for _, url := range invalidURLs {
+		err := ValidateRepoURL(url)
+		if err == nil {
+			t.Fatalf("Expected repo url: %v to be invalid but did not throw an error")
+		}
+	}
+}