Add repository location parsing code

backend/sftp/uri.go

@@ -0,0 +1,68 @@
+package sftp
+
+import (
+	"errors"
+	"net/url"
+	"strings"
+)
+
+// Config collects all information required to connect to an sftp server.
+type Config struct {
+	User, Host, Dir string
+}
+
+// ParseConfig extracts all information for the sftp connection from the string s.
+func ParseConfig(s string) (interface{}, error) {
+	if strings.HasPrefix(s, "sftp://") {
+		return parseFormat1(s)
+	}
+
+	// otherwise parse in the sftp:user@host:path format, which means we'll get
+	// "user@host:path" in s
+	return parseFormat2(s)
+}
+
+// parseFormat1 parses the first format, starting with a slash, so the user
+// either specified "sftp://host/path", so we'll get everything after the first
+// colon character
+func parseFormat1(s string) (Config, error) {
+	url, err := url.Parse(s)
+	if err != nil {
+		return Config{}, err
+	}
+
+	cfg := Config{
+		Host: url.Host,
+		Dir:  url.Path[1:],
+	}
+	if url.User != nil {
+		cfg.User = url.User.Username()
+	}
+	return cfg, nil
+}
+
+// parseFormat2 parses the second format, sftp:user@host:path
+func parseFormat2(s string) (cfg Config, err error) {
+	// split user/host and path at the second colon
+	data := strings.SplitN(s, ":", 3)
+	if len(data) < 3 {
+		return Config{}, errors.New("sftp: invalid format, hostname or path not found")
+	}
+
+	if data[0] != "sftp" {
+		return Config{}, errors.New(`invalid format, does not start with "sftp:"`)
+	}
+
+	userhost := data[1]
+	cfg.Dir = data[2]
+
+	data = strings.SplitN(userhost, "@", 2)
+	if len(data) == 2 {
+		cfg.User = data[0]
+		cfg.Host = data[1]
+	} else {
+		cfg.Host = userhost
+	}
+
+	return cfg, nil
+}

backend/sftp/uri_test.go

@@ -0,0 +1,52 @@
+package sftp
+
+import "testing"
+
+var uriTests = []struct {
+	s   string
+	cfg Config
+}{
+	// first form, user specified sftp://user@host/dir
+	{
+		"sftp://user@host/dir/subdir",
+		Config{User: "user", Host: "host", Dir: "dir/subdir"},
+	},
+	{
+		"sftp://host/dir/subdir",
+		Config{Host: "host", Dir: "dir/subdir"},
+	},
+	{
+		"sftp://host//dir/subdir",
+		Config{Host: "host", Dir: "/dir/subdir"},
+	},
+
+	// second form, user specified sftp:user@host:/dir
+	{
+		"sftp:foo@bar:/baz/quux",
+		Config{User: "foo", Host: "bar", Dir: "/baz/quux"},
+	},
+	{
+		"sftp:bar:../baz/quux",
+		Config{Host: "bar", Dir: "../baz/quux"},
+	},
+	{
+		"sftp:fux@bar:baz/qu:ux",
+		Config{User: "fux", Host: "bar", Dir: "baz/qu:ux"},
+	},
+}
+
+func TestParseConfig(t *testing.T) {
+	for i, test := range uriTests {
+		cfg, err := ParseConfig(test.s)
+		if err != nil {
+			t.Errorf("test %d failed: %v", i, err)
+			continue
+		}
+
+		if cfg != test.cfg {
+			t.Errorf("test %d: wrong config, want:\n  %v\ngot:\n  %v",
+				i, test.cfg, cfg)
+			continue
+		}
+	}
+}