Merge pull request #7 from moznion/option_to_get_by_ssh

Support the option to `get` and `import` by SSH with GitHub

commands.go

@@ -33,6 +33,7 @@ var commandGet = cli.Command{
 	Action: doGet,
 	Flags: []cli.Flag{
 		cli.BoolFlag{"update, u", "Update local repository if cloned already"},
+		cli.BoolFlag{"p", "Clone with SSH"},
 	},
 }
 
@@ -82,6 +83,7 @@ var commandImportStarred = cli.Command{
 	Action: doImportStarred,
 	Flags: []cli.Flag{
 		cli.BoolFlag{"update, u", "Update local repository if cloned already"},
+		cli.BoolFlag{"p", "Clone with SSH"},
 	},
 }
 
@@ -104,11 +106,11 @@ type commandDoc struct {
 }
 
 var commandDocs = map[string]commandDoc{
-	"get":     {"", "[-u] <repository URL> | <user>/<project>"},
+	"get":     {"", "[-u] <repository URL> | [-u] [-p] <user>/<project>"},
 	"list":    {"", "[-p] [-e] [<query>]"},
 	"look":    {"", "<project> | <user>/<project> | <host>/<user>/<project>"},
-	"import":  {"", "[-u] starred <user> | pocket"},
-	"starred": {"import", "[-u] <user>"},
+	"import":  {"", "[-u] [-p] starred <user> | [-u] pocket"},
+	"starred": {"import", "[-u] [-p] <user>"},
 	"pocket":  {"import", "[-u]"},
 }
 
@@ -157,6 +159,12 @@ func doGet(c *cli.Context) {
 		if url.Path[0] != '/' {
 			url.Path = "/" + url.Path
 		}
+
+		isSSH := c.Bool("p")
+		if isSSH {
+			url, err = ConvertGitHubURLHTTPToSSH(url)
+			utils.DieIf(err)
+		}
 	}
 
 	remote, err := NewRemoteRepository(url)
@@ -323,6 +331,8 @@ func doLook(c *cli.Context) {
 func doImportStarred(c *cli.Context) {
 	user := c.Args().First()
 
+	isSSH := c.Bool("p")
+
 	if user == "" {
 		cli.ShowCommandHelp(c, "starred")
 		os.Exit(1)
@@ -344,6 +354,13 @@ func doImportStarred(c *cli.Context) {
 				utils.Log("error", fmt.Sprintf("Could not parse URL <%s>: %s", repo.HTMLURL, err))
 				continue
 			}
+			if isSSH {
+				url, err = ConvertGitHubURLHTTPToSSH(url)
+				if err != nil {
+					utils.Log("error", fmt.Sprintf("Could not convert URL <%s>: %s", repo.HTMLURL, err))
+					continue
+				}
+			}
 
 			remote, err := NewRemoteRepository(url)
 			if utils.ErrorIf(err) {

url.go

@@ -24,3 +24,8 @@ func NewURL(ref string) (*url.URL, error) {
 
 	return url.Parse(ref)
 }
+
+func ConvertGitHubURLHTTPToSSH(url *url.URL) (*url.URL, error) {
+	sshURL := fmt.Sprintf("ssh://git@github.com/%s", url.Path)
+	return url.Parse(sshURL)
+}

url_test.go

@@ -25,3 +25,12 @@ func TestNewURL(t *testing.T) {
 	Expect(scpUrlWithoutUser.Host).To(Equal("github.com"))
 	Expect(err).To(BeNil())
 }
+
+func TestConvertGitHubURLHTTPToSSH(t *testing.T) {
+	RegisterTestingT(t)
+
+	httpsURL, err := NewURL("https://github.com/motemen/pusheen-explorer")
+	sshURL, err := ConvertGitHubURLHTTPToSSH(httpsURL)
+	Expect(err).To(BeNil())
+	Expect(sshURL.String()).To(Equal("ssh://git@github.com/motemen/pusheen-explorer"))
+}