Add option to use SSH for cloning repo

main.go

@@ -23,6 +23,11 @@ func main() {
 			Usage:   "git remote url",
 			EnvVars: []string{"PLUGIN_REMOTE", "CI_REPO_CLONE_URL"},
 		},
+		&cli.StringFlag{
+			Name:    "remote-ssh",
+			Usage:   "git clone ssh url",
+			EnvVars: []string{"CI_REPO_CLONE_SSH_URL"},
+		},
 		&cli.StringFlag{
 			Name:    "path",
 			Usage:   "git clone path",
@@ -140,6 +145,22 @@ func main() {
 			Usage:   "Define safe directories",
 			EnvVars: []string{"PLUGIN_SAFE_DIRECTORY", "CI_WORKSPACE"},
 		},
+		&cli.BoolFlag{
+			Name:    "use-ssh",
+			Usage:   "Using ssh for git clone",
+			EnvVars: []string{"PLUGIN_USE_SSH"},
+			Value:   false,
+		},
+		&cli.StringFlag{
+			Name:    "ssh-key",
+			Usage:   "SSH key for ssh clone",
+			EnvVars: []string{"PLUGIN_SSH_KEY"},
+		},
+		&cli.StringFlag{
+			Name:    "ssh-user",
+			Usage:   "Username for ssh clone",
+			EnvVars: []string{"PLUGIN_SSH_USER"},
+		},
 	}
 
 	if err := app.Run(os.Args); err != nil {
@@ -154,7 +175,8 @@ func run(c *cli.Context) error {
 
 	plugin := Plugin{
 		Repo: Repo{
-			Clone: c.String("remote"),
+			Clone:    c.String("remote"),
+			CloneSSH: c.String("remote-ssh"),
 		},
 		Pipeline: Pipeline{
 			Commit: c.String("sha"),
@@ -180,6 +202,8 @@ func run(c *cli.Context) error {
 			Partial:         c.Bool("partial"),
 			Home:            c.String("home"),
 			SafeDirectory:   c.String("safe-directory"),
+			UseSSH:          c.Bool("use-ssh"),
+			SSHKey:          c.String("ssh-key"),
 		},
 		Backoff: Backoff{
 			Attempts: c.Int("backoff-attempts"),

plugin.go

@@ -63,7 +63,17 @@ func (p Plugin) Exec() error {
 	if isDirEmpty(filepath.Join(p.Pipeline.Path, ".git")) {
 		cmds = append(cmds, initGit(p.Config.Branch))
 		cmds = append(cmds, safeDirectory(p.Config.SafeDirectory))
-		cmds = append(cmds, remote(p.Repo.Clone))
+		if p.Config.UseSSH {
+			// If env var PLUGIN_USE_SSH is set to true, use SSH instead of HTTPS
+			cmds = append(cmds, remote(p.Repo.CloneSSH))
+			if p.Config.SSHKey != "" {
+				// If env var PLUGIN_SSH_KEY is set, use it as the SSH key
+				cmds = append(cmds, sshKeyHandler(p.Config.SSHKey))
+			}
+		} else {
+			cmds = append(cmds, remote(p.Repo.Clone))
+		}
+
 	}
 
 	// fetch ref in any case
@@ -210,6 +220,11 @@ func safeDirectory(safeDirectory string) *exec.Cmd {
 	return appendEnv(exec.Command("git", "config", "--global", "safe.directory", safeDirectory), defaultEnvVars...)
 }
 
+// Use custom SSH Key thanks to core.sshCommand
+func sshKeyHandler(sshKey string) *exec.Cmd {
+	return appendEnv(exec.Command("git", "config", "core.sshCommand", "ssh -i "+sshKey), defaultEnvVars...)
+}
+
 // Sets the remote origin for the repository.
 func remote(remote string) *exec.Cmd {
 	return appendEnv(exec.Command(

types.go

@@ -7,7 +7,8 @@ import (
 
 type (
 	Repo struct {
-		Clone string
+		Clone    string
+		CloneSSH string
 	}
 
 	Pipeline struct {
@@ -38,6 +39,9 @@ type (
 		Partial         bool
 		filter          string
 		SafeDirectory   string
+		UseSSH          bool
+		SSHKey          string
+		SSHUser         string
 	}
 
 	Backoff struct {