add install command

Author: stang

cmd/git-remote-https+iap/main.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	_url "net/url"
 	"os"
+	"strings"
 	"time"
 
 	"github.com/adohkan/git-remote-https-iap/internal/git"
@@ -30,14 +31,19 @@ var (
 	versionCmd = &cobra.Command{
 		Use:   "version",
 		Short: "Print version number",
-		Run: func(cmd *cobra.Command, args []string) {
-			fmt.Printf("%s %s\n", BinaryName, version)
-		},
+		Run:   printVersion,
+	}
+
+	installProtocolCmd = &cobra.Command{
+		Use:   "install",
+		Short: "Install protocol in Git config",
+		Run:   installGitProtocol,
 	}
 )
 
 func init() {
 	rootCmd.AddCommand(versionCmd)
+	rootCmd.AddCommand(installProtocolCmd)
 
 	// set default log level
 	zerolog.SetGlobalLevel(zerolog.DebugLevel)
@@ -58,6 +64,16 @@ func execute(cmd *cobra.Command, args []string) {
 	git.PassThruRemoteHTTPSHelper(remote, url)
 }
 
+func printVersion(cmd *cobra.Command, args []string) {
+	fmt.Printf("%s %s\n", BinaryName, version)
+}
+
+func installGitProtocol(cmd *cobra.Command, args []string) {
+	p := strings.TrimLeft(BinaryName, "git-remote-")
+	git.InstallProtocol(p)
+	log.Info().Msgf("%s protocol configured in git!", p)
+}
+
 func handleIAPAuthCookieFor(url string) {
 	// All our work will be based on the basedomain of the provided URL
 	// as IAP would be setup for the whole domain.

internal/git/git.go

@@ -91,3 +91,12 @@ func GetCredentials(protocol, host, username string) (string, error) {
 
 	return "", fmt.Errorf("GetCredentials - not found for protocol=%s,host=%s,username=%s", protocol, host, username)
 }
+
+func InstallProtocol(protocol string) {
+	protocol = fmt.Sprintf("protocol.%s.allow", protocol)
+	args := []string{"config", "--global", protocol, "always"}
+	cmd := exec.Command(GitBinary, args...)
+	if err := cmd.Run(); err != nil {
+		log.Fatal().Msgf("InstallProtocol - %s", err)
+	}
+}