Add sub-command mirror to set the mirror address (#15)

Author: LinuxSuRen

README.md

@@ -12,6 +12,14 @@ cgit is a wrapper of git.
 * GitHub proxy transparent support
 * Git command alias support 
 
+## Mirror
+
+`cgit` can set a mirror address for you if it's very slow with fetching data from GitHub.
+
+Run this command `cgit mirror` in your local git repository directory, 
+it'll change the fetch address to `github.com.cnpmjs.org`. Reversing it is very easy, 
+just run command `cigt mirror --enable=false`.
+
 # Install
 
 ```
@@ -43,3 +51,4 @@ Add a command alias: `cgit alias set cm 'checkout master'`
 Use an alias: `cgit cm`
 
 List all alias commands: `cgit alias list`
+

cmd/mirror.go

@@ -0,0 +1,78 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+	"github.com/linuxsuren/cgit/pkg"
+	"github.com/spf13/cobra"
+	"net/url"
+	"strings"
+)
+
+type mirrorOption struct {
+	enable bool
+	remote string
+}
+
+// NewMirrorCmd returns the mirror command
+func NewMirrorCmd(ctx context.Context) (cmd *cobra.Command) {
+	opt := mirrorOption{}
+
+	cmd = &cobra.Command{
+		Use:   "mirror",
+		Short: "Toggle the git mirror",
+		RunE:  opt.runE,
+	}
+
+	flags := cmd.Flags()
+	flags.BoolVarP(&opt.enable, "enable", "e", true, "Enable/disable the git mirror")
+	flags.StringVarP(&opt.remote, "remote", "r", "origin", "The remote of git repository")
+	return
+}
+
+func (o *mirrorOption) runE(cmd *cobra.Command, args []string) (err error) {
+	var remoteURLStr string
+	var remotePushURLStr string
+	if remoteURLStr, err = pkg.ExecCommandWithOutput("git", "", "remote", "get-url", o.remote); err != nil {
+		return
+	}
+
+	if remotePushURLStr, err = pkg.ExecCommandWithOutput("git", "", "remote", "get-url", "--push", o.remote); err != nil {
+		return
+	}
+	// remove the .git tail
+	remoteURLStr = strings.TrimSuffix(remoteURLStr, ".git")
+
+	gitProtocol := strings.HasPrefix(remoteURLStr, "git@")
+	if gitProtocol {
+		remoteURLStr = strings.ReplaceAll(remoteURLStr, "git@github.com:", "https://github.com/")
+	}
+
+	var remoteURL *url.URL
+	if remoteURL, err = url.Parse(remoteURLStr); err != nil {
+		cmd.Println("error with parse URL", remoteURLStr)
+		return
+	}
+
+	if o.enable {
+		remoteURL.Host = "github.com.cnpmjs.org"
+	} else {
+		remoteURL.Host = "github.com"
+	}
+
+	targetRemoteURLStr := remoteURL.String()
+	if strings.HasPrefix(remotePushURLStr, "git@") && !o.enable {
+		// the mirror git server does not support git protocol
+		targetRemoteURLStr = strings.ReplaceAll(targetRemoteURLStr, fmt.Sprintf("https://%s/", remoteURL.Host),
+			fmt.Sprintf("git@%s:", remoteURL.Host))
+		targetRemoteURLStr += ".git"
+	}
+
+	if err = pkg.ExecCommandInDir("git", "", "remote", "set-url", o.remote, targetRemoteURLStr); err != nil {
+		return
+	}
+	if err = pkg.ExecCommandInDir("git", "", "remote", "set-url", "--push", o.remote, remotePushURLStr); err != nil {
+		return
+	}
+	return
+}

go.sum

@@ -158,15 +158,9 @@ github.com/kr/pty v1.1.4/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/pty v1.1.8/go.mod h1:O1sed60cT9XZ5uDucP5qwvh+TE3NnUj51EiZO/lmSfw=
 github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
-github.com/linuxsuren/cobra-extension v0.0.1 h1:e288zsm1iWUaAOjfiw42hiWJsrrXZnJ0vKApWPyTfkQ=
-github.com/linuxsuren/cobra-extension v0.0.1/go.mod h1:UqHi31q7Dj+46nzcVWNZz5Z1JcHBJcmCyPAZRoM8LXo=
 github.com/linuxsuren/cobra-extension v0.0.3/go.mod h1:UqHi31q7Dj+46nzcVWNZz5Z1JcHBJcmCyPAZRoM8LXo=
 github.com/linuxsuren/cobra-extension v0.0.5 h1:3go926vpXiizRzBEwt8CddCDerRsG5hlIFr0JJvGaKc=
 github.com/linuxsuren/cobra-extension v0.0.5/go.mod h1:qcEJv7BbL0UpK6MbrTESP/nKf1+z1wQdMAnE1NBl3QQ=
-github.com/linuxsuren/go-cli-alias v0.0.3-0.20201208135826-ad33122f181c h1:YNpS+IdkZA9fvNa9gi7xA45FrVSNsEamrzdyezJndjM=
-github.com/linuxsuren/go-cli-alias v0.0.3-0.20201208135826-ad33122f181c/go.mod h1:l2/nzdvung0zorpCxI1CT+Fuuc/W1sFEWS0jt3/b2ZM=
-github.com/linuxsuren/go-cli-alias v0.0.3 h1:IbqC+4uOTCRM27WCII7Z35pXP4bwqck0J5jg2xmIqr4=
-github.com/linuxsuren/go-cli-alias v0.0.3/go.mod h1:l2/nzdvung0zorpCxI1CT+Fuuc/W1sFEWS0jt3/b2ZM=
 github.com/linuxsuren/go-cli-alias v0.0.4 h1:+otarDOeSZzzbTCr9CllAOQCQNpaf/HI41iDvWWUo/w=
 github.com/linuxsuren/go-cli-alias v0.0.4/go.mod h1:dfwOx8H0iVpdS9gtLC80GCC4cvDvzWejxjxXn7sFWUs=
 github.com/linuxsuren/go-cli-plugin v0.0.1/go.mod h1:uyO09KK8otYfDV5LVTfcWX2UbAi3kEz3PrkdfIxnDlg=

main.go

@@ -1,15 +1,16 @@
 package main
 
 import (
+	"context"
 	"fmt"
+	"github.com/linuxsuren/cgit/cmd"
+	"github.com/linuxsuren/cgit/pkg"
 	ext "github.com/linuxsuren/cobra-extension"
 	extver "github.com/linuxsuren/cobra-extension/version"
 	aliasCmd "github.com/linuxsuren/go-cli-alias/pkg/cmd"
 	"github.com/spf13/cobra"
-	"os"
 	"os/exec"
 	"strings"
-	"syscall"
 )
 
 const (
@@ -18,28 +19,30 @@ const (
 )
 
 func main() {
-	cmd := &cobra.Command{
+	command := &cobra.Command{
 		Use: AliasCLI,
-		RunE: func(cmd *cobra.Command, args []string) (err error) {
-			env := os.Environ()
-
+		RunE: func(command *cobra.Command, args []string) (err error) {
+			fmt.Println(args, "sdfs")
 			preHook(args)
 
+			command.Println(args)
+
 			var gitBinary string
 			if gitBinary, err = exec.LookPath(TargetCLI); err == nil {
-				syscall.Exec(gitBinary, append([]string{TargetCLI}, args...), env)
+				err = pkg.ExecCommandInDir(gitBinary, "", args...)
 			}
 			return
 		},
 	}
 
-	cmd.AddCommand(extver.NewVersionCmd("linuxsuren", AliasCLI, AliasCLI, nil))
+	command.AddCommand(extver.NewVersionCmd("linuxsuren", AliasCLI, AliasCLI, nil))
 
-	aliasCmd.AddAliasCmd(cmd, getAliasList())
+	aliasCmd.AddAliasCmd(command, getAliasList())
 
-	cmd.AddCommand(ext.NewCompletionCmd(cmd))
+	command.AddCommand(ext.NewCompletionCmd(command),
+		cmd.NewMirrorCmd(context.TODO()))
 
-	aliasCmd.Execute(cmd, TargetCLI, getAliasList(), preHook)
+	aliasCmd.Execute(command, TargetCLI, getAliasList(), preHook)
 }
 
 func preHook(args []string) {

pkg/exec.go

@@ -0,0 +1,93 @@
+package pkg
+
+import (
+	"io"
+	"io/ioutil"
+	"os"
+	"os/exec"
+	"strings"
+	"sync"
+)
+
+// ExecCommandWithOutput run a command than returns the output
+func ExecCommandWithOutput(name, dir string, args ...string) (output string, err error) {
+	command := exec.Command(name, args...)
+	if dir != "" {
+		command.Dir = dir
+	}
+
+	var stdoutIn io.ReadCloser
+	if stdoutIn, err = command.StdoutPipe(); err != nil {
+		return
+	}
+	if err = command.Start(); err != nil {
+		return
+	}
+
+	var data []byte
+	if data, err = ioutil.ReadAll(stdoutIn); err == nil {
+		output = strings.TrimSpace(string(data))
+	}
+	return
+}
+
+// ExecCommandInDir run a command in a directory
+func ExecCommandInDir(name, dir string, args ...string) (err error) {
+	command := exec.Command(name, args...)
+	if dir != "" {
+		command.Dir = dir
+	}
+
+	//var stdout []byte
+	//var errStdout error
+	stdoutIn, _ := command.StdoutPipe()
+	stderrIn, _ := command.StderrPipe()
+	err = command.Start()
+	if err != nil {
+		return err
+	}
+
+	// cmd.Wait() should be called only after we finish reading
+	// from stdoutIn and stderrIn.
+	// wg ensures that we finish
+	var wg sync.WaitGroup
+	wg.Add(1)
+	go func() {
+		_, _ = copyAndCapture(os.Stdout, stdoutIn)
+		wg.Done()
+	}()
+
+	_, _ = copyAndCapture(os.Stderr, stderrIn)
+
+	wg.Wait()
+
+	err = command.Wait()
+	return
+}
+
+func execCommand(name string, arg ...string) (err error) {
+	return ExecCommandInDir(name, "", arg...)
+}
+
+func copyAndCapture(w io.Writer, r io.Reader) ([]byte, error) {
+	var out []byte
+	buf := make([]byte, 1024, 1024)
+	for {
+		n, err := r.Read(buf[:])
+		if n > 0 {
+			d := buf[:n]
+			out = append(out, d...)
+			_, err := w.Write(d)
+			if err != nil {
+				return out, err
+			}
+		}
+		if err != nil {
+			// Read returns io.EOF at the end of file, which is not an error for us
+			if err == io.EOF {
+				err = nil
+			}
+			return out, err
+		}
+	}
+}