Add GitHub proxy support to download file (#119)

* Add GitHub proxy support to download file

* Use strings.Replace instead of strings.ReplaceAll

Author: LinuxSuRen

cmd/get.go

@@ -36,6 +36,10 @@ func newGetCmd(ctx context.Context) (cmd *cobra.Command) {
 		"If you accept preRelease as the binary asset from GitHub")
 	flags.BoolVarP(&opt.AcceptPreRelease, "pre", "", false,
 		"Same with option --accept-preRelease")
+	flags.StringVarP(&opt.ProxyGitHub, "proxy-github", "", "",
+		`The proxy address of github.com, the proxy address will be the prefix of the final address.
+Available proxy: gh.api.99988866.xyz
+Thanks to https://github.com/hunshcn/gh-proxy`)
 
 	flags.IntVarP(&opt.Timeout, "time", "", 10,
 		`The default timeout in seconds with the HTTP request`)
@@ -52,6 +56,8 @@ func newGetCmd(ctx context.Context) (cmd *cobra.Command) {
 	flags.StringVarP(&opt.Arch, "arch", "", runtime.GOARCH, "The arch of target binary file")
 	flags.BoolVarP(&opt.PrintSchema, "print-schema", "", false,
 		"Print the schema of HDConfig if the flag is true without other function")
+
+	_ = cmd.RegisterFlagCompletionFunc("proxy-github", ArrayCompletion("gh.api.99988866.xyz"))
 	return
 }
 
@@ -64,6 +70,7 @@ type downloadOption struct {
 	MaxAttempts      int
 	AcceptPreRelease bool
 	RoundTripper     http.RoundTripper
+	ProxyGitHub      string
 
 	ContinueAt int64
 
@@ -119,6 +126,10 @@ func (o *downloadOption) preRunE(cmd *cobra.Command, args []string) (err error)
 	}
 	o.URL = targetURL
 
+	if o.ProxyGitHub != "" {
+		o.URL = strings.Replace(o.URL, "github.com", fmt.Sprintf("%s/github.com", o.ProxyGitHub), 1)
+	}
+
 	if o.Output == "" {
 		var urlObj *url.URL
 		if urlObj, err = url.Parse(o.URL); err == nil {

cmd/install.go

@@ -42,6 +42,10 @@ func newInstallCmd(ctx context.Context) (cmd *cobra.Command) {
 		"Indicate if install it via go install github.com/xxx/xxx")
 	flags.StringVarP(&opt.fromBranch, "from-branch", "", "master",
 		"Only works if the flag --from-source is true")
+	flags.StringVarP(&opt.ProxyGitHub, "proxy-github", "", "",
+		`The proxy address of github.com, the proxy address will be the prefix of the final address.
+Available proxy: gh.api.99988866.xyz
+Thanks to https://github.com/hunshcn/gh-proxy`)
 
 	flags.BoolVarP(&opt.Download, "download", "", true,
 		"If download the package")

cmd/util.go

@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"context"
+	"github.com/spf13/cobra"
 	"io"
 	"net/http"
 	"os"
@@ -87,3 +88,13 @@ func copyAndCapture(w io.Writer, r io.Reader) ([]byte, error) {
 		}
 	}
 }
+
+// CompletionFunc is the function for command completion
+type CompletionFunc func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective)
+
+// ArrayCompletion return a completion  which base on an array
+func ArrayCompletion(array ...string) CompletionFunc {
+	return func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		return array, cobra.ShellCompDirectiveNoFileComp
+	}
+}