Add setup command to init the config (#184)

Author: LinuxSuRen

cmd/get.go

@@ -65,9 +65,9 @@ func newGetCmd(ctx context.Context) (cmd *cobra.Command) {
 type downloadOption struct {
 	searchOption
 
-	URL          string
-	Output       string
-	ShowProgress bool
+	URL              string
+	Output           string
+	ShowProgress     bool
 	Timeout          int
 	MaxAttempts      int
 	AcceptPreRelease bool

cmd/root.go

@@ -2,10 +2,12 @@ package cmd
 
 import (
 	"context"
+	"fmt"
 	extpkg "github.com/linuxsuren/cobra-extension/pkg"
 	extver "github.com/linuxsuren/cobra-extension/version"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+	"os"
 	"runtime"
 )
 
@@ -21,7 +23,7 @@ func NewRoot(cxt context.Context) (cmd *cobra.Command) {
 	}
 
 	cmd.AddCommand(
-		newGetCmd(cxt), newInstallCmd(cxt), newFetchCmd(cxt), newSearchCmd(cxt), newTestCmd(),
+		newGetCmd(cxt), newInstallCmd(cxt), newFetchCmd(cxt), newSearchCmd(cxt), newTestCmd(), newSetupCommand(),
 		extver.NewVersionCmd("linuxsuren", "http-downloader", "hd", nil),
 		extpkg.NewCompletionCmd(cmd))
 	return
@@ -36,6 +38,8 @@ func loadConfig() (err error) {
 		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
 			// Config file not found; ignore error if desired
 			err = nil
+		} else {
+			err = fmt.Errorf("failed to load config: %s, error: %v", os.ExpandEnv("$HOME/.config/hd.yaml"), err)
 		}
 	}
 	viper.SetDefault("provider", ProviderGitHub)

cmd/setup.go

@@ -0,0 +1,37 @@
+package cmd
+
+import (
+	"github.com/AlecAivazis/survey/v2"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+func newSetupCommand() (cmd *cobra.Command) {
+	opt := &setupOption{}
+	cmd = &cobra.Command{
+		Use:   "setup",
+		Short: "Init the configuration of hd",
+		RunE:  opt.runE,
+	}
+	return
+}
+
+type setupOption struct {
+}
+
+func (o *setupOption) runE(cmd *cobra.Command, args []string) (err error) {
+	selector := &survey.Select{
+		Message: "Select proxy-github",
+		Options: []string{"gh.api.99988866.xyz", "ghproxy.com", "mirror.ghproxy.com", ""},
+	}
+
+	var choose string
+	if err = survey.AskOne(selector, &choose); err == nil {
+		viper.Set("proxy-github", choose)
+	} else {
+		return
+	}
+
+	err = viper.WriteConfig()
+	return
+}

go.mod

@@ -3,7 +3,7 @@ module github.com/linuxsuren/http-downloader
 go 1.16
 
 require (
-	github.com/AlecAivazis/survey/v2 v2.3.1
+	github.com/AlecAivazis/survey/v2 v2.3.2
 	github.com/go-git/go-git/v5 v5.4.2
 	github.com/golang/mock v1.6.0
 	github.com/google/go-github/v29 v29.0.3

go.sum

@@ -46,6 +46,8 @@ dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7
 github.com/AlecAivazis/survey/v2 v2.2.2/go.mod h1:9FJRdMdDm8rnT+zHVbvQT2RTSTLq0Ttd6q3Vl2fahjk=
 github.com/AlecAivazis/survey/v2 v2.3.1 h1:lzkuHA60pER7L4eYL8qQJor4bUWlJe4V0gqAT19tdOA=
 github.com/AlecAivazis/survey/v2 v2.3.1/go.mod h1:TH2kPCDU3Kqq7pLbnCWwZXDBjnhZtmsCle5EiYDJ2fg=
+github.com/AlecAivazis/survey/v2 v2.3.2 h1:TqTB+aDDCLYhf9/bD2TwSO8u8jDSmMUd2SUVO4gCnU8=
+github.com/AlecAivazis/survey/v2 v2.3.2/go.mod h1:TH2kPCDU3Kqq7pLbnCWwZXDBjnhZtmsCle5EiYDJ2fg=
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
 github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=
 github.com/Microsoft/go-winio v0.4.14/go.mod h1:qXqCSQ3Xa7+6tgxaGTIe4Kpcdsi+P8jBhyzoq1bpyYA=