add cobra support for gs tool

.gitignore

@@ -14,4 +14,6 @@
 # Dependency directories (remove the comment below to include it)
 vendor/
 
-.idea/
+.idea/
+.vscode/
+.history/

README.md

@@ -13,4 +13,6 @@ gs release tag
 | pull      | 新建并拉取 go-spring 子项目      |
 | push      | 将修改推送到 go-spring 子项目    |
 | remove    | 本地移除 go-spring 子项目       |
-| release   | 本地所有子项目发布远程标签        |
+| release   | 本地所有子项目发布远程标签        |
+| repair    | 修复远程项目的链接               |
+| backup    | 备份本地项目文件               |

cmd/gs/backup.go

@@ -0,0 +1,26 @@
+package gs
+
+import (
+	"os"
+
+	"github.com/go-spring/gs/internal"
+	"github.com/spf13/cobra"
+)
+
+var backpupCmd = &cobra.Command{
+	Use:     "backup",
+	Aliases: []string{"bak"},
+	Short:   "backup a project",
+	Args:    cobra.ExactArgs(0),
+	Run: func(cmd *cobra.Command, args []string) {
+		rootDir, err := os.Getwd()
+		if err != nil {
+			panic(err)
+		}
+		internal.Zip(rootDir)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(backpupCmd)
+}

cmd/gs/pull.go

@@ -0,0 +1,64 @@
+package gs
+
+import (
+	"os"
+
+	"github.com/go-spring/gs/internal"
+	"github.com/spf13/cobra"
+)
+
+var pullCmd = &cobra.Command{
+	Use:     "pull spring-*/starter-* [branch]",
+	Aliases: []string{"pl"},
+	Short:   "pull remote code",
+	Args:    cobra.ExactArgs(2),
+	Run: func(cmd *cobra.Command, args []string) {
+		projectName := args[0]
+		branch := args[1]
+		if branch == "" {
+			branch = "main"
+		}
+		rootDir, err := os.Getwd()
+		if err != nil {
+			panic(err)
+		}
+
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			panic("illegal backup value,must")
+		}
+		if backup {
+			internal.Zip(rootDir)
+		}
+		pull(rootDir, projectName, branch)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(pullCmd)
+}
+
+// pull 拉取远程项目
+func pull(rootDir string, projectName string, branch string) {
+	_, dir, project := validProject(projectName)
+	internal.SafeStash(rootDir, func() {
+		remotes := internal.Remotes(rootDir)
+		if internal.ContainsString(remotes, project) < 0 {
+			add := false
+			defer func() {
+				if !add {
+					remove(rootDir, projectName)
+				}
+			}()
+			repository := internal.Add(rootDir, project, dir, branch)
+			projectXml.Add(internal.Project{
+				Name:   project,
+				Dir:    dir,
+				Url:    repository,
+				Branch: branch,
+			})
+			add = true
+		}
+		internal.Sync(rootDir, project, dir, branch)
+	})
+}

cmd/gs/push.go

@@ -0,0 +1,47 @@
+package gs
+
+import (
+	"os"
+
+	"github.com/go-spring/gs/internal"
+	"github.com/spf13/cobra"
+)
+
+var pushCmd = &cobra.Command{
+	Use:     "push spring-*/starter-*",
+	Aliases: []string{"ps"},
+	Short:   "push local code to remote repo",
+	Args:    cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		projectName := args[0]
+		rootDir, err := os.Getwd()
+		if err != nil {
+			panic(err)
+		}
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			panic("illegal backup value,must")
+		}
+		if backup {
+			internal.Zip(rootDir)
+		}
+		push(rootDir, projectName)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(pushCmd)
+}
+
+// push 推送远程项目
+func push(rootDir string, projectName string) {
+
+	_, dir, project := validProject(projectName)
+	internal.SafeStash(rootDir, func() {
+
+		// 将修改提交到远程项目，不需要往回合并
+		if p, ok := projectXml.Find(project); ok {
+			internal.Push(rootDir, project, dir, p.Branch)
+		}
+	})
+}

cmd/gs/release.go

@@ -0,0 +1,126 @@
+package gs
+
+import (
+	"bufio"
+	"bytes"
+	"errors"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/go-spring/gs/internal"
+	"github.com/spf13/cobra"
+)
+
+var releaseCmd = &cobra.Command{
+	Use:     "release tag",
+	Aliases: []string{"rs"},
+	Short:   "release tag",
+	Args:    cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		tag := args[0]
+		rootDir, err := os.Getwd()
+		if err != nil {
+			panic(err)
+		}
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			panic("illegal backup value,must")
+		}
+		if backup {
+			internal.Zip(rootDir)
+		}
+		release(rootDir, tag)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(releaseCmd)
+}
+
+// release 发布所有远程项目
+func release(rootDir string, tag string) {
+	// tag := arg(2)
+	err := filepath.Walk(rootDir, func(walkFile string, _ os.FileInfo, err error) error {
+
+		if err != nil {
+			return err
+		}
+
+		if path.Base(walkFile) != "go.mod" {
+			return nil
+		}
+
+		fmt.Println(walkFile)
+		fileData, e0 := ioutil.ReadFile(walkFile)
+		if e0 != nil {
+			return nil
+		}
+
+		outBuf := bytes.NewBuffer(nil)
+		r := bufio.NewReader(strings.NewReader(string(fileData)))
+		for {
+			line, isPrefix, e1 := r.ReadLine()
+			if len(line) > 0 && e1 != nil {
+				panic(e1)
+			}
+			if isPrefix {
+				panic(errors.New("ReadLine returned prefix"))
+			}
+			if e1 != nil {
+				if e1 != io.EOF {
+					panic(err)
+				}
+				break
+			}
+			s := strings.TrimSpace(string(line))
+			if strings.HasPrefix(s, "github.com/go-spring/spring-") ||
+				strings.HasPrefix(s, "github.com/go-spring/starter-") {
+				index := strings.LastIndexByte(s, ' ')
+				if index <= 0 {
+					panic(errors.New(s))
+				}
+				b := append(line[:index+2], []byte(tag)...)
+				outBuf.Write(b)
+			} else {
+				outBuf.Write(line)
+			}
+			outBuf.WriteString("\n")
+		}
+
+		fmt.Println(outBuf.String())
+		return ioutil.WriteFile(walkFile, outBuf.Bytes(), os.ModePerm)
+	})
+
+	if err != nil {
+		panic(err)
+	}
+
+	// 提交代码更新
+	internal.Commit(rootDir, "publish "+tag)
+
+	// 遍历所有项目，推送远程更新
+	for _, project := range projectXml.Projects {
+		_, dir, _ := validProject(project.Name)
+		internal.Push(rootDir, project.Name, dir, project.Branch)
+	}
+
+	// 创建临时目录
+	now := time.Now().Format("20060102150405")
+	buildDir := path.Join(rootDir, "..", "go-spring-build-"+now)
+	err = os.MkdirAll(buildDir, os.ModePerm)
+	if err != nil {
+		panic(err)
+	}
+
+	// 遍历所有项目，推送远程标签
+	for _, project := range projectXml.Projects {
+		projectDir := internal.Clone(buildDir, project.Name, project.Url)
+		internal.Release(projectDir, tag)
+	}
+}

cmd/gs/remove.go

@@ -0,0 +1,53 @@
+package gs
+
+import (
+	"os"
+	"path"
+
+	"github.com/go-spring/gs/internal"
+	"github.com/spf13/cobra"
+)
+
+var removeCmd = &cobra.Command{
+	Use:     "remove spring-*/starter-*",
+	Aliases: []string{"pl"},
+	Short:   "pull remote code",
+	Args:    cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		projectName := args[0]
+		rootDir, err := os.Getwd()
+		if err != nil {
+			panic(err)
+		}
+		backup, err := cmd.Flags().GetBool("backup")
+		if err != nil {
+			panic("illegal backup value,must")
+		}
+		if backup {
+			internal.Zip(rootDir)
+		}
+
+		remove(rootDir, projectName)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(removeCmd)
+}
+
+// remove 移除远程项目
+func remove(rootDir string, projectName string) {
+
+	_, dir, project := validProject(projectName)
+	internal.Remove(rootDir, project)
+
+	projectDir := path.Join(rootDir, dir)
+	_ = os.RemoveAll(projectDir)
+
+	if _, err := os.Stat(projectDir); !os.IsNotExist(err) {
+		panic(err)
+	}
+
+	projectXml.Remove(project)
+	internal.Remotes(rootDir)
+}

cmd/gs/repair.go

@@ -0,0 +1,37 @@
+package gs
+
+import (
+	"os"
+
+	"github.com/go-spring/gs/internal"
+	"github.com/spf13/cobra"
+)
+
+var repairCmd = &cobra.Command{
+	Use:     "repair spring-*/starter-* [branch]",
+	Aliases: []string{"rp"},
+	Short:   "repair project dir",
+	Args:    cobra.ExactArgs(2),
+	Run: func(cmd *cobra.Command, args []string) {
+		projectName := args[0]
+		branch := args[1]
+		if branch == "" {
+			branch = "main"
+		}
+		rootDir, err := os.Getwd()
+		if err != nil {
+			panic(err)
+		}
+		repair(rootDir, projectName, branch)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(repairCmd)
+}
+
+// repair 修复远程项目的链接
+func repair(rootDir string, projectName string, branch string) {
+	_, dir, project := validProject(projectName)
+	internal.Add(rootDir, project, dir, branch)
+}