You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
package migrate
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/zeromicro/go-zero/core/stringx"
|
|
"github.com/zeromicro/go-zero/tools/goctl/rpc/execx"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/console"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
|
|
)
|
|
|
|
const (
|
|
deprecatedGoZeroMod = "github.com/tal-tech/go-zero"
|
|
deprecatedBuilderx = "github.com/tal-tech/go-zero/tools/goctl/model/sql/builderx"
|
|
replacementBuilderx = "github.com/zeromicro/go-zero/core/stores/builder"
|
|
goZeroMod = "github.com/zeromicro/go-zero"
|
|
)
|
|
|
|
var errInvalidGoMod = errors.New("it's only working for go module")
|
|
|
|
func editMod(version string, verbose bool) error {
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
isGoMod, _ := ctx.IsGoMod(wd)
|
|
if !isGoMod {
|
|
return nil
|
|
}
|
|
|
|
latest, err := getLatest(goZeroMod, verbose)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !stringx.Contains(latest, version) {
|
|
return fmt.Errorf("release version %q is not found", version)
|
|
}
|
|
|
|
mod := fmt.Sprintf("%s@%s", goZeroMod, version)
|
|
err = removeRequire(deprecatedGoZeroMod, verbose)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return addRequire(mod, verbose)
|
|
}
|
|
|
|
func addRequire(mod string, verbose bool) error {
|
|
if verbose {
|
|
console.Info("adding require %s ...", mod)
|
|
time.Sleep(200 * time.Millisecond)
|
|
}
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
isGoMod, _ := ctx.IsGoMod(wd)
|
|
if !isGoMod {
|
|
return errInvalidGoMod
|
|
}
|
|
|
|
_, err = execx.Run("go mod edit -require "+mod, wd)
|
|
return err
|
|
}
|
|
|
|
func removeRequire(mod string, verbose bool) error {
|
|
if verbose {
|
|
console.Info("remove require %s ...", mod)
|
|
time.Sleep(200 * time.Millisecond)
|
|
}
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = execx.Run("go mod edit -droprequire "+mod, wd)
|
|
return err
|
|
}
|
|
|
|
func tidy(verbose bool) error {
|
|
if verbose {
|
|
console.Info("go mod tidy ...")
|
|
time.Sleep(200 * time.Millisecond)
|
|
}
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
isGoMod, _ := ctx.IsGoMod(wd)
|
|
if !isGoMod {
|
|
return nil
|
|
}
|
|
|
|
_, err = execx.Run("go mod tidy", wd)
|
|
return err
|
|
}
|