hotgo/server/internal/cmd/up.go

66 lines
1.4 KiB
Go

// Package cmd
// @Link https://github.com/bufanyun/hotgo
// @Copyright Copyright (c) 2023 HotGo CLI
// @Author Ms <133814250@qq.com>
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
package cmd
import (
"context"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gcmd"
"hotgo/internal/cmd/upgrade/fix"
)
var (
Up = &gcmd.Command{
Name: "up",
Brief: "处理hotgo版本升级更新带来的兼容问题",
Description: ``,
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
args := parser.GetOptAll()
g.Log().Debugf(ctx, "up args:%v", args)
if len(args) == 0 {
err = gerror.New("up args cannot be empty.")
return
}
method, ok := args["m"]
if !ok {
err = gerror.New("up method cannot be empty.")
return
}
switch method {
case "fix":
err = handleUpgradeFix(ctx, args)
default:
err = gerror.Newf("up method[%v] does not exist", method)
}
if err == nil {
g.Log().Info(ctx, "up exec successful!")
}
return
},
}
)
// handleUpgradeFix 处理修复脚本
func handleUpgradeFix(ctx context.Context, args map[string]string) (err error) {
a1, ok := args["a1"]
if !ok {
err = gerror.New("fix args cannot be empty.")
return
}
switch a1 {
case "menuTree":
fix.UpdateAdminMenuTree(ctx)
default:
err = gerror.Newf("fix a1 is invalid, a1:%v", a1)
}
return
}