// 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
}