package cms

import (
	"fmt"
	"time"

	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
	"git.rosy.net.cn/jx-callback/business/model"
	"git.rosy.net.cn/jx-callback/business/model/dao"
)

func GetMenu(ctx *jxcontext.Context) (menus []*model.Menu, err error) {
	return dao.GetMenu(dao.GetDB(), "", 0)
}

func AddMenu(ctx *jxcontext.Context, menu *model.Menu) (err error) {
	var (
		db = dao.GetDB()
	)
	if menu == nil {
		return fmt.Errorf("添加失败！menu nil")
	}
	if menu.Name == "" || menu.Level == 0 {
		return fmt.Errorf("添加失败！menu 名称和等级必须有值！")
	}
	menus, err := dao.GetMenu(db, menu.Name, menu.Level)
	if len(menus) > 0 {
		return fmt.Errorf("添加失败！已存在相同名称的 menu name : %v", menu.Name)
	}
	dao.Begin(db)
	defer func() {
		if r := recover(); r != nil {
			dao.Rollback(db)
			panic(r)
		}
	}()
	dao.WrapAddIDCULDEntity(menu, ctx.GetUserName())
	err = dao.CreateEntity(db, menu)
	dao.Commit(db)
	return err
}

func UpdateMenu(ctx *jxcontext.Context, menuID int, payload map[string]interface{}, isDelete bool) (num int64, err error) {
	var (
		db = dao.GetDB()
	)
	menu := &model.Menu{}
	menu.ID = menuID
	err = dao.GetEntity(db, menu)
	if err != nil {
		return 0, err
	}
	dao.Begin(db)
	defer func() {
		if r := recover(); r != nil {
			dao.Rollback(db)
			panic(r)
		}
	}()
	if !isDelete {
		valid := dao.StrictMakeMapByStructObject(payload, menu, ctx.GetUserName())
		if len(valid) > 0 {
			if num, err = dao.UpdateEntityLogically(db, menu, valid, ctx.GetUserName(), nil); err != nil {
				dao.Rollback(db)
				return 0, err
			}
		}
	} else {
		menu.DeletedAt = time.Now()
		num, err = dao.UpdateEntity(db, menu, "DeletedAt")
	}
	dao.Commit(db)
	return num, err
}