package models

import (
	"errors"
	"github.com/go-admin-team/go-admin-core/sdk/pkg"
	"go-admin/common/models"

	"gorm.io/gorm"

	"go-admin/app/admin/models/system"
)

type Menu struct {
	MenuId     int    `json:"menuId" gorm:"primaryKey;autoIncrement"`
	MenuName   string `json:"menuName" gorm:"size:128;"`
	Title      string `json:"title" gorm:"size:128;"`
	Icon       string `json:"icon" gorm:"size:128;"`
	Path       string `json:"path" gorm:"size:128;"`
	Paths      string `json:"paths" gorm:"size:128;"`
	MenuType   string `json:"menuType" gorm:"size:1;"`
	Action     string `json:"action" gorm:"size:16;"`
	Permission string `json:"permission" gorm:"size:255;"`
	ParentId   int    `json:"parentId" gorm:"size:11;"`
	NoCache    bool   `json:"noCache" gorm:"size:8;"`
	Breadcrumb string `json:"breadcrumb" gorm:"size:255;"`
	Component  string `json:"component" gorm:"size:255;"`
	Sort       int    `json:"sort" gorm:"size:4;"`
	Visible    string `json:"visible" gorm:"size:1;"`
	CreateBy   string `json:"createBy" gorm:"size:128;"`
	UpdateBy   string `json:"updateBy" gorm:"size:128;"`
	IsFrame    string `json:"isFrame" gorm:"size:1;DEFAULT:0;"`
	DataScope  string `json:"dataScope" gorm:"-"`
	Params     string `json:"params" gorm:"-"`
	RoleId     int    `gorm:"-"`
	Children   []Menu `json:"children" gorm:"-"`
	IsSelect   bool   `json:"is_select" gorm:"-"`
	models.ModelTime
}

func (Menu) TableName() string {
	return "sys_menu"
}

type MenuLable struct {
	Id       int         `json:"id" gorm:"-"`
	Label    string      `json:"label" gorm:"-"`
	Children []MenuLable `json:"children" gorm:"-"`
}

type Menus struct {
	MenuId     int    `json:"menuId" gorm:"column:menu_id;primaryKey;autoIncrement;"`
	MenuName   string `json:"menuName" gorm:"column:menu_name"`
	Title      string `json:"title" gorm:"column:title"`
	Icon       string `json:"icon" gorm:"column:icon"`
	Path       string `json:"path" gorm:"column:path"`
	MenuType   string `json:"menuType" gorm:"column:menu_type"`
	Action     string `json:"action" gorm:"column:action"`
	Permission string `json:"permission" gorm:"column:permission"`
	ParentId   int    `json:"parentId" gorm:"column:parent_id"`
	NoCache    bool   `json:"noCache" gorm:"column:no_cache"`
	Breadcrumb string `json:"breadcrumb" gorm:"column:breadcrumb"`
	Component  string `json:"component" gorm:"column:component"`
	Sort       int    `json:"sort" gorm:"column:sort"`

	Visible  string `json:"visible" gorm:"column:visible"`
	Children []Menu `json:"children" gorm:"-"`

	CreateBy  string `json:"createBy" gorm:"column:create_by"`
	UpdateBy  string `json:"updateBy" gorm:"column:update_by"`
	DataScope string `json:"dataScope" gorm:"-"`
	Params    string `json:"params" gorm:"-"`
	BaseModel
}

func (Menus) TableName() string {
	return "sys_menu"
}

type MenuRole struct {
	Menus
	IsSelect bool `json:"is_select" gorm:"-"`
}

type MS []Menu

//func (e *Menu) GetByMenuId() (Menu Menu, err error) {
//
//	table := orm.Eloquent.Table(e.TableName())
//	table = table.Where("menu_id = ?", e.MenuId)
//	if err = table.Find(&Menu).Error; err != nil {
//		return
//	}
//	return
//}

//func (e *Menu) SetMenu() (m []Menu, err error) {
//	menulist, err := e.GetPage()
//
//	m = make([]Menu, 0)
//	for i := 0; i < len(menulist); i++ {
//		if menulist[i].ParentId != 0 {
//			continue
//		}
//		menusInfo := DiguiMenu(&menulist, menulist[i])
//
//		m = append(m, menusInfo)
//	}
//	return
//}

//func DiguiMenu(menulist *[]Menu, menu Menu) Menu {
//	list := *menulist
//
//	min := make([]Menu, 0)
//	for j := 0; j < len(list); j++ {
//
//		if menu.MenuId != list[j].ParentId {
//			continue
//		}
//		mi := Menu{}
//		mi.MenuId = list[j].MenuId
//		mi.MenuName = list[j].MenuName
//		mi.Title = list[j].Title
//		mi.Icon = list[j].Icon
//		mi.Path = list[j].Path
//		mi.MenuType = list[j].MenuType
//		mi.Action = list[j].Action
//		mi.Permission = list[j].Permission
//		mi.ParentId = list[j].ParentId
//		mi.NoCache = list[j].NoCache
//		mi.Breadcrumb = list[j].Breadcrumb
//		mi.Component = list[j].Component
//		mi.Sort = list[j].Sort
//		mi.Visible = list[j].Visible
//		mi.CreatedAt = list[j].CreatedAt
//		mi.Children = []Menu{}
//
//		if mi.MenuType != "F" {
//			ms := DiguiMenu(menulist, mi)
//			min = append(min, ms)
//
//		} else {
//			min = append(min, mi)
//		}
//
//	}
//	menu.Children = min
//	return menu
//}

//func (e *Menu) SetMenuLabel() (m *[]MenuLable, err error) {
//	menulist, err := e.Get()
//
//	ml := make([]MenuLable, 0)
//	for i := 0; i < len(menulist); i++ {
//		if menulist[i].ParentId != 0 {
//			continue
//		}
//		e := MenuLable{}
//		e.Id = menulist[i].MenuId
//		e.Label = menulist[i].Title
//		menusInfo := MenuLabelCall(&menulist, e)
//
//		ml = append(ml, menusInfo)
//	}
//	return &ml, err
//}

//func MenuLabelCall(menulist *[]Menu, menu MenuLable) MenuLable {
//	list := *menulist
//
//	min := make([]MenuLable, 0)
//	for j := 0; j < len(list); j++ {
//
//		if menu.Id != list[j].ParentId {
//			continue
//		}
//		mi := MenuLable{}
//		mi.Id = list[j].MenuId
//		mi.Label = list[j].Title
//		mi.Children = []MenuLable{}
//		if list[j].MenuType != "F" {
//			ms := MenuLabelCall(menulist, mi)
//			min = append(min, ms)
//		} else {
//			min = append(min, mi)
//		}
//
//	}
//	if len(min) > 0 {
//		menu.Children = min
//	} else {
//		menu.Children = nil
//	}
//	return menu
//}

//func (e *Menu) SetMenuRole(roleName string) (m []Menu, err error) {
//
//	menus, err := e.GetByRoleName(roleName)
//
//	m = make([]Menu, 0)
//	for i := 0; i < len(menus); i++ {
//		if menus[i].ParentId != 0 {
//			continue
//		}
//		menusInfo := DiguiMenu(&menus, menus[i])
//
//		m = append(m, menusInfo)
//	}
//	return
//}

//func (e *MenuRole) Get(tx *gorm.DB) (Menus []MenuRole, err error) {
//	table := tx.Table(e.TableName())
//	if e.MenuName != "" {
//		table = table.Where("menu_name = ?", e.MenuName)
//	}
//	if err = table.Order("sort").Find(&Menus).Error; err != nil {
//		return
//	}
//	return
//}

//func (e *Menu) GetByRoleName(roleName string) (Menus []Menu, err error) {
//	var table *gorm.DB
//	if roleName == "admin" {
//		table = orm.Eloquent.Table(e.TableName()).Select("sys_menu.*")
//		table = table.Where(" menu_type in ('M','C')")
//	} else {
//		table = orm.Eloquent.Table(e.TableName()).Select("sys_menu.*").Joins("left join sys_role_menu on sys_role_menu.menu_id=sys_menu.menu_id")
//		table = table.Where("sys_role_menu.role_name=? and menu_type in ('M','C')", roleName)
//	}
//	if err = table.Order("sort").Find(&Menus).Error; err != nil {
//		return
//	}
//	return
//}

func (e *Menu) Get(tx *gorm.DB) (Menus []Menu, err error) {
	table := tx.Table(e.TableName())
	if e.MenuName != "" {
		table = table.Where("menu_name = ?", e.MenuName)
	}
	if e.Path != "" {
		table = table.Where("path = ?", e.Path)
	}
	if e.Action != "" {
		table = table.Where("action = ?", e.Action)
	}
	if e.MenuType != "" {
		table = table.Where("menu_type = ?", e.MenuType)
	}

	if err = table.Order("sort").Find(&Menus).Error; err != nil {
		return
	}
	return
}

func (e *Menu) GetPage(tx *gorm.DB) (Menus []Menu, err error) {
	table := tx.Table(e.TableName())
	if e.MenuName != "" {
		table = table.Where("menu_name = ?", e.MenuName)
	}
	if e.Title != "" {
		table = table.Where("title = ?", e.Title)
	}
	if e.Visible != "" {
		table = table.Where("visible = ?", e.Visible)
	}
	if e.MenuType != "" {
		table = table.Where("menu_type = ?", e.MenuType)
	}

	// 数据权限控制
	dataPermission := new(system.DataPermission)
	dataPermission.UserId, _ = pkg.StringToInt(e.DataScope)
	table, err = dataPermission.GetDataScope("sys_menu", table)
	if err != nil {
		return nil, err
	}
	if err = table.Order("sort").Find(&Menus).Error; err != nil {
		return
	}
	return
}

func (e *Menu) Create(tx *gorm.DB) (id int, err error) {
	result := tx.Table(e.TableName()).Create(&e)
	if result.Error != nil {
		err = result.Error
		return
	}
	err = InitPaths(tx, e)
	if err != nil {
		return
	}
	id = e.MenuId
	return
}

func InitPaths(tx *gorm.DB, menu *Menu) (err error) {
	parentMenu := new(Menu)
	if menu.ParentId != 0 {
		tx.Table("sys_menu").Where("menu_id = ?", menu.ParentId).First(parentMenu)
		if parentMenu.Paths == "" {
			err = errors.New("父级paths异常，请尝试对当前节点父级菜单进行更新操作！")
			return
		}
		menu.Paths = parentMenu.Paths + "/" + pkg.IntToString(menu.MenuId)
	} else {
		menu.Paths = "/0/" + pkg.IntToString(menu.MenuId)
	}
	tx.Table("sys_menu").Where("menu_id = ?", menu.MenuId).Update("paths", menu.Paths)
	return
}

//func (e *Menu) Update(tx *gorm.DB, id int) (update Menu, err error) {
//	if err = tx.Table(e.TableName()).First(&update, id).Error; err != nil {
//		return
//	}
//
//	//参数1:是要修改的数据
//	//参数2:是修改的数据
//	if err = tx.Table(e.TableName()).Model(&update).Updates(&e).Error; err != nil {
//		return
//	}
//	err = InitPaths(tx, e)
//	if err != nil {
//		return
//	}
//	return
//}

//func (e *Menu) Delete(tx *gorm.DB, id int) (success bool, err error) {
//	if err = tx.Table(e.TableName()).Where("menu_id = ?", id).Delete(&Menu{}).Error; err != nil {
//		success = false
//		return
//	}
//	success = true
//	return
//}