package service

import (
	"basic.com/valib/logger.git"
	"errors"
	"sort"
	"strings"
	"vamicro/extend/util"
	"vamicro/system-service/models"
	"vamicro/system-service/vo"
)

type SysMenuService struct {

}

func (sms *SysMenuService) Save(menu models.SysMenu) (bool,models.SysMenu){
	if menu.ParentId == "" || menu.ParentId == "0" {
		menu.ParentId = "0"
		menu.ParentIds= "0"
	} else {
		var pMenu models.SysMenu

		rows,_ := pMenu.SelectById(menu.ParentId)
		if rows > 0 {
			menu.ParentIds = pMenu.ParentIds+","+pMenu.Id
		} else {
			menu.ParentId = "0"
			menu.ParentIds= "0"
		}
	}
	if menu.Id == "" {
		menu.Id = util.PseudoUuid()
		if menu.Insert() {
			return true,menu
		} else {
			return false,menu
		}
	} else {
		if menu.Update() {
			return true,menu
		} else {
			return false,menu
		}
	}
}

/*
	默认应用可以具体到内部的按钮权限
	可选应用只控制到菜单的显示
*/
func (sms *SysMenuService) GetMenuTree(userId string, myMenuIds map[string]string) ([]vo.SysMenuTree,error) {
	var user models.SysUser
	rows, _ := user.SelectById(userId)
	if rows >0 {
		var menuE models.SysMenu
		allMenus := menuE.FindAll()

		var menuM = make(map[string]models.SysMenu, 0)
		if allMenus !=nil {
			for _,mu :=range allMenus {
				menuM[mu.Id] = mu
			}
		}

		var role models.SysRole
		roles := role.FindByUserId(user.Id)
		logger.Debug("len(roles):", len(roles))
		if roles !=nil && len(roles) >0 {
			var userMenus []models.SysMenu
			if roles[0].Name == "超级管理员" {
				userMenus = allMenus
			} else if roles[0].Name ==  "管理员" || roles[0].Name == "普通用户" {
				//查本人可以看到的菜单
				userMenus = menuE.FindByUserId(user.Id)
				if len(userMenus) == 0 {
					userMenus = menuE.FindByRoleId(roles[0].Id)
				}
				//userMenus = allMenus
			}
			logger.Debug("len(userMenus):", len(userMenus))
			var uM = make(map[string]models.SysMenu,0)
			if len(userMenus) >0 {
				for _,srmE := range userMenus{
					if _,ok :=uM[srmE.Id];!ok{
						menuModel := menuM[srmE.Id]
						uM[menuModel.Id] = menuModel

						pidArr := strings.Split(menuModel.ParentIds, ",")
						for _,pid :=range pidArr {
							if pid !="" && pid !="0" {
								if _,k :=uM[pid];!k{
									uM[pid] = menuM[pid]
								}
							}
						}
					}
				}

				var allUserMenus []models.SysMenu
				for _,v :=range uM {
					allUserMenus = append(allUserMenus, v)
				}
				uS := menuFormat(&allUserMenus,"0", myMenuIds)
				var sUs uMenuT
				sUs = uS
				sort.Sort(sUs)
				return sUs,nil
			} else {
				return []vo.SysMenuTree{},nil
			}
		} else {
			return nil,errors.New("user role wrong")
		}
	} else {
		return nil,errors.New("user not found")
	}
}

func menuFormat(dbArr *[]models.SysMenu, parentId string, myMenuIds map [string]string) []vo.SysMenuTree {
	var arr []vo.SysMenuTree
	for _, m := range *dbArr {
		if parentId == m.ParentId {
			tmp := vo.SysMenuTree{}
			tmp.Id = m.Id
			tmp.Name = m.Name
			tmp.Sort = m.Sort
			_,ok := myMenuIds[m.Id]
			tmp.Selected = ok

			children := menuFormat(dbArr, m.Id, myMenuIds)
			tmp.Children = children
			arr = append(arr, tmp)
		}
	}
    var sUs uMenuT
    sUs = arr
    sort.Sort(sUs)
    return sUs
}

type uMenuT []vo.SysMenuTree

func (ul uMenuT) Len()int {
	return len(ul)
}
func (ul uMenuT) Swap(i,j int) {
	ul[i],ul[j] = ul[j],ul[i]
}
func (ul uMenuT) Less(i,j int) bool {
	return ul[i].Sort < ul[j].Sort
}