ERP_user/controllers/Power.go

package controllers

import (
	"ERP_user/Nats/NatsServer"
	"ERP_user/conf"
	"ERP_user/models/Account"
	"ERP_user/models/System"
	"fmt"
	"math"
	"strings"

	"github.com/astaxie/beego/logs"
	"github.com/beego/beego/v2/adapter/orm"
	beego "github.com/beego/beego/v2/server/web"
	menulibs "gogs.baozhida.cn/zoie/ERP_libs/Menu"
	powerlibs "gogs.baozhida.cn/zoie/ERP_libs/Power"
	"gogs.baozhida.cn/zoie/ERP_libs/lib"
)

type PowerController struct {
	beego.Controller
	User Account.User
}

func (c *PowerController) Prepare() {
	c.User = *Account.User_r
}

// 列表 -
func (c *PowerController) List() {

	// 分页参数 初始化
	page, _ := c.GetInt("page")
	if page < 1 {
		page = 1
	}
	page_z, _ := c.GetInt("page_z")
	if page_z < 1 {
		page_z = conf.Page_size
	}

	// 查询
	T_name := c.GetString("T_name")
	R_List, R_cnt := Account.Read_Power_List(T_name, page, page_z)

	var r_jsons lib.R_JSONS
	r_jsons.Num = R_cnt
	r_jsons.Data = R_List
	r_jsons.Page = page
	r_jsons.Page_size = int(math.Ceil(float64(R_cnt) / float64(page_z)))

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
	c.ServeJSON()
	return
}

func (c *PowerController) Get() {

	type Data struct {
		Power        powerlibs.Power_R
		Menu         []menulibs.Menu
		Menu_checked []int
	}

	var data Data

	T_id := c.GetString("T_id")
	T_code := c.GetString("T_code")
	if len(T_code) == 0 {
		T_code = conf.ERP_ACCOUNT_Sys
	}

	if len(T_id) < 1 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	sysName := Account.Get_Sys_Name(T_code)

	var menu []menulibs.Menu
	var power powerlibs.Power
	var err error

	power, err = Account.Read_Power_ByT_id(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
		c.ServeJSON()
		return
	}

	switch T_code {
	case conf.ERP_ACCOUNT_Sys:
		menu, err = Account.Read_Menu_List()
		if err != nil {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: "获取用户系统菜单失败！"}
			c.ServeJSON()
			return
		}
	default:
		power, err = NatsServer.ERP_Read_Power_ByT_id(T_code, T_id)
		if err != nil {
			if err.Error() == orm.ErrNoRows.Error() {
				var_ := powerlibs.Power{
					T_id:   power.T_id,
					T_name: power.T_name,
				}
				_, err = NatsServer.ERP_Add_Power(T_code, var_)
				if err != nil {
					logs.Error(lib.FuncName(), err)
				}
			} else {
				c.Data["json"] = lib.JSONS{Code: 202, Msg: fmt.Sprintf("获取%s权限失败！", sysName)}
				c.ServeJSON()
				return
			}
		}

		menu, err = NatsServer.ERP_Read_Menu_List(T_code)
		if err != nil {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: fmt.Sprintf("获取%s菜单失败！", sysName)}
			c.ServeJSON()
			return
		}
	}

	data.Power = powerlibs.PowerToPower_R(power)
	data.Menu = menu

	// 处理 Menu_checked
	// 如果角色没有某个父菜单下的所有子菜单权限，则不返回该父菜单给前端
	if power.T_menu != "*" {
		// 先获取已勾选的菜单ID列表
		checkedIds := lib.SplitStringToIntIds(power.T_menu, "M")

		// 过滤父级：仅当父菜单的所有子菜单都被勾选时才保留父菜单
		data.Menu_checked = filterParentMenusFromChecked(menu, checkedIds)
	}
	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: data}
	c.ServeJSON()
	return

}

func (c *PowerController) Add() {

	T_name := c.GetString("T_name")
	rand_x := 0
	T_id := ""
	for true {
		T_id = lib.GetRandstring(8, "", int64(rand_x))
		_, err := Account.Read_Power_ByT_id(T_id)
		if err != nil {
			break
		}
		rand_x += 1
	}

	var_ := powerlibs.Power{
		T_id:   T_id,
		T_name: T_name,
	}

	id, err := Account.Add_Power(var_)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败!"}
		c.ServeJSON()
		return
	}

	for _, prefix := range Account.ERP_Sys_prefix {
		i := 3
		for {
			// Nats 发送失败 重试3次
			_, err = NatsServer.ERP_Add_Power(prefix, var_)
			if err == nil || i == 0 {
				break
			}
			i--
		}

		if err != nil {
			sysName := Account.Get_Sys_Name(prefix)
			logs.Error(lib.FuncName(), sysName+"添加失败: ", err)
			c.Data["json"] = lib.JSONS{Code: 202, Msg: sysName + "添加失败!"}
			c.ServeJSON()
			return
		}
	}

	System.Add_UserLogs_T(c.User.T_uuid, "权限", "新增", var_)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: id}
	c.ServeJSON()
	return

}

func (c *PowerController) Edit() {

	T_id := c.GetString("T_id")
	T_name := c.GetString("T_name")

	if len(T_id) == 0 {
		c.Data["json"] = lib.JSONS{Code: 201, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	userPower, err := Account.Read_Power_ByT_id(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	userPower.T_name = T_name

	id, err := Account.Update_Power(userPower)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
		c.ServeJSON()
		return
	}

	for _, prefix := range Account.ERP_Sys_prefix {
		go func(sys string) {
			// Nats 发送失败 重试10次
			var power powerlibs.Power
			i := 10
			for {
				power, err = NatsServer.ERP_Read_Power_ByT_id(sys, T_id)
				if err != nil {
					if err.Error() == orm.ErrNoRows.Error() {
						var_ := powerlibs.Power{
							T_id:   userPower.T_id,
							T_name: T_name,
						}
						_, err = NatsServer.ERP_Add_Power(sys, var_)
						if err == nil || i == 0 {
							break
						}
					} else {
						continue
					}
				}
				if power.Id > 0 {
					power.T_name = T_name
					_, err = NatsServer.ERP_Update_Power(sys, power)
					if err == nil || i == 0 {
						break
					}
				}
				i--
			}
		}(prefix)
	}

	System.Add_UserLogs_T(c.User.T_uuid, "权限", "修改", userPower)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: id}
	c.ServeJSON()
	return

}

func (c *PowerController) Sys_List() {

	type Sys struct {
		T_sys  string
		T_name string
	}
	var syslist []Sys

	list := strings.Split(conf.ERP_Sys, ",")
	for _, v := range list {
		sys, name := strings.Split(v, "|")[0], strings.Split(v, "|")[1]
		syslist = append(syslist, Sys{
			T_sys:  sys,
			T_name: name,
		})
	}

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: syslist}
	c.ServeJSON()
	return

}

func (c *PowerController) Edit_Menu() {

	T_id := c.GetString("T_id")
	T_code := c.GetString("T_code")
	T_menu := c.GetString("T_menu")

	if len(T_id) == 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	sysName := Account.Get_Sys_Name(T_code)
	if sysName == T_code {
		// 不存在这个code
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_code Err！"}
		c.ServeJSON()
		return
	}

	power, err := Account.Read_Power_ByT_id(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}
	var id int
	switch T_code {
	case conf.ERP_ACCOUNT_Sys:
		power.T_menu = T_menu
		id, err = Account.Update_Power(power)
		if err != nil {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
			c.ServeJSON()
			return
		}
	default:
		var power2 powerlibs.Power
		power2, err = NatsServer.ERP_Read_Power_ByT_id(T_code, T_id)
		if err != nil {
			if err.Error() == orm.ErrNoRows.Error() {
				var_ := powerlibs.Power{
					T_id:   power.T_id,
					T_name: power.T_name,
					T_menu: T_menu,
				}
				_, err = NatsServer.ERP_Add_Power(T_code, var_)
				if err != nil {
					c.Data["json"] = lib.JSONS{Code: 202, Msg: fmt.Sprintf("%s 修改失败!", sysName)}
					c.ServeJSON()
					return
				}
			} else {
				c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
				c.ServeJSON()
				return
			}
		}
		if power2.Id > 0 {
			power2.T_menu = T_menu
			id, err = NatsServer.ERP_Update_Power(T_code, power2)
			if err != nil {
				c.Data["json"] = lib.JSONS{Code: 202, Msg: fmt.Sprintf("%s 修改失败!", sysName)}
				c.ServeJSON()
				return
			}
		}
	}
	Account.Redis_Menu_DelK(T_id)
	System.Add_UserLogs_T(c.User.T_uuid, "权限", sysName+" 修改菜单", power)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: sysName + " ok!", Data: id}
	c.ServeJSON()
	return

}

func (c *PowerController) Del() {

	T_id := c.GetString("T_id")
	if len(T_id) == 0 {
		c.Data["json"] = lib.JSONS{Code: 201, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}
	R, err := Account.Read_Power_ByT_id(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	if R.Id == 1 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "禁止删除系统角色！"}
		c.ServeJSON()
		return
	}

	id, err := Account.Delete_Power(R)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
		c.ServeJSON()
		return
	}

	//for _, prefix := range Account.ERP_Sys_prefix {
	//	i := 3
	//	for {
	//		R, err = NatsServer.ERP_Read_Power_ByT_id(prefix, T_id)
	//		if err != nil {
	//			if err.Error() == orm.ErrNoRows.Error() {
	//				break
	//			}
	//			c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
	//			c.ServeJSON()
	//			return
	//		}
	//		// Nats 发送失败 重试3次
	//		_, err = NatsServer.ERP_Delete_Power(prefix, R)
	//		if err == nil || i == 0 {
	//			break
	//		}
	//		i--
	//	}
	//
	//	if err != nil {
	//		name := Account.Get_Sys_Name(prefix)
	//		c.Data["json"] = lib.JSONS{Code: 202, Msg: name + " 修改失败!"}
	//		c.ServeJSON()
	//		return
	//	}
	//}

	for _, prefix := range Account.ERP_Sys_prefix {
		go func(sys string) {
			var power powerlibs.Power
			i := 10
			for {
				power, err = NatsServer.ERP_Read_Power_ByT_id(sys, T_id)
				if err != nil {
					if err.Error() == orm.ErrNoRows.Error() {
						break
					}
					continue
				}
				// Nats 发送失败 重试3次
				_, err = NatsServer.ERP_Delete_Power(sys, power)
				if err == nil || i == 0 {
					break
				}
				i--
			}
		}(prefix)

	}

	System.Add_UserLogs_T(c.User.T_uuid, "权限", "删除", T_id)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: id}
	c.ServeJSON()
	return

}

// 规则：父菜单只有在其所有直接子菜单都被勾选时，才可被返回；
// 否则移除该父菜单。该规则会向上递归传播，确保多层级结构正确处理。
func filterParentMenusFromChecked(menu []menulibs.Menu, checkedIds []int) []int {
	// 先将树形菜单结构扁平化，提取所有菜单节点
	allMenus := flattenMenuTree(menu)

	// 构建父子关系映射
	parentToChildren := make(map[int][]int)
	for _, m := range allMenus {
		if m.T_mid > 0 {
			parentToChildren[m.T_mid] = append(parentToChildren[m.T_mid], m.Id)
		}
	}

	// 勾选集合
	checked := make(map[int]bool)
	for _, id := range checkedIds {
		checked[id] = true
	}

	// 计算叶子集合（无子节点的菜单）
	isLeaf := make(map[int]bool)
	for _, m := range allMenus {
		isLeaf[m.Id] = len(parentToChildren[m.Id]) == 0
	}

	// DFS 获取某父级下的所有叶子后代
	var getDescendantLeaves func(int, map[int]bool)
	getDescendantLeaves = func(id int, leaves map[int]bool) {
		children := parentToChildren[id]
		if len(children) == 0 {
			// 自身为叶子
			leaves[id] = true
			return
		}
		for _, child := range children {
			if len(parentToChildren[child]) == 0 {
				leaves[child] = true
			} else {
				getDescendantLeaves(child, leaves)
			}
		}
	}

	// 结果集合：
	// 1) 始终包含勾选的叶子节点
	// 2) 对于勾选的父级，仅当其所有叶子后代都被勾选时，才包含该父级
	include := make(map[int]bool)
	for id := range checked {
		if isLeaf[id] {
			include[id] = true
		}
	}

	for id := range checked {
		if isLeaf[id] {
			continue
		}
		leaves := make(map[int]bool)
		getDescendantLeaves(id, leaves)
		allLeavesChecked := true
		for leaf := range leaves {
			if !checked[leaf] {
				allLeavesChecked = false
				break
			}
		}
		if allLeavesChecked {
			include[id] = true
		}
	}

	// 输出结果
	result := make([]int, 0, len(include))
	for id := range include {
		result = append(result, id)
	}
	return result
}

// flattenMenuTree 将树形菜单结构扁平化为一维数组
func flattenMenuTree(menu []menulibs.Menu) []menulibs.Menu {
	result := make([]menulibs.Menu, 0)
	for _, m := range menu {
		result = append(result, m)
		if len(m.Children) > 0 {
			result = append(result, flattenMenuTree(m.Children)...)
		}
	}
	return result
}