ERP_user/controllers/Power.go

package controllers

import (
	"ERP_user/Nats/NatsServer"
	"ERP_user/conf"
	"ERP_user/models/Account"
	"ERP_user/models/System"
	"fmt"
	menulibs "git.baozhida.cn/ERP_libs/Menu"
	powerlibs "git.baozhida.cn/ERP_libs/Power"
	"git.baozhida.cn/ERP_libs/lib"
	"github.com/astaxie/beego/logs"
	"github.com/beego/beego/v2/adapter/orm"

	beego "github.com/beego/beego/v2/server/web"
	"math"
)

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
	if power.T_menu != "*" {
		data.Menu_checked = lib.SplitStringToIntIds(power.T_menu, "M")
	}

	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
	}

	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
	}

	power.T_name = T_name

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

	// 向除开用户系统的其他系统修改权限
	// 查询新系统是否存在与用户系统powerT_id相同的power 存在=>修改 不存在=>添加
	for _, prefix := range Account.ERP_Sys_prefix {
		// Nats 发送失败 重试3次
		i := 3
		for {
			power, err = NatsServer.ERP_Read_Power_ByT_id(prefix, T_id)
			if err != nil {
				if err.Error() == orm.ErrNoRows.Error() {
					var_ := powerlibs.Power{
						T_id:   power.T_id,
						T_name: T_name,
					}
					_, err = NatsServer.ERP_Add_Power(prefix, var_)
					if err == nil || i == 0 {
						break
					}
				} else {
					c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
					c.ServeJSON()
					return
				}
			}
			if power.Id > 0 {
				power.T_name = T_name
				_, err = NatsServer.ERP_Update_Power(prefix, power)
				if err == nil || i == 0 {
					break
				}
			}
			i--
		}

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

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

	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 list []Sys
	for k, v := range conf.ERP_Sys_List {
		list = append(list, Sys{
			T_sys:  k,
			T_name: v,
		})
	}

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: list}
	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 int64
	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:
		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,
					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 power.Id > 0 {
			power.T_menu = T_menu
			id, err = NatsServer.ERP_Update_Power(T_code, power)
			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
	}

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

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

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

}