package controllers

import (
	"encoding/json"
	"oppmg/common/log"
	"oppmg/models"
	"oppmg/protocol"
	serverbac "oppmg/services/rbac"
)

// RbacController 角色权限
type RbacController struct {
	BaseController
}

//URLMapping 实现ControllerInterface中的URLMapping
func (c *RbacController) URLMapping() {
	//c.Mapping("AccessToken", c.AccessToken)
}

// RoleAdd 添加部门
// @router /role [post]
func (c *RbacController) RoleAdd() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestRoleAdd
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.Types = models.ROLETYPES_ROLE
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}
	n := []rune(param.Name)
	if len(n) > 10 {
		msg = protocol.BadRequestParam("10008")
		return
	}
	if len(n) <= 0 {
		msg = protocol.BadRequestParam("10009")
		return
	}
	r, err := serverbac.RoleAdd(param)
	msg = protocol.NewReturnResponse(r, err)
	return
}

// RoleUpdate 编辑部门
// @router /role/edit [post]
func (c *RbacController) RoleUpdate() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestRoleEdit
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}
	n := []rune(param.Name)
	if len(n) > 10 {
		msg = protocol.BadRequestParam("10008")
		return
	}
	if len(n) <= 0 {
		msg = protocol.BadRequestParam("10009")
		return
	}
	param.Types = models.ROLETYPES_ROLE
	r, err := serverbac.RoleEdit(param)
	msg = protocol.NewReturnResponse(r, err)
	return
}

// RoleDelete 删除角色
// @router /role[delete]
func (c *RbacController) RoleDelete() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestRoleDelete
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}

	err := serverbac.RoleDelete(param)

	msg = protocol.NewReturnResponse(nil, err)
	return
}

// RoleGroupAdd 角色组
// @router /role_group [post]
func (c *RbacController) RoleGroupAdd() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestRoleAdd
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}
	n := []rune(param.Name)
	if len(n) > 10 {
		msg = protocol.BadRequestParam("10007")
		return
	}
	if len(n) <= 0 {
		msg = protocol.BadRequestParam("10081")
		return
	}
	param.Pid = 0
	param.Types = models.ROLETYPES_GROUP
	roleinfo, err := serverbac.RoleAdd(param)
	msg = protocol.NewReturnResponse(roleinfo, err)
	return
}

// RoleGroupEdit 角色组
// @router /role_group [put]
func (c *RbacController) RoleGroupUpdate() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestRoleEdit
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}
	n := []rune(param.Name)
	if len(n) > 10 {
		msg = protocol.BadRequestParam("10007")
		return
	}
	if len(n) <= 0 {
		msg = protocol.BadRequestParam("10081")
		return
	}
	param.Pid = 0
	param.Types = models.ROLETYPES_GROUP
	roleinfo, err := serverbac.RoleEdit(param)
	msg = protocol.NewReturnResponse(roleinfo, err)
	return
}

// RoleGroupEdit 角色组
// @router /role_group [delete]
func (c *RbacController) RoleGroupDelete() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestRoleDelete
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}

	err := serverbac.RoleGroupDelete(param)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

// RoleList ...
// @router /rbac/role/list [post]
func (c *RbacController) RoleList() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var companyid int64
	companyid = c.GetCompanyId()
	if companyid <= 0 {
		msg = protocol.BadRequestParam("1")
		return
	}
	userid := c.GetUserId()
	rolelist, err := serverbac.GetRoleList(companyid, userid)
	msg = protocol.NewReturnResponse(rolelist, err)
	return
}

//PermissionList
//@router  /menu/list [post]
func (c *RbacController) MenuList() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	list, err := serverbac.GetMenuAll()
	data := protocol.ResponsePermissionList{
		Lists: list,
	}
	msg = protocol.NewReturnResponse(data, err)
	return
}

//GetRoleHasMenu  获取角色下的菜单
//@router /role/menu
func (c *RbacController) RoleHasMenu() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Parameter struct {
		RoleId int64 `json:"role_id"`
	}
	var param Parameter
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	companyid := c.GetCompanyId()
	list, err := serverbac.GetRoleHasMenu(param.RoleId, companyid)
	msg = protocol.NewReturnResponse(list, err)
	return
}

//TODO
func (c *RbacController) GetRoleMenuAll() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	log.Info("====>")
	list, err := serverbac.GetRoleHasMenuAll(1, 1)
	msg = protocol.NewReturnResponse(list, err)
	return
}

//RoleMenuEdit 设置角色的菜单
//@router /role/menu/edit
func (c *RbacController) RoleMenuEdit() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Parameter struct {
		RoleId  int64   `json:"role_id"`
		MenuIds []int64 `json:"menu_ids"`
	}
	var param Parameter
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	companyid := c.GetCompanyId()
	err := serverbac.RoleMenuEdit(companyid, param.RoleId, param.MenuIds)
	msg = protocol.NewReturnResponse(nil, err)
	return
}