wehub-resource-sync
358 lines
9.8 KiB
Go
358 lines
9.8 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/phachon/mm-wiki/app/models"
|
|
"github.com/phachon/mm-wiki/app/utils"
|
|
"strings"
|
|
)
|
|
|
|
type RoleController struct {
|
|
BaseController
|
|
}
|
|
|
|
func (this *RoleController) Add() {
|
|
this.viewLayout("role/form", "role")
|
|
}
|
|
|
|
func (this *RoleController) Save() {
|
|
|
|
if !this.IsPost() {
|
|
this.ViewError("请求方式有误！", "/system/role/list")
|
|
}
|
|
name := strings.TrimSpace(this.GetString("name", ""))
|
|
if name == "" {
|
|
this.jsonError("角色名称不能为空！")
|
|
}
|
|
|
|
ok, err := models.RoleModel.HasRoleName(name)
|
|
if err != nil {
|
|
this.ErrorLog("添加角色失败：" + err.Error())
|
|
this.jsonError("添加角色失败！")
|
|
}
|
|
if ok {
|
|
this.jsonError("角色名已经存在！")
|
|
}
|
|
|
|
roleId, err := models.RoleModel.Insert(map[string]interface{}{
|
|
"name": name,
|
|
})
|
|
|
|
if err != nil {
|
|
this.ErrorLog("添加角色失败：" + err.Error())
|
|
this.jsonError("添加角色失败")
|
|
}
|
|
this.InfoLog("添加角色 " + utils.Convert.IntToString(roleId, 10) + " 成功")
|
|
this.jsonSuccess("添加角色成功", nil, "/system/role/list")
|
|
}
|
|
|
|
func (this *RoleController) List() {
|
|
|
|
page, _ := this.GetInt("page", 1)
|
|
keyword := strings.TrimSpace(this.GetString("keyword", ""))
|
|
number, _ := this.GetRangeInt("number", 20, 10, 100)
|
|
limit := (page - 1) * number
|
|
|
|
var err error
|
|
var count int64
|
|
var roles []map[string]string
|
|
if keyword != "" {
|
|
count, err = models.RoleModel.CountRolesByKeyword(keyword)
|
|
roles, err = models.RoleModel.GetRolesByKeywordAndLimit(keyword, limit, number)
|
|
} else {
|
|
count, err = models.RoleModel.CountRoles()
|
|
roles, err = models.RoleModel.GetRolesByLimit(limit, number)
|
|
}
|
|
if err != nil {
|
|
this.ErrorLog("获取角色列表失败: " + err.Error())
|
|
this.ViewError("获取角色列表失败", "/system/main/index")
|
|
}
|
|
|
|
this.Data["roles"] = roles
|
|
this.Data["keyword"] = keyword
|
|
this.SetPaginator(number, count)
|
|
this.viewLayout("role/list", "role")
|
|
}
|
|
|
|
func (this *RoleController) Edit() {
|
|
|
|
roleId := this.GetString("role_id", "")
|
|
if roleId == "" {
|
|
this.ViewError("角色不存在", "/system/role/list")
|
|
}
|
|
if roleId == fmt.Sprintf("%d", models.Role_Root_Id) {
|
|
this.ViewError("不能修改超级管理员", "/system/role/list")
|
|
}
|
|
|
|
role, err := models.RoleModel.GetRoleByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("查找角色错误：" + err.Error())
|
|
this.ViewError("查找角色错误", "/system/role/list")
|
|
}
|
|
if len(role) == 0 {
|
|
this.ViewError("角色不存在", "/system/role/list")
|
|
}
|
|
|
|
this.Data["role"] = role
|
|
this.viewLayout("role/form", "role")
|
|
}
|
|
|
|
func (this *RoleController) Modify() {
|
|
|
|
if !this.IsPost() {
|
|
this.ViewError("请求方式有误！", "/system/role/list")
|
|
}
|
|
roleId := this.GetString("role_id", "")
|
|
name := strings.TrimSpace(this.GetString("name", ""))
|
|
|
|
if roleId == "" {
|
|
this.jsonError("角色不存在！")
|
|
}
|
|
if roleId == fmt.Sprintf("%d", models.Role_Root_Id) {
|
|
this.jsonError("超级管理员不能修改！")
|
|
}
|
|
if name == "" {
|
|
this.jsonError("角色名称不能为空！")
|
|
}
|
|
|
|
role, err := models.RoleModel.GetRoleByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("修改角色 " + roleId + " 失败: " + err.Error())
|
|
this.jsonError("修改角色失败！")
|
|
}
|
|
if len(role) == 0 {
|
|
this.jsonError("角色不存在！")
|
|
}
|
|
if role["role_id"] == fmt.Sprintf("%d", models.Role_Root_Id) {
|
|
this.jsonError("超级管理员角色不能修改！")
|
|
}
|
|
|
|
ok, _ := models.RoleModel.HasSameName(roleId, name)
|
|
if ok {
|
|
this.jsonError("角色名已经存在！")
|
|
}
|
|
_, err = models.RoleModel.Update(roleId, map[string]interface{}{
|
|
"name": name,
|
|
})
|
|
|
|
if err != nil {
|
|
this.ErrorLog("修改角色 " + roleId + " 失败：" + err.Error())
|
|
this.jsonError("修改角色" + roleId + "失败")
|
|
}
|
|
this.InfoLog("修改角色 " + roleId + " 成功")
|
|
this.jsonSuccess("修改角色成功", nil, "/system/role/list")
|
|
}
|
|
|
|
func (this *RoleController) User() {
|
|
|
|
keywords := map[string]string{}
|
|
page, _ := this.GetInt("page", 1)
|
|
roleId := strings.TrimSpace(this.GetString("role_id", ""))
|
|
|
|
if roleId == "" {
|
|
this.ViewError("没有选择角色！")
|
|
}
|
|
keywords["role_id"] = roleId
|
|
|
|
number, _ := this.GetRangeInt("number", 15, 10, 100)
|
|
limit := (page - 1) * number
|
|
var err error
|
|
var count int64
|
|
var users []map[string]string
|
|
count, err = models.UserModel.CountUsersByKeywords(keywords)
|
|
if err != nil {
|
|
this.ErrorLog("获取角色用户列表失败: " + err.Error())
|
|
this.ViewError("获取角色用户列表失败！", "/system/role/list")
|
|
}
|
|
users, err = models.UserModel.GetUsersByKeywordsAndLimit(keywords, limit, number)
|
|
if err != nil {
|
|
this.ErrorLog("获取用户列表失败: " + err.Error())
|
|
this.ViewError("获取用户列表失败！", "/system/role/list")
|
|
}
|
|
|
|
role, err := models.RoleModel.GetRoleByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("获取用户列表失败: " + err.Error())
|
|
this.ViewError("获取角色用户列表失败！", "/system/main/index")
|
|
}
|
|
for _, user := range users {
|
|
user["role_name"] = role["name"]
|
|
}
|
|
|
|
this.Data["users"] = users
|
|
this.Data["roleId"] = roleId
|
|
this.SetPaginator(number, count)
|
|
this.viewLayout("role/user", "role")
|
|
}
|
|
|
|
func (this *RoleController) Privilege() {
|
|
|
|
roleId := this.GetString("role_id", "")
|
|
if roleId == "" {
|
|
this.ViewError("角色不存在", "/system/role/list")
|
|
}
|
|
|
|
role, err := models.RoleModel.GetRoleByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("查找角色权限失败：" + err.Error())
|
|
this.ViewError("查看角色权限失败！", "/system/role/list")
|
|
}
|
|
if len(role) == 0 {
|
|
this.ViewError("角色不存在", "/system/role/list")
|
|
}
|
|
|
|
menus, controllers, err := models.PrivilegeModel.GetTypePrivileges()
|
|
if err != nil {
|
|
this.ViewError("查找角色权限失败！")
|
|
}
|
|
|
|
var rolePrivileges = []map[string]string{}
|
|
if role["role_id"] == fmt.Sprintf("%d", models.Role_Root_Id) {
|
|
rolePrivileges, err = models.RolePrivilegeModel.GetRootRolePrivileges()
|
|
} else {
|
|
rolePrivileges, err = models.RolePrivilegeModel.GetRolePrivilegesByRoleId(roleId)
|
|
}
|
|
if err != nil {
|
|
this.ViewError("查找用户权限出错")
|
|
}
|
|
|
|
this.Data["role"] = role
|
|
this.Data["menus"] = menus
|
|
this.Data["controllers"] = controllers
|
|
this.Data["rolePrivileges"] = rolePrivileges
|
|
this.Data["disabledPrivilegeIds"] = models.Privilege_Default_Ids
|
|
|
|
this.viewLayout("role/privilege", "role")
|
|
}
|
|
|
|
func (this *RoleController) GrantPrivilege() {
|
|
|
|
if !this.IsPost() {
|
|
this.ViewError("请求方式有误！", "/system/role/list")
|
|
}
|
|
privilegeIds := this.GetStrings("privilege_id", []string{})
|
|
roleId := this.GetString("role_id", "")
|
|
|
|
if roleId == "" {
|
|
this.jsonError("没有选择角色!")
|
|
}
|
|
//if len(privilegeIds) == 0 {
|
|
// this.jsonError("没有选择权限!")
|
|
//}
|
|
// add default privileges
|
|
privilegeIds = append(privilegeIds, models.Privilege_Default_Ids...)
|
|
|
|
role, err := models.RoleModel.GetRoleByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("角色 " + roleId + " 授权失败：" + err.Error())
|
|
this.jsonError("角色不存在")
|
|
}
|
|
if len(role) == 0 {
|
|
this.jsonError("角色不存在")
|
|
}
|
|
|
|
if role["role_id"] == fmt.Sprintf("%d", models.Role_Root_Id) {
|
|
this.jsonError("超级管理员不需要授权！")
|
|
}
|
|
|
|
res, err := models.RolePrivilegeModel.GrantRolePrivileges(roleId, privilegeIds)
|
|
if err != nil {
|
|
this.ErrorLog("角色 " + roleId + " 授权失败：" + err.Error())
|
|
this.jsonError("角色授权失败！")
|
|
}
|
|
if !res {
|
|
this.jsonError("角色授权失败")
|
|
}
|
|
|
|
this.InfoLog("角色 " + roleId + " 授权成功")
|
|
this.jsonSuccess("角色授权成功", nil, "/system/role/list")
|
|
}
|
|
|
|
func (this *RoleController) Delete() {
|
|
if !this.IsPost() {
|
|
this.ViewError("请求方式有误！", "/system/role/list")
|
|
}
|
|
|
|
roleId := this.GetString("role_id", "")
|
|
if roleId == "" {
|
|
this.jsonError("没有选择角色！")
|
|
}
|
|
if roleId == fmt.Sprintf("%d", models.Role_Root_Id) {
|
|
this.jsonError("超级管理员不能删除！")
|
|
}
|
|
|
|
role, err := models.RoleModel.GetRoleByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("删除角色 " + roleId + " 失败: " + err.Error())
|
|
this.jsonError("删除角色失败")
|
|
}
|
|
if len(role) == 0 {
|
|
this.jsonError("角色不存在")
|
|
}
|
|
if role["type"] == fmt.Sprintf("%d", models.Role_Type_System) {
|
|
this.jsonError("系统角色不能删除！")
|
|
}
|
|
|
|
// check role user
|
|
users, err := models.UserModel.GetUsersByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("删除角色 " + roleId + " 失败: " + err.Error())
|
|
this.jsonError("删除角色失败")
|
|
}
|
|
if len(users) > 0 {
|
|
this.jsonError("不能删除角色，请先移除该角色下用户!")
|
|
}
|
|
|
|
// delete role privilege by role id
|
|
err = models.RolePrivilegeModel.DeleteByRoleId(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("删除角色 " + roleId + " 权限失败: " + err.Error())
|
|
this.jsonError("删除角色失败")
|
|
}
|
|
|
|
// delete role by role id
|
|
err = models.RoleModel.Delete(roleId)
|
|
if err != nil {
|
|
this.ErrorLog("删除角色 " + roleId + " 失败: " + err.Error())
|
|
this.jsonError("删除角色失败")
|
|
}
|
|
|
|
this.InfoLog("删除角色 " + roleId + " 成功")
|
|
this.jsonSuccess("删除角色成功", nil, "/system/role/list")
|
|
}
|
|
|
|
func (this *RoleController) ResetUser() {
|
|
|
|
if !this.IsPost() {
|
|
this.ViewError("请求方式有误！", "/system/role/list")
|
|
}
|
|
userId := this.GetString("user_id", "")
|
|
if userId == "" {
|
|
this.jsonError("用户不存在")
|
|
}
|
|
|
|
if this.UserId == "1" {
|
|
this.jsonError("root 用户不能重置角色！")
|
|
}
|
|
|
|
user, err := models.UserModel.GetUserByUserId(userId)
|
|
if err != nil {
|
|
this.ErrorLog("重置用户 " + userId + " 角色失败: " + err.Error())
|
|
this.jsonError("重置用户角色失败")
|
|
}
|
|
if len(user) == 0 {
|
|
this.jsonError("用户不存在")
|
|
}
|
|
|
|
_, err = models.UserModel.Update(userId, map[string]interface{}{
|
|
"role_id": models.Role_Default_Id,
|
|
})
|
|
if err != nil {
|
|
this.ErrorLog("重置用户 " + userId + " 角色失败: " + err.Error())
|
|
this.jsonError("重置用户角色失败")
|
|
}
|
|
|
|
this.InfoLog("重置用户 " + userId + " 角色成功")
|
|
this.jsonSuccess("重置用户角色成功", nil, "/system/role/list")
|
|
}
|