construction_management/web/middleware/accessAuth.go

/*
 * @description: 访问权限认证
 * @Author: CP
 * @Date: 2020-10-09 10:43:39
 * @FilePath: \construction_management\web\middleware\accessAuth.go
 */
package middleware

import (
	"errors"

	"github.com/kataras/iris/v12"
)

// 员工表权限解析
type permission struct {
	Add    int ` json:"add" `
	Delete int ` json:"delete" `
	Access int ` json:"access" `
}

// 权限JSON对象
type permissionPath struct {
	Contract       path     `json:"contract" `
	Safe           path     `json:"safe" `
	Quality        path     `json:"quality" `
	ProjectSetting []string `json:"projectSetting" `
}

// 对象中地址列表
type path struct {
	Add    []string `json:"add" `
	Access []string `json:"access" `
	Delete []string `json:"delete" `
}

// 权限验证中间件
func AccessAuth(ctx iris.Context) {
	// // 1.获得成员信息
	// account := ctx.Values().Get("account").(*models.CmProjectAccount)
	// // 1-1.是管理员- 拥有所有权限
	// if account.IsAdmin != 1 {
	// 	// if account.IsAdmin == 1 {
	// 	// 2.获得员工可访问的权限
	// 	contractPermission := permission{}
	// 	if account.ContractPermission != "" {
	// 		err := json.Unmarshal([]byte(account.ContractPermission), &contractPermission)
	// 		// 错误后 全部权限默认为0
	// 		if err != nil {
	// 			log.Println("合同权限解析错误:err=", err)
	// 		}
	// 	}
	// 	safePermission := permission{}
	// 	if account.SafePermission != "" {
	// 		err := json.Unmarshal([]byte(account.SafePermission), &safePermission)
	// 		if err != nil {
	// 			log.Println("安全权限解析错误:err=", err)
	// 		}
	// 	}
	// 	qualityPermission := permission{}
	// 	if account.QualityPermission != "" {
	// 		err := json.Unmarshal([]byte(account.QualityPermission), &qualityPermission)
	// 		if err != nil {
	// 			log.Println("质量权限解析错误:err=", err)
	// 		}
	// 	}

	// 	// 2-1获得权限列表
	// 	permissionPath := permissionPath{}
	// 	permissionPathData, err := ioutil.ReadFile("../lib/permission.json")
	// 	err = json.Unmarshal(permissionPathData, &permissionPath)
	// 	if err != nil {
	// 		log.Println("权限解析错误:err=", err)
	// 		ctx.JSON(iris.Map{"code": 2, "msg": "权限解析错误"})
	// 		return
	// 	}
	// 	// 2-2 不容许访问的权限--比对访问路径
	// 	requestPath := ctx.Path()
	// 	// 合同权限
	// 	err = verifyAuth(contractPermission, permissionPath.Contract, requestPath)
	// 	if err != nil {
	// 		ctx.JSON(iris.Map{"code": 2, "msg": fmt.Sprintf("%s", err)})
	// 		return
	// 	}
	// 	// 安全权限
	// 	err = verifyAuth(contractPermission, permissionPath.Safe, requestPath)
	// 	if err != nil {
	// 		ctx.JSON(iris.Map{"code": 2, "msg": fmt.Sprintf("%s", err)})
	// 		return
	// 	}
	// 	// 质量权限
	// 	err = verifyAuth(contractPermission, permissionPath.Quality, requestPath)
	// 	if err != nil {
	// 		ctx.JSON(iris.Map{"code": 2, "msg": fmt.Sprintf("%s", err)})
	// 		return
	// 	}
	// 	// 项目设置 -只有管理员才能访问
	// 	for _, path := range permissionPath.ProjectSetting {
	// 		if path == requestPath {
	// 			ctx.JSON(iris.Map{"code": 2, "msg": "无权访问"})
	// 			return
	// 		}
	// 	}

	// }

	ctx.Next()
}

// 验证权限路径
func verifyAuth(permission permission, pathList path, requestPath string) error {
	if permission.Add == 0 {
		for _, path := range pathList.Add {
			if path == requestPath {
				return errors.New("无权访问")
			}
		}
	}
	if permission.Access == 0 {
		for _, path := range pathList.Access {
			if path == requestPath {
				return errors.New("无权访问")
			}
		}
	}
	if permission.Delete == 0 {
		for _, path := range pathList.Delete {
			if path == requestPath {
				return errors.New("无权访问")
			}
		}
	}
	return nil
}