yibu/server/model/sys_casbin.go

package model

import (
	"errors"
	"gin-vue-admin/global"
	"github.com/casbin/casbin"
	"github.com/casbin/casbin/util"
	gormadapter "github.com/casbin/gorm-adapter"
	"strings"
)

type CasbinModel struct {
	ID          uint   `json:"id" gorm:"column:_id"`
	Ptype       string `json:"ptype" gorm:"column:ptype"`
	AuthorityId string `json:"rolename" gorm:"column:v0"`
	Path        string `json:"path" gorm:"column:v1"`
	Method      string `json:"method" gorm:"column:v2"`
}

// 供入参使用
type CasbinInfo struct {
	Path   string `json:"path"`
	Method string `json:"method"`
}

// 供入参使用
type CasbinInReceive struct {
	AuthorityId string       `json:"authorityId"`
	CasbinInfos []CasbinInfo `json:"casbinInfos"`
}

// @title    UpdateCasbin
// @description   update casbin authority, 更新casbin权限
// @auth                     （2020/04/05  20:22 ）
// @param     authorityId      string
// @param     casbinInfos      []CasbinInfo
// @return                     error
func (c *CasbinModel) UpdateCasbin(authorityId string, casbinInfos []CasbinInfo) error {
	c.ClearCasbin(0, authorityId)
	for _, v := range casbinInfos {
		cm := CasbinModel{
			ID:          0,
			Ptype:       "p",
			AuthorityId: authorityId,
			Path:        v.Path,
			Method:      v.Method,
		}
		addflag := c.AddCasbin(cm)
		if addflag == false {
			return errors.New("存在相同api,添加失败,请联系管理员")
		}
	}
	return nil
}

// @title    UpdateCasbinApi
// @description   update casbin apis, API更新随动
// @auth                     （2020/04/05  20:22 ）
// @param     oldPath          string
// @param     newPath          string
// @return                     error
func (c *CasbinModel) UpdateCasbinApi(oldPath string, newPath string) error {
	var cs []CasbinModel
	err := global.GVA_DB.Table("casbin_rule").Where("v1 = ?", oldPath).Find(&cs).Update("v1", newPath).Error
	return err
}

// @title    AddCasbin
// @description   add casbin authority, 添加权限
// @auth                     （2020/04/05  20:22 ）
// @param     cm              CasbinModel
// @return                    bool
func (c *CasbinModel) AddCasbin(cm CasbinModel) bool {
	e := Casbin()
	return e.AddPolicy(cm.AuthorityId, cm.Path, cm.Method)
}

// @title    GetPolicyPathByAuthorityId
// @description   get policy path by authorityId, 获取权限列表
// @auth                     （2020/04/05  20:22 ）
// @param     authorityId     string
// @return                    []string
func (c *CasbinModel) GetPolicyPathByAuthorityId(authorityId string) []string {
	e := Casbin()
	var pathList []string
	list := e.GetFilteredPolicy(0, authorityId)
	for _, v := range list {
		pathList = append(pathList, v[1])
	}
	return pathList
}

// @title    ClearCasbin
// @description   清除匹配的权限
// @auth                     （2020/04/05  20:22 ）
// @param     v               int
// @param     p               string
// @return                    bool
func (c *CasbinModel) ClearCasbin(v int, p string) bool {
	e := Casbin()
	return e.RemoveFilteredPolicy(v, p)

}

// @title    ParamsMatch
// @description   customized rule, 自定义规则函数
// @auth                     （2020/04/05  20:22 ）
// @param     fullNameKey1    string
// @param     key2            string
// @return                    bool
func ParamsMatch(fullNameKey1 string, key2 string) bool {
	key1 := strings.Split(fullNameKey1, "?")[0]
	//剥离路径后再使用casbin的keyMatch2
	return util.KeyMatch2(key1, key2)
}

// @title    ParamsMatchFunc
// @description   customized function, 自定义规则函数
// @auth                     （2020/04/05  20:22 ）
// @param     args            ...interface{}
// @return                    interface{}
// @return                    error
func ParamsMatchFunc(args ...interface{}) (interface{}, error) {
	name1 := args[0].(string)
	name2 := args[1].(string)

	return (bool)(ParamsMatch(name1, name2)), nil
}

// @title    Casbin
// @description   store to DB, 持久化到数据库  引入自定义规则
// @auth                     （2020/04/05  20:22 ）
func Casbin() *casbin.Enforcer {
	a := gormadapter.NewAdapterByDB(global.GVA_DB)
	e := casbin.NewEnforcer(global.GVA_CONFIG.Casbin.ModelPath, a)
	e.AddFunction("ParamsMatch", ParamsMatchFunc)
	_ = e.LoadPolicy()
	return e
}