mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
package tool
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"github.com/PandaXGO/PandaKit/biz"
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
"pandax/apps/system/entity"
|
|
"pandax/apps/system/services"
|
|
"pandax/pkg/global"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type DeviceAuth struct {
|
|
Owner string `json:"owner"`
|
|
OrgId int64 `json:"orgId"`
|
|
DeviceId string `json:"device_id"`
|
|
DeviceType string `json:"device_type"`
|
|
ProductId string `json:"product_id"`
|
|
RuleChainId string `json:"rule_chain_id"`
|
|
Name string `json:"name"`
|
|
Token string `json:"token"`
|
|
CreatedAt int64 `json:"created_at"`
|
|
ExpiredAt int64 `json:"expired_at"`
|
|
}
|
|
|
|
func (entity *DeviceAuth) CreateDeviceToken() (err error) {
|
|
|
|
return nil
|
|
}
|
|
|
|
func (entity *DeviceAuth) GetDeviceToken(key string) error {
|
|
if err := global.RedisDb.Get(key, entity); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (token *DeviceAuth) MD5ID() string {
|
|
buf := bytes.NewBufferString(token.DeviceId)
|
|
buf.WriteString(token.DeviceType)
|
|
buf.WriteString(strconv.FormatInt(token.CreatedAt, 10))
|
|
access := base64.URLEncoding.EncodeToString([]byte(uuid.NewMD5(uuid.Must(uuid.NewRandom()), buf.Bytes()).String()))
|
|
access = strings.TrimRight(access, "=")
|
|
return access
|
|
}
|
|
func (token *DeviceAuth) GetMarshal() string {
|
|
marshal, _ := json.Marshal(*token)
|
|
return string(marshal)
|
|
}
|
|
|
|
func (token *DeviceAuth) GetUnMarshal(data []byte) error {
|
|
return json.Unmarshal(data, token)
|
|
}
|
|
|
|
// 序列化
|
|
func (m *DeviceAuth) MarshalBinary() (data []byte, err error) {
|
|
return json.Marshal(m)
|
|
}
|
|
|
|
// 反序列化
|
|
func (m *DeviceAuth) UnmarshalBinary(data []byte) error {
|
|
return json.Unmarshal(data, m)
|
|
}
|
|
|
|
func OrgAuthSet(tx *gorm.DB, roleId int64, owner string) {
|
|
//TODO 使用缓存
|
|
role, err := services.SysRoleModelDao.FindOrganizationsByRoleId(roleId)
|
|
biz.ErrIsNil(err, "查询角色数据权限失败")
|
|
if role.DataScope != entity.SELFDATASCOPE {
|
|
biz.IsTrue(len(role.Org) > 0, "该角色下未分配组织权限")
|
|
tx.Where("org_id in (?)", strings.Split(role.Org, ","))
|
|
} else {
|
|
tx.Where("owner = ?", owner)
|
|
}
|
|
|
|
}
|