73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
|
package logic
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
|
|||
|
|
"github.com/gogf/gf/v2/frame/g"
|
|||
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|||
|
|
|
|||
|
|
v1 "tool-api/api/user/v1"
|
|||
|
|
"tool-api/internal/consts"
|
|||
|
|
"tool-api/internal/model/entity"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ===== 等级有效期（会员）判定 =====
|
|||
|
|
//
|
|||
|
|
// 背景：users.level_key 是单值，无法表达「会员到期」。引入 users.level_expire_at 后，
|
|||
|
|
// 所有依赖等级的地方都必须先经过 EffectiveLevelKey 收敛，否则会出现「会员已过期但
|
|||
|
|
// 工具仍可用」的越权问题。
|
|||
|
|
//
|
|||
|
|
// 约定：
|
|||
|
|
// - level_expire_at IS NULL → 永久有效（管理员手动分配 / 种子默认等级）
|
|||
|
|
// - level_expire_at <= now → 已过期，回落 consts.DefaultLevelKey
|
|||
|
|
// - level_expire_at > now → 正常返回 users.level_key
|
|||
|
|
|
|||
|
|
// EffectiveLevelKey 计算用户当前实际生效的等级 key。一处判定，全处复用。
|
|||
|
|
func EffectiveLevelKey(user *entity.Users) string {
|
|||
|
|
if user == nil || user.LevelKey == "" {
|
|||
|
|
return consts.DefaultLevelKey
|
|||
|
|
}
|
|||
|
|
if IsLevelExpired(user) {
|
|||
|
|
return consts.DefaultLevelKey
|
|||
|
|
}
|
|||
|
|
return user.LevelKey
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsLevelExpired 等级是否已过期
|
|||
|
|
func IsLevelExpired(user *entity.Users) bool {
|
|||
|
|
if user == nil || user.LevelExpireAt == nil || user.LevelExpireAt.IsZero() {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
return user.LevelExpireAt.Before(gtime.Now())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// EffectiveLevelInfo 当前生效等级（含已授权模块），已过期时返回默认等级
|
|||
|
|
func EffectiveLevelInfo(ctx context.Context, user *entity.Users) (*v1.LevelInfo, error) {
|
|||
|
|
return GetLevelByKey(ctx, EffectiveLevelKey(user))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// levelWeight 等级权限权重，取 levels.sort（越大权限越高）。
|
|||
|
|
// 查询失败时回落 0，保证不因权重查询异常而误升权。
|
|||
|
|
func levelWeight(ctx context.Context, levelKey string) int {
|
|||
|
|
if levelKey == "" {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
value, err := g.Model(consts.TableLevels).
|
|||
|
|
Where("level_key", levelKey).
|
|||
|
|
Where("is_enabled", 1).
|
|||
|
|
Value("sort")
|
|||
|
|
if err != nil || value == nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return value.Int()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CanLevelAccess 判断某个等级是否包含指定模块（供发货后自检 / 管理端复用）
|
|||
|
|
func CanLevelAccess(ctx context.Context, levelKey, moduleKey string) bool {
|
|||
|
|
level, err := GetLevelByKey(ctx, levelKey)
|
|||
|
|
if err != nil {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
return slicesContain(level.Modules, moduleKey)
|
|||
|
|
}
|