perms-system-server/internal/logic/user/setUserPermsLogic.go

package user

import (
	"context"
	"fmt"
	"strings"
	"time"

	"perms-system-server/internal/consts"
	authHelper "perms-system-server/internal/logic/auth"
	"perms-system-server/internal/middleware"
	"perms-system-server/internal/model/userperm"
	"perms-system-server/internal/response"
	"perms-system-server/internal/svc"
	"perms-system-server/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type SetUserPermsLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewSetUserPermsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SetUserPermsLogic {
	return &SetUserPermsLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

// SetUserPerms 设置用户个性化权限。对指定用户在当前产品下做权限全量覆盖，支持 ALLOW（附加）和 DENY（拒绝）两种效果，用于角色权限之外的细粒度调整。
func (l *SetUserPermsLogic) SetUserPerms(req *types.SetPermsReq) error {
	targetUser, err := l.svcCtx.SysUserModel.FindOne(l.ctx, req.UserId)
	if err != nil {
		return response.ErrNotFound("用户不存在")
	}

	productCode := middleware.GetProductCode(l.ctx)

	if err := authHelper.RequireProductAdminFor(l.ctx, productCode); err != nil {
		return err
	}

	product, err := l.svcCtx.SysProductModel.FindOneByCode(l.ctx, productCode)
	if err != nil {
		return response.ErrNotFound("产品不存在")
	}
	if product.Status != consts.StatusEnabled {
		return response.ErrBadRequest("产品已被禁用，无法设置权限")
	}

	if err := authHelper.CheckManageAccess(l.ctx, l.svcCtx, req.UserId, productCode, authHelper.WithPrefetchedTarget(targetUser)); err != nil {
		return err
	}

	member, memberErr := l.svcCtx.SysProductMemberModel.FindOneByProductCodeUserId(l.ctx, productCode, req.UserId)
	if memberErr != nil {
		return response.ErrBadRequest("目标用户不是当前产品的成员")
	}
	if member.Status != consts.StatusEnabled {
		return response.ErrBadRequest("目标用户的成员资格已被禁用")
	}

	for _, p := range req.Perms {
		if p.Effect != consts.PermEffectAllow && p.Effect != consts.PermEffectDeny {
			return response.ErrBadRequest("effect值无效，仅支持 ALLOW 和 DENY")
		}
	}

	perms := req.Perms
	if len(perms) > 0 {
		seen := make(map[int64]string, len(perms))
		uniquePerms := make([]types.UserPermItem, 0, len(perms))
		for _, p := range perms {
			if prev, ok := seen[p.PermId]; ok {
				if prev != p.Effect {
					return response.ErrBadRequest("同一权限ID不能同时为 ALLOW 和 DENY")
				}
				continue
			}
			seen[p.PermId] = p.Effect
			uniquePerms = append(uniquePerms, p)
		}
		perms = uniquePerms
	}

	if len(perms) > 0 {
		permIds := make([]int64, 0, len(perms))
		for _, p := range perms {
			permIds = append(permIds, p.PermId)
		}
		dbPerms, err := l.svcCtx.SysPermModel.FindByIds(l.ctx, permIds)
		if err != nil {
			return err
		}
		if len(dbPerms) != len(perms) {
			return response.ErrBadRequest("包含无效的权限ID")
		}
		for _, p := range dbPerms {
			if p.ProductCode != productCode {
				return response.ErrBadRequest("不能设置其他产品的权限")
			}
			if p.Status != consts.StatusEnabled {
				return response.ErrBadRequest(fmt.Sprintf("权限 %s 已被禁用，无法设置", p.Code))
			}
		}
	}

	if err := l.svcCtx.SysUserPermModel.TransactCtx(l.ctx, func(ctx context.Context, session sqlx.Session) error {
		if err := l.svcCtx.SysUserPermModel.DeleteByUserIdForProductTx(ctx, session, req.UserId, productCode); err != nil {
			return err
		}
		if len(perms) == 0 {
			return nil
		}
		now := time.Now().Unix()
		data := make([]*userperm.SysUserPerm, 0, len(perms))
		permIds := make([]int64, 0, len(perms))
		for _, p := range perms {
			data = append(data, &userperm.SysUserPerm{
				UserId:     req.UserId,
				PermId:     p.PermId,
				Effect:     p.Effect,
				CreateTime: now,
				UpdateTime: now,
			})
			permIds = append(permIds, p.PermId)
		}
		if err := l.svcCtx.SysUserPermModel.BatchInsertWithTx(ctx, session, data); err != nil {
			return err
		}
		// 审计 L-4：事务末对 sys_perm 的 status 再做一次 COUNT 复核，把 "入事务前 FindByIds
		// 校验通过 → 事务外某次 SyncPermissions 把这些 permId 置为 DISABLED → 我们仍旧把脏
		// user_perm 写进去" 的 TOCTOU 窗口收紧到零。校验失败主动返回错误触发事务回滚，外层
		// 映射为 409（数据被并发改动，前端建议重试）。
		//   复核手法：不在这里加 FOR SHARE（会被 SyncPerms 的 LockByCodeTx X 锁阻塞，增大 SyncPerms 尾延迟）；
		//   COUNT 只读 sys_perm 最新可见版本即可：
		//     - 同一事务读到的是事务开始时的快照（InnoDB RR），若 SyncPermissions 已提交，
		//       COUNT 结果会反映 DISABLED 行，与 BatchInsertWithTx 真实落盘行数对不上 → 回滚。
		//     - 若 SyncPermissions 与本事务重叠但尚未提交，则两边按常规 gap lock 规则互不可见，
		//       本次依然按事务开始时的 Enabled 快照落盘；SyncPermissions 提交后其 UPDATE 会
		//       覆盖 sys_perm.status，本 user_perm 行在 loadPerms JOIN sys_perm.status = ? 时
		//       自动被过滤，与 "脏行不生效" 的不变式一致。
		placeholders := make([]string, len(permIds))
		args := make([]interface{}, 0, len(permIds)+2)
		for i, id := range permIds {
			placeholders[i] = "?"
			args = append(args, id)
		}
		args = append(args, productCode, consts.StatusEnabled)
		countQuery := fmt.Sprintf(
			"SELECT COUNT(*) FROM sys_perm WHERE id IN (%s) AND `productCode` = ? AND `status` = ?",
			strings.Join(placeholders, ","),
		)
		var enabled int64
		if err := session.QueryRowCtx(ctx, &enabled, countQuery, args...); err != nil {
			return err
		}
		if enabled != int64(len(permIds)) {
			return response.ErrConflict("部分权限在提交时已被禁用，请刷新后重试")
		}
		return nil
	}); err != nil {
		return err
	}

	l.svcCtx.UserDetailsLoader.Del(l.ctx, req.UserId, productCode)
	return nil
}