任务三

internal/data/system/authority_btn.go

@@ -0,0 +1,71 @@
+package system
+
+import (
+	"context"
+	"errors"
+
+	"kra/internal/biz/system"
+	"kra/internal/data/model"
+
+	"gorm.io/gorm"
+)
+
+type authorityBtnRepo struct {
+	db *gorm.DB
+}
+
+// NewAuthorityBtnRepo 创建角色按钮权限仓储
+func NewAuthorityBtnRepo(db *gorm.DB) system.AuthorityBtnRepo {
+	return &authorityBtnRepo{db: db}
+}
+
+func (r *authorityBtnRepo) GetAuthorityBtn(ctx context.Context, authorityId, menuId uint) ([]uint, error) {
+	var btns []model.SysAuthorityBtn
+	if err := r.db.WithContext(ctx).
+		Where("authority_id = ? AND sys_menu_id = ?", authorityId, menuId).
+		Find(&btns).Error; err != nil {
+		return nil, err
+	}
+	selected := make([]uint, len(btns))
+	for i, b := range btns {
+		selected[i] = uint(b.SysBaseMenuBtnID)
+	}
+	return selected, nil
+}
+
+func (r *authorityBtnRepo) SetAuthorityBtn(ctx context.Context, authorityId, menuId uint, btnIds []uint) error {
+	return r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		// 删除旧权限
+		if err := tx.Where("authority_id = ? AND sys_menu_id = ?", authorityId, menuId).
+			Delete(&model.SysAuthorityBtn{}).Error; err != nil {
+			return err
+		}
+		// 创建新权限
+		if len(btnIds) > 0 {
+			btns := make([]model.SysAuthorityBtn, len(btnIds))
+			for i, btnId := range btnIds {
+				btns[i] = model.SysAuthorityBtn{
+					AuthorityID:      int64(authorityId),
+					SysMenuID:        int64(menuId),
+					SysBaseMenuBtnID: int64(btnId),
+				}
+			}
+			if err := tx.Create(&btns).Error; err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
+func (r *authorityBtnRepo) CanRemoveAuthorityBtn(ctx context.Context, btnId uint) error {
+	var count int64
+	if err := r.db.WithContext(ctx).Model(&model.SysAuthorityBtn{}).
+		Where("sys_base_menu_btn_id = ?", btnId).Count(&count).Error; err != nil {
+		return err
+	}
+	if count > 0 {
+		return errors.New("此按钮正在被使用无法删除")
+	}
+	return nil
+}