94 lines
3.1 KiB
Go
94 lines
3.1 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"kra/internal/biz"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type permissionRepo struct{ data *Data }
|
|
|
|
func NewPermissionRepo(data *Data) biz.PermissionRepo { return &permissionRepo{data: data} }
|
|
|
|
type menuButtonPO struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
DeletedAt gorm.DeletedAt `gorm:"index"`
|
|
Name string
|
|
Description string `gorm:"column:desc"`
|
|
MenuID uint `gorm:"column:sys_base_menu_id"`
|
|
}
|
|
|
|
func (menuButtonPO) TableName() string { return "sys_base_menu_btns" }
|
|
|
|
type authorityButtonPO struct {
|
|
AuthorityID uint `gorm:"primaryKey;column:authority_id"`
|
|
MenuID uint `gorm:"primaryKey;column:sys_menu_id"`
|
|
ButtonID uint `gorm:"primaryKey;column:sys_base_menu_btn_id"`
|
|
}
|
|
|
|
func (authorityButtonPO) TableName() string { return "sys_authority_btns" }
|
|
|
|
func (r *permissionRepo) Buttons(ctx context.Context, menuID uint) ([]*biz.MenuButton, error) {
|
|
var pos []menuButtonPO
|
|
if err := r.data.gormDB.WithContext(ctx).Where("sys_base_menu_id = ?", menuID).Find(&pos).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
out := make([]*biz.MenuButton, 0, len(pos))
|
|
for _, po := range pos {
|
|
out = append(out, menuButtonFromPO(po))
|
|
}
|
|
return out, nil
|
|
}
|
|
func (r *permissionRepo) SetAuthorityButtons(ctx context.Context, aid uint, buttons map[uint][]uint) error {
|
|
return r.data.gormDB.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Where("authority_id = ?", aid).Delete(&authorityButtonPO{}).Error; err != nil {
|
|
return err
|
|
}
|
|
var links []authorityButtonPO
|
|
for menuID, ids := range buttons {
|
|
for _, id := range ids {
|
|
links = append(links, authorityButtonPO{AuthorityID: aid, MenuID: menuID, ButtonID: id})
|
|
}
|
|
}
|
|
if len(links) > 0 {
|
|
return tx.Create(&links).Error
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
func (r *permissionRepo) SelectedButtons(ctx context.Context, aid, menuID uint) ([]uint, error) {
|
|
var ids []uint
|
|
err := r.data.gormDB.WithContext(ctx).Model(&authorityButtonPO{}).Where("authority_id = ? AND sys_menu_id = ?", aid, menuID).Pluck("sys_base_menu_btn_id", &ids).Error
|
|
return ids, err
|
|
}
|
|
func (r *permissionRepo) SetSelectedButtons(ctx context.Context, aid, menuID uint, ids []uint) error {
|
|
return r.data.gormDB.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Where("authority_id = ? AND sys_menu_id = ?", aid, menuID).Delete(&authorityButtonPO{}).Error; err != nil {
|
|
return err
|
|
}
|
|
links := make([]authorityButtonPO, 0, len(ids))
|
|
for _, id := range ids {
|
|
links = append(links, authorityButtonPO{AuthorityID: aid, MenuID: menuID, ButtonID: id})
|
|
}
|
|
if len(links) > 0 {
|
|
return tx.Create(&links).Error
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
func (r *permissionRepo) CanRemoveButton(ctx context.Context, id uint) (bool, error) {
|
|
var count int64
|
|
err := r.data.gormDB.WithContext(ctx).Model(&authorityButtonPO{}).Where("sys_base_menu_btn_id = ?", id).Count(&count).Error
|
|
return count == 0, err
|
|
}
|
|
func (r *permissionRepo) AuthorityButtonIDs(ctx context.Context, aid uint) ([]uint, error) {
|
|
var ids []uint
|
|
err := r.data.gormDB.WithContext(ctx).Model(&authorityButtonPO{}).Where("authority_id = ?", aid).Pluck("sys_base_menu_btn_id", &ids).Error
|
|
return ids, err
|
|
}
|