package admin

import (
	"context"

	"gorm.io/gorm"

	"portal-mini-go/internal/model/admin"
	"portal-mini-go/internal/pkg/errors"
)

// PermissionRepository 权限（接口 / 菜单）的数据访问。
type PermissionRepository struct {
	db *gorm.DB
}

func NewPermissionRepository(db *gorm.DB) *PermissionRepository {
	return &PermissionRepository{db: db}
}

// GetByID 按 ID 查询权限，不存在返回 ErrNotFound。
func (r *PermissionRepository) GetByID(ctx context.Context, id uint64) (*admin.Permission, error) {
	var m admin.Permission
	if err := r.db.WithContext(ctx).Where("id = ?", id).First(&m).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, errors.ErrNotFound
		}
		return nil, errors.Wrap(50000, "query permission", err)
	}
	return &m, nil
}

// Create 新建权限。
func (r *PermissionRepository) Create(ctx context.Context, m *admin.Permission) error {
	if err := r.db.WithContext(ctx).Create(m).Error; err != nil {
		return errors.Wrap(50000, "create permission", err)
	}
	return nil
}

// PermListFilter 权限列表过滤条件。Type 为空表示不过滤类型。
type PermListFilter struct {
	Type    string
	Keyword string
	Page    int
	Size    int
}

// List 分页查询权限，支持类型与关键字（名称 / code / 路径）过滤。
func (r *PermissionRepository) List(ctx context.Context, f PermListFilter) ([]admin.Permission, int64, error) {
	q := r.db.WithContext(ctx).Model(&admin.Permission{})
	if f.Type != "" {
		q = q.Where("type = ?", f.Type)
	}
	if f.Keyword != "" {
		kw := "%" + f.Keyword + "%"
		q = q.Where("name LIKE ? OR code LIKE ? OR path LIKE ?", kw, kw, kw)
	}
	var total int64
	if err := q.Count(&total).Error; err != nil {
		return nil, 0, errors.Wrap(50000, "count permission", err)
	}
	var list []admin.Permission
	if err := q.Order("sort ASC, id ASC").Offset((f.Page - 1) * f.Size).Limit(f.Size).Find(&list).Error; err != nil {
		return nil, 0, errors.Wrap(50000, "list permission", err)
	}
	return list, total, nil
}

// Update 按 ID 局部更新权限非零值字段。
func (r *PermissionRepository) Update(ctx context.Context, id uint64, cols map[string]interface{}) error {
	res := r.db.WithContext(ctx).Model(&admin.Permission{}).Where("id = ?", id).Updates(cols)
	if res.Error != nil {
		return errors.Wrap(50000, "update permission", res.Error)
	}
	if res.RowsAffected == 0 {
		return errors.ErrNotFound
	}
	return nil
}

// Delete 删除权限，并清理角色-权限关联。
func (r *PermissionRepository) Delete(ctx context.Context, id uint64) error {
	if err := r.db.WithContext(ctx).Where("permission_id = ?", id).Delete(&admin.RolePermission{}).Error; err != nil {
		return errors.Wrap(50000, "delete role permissions", err)
	}
	res := r.db.WithContext(ctx).Where("id = ?", id).Delete(&admin.Permission{})
	if res.Error != nil {
		return errors.Wrap(50000, "delete permission", res.Error)
	}
	if res.RowsAffected == 0 {
		return errors.ErrNotFound
	}
	return nil
}

// AllMenuPermissions 返回全部菜单类型权限（用于组装菜单树）。
func (r *PermissionRepository) AllMenuPermissions(ctx context.Context) ([]admin.Permission, error) {
	var list []admin.Permission
	if err := r.db.WithContext(ctx).
		Where("type = ?", admin.PermTypeMenu).
		Order("sort ASC, id ASC").
		Find(&list).Error; err != nil {
		return nil, errors.Wrap(50000, "list menu permissions", err)
	}
	return list, nil
}

// AllCodes 返回全部权限 code（超级管理员短路放行时使用）。
func (r *PermissionRepository) AllCodes(ctx context.Context) ([]string, error) {
	var codes []string
	if err := r.db.WithContext(ctx).Model(&admin.Permission{}).Pluck("code", &codes).Error; err != nil {
		return nil, errors.Wrap(50000, "list permission codes", err)
	}
	return codes, nil
}