package repository

import (
	"gofaster/internal/auth/model"
	"gofaster/internal/shared/repository"

	"gorm.io/gorm"
)

// MenuRepository 菜单仓库
type MenuRepository struct {
	repository.BaseRepo
}

// NewMenuRepository 创建菜单仓库实例
func NewMenuRepository(db *gorm.DB) *MenuRepository {
	return &MenuRepository{
		BaseRepo: *repository.NewBaseRepo(db),
	}
}

// Create 创建菜单
func (r *MenuRepository) Create(menu *model.Menu) error {
	return r.DB().Create(menu).Error
}

// Update 更新菜单
func (r *MenuRepository) Update(menu *model.Menu) error {
	return r.DB().Save(menu).Error
}

// Delete 删除菜单
func (r *MenuRepository) Delete(id uint) error {
	return r.DB().Delete(&model.Menu{}, id).Error
}

// FindByID 根据ID查找菜单
func (r *MenuRepository) FindByID(id uint) (*model.Menu, error) {
	var menu model.Menu
	err := r.DB().First(&menu, id).Error
	if err != nil {
		return nil, err
	}
	return &menu, nil
}

// FindByPath 根据路径查找菜单
func (r *MenuRepository) FindByPath(path string) (*model.Menu, error) {
	var menu model.Menu
	err := r.DB().Where("path = ?", path).First(&menu).Error
	if err != nil {
		return nil, err
	}
	return &menu, nil
}

// FindAll 查找所有菜单
func (r *MenuRepository) FindAll() ([]model.Menu, error) {
	var menus []model.Menu
	err := r.DB().Order("sort ASC, id ASC").Find(&menus).Error
	return menus, err
}

// FindByModule 根据模块查找菜单
func (r *MenuRepository) FindByModule(module string) ([]model.Menu, error) {
	var menus []model.Menu
	err := r.DB().Where("module = ?", module).Order("sort ASC, id ASC").Find(&menus).Error
	return menus, err
}

// FindByParentID 根据父ID查找子菜单
func (r *MenuRepository) FindByParentID(parentID *uint) ([]model.Menu, error) {
	var menus []model.Menu
	query := r.DB()
	if parentID == nil {
		query = query.Where("parent_id IS NULL")
	} else {
		query = query.Where("parent_id = ?", *parentID)
	}
	err := query.Order("sort ASC, id ASC").Find(&menus).Error
	return menus, err
}

// FindTree 查找菜单树
func (r *MenuRepository) FindTree() ([]model.Menu, error) {
	var menus []model.Menu
	err := r.DB().Order("sort ASC, id ASC").Find(&menus).Error
	return menus, err
}