package repository

import (
	"seeyon-filesystem/config"
	"seeyon-filesystem/model"
)

// DepartmentRepository 部门数据访问
type DepartmentRepository struct{}

func NewDepartmentRepository() *DepartmentRepository {
	return &DepartmentRepository{}
}

// Create 创建部门
func (r *DepartmentRepository) Create(dept *model.Department) error {
	return config.DB.Create(dept).Error
}

// FindByID 根据ID查找部门
func (r *DepartmentRepository) FindByID(id uint64) (*model.Department, error) {
	var dept model.Department
	err := config.DB.First(&dept, id).Error
	if err != nil {
		return nil, err
	}
	return &dept, nil
}

// List 查询所有部门
func (r *DepartmentRepository) List() ([]model.Department, error) {
	var depts []model.Department
	err := config.DB.Order("sort_order ASC, id ASC").Find(&depts).Error
	return depts, err
}

// ListByParent 查询子部门
func (r *DepartmentRepository) ListByParent(parentID uint64) ([]model.Department, error) {
	var depts []model.Department
	err := config.DB.Where("parent_id = ?", parentID).Order("sort_order ASC, id ASC").Find(&depts).Error
	return depts, err
}

// Update 更新部门
func (r *DepartmentRepository) Update(dept *model.Department) error {
	return config.DB.Save(dept).Error
}

// Delete 删除部门
func (r *DepartmentRepository) Delete(id uint64) error {
	// 检查是否有子部门
	var count int64
	config.DB.Model(&model.Department{}).Where("parent_id = ?", id).Count(&count)
	if count > 0 {
		return nil // 有子部门不允许删除, 由service层处理错误
	}
	return config.DB.Delete(&model.Department{}, id).Error
}

// HasChildren 检查是否有子部门
func (r *DepartmentRepository) HasChildren(id uint64) (bool, error) {
	var count int64
	err := config.DB.Model(&model.Department{}).Where("parent_id = ?", id).Count(&count).Error
	return count > 0, err
}

// CountUsers 统计部门下的用户数
func (r *DepartmentRepository) CountUsers(deptID uint64) (int64, error) {
	var count int64
	err := config.DB.Model(&model.User{}).Where("dept_id = ?", deptID).Count(&count).Error
	return count, err
}