package com.ruoyi.system.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.entity.TDept;
import com.ruoyi.system.mapper.TDeptMapper;
import com.ruoyi.system.service.TDeptService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 点位管理 服务实现类
 * </p>
 *
 * @author xiaochen
 * @since 2025-05-28
 */
@Service
public class TDeptServiceImpl extends ServiceImpl<TDeptMapper, TDept> implements TDeptService {

    /**
     * 获取部门树结构
     * @return 部门树列表
     */
    public List<TDept> selectDeptTreeList() {
        // 查询所有部门
        List<TDept> depts = this.list();
        // 构建树结构
        return buildDeptTree(depts);
    }

    @Override
    public List<Integer> getAllSubDeptIds(String deptId) {
        List<Integer> allSubIds = new ArrayList<>();
        getSubDeptIdsRecursive(Integer.valueOf(deptId), allSubIds);
        return allSubIds.stream().distinct().collect(Collectors.toList());
    }
    private void getSubDeptIdsRecursive(Integer parentId, List<Integer> allSubIds) {
        // 查询直接下级
        List<Integer> directSubIds = this.baseMapper.selectList(new LambdaQueryWrapper<TDept>().eq(TDept::getParentId, parentId)).stream()
                .map(TDept::getId).collect(Collectors.toList());
        for (Integer subId : directSubIds) {
            allSubIds.add(subId);
            // 递归查询下级的下级
            getSubDeptIdsRecursive(subId, allSubIds);
        }
        allSubIds.add(Integer.valueOf(parentId));

    }
    /**
     * 构建部门树结构
     * @param depts 部门列表
     * @return 树结构的部门列表
     */
    private List<TDept> buildDeptTree(List<TDept> depts) {
        return depts.stream()
                // 筛选出顶级部门（parentId为null或0的部门）
                .filter(dept -> dept.getParentId() == null || dept.getParentId() == 0)
                // 为每个顶级部门构建子树
                .peek(dept -> dept.setChildren(getChildren(dept, depts)))
                .collect(Collectors.toList());
    }

    /**
     * 递归获取部门的子部门
     * @param root 当前部门
     * @param depts 所有部门列表
     * @return 子部门列表
     */
    private List<TDept> getChildren(TDept root, List<TDept> depts) {
        return depts.stream()
                // 筛选出当前部门的直接子部门
                .filter(dept -> dept.getParentId() != null && dept.getParentId().equals(root.getId()))
                // 为每个子部门递归构建其子树
                .peek(dept -> dept.setChildren(getChildren(dept, depts)))
                .collect(Collectors.toList());
    }
}