package com.ruoyi.system.service.impl;

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.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);
    }

    /**
     * 构建部门树结构
     * @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());
    }
}