package com.cloudroam.service.impl;

import cn.hutool.core.util.IdUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.cloudroam.bo.ProjectInfoBO;
import com.cloudroam.bo.projectModule.ProjectModuleBO;
import com.cloudroam.bo.projectModule.ProjectModuleTreeBO;
import com.cloudroam.common.LocalUser;
import com.cloudroam.common.enumeration.DeletedEnum;
import com.cloudroam.common.mybatis.LinPage;
import com.cloudroam.dto.projectInfo.QueryProjectInfoDTO;
import com.cloudroam.dto.projectModule.CreateProjectModuleDTO;
import com.cloudroam.dto.projectModule.ProjectModulePageDTO;
import com.cloudroam.mapper.ProjectModuleMapperCustom;
import com.cloudroam.model.*;
import com.cloudroam.mapper.ProjectModuleMapper;
import com.cloudroam.service.ProjectInfoService;
import com.cloudroam.service.ProjectModuleService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author generator@TaoJie
 * @since 2024-08-16
 */
@Service
public class ProjectModuleServiceImpl extends ServiceImpl<ProjectModuleMapper, ProjectModuleDO> implements ProjectModuleService {

    @Autowired
    private ProjectInfoService projectInfoService;

    @Autowired
    private ProjectModuleMapperCustom projectModuleMapperCustom;

    @Override
    public ProjectModuleDO createProjectModule(CreateProjectModuleDTO validator) {

        UserDO user = LocalUser.getLocalUser();
        ProjectModuleDO projectModuleDO=new ProjectModuleDO();
        BeanUtils.copyProperties(validator,projectModuleDO);

        // 根据parentId来查找是否有当前模块，如果有则设置，没有则查询项目表
        if(StringUtils.isNotBlank(validator.getParentId())){
            final ProjectModuleDO existsModuleDO = this.getById(validator.getParentId());
            if(null!=existsModuleDO){
                projectModuleDO.setParentId(existsModuleDO.getId());
                projectModuleDO.setProjectId(existsModuleDO.getProjectId());
            }else{
                ProjectInfoDO projectInfoDO=projectInfoService.getById(validator.getParentId());

                if(null!=projectInfoDO){
                    projectModuleDO.setParentId(projectInfoDO.getId());
                    projectModuleDO.setProjectId(projectInfoDO.getId());
                }

            }
        }


        projectModuleDO.setId(IdUtil.simpleUUID());
        projectModuleDO.setCreateTime(new Date());
        projectModuleDO.setCreateUserId(String.valueOf(user.getId()));
        projectModuleDO.setCreateUserName(user.getNickname());

        projectModuleDO.setUpdateTime(new Date());
        projectModuleDO.setUpdateUserId(String.valueOf(user.getId()));
        projectModuleDO.setUpdateUserName(user.getNickname());

        projectModuleDO.setIsDeleted(DeletedEnum.DELETED_NO.getCode());
        projectModuleDO.setDeleteUserId(String.valueOf(user.getId()));
        projectModuleDO.setDeleteUserName(user.getNickname());


        this.baseMapper.insert(projectModuleDO);


        return this.baseMapper.selectById(projectModuleDO.getId());
    }

    @Override
    public boolean updateProjectModule(CreateProjectModuleDTO validator) {

        UserDO user = LocalUser.getLocalUser();
        ProjectModuleDO projectModuleDO=new ProjectModuleDO();
        BeanUtils.copyProperties(validator,projectModuleDO);

        final ProjectModuleDO existProjectModuleDO = baseMapper.selectById(validator.getId());

        String prevProjectId= existProjectModuleDO.getProjectId();


        if(StringUtils.isNotBlank(validator.getParentId())){
            final ProjectModuleDO existsModuleDO = this.getById(validator.getParentId());
            if(null!=existsModuleDO){
                projectModuleDO.setParentId(existsModuleDO.getId());
                projectModuleDO.setProjectId(existsModuleDO.getProjectId());
            }else{
                ProjectInfoDO projectInfoDO=projectInfoService.getById(validator.getParentId());

                if(null!=projectInfoDO){
                    projectModuleDO.setParentId(projectInfoDO.getId());
                    projectModuleDO.setProjectId(projectInfoDO.getId());
                }
            }
        }

        projectModuleDO.setUpdateTime(new Date());
        projectModuleDO.setUpdateUserId(String.valueOf(user.getId()));
        projectModuleDO.setUpdateUserName(user.getNickname());

        // 获取当前所有节点下的子节点，更行项目id和parentId

        // 获取所有projectId的模块
        final List<ProjectModuleDO> projectModuleDOSByProjectId = getProjectModuleDOSByProjectId(prevProjectId);
        // 获取当前moduleId下的所有子节点
        final List<ProjectModuleDO> allSubModules = getAllSubModules(existProjectModuleDO.getId(), projectModuleDOSByProjectId);
        final List<ProjectModuleDO> needUpdateSubList = allSubModules.stream().map(module -> {
            ProjectModuleDO subProjectModuleDO = new ProjectModuleBO();
            subProjectModuleDO.setId(module.getId());
            subProjectModuleDO.setProjectId(projectModuleDO.getProjectId());
            return subProjectModuleDO;
        }).collect(Collectors.toList());

        updateBatchById(needUpdateSubList);
        


        return baseMapper.updateById(projectModuleDO)>0;



    }

    public List<ProjectModuleDO> getAllSubModules(String moduleId, List<ProjectModuleDO> projectModules) {
        List<ProjectModuleDO> subModules = new ArrayList<>();
        findSubModules(moduleId, projectModules, subModules);
        return subModules;
    }

    private void findSubModules(String parentId, List<ProjectModuleDO> projectModules, List<ProjectModuleDO> subModules) {
        for (ProjectModuleDO module : projectModules) {
            if (StringUtils.isNotBlank(module.getParentId())
                    &&module.getParentId().equals(parentId)) {
                subModules.add(module);
                // 递归查找子节点
                findSubModules(module.getId(), projectModules, subModules);
            }
        }
    }


    @Override
    public void deleteProjectModuleById(String id) {
        UserDO user = LocalUser.getLocalUser();

        ProjectModuleDO projectModuleDO=new ProjectModuleDO();
        projectModuleDO.setId(id);
        projectModuleDO.setIsDeleted(DeletedEnum.DELETED_YES.getCode());
        projectModuleDO.setDeleteTime(new Date());
        projectModuleDO.setDeleteUserId(String.valueOf(user.getId()));
        projectModuleDO.setDeleteUserName(user.getNickname());
        baseMapper.updateById(projectModuleDO);

    }


    @Override
    public List<ProjectModuleTreeBO> listTree(ProjectModulePageDTO dto) {

        QueryProjectInfoDTO queryProjectInfoDTO=new QueryProjectInfoDTO();
        if(StringUtils.isNotBlank(dto.getProjectId())) queryProjectInfoDTO.setProjectId(dto.getProjectId());
        // 查询所有的项目
        final List<ProjectInfoDO> projectInfoList = projectInfoService.getProjectInfoList(queryProjectInfoDTO);


        final List<ProjectModuleTreeBO> projectModuleTreeBOS = getProjectProjectModuleTreeBOS(projectInfoList);

        return projectModuleTreeBOS;
    }

    private List<ProjectModuleTreeBO> getProjectProjectModuleTreeBOS(List<ProjectInfoDO> projectInfoList) {
        final List<ProjectModuleTreeBO> projectModuleTreeBOS = projectInfoList.stream().map(projectInfoDO -> {

            ProjectModuleTreeBO projectModuleTreeBO = new ProjectModuleTreeBO();
            projectModuleTreeBO.setProjectFlag(1);
            projectModuleTreeBO.setName(projectInfoDO.getProjectName());
            projectModuleTreeBO.setId(projectInfoDO.getId());


            final List<ProjectModuleDO> projectModuleDOList = getProjectModuleDOSByProjectId(projectInfoDO.getId());


            projectModuleTreeBO.setChildren(buildModuleTree(projectModuleDOList));

            return projectModuleTreeBO;
        }).collect(Collectors.toList());
        return projectModuleTreeBOS;
    }

    private List<ProjectModuleDO> getProjectModuleDOSByProjectId(String projectId) {
        // 获取该项目下的所有模块树
        QueryWrapper<ProjectModuleDO> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(ProjectModuleDO::getIsDeleted, DeletedEnum.DELETED_NO.getCode())
                .eq(ProjectModuleDO::getProjectId, projectId);
        List<ProjectModuleDO> projectModuleDOList = this.list(queryWrapper);
        return projectModuleDOList;
    }

    @Override
    public IPage<ProjectModuleTreeBO> getProjectModulePage(QueryProjectInfoDTO dto) {
        final IPage<ProjectInfoBO> projectInfoPage = projectInfoService.getProjectInfoPage(dto);
        final List<ProjectInfoBO> records = projectInfoPage.getRecords();
        final List<ProjectInfoDO> projectList = records.stream().map(ProjectInfoBO -> {
            ProjectInfoDO projectInfoDO = new ProjectInfoDO();
            BeanUtils.copyProperties(ProjectInfoBO, projectInfoDO);
            return projectInfoDO;
        }).collect(Collectors.toList());
        final List<ProjectModuleTreeBO> projectProjectModuleTreeBOS = getProjectProjectModuleTreeBOS(projectList);

        IPage<ProjectModuleTreeBO> iPage=new LinPage<>(dto.getPage(),dto.getCount());
        iPage.setTotal(projectInfoPage.getTotal());
        iPage.setCurrent(projectInfoPage.getCurrent());

        return iPage.setRecords(projectProjectModuleTreeBOS);
    }

    @Override
    public ProjectModuleTreeBO getProjectModuleTreeBOById(String id) {
        final ProjectModuleDO projectModuleDO = this.getById(id);
        return convertToTreeBO(projectModuleDO);
    }

    @Override
    public List<ProjectModuleDO> list(ProjectModulePageDTO dto) {
        LambdaQueryWrapper<ProjectModuleDO> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(ProjectModuleDO::getIsDeleted, DeletedEnum.DELETED_NO.getCode())
                .eq(StringUtils.isNotBlank(dto.getProjectId()), ProjectModuleDO::getProjectId, dto.getProjectId())
                .like(StringUtils.isNotBlank(dto.getModuleName()), ProjectModuleDO::getModuleName, dto.getModuleName());
        return list(queryWrapper);
    }

    @Override
    public List<ProjectModuleDO> getChildrenModuleIds(String moduleId) {

        // 根据模块id查找当前项目下的所有子模块
        List<ProjectModuleDO> list=projectModuleMapperCustom.selectProjectModuleListByModuleId(moduleId);

        // 获取当前模块的子模块
        if(CollectionUtils.isNotEmpty(list)){
            // 查找当前父节点是moduleId下的所有子模块
            List<ProjectModuleDO> subModules = getAllSubModules(moduleId, list);

            return subModules;
        }

        return null;
    }

    public List<ProjectModuleTreeBO> buildModuleTree(List<ProjectModuleDO> moduleList) {
        // 将所有模块数据转换为 Map 以便于快速查找
        Map<String, ProjectModuleTreeBO> moduleMap = moduleList.stream()
                .collect(Collectors.toMap(ProjectModuleDO::getId, this::convertToTreeBO));

        // 构建树形结构
        List<ProjectModuleTreeBO> rootModules = new ArrayList<>();
        for (ProjectModuleDO module : moduleList) {
            ProjectModuleTreeBO treeBO = moduleMap.get(module.getId());
            if (module.getParentId().equals(module.getProjectId())) {
                rootModules.add(treeBO);
            } else {
                ProjectModuleTreeBO parentBO = moduleMap.get(module.getParentId());
                if (parentBO != null) {
                    parentBO.getChildren().add(treeBO);
                }
            }
        }
        return rootModules;
    }

    private ProjectModuleTreeBO convertToTreeBO(ProjectModuleDO module) {
        ProjectModuleTreeBO treeBO = new ProjectModuleTreeBO();
        treeBO.setId(module.getId());
        treeBO.setName(module.getModuleName());
        treeBO.setParentId(module.getParentId());
        treeBO.setChildren(new ArrayList<>());
        treeBO.setProjectFlag(0); // 你可以根据实际需求设置 projectFlag 的值
        return treeBO;
    }

}