package com.ruoyi.production.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.production.bean.dto.ProductionTeamDto;
import com.ruoyi.production.bean.vo.ProductionTeamVo;
import com.ruoyi.production.mapper.ProductionTeamMapper;
import com.ruoyi.production.pojo.ProductionTeam;
import com.ruoyi.production.pojo.ProductionTeamUserRel;
import com.ruoyi.production.service.ProductionTeamService;
import com.ruoyi.production.service.ProductionTeamUserRelService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

import java.util.List;

@Service
@RequiredArgsConstructor
public class ProductionTeamServiceImpl extends ServiceImpl<ProductionTeamMapper, ProductionTeam> implements ProductionTeamService {

    private final ProductionTeamUserRelService teamUserRelService;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean createTeam(ProductionTeamDto dto) {
        // 创建班组
        ProductionTeam team = new ProductionTeam();
        team.setTeamName(dto.getTeamName());
        team.setRemark(dto.getRemark());
        this.save(team);

        // 保存成员关系
        saveTeamUserRel(team.getId(), dto.getLeaderId(), dto.getMemberIds());

        return true;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean updateTeam(ProductionTeamDto dto) {
        // 更新班组
        ProductionTeam team = new ProductionTeam();
        team.setId(dto.getId());
        team.setTeamName(dto.getTeamName());
        team.setRemark(dto.getRemark());
        this.updateById(team);

        // 删除原有成员关系
        QueryWrapper<ProductionTeamUserRel> wrapper = new QueryWrapper<>();
        wrapper.eq("production_team_id", dto.getId());
        teamUserRelService.remove(wrapper);

        // 保存新的成员关系
        saveTeamUserRel(dto.getId(), dto.getLeaderId(), dto.getMemberIds());

        return true;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean deleteTeam(Long id) {
        // 删除成员关系
        QueryWrapper<ProductionTeamUserRel> wrapper = new QueryWrapper<>();
        wrapper.eq("production_team_id", id);
        teamUserRelService.remove(wrapper);

        // 删除班组
        return this.removeById(id);
    }

    @Override
    public ProductionTeamVo getTeamDetail(Long id) {
        ProductionTeamVo teamVo = getBaseMapper().selectTeamDetail(id);
        if (teamVo != null) {
            // 加载成员列表
            List<ProductionTeamVo.MemberVo> members = getBaseMapper().selectTeamMembers(id);
            teamVo.setMembers(members);
        }
        return teamVo;
    }

    @Override
    public List<ProductionTeamVo> getTeamList(ProductionTeamDto dto) {
        List<ProductionTeamVo> teamList = getBaseMapper().selectTeamList(dto);
        // 为每个班组加载成员列表
        for (ProductionTeamVo teamVo : teamList) {
            List<ProductionTeamVo.MemberVo> members = getBaseMapper().selectTeamMembers(teamVo.getId());
            teamVo.setMembers(members);
        }
        return teamList;
    }

    @Override
    public IPage<ProductionTeamVo> listPage(Page<ProductionTeamDto> page, ProductionTeamDto dto) {
        IPage<ProductionTeamVo> resultPage = getBaseMapper().selectTeamPage(page, dto);
        // 为每个班组加载成员列表
        for (ProductionTeamVo teamVo : resultPage.getRecords()) {
            List<ProductionTeamVo.MemberVo> members = getBaseMapper().selectTeamMembers(teamVo.getId());
            teamVo.setMembers(members);
        }
        return resultPage;
    }

    /**
     * 保存班组用户关系
     */
    private void saveTeamUserRel(Long teamId, Long leaderId, List<Long> memberIds) {
        // 保存班组长
        if (leaderId != null) {
            ProductionTeamUserRel leaderRel = new ProductionTeamUserRel();
            leaderRel.setProductionTeamId(teamId);
            leaderRel.setSysUserId(leaderId);
            leaderRel.setIsLeader((byte) 1);
            teamUserRelService.save(leaderRel);
        }

        // 保存成员
        if (!CollectionUtils.isEmpty(memberIds)) {
            for (Long memberId : memberIds) {
                // 跳过班组长（已保存）
                if (leaderId != null && leaderId.equals(memberId)) {
                    continue;
                }

                ProductionTeamUserRel memberRel = new ProductionTeamUserRel();
                memberRel.setProductionTeamId(teamId);
                memberRel.setSysUserId(memberId);
                memberRel.setIsLeader((byte) 0);
                teamUserRelService.save(memberRel);
            }
        }
    }
}