package com.ruoyi.basic.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.basic.dto.CoalPlanDto;
import com.ruoyi.basic.entity.CoalField;
import com.ruoyi.basic.entity.CoalPlan;
import com.ruoyi.basic.mapper.CoalFieldMapper;
import com.ruoyi.basic.mapper.CoalPlanMapper;
import com.ruoyi.basic.service.CoalPlanService;
import com.ruoyi.basic.vo.CoalFieldVo;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.bean.BeanUtils;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 * 煤质信息表，记录煤炭质量检测相关数据 服务实现类
 * </p>
 *
 * @author ruoyi
 * @since 2025-06-09
 */
@Service
@RequiredArgsConstructor
public class CoalPlanServiceImpl extends ServiceImpl<CoalPlanMapper, CoalPlan> implements CoalPlanService {

    private final CoalPlanMapper coalPlanMapper;

    private final CoalFieldMapper coalFieldMapper;

    @Override
    public IPage<CoalPlan> selectCoalFieldList(Page page, CoalPlanDto coalPlanDto) {
        LambdaQueryWrapper<CoalPlan> queryWrapper = new LambdaQueryWrapper<>();
        if (StringUtils.hasText(coalPlanDto.getSearchAll())){
            queryWrapper.like(CoalPlan::getPlan,coalPlanDto.getSearchAll());
        }
        return coalPlanMapper.selectPage(page, queryWrapper);
    }

    @Override
    public int addOrEditCoalPlan(CoalPlanDto coalPlanDto) {
        CoalPlan coalPlan = new CoalPlan();
        BeanUtils.copyProperties(coalPlanDto, coalPlan);
        if (Objects.isNull(coalPlanDto.getId())) {
            return coalPlanMapper.insert(coalPlan);
        } else {
            return coalPlanMapper.updateById(coalPlan);
        }
    }

    @Override
    public int delCoalPlanByIds(Long[] ids) {
        // 检查参数
        if (ids == null || ids.length == 0) {
            return 0;
        }
        // 构造更新条件
        UpdateWrapper<CoalPlan> updateWrapper = new UpdateWrapper<>();
        updateWrapper.in("id", ids)
                .set("deleted", 1);  // 设置 deleted 为 1 表示已删除
        // 执行批量逻辑删除
        return coalPlanMapper.update(null, updateWrapper);
    }

    @Override
    public List<CoalPlan> selectAllList() {
        return coalPlanMapper.selectList(null);
    }

    @Override
    public List<CoalFieldVo> selectCoalPlanById(Long id) {
        CoalPlan coalPlan = coalPlanMapper.selectById(id);
        LambdaQueryWrapper<CoalField> coalFieldLambdaQueryWrapper = new LambdaQueryWrapper<>();
        List<Long> ids = Arrays.stream(coalPlan.getFieldIds().split(",")).map(String::trim).map(Long::parseLong).collect(Collectors.toList());
        coalFieldLambdaQueryWrapper.in(CoalField::getId, ids);
        List<CoalField> coalFields = coalFieldMapper.selectList(coalFieldLambdaQueryWrapper);
        return coalFields.stream()
                .map(coalField -> {
                    CoalFieldVo dto = new CoalFieldVo();
                    dto.setId(coalField.getId());
                    dto.setFields(coalField.getFields());
                    dto.setFieldName(coalField.getFieldName());
                    return dto;
                })
                .collect(Collectors.toList());
    }
}