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.CoalFieldDto;
import com.ruoyi.basic.entity.CoalField;
import com.ruoyi.basic.mapper.CoalFieldMapper;
import com.ruoyi.basic.service.CoalFieldService;
import com.ruoyi.common.exception.base.BaseException;
import com.ruoyi.common.utils.bean.BeanUtils;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

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

    private final CoalFieldMapper coalFieldMapper;

    // 正则表达式：匹配 "fieldXX" 格式，XX 为两位数字
    private static final Pattern FIELD_PATTERN = Pattern.compile("^field(\\d{2})$");

    @Override
    public IPage<CoalField> selectCoalFieldList(Page page, CoalFieldDto coalFieldDto) {
        LambdaQueryWrapper<CoalField> queryWrapper = new LambdaQueryWrapper<>();
        if (StringUtils.hasText(coalFieldDto.getSearchAll())) {
            queryWrapper.like(CoalField::getFieldName, coalFieldDto.getSearchAll());
        }
        queryWrapper.orderByDesc(CoalField::getCreateTime);
        return coalFieldMapper.selectPage(page, queryWrapper);
    }

    @Override
    public int addOrEditCoalField(CoalFieldDto coalFieldDto) {
        CoalField coalField = new CoalField();
        BeanUtils.copyProperties(coalFieldDto, coalField);
        if (Objects.isNull(coalFieldDto.getId())) {
            // 新增操作：生成唯一的 fieldXX 编号
            String nextFieldNumber = generateNextFieldNumber();
            coalField.setFields(nextFieldNumber);
            return coalFieldMapper.insert(coalField);
        } else {
            // 更新操作：不修改 fieldNumber
            return coalFieldMapper.updateById(coalField);
        }
    }

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

    @Override
    public List<CoalField> selectAllList(CoalFieldDto coalFieldDto) {
        return coalFieldMapper.selectList(null);
    }

    @Override
    public List<CoalFieldDto> getFieldsByNames(Set<String> fieldNames) {
        // 1. 参数校验
        if (fieldNames == null || fieldNames.isEmpty()) {
            throw new IllegalArgumentException("字段名集合不能为空");
        }

        // 2. 过滤空值
        Set<String> filteredNames = fieldNames.stream()
                .filter(name -> name != null && !name.trim().isEmpty())
                .collect(Collectors.toSet());

        if (filteredNames.isEmpty()) {
            return Collections.emptyList();
        }

        // 3. 查询数据库
        try {
            List<CoalField> entities = coalFieldMapper.getFieldsByNames(filteredNames);

            // 4. 实体转DTO
            return entities.stream()
                    .map(this::convertToDto)
                    .collect(Collectors.toList());

        } catch (Exception e) {
            throw new BaseException("查询煤质字段信息失败，请稍后重试");
        }
    }

    private CoalFieldDto convertToDto(CoalField entity) {
        CoalFieldDto dto = new CoalFieldDto();
        dto.setId(entity.getId());
        dto.setFields(entity.getFields());
        dto.setFieldName(entity.getFieldName());
        return dto;
    }

    @Override
    public Set<String> getFieldNamesByNames(Set<String> fieldNames) {
        return coalFieldMapper.getFieldNamesByNames(fieldNames);
    }

    private String generateNextFieldNumber() {
        // 获取所有已存在的 CoalField 记录，包括已删除的
        LambdaQueryWrapper<CoalField> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.in(CoalField::getDeleted, 0, 1);
        List<CoalField> existingFields = coalFieldMapper.selectList(queryWrapper);

        // 提取并解析所有已存在的 fieldXX 编号
        int maxNumber = 0;
        for (CoalField field : existingFields) {
            String fields = field.getFields();
            if (fields != null) {
                Matcher matcher = FIELD_PATTERN.matcher(fields);
                if (matcher.matches()) {
                    String numberPart = matcher.group(1);
                    int currentNumber = Integer.parseInt(numberPart);
                    maxNumber = Math.max(maxNumber, currentNumber);
                }
            }
        }

        // 生成下一个编号（格式为两位数字，不足补零）
        int nextNumber = maxNumber + 1;
        return String.format("field%02d", nextNumber);
    }
}