package com.ruoyi.web.modules.industryservice.service.impl;

import java.util.ArrayList;
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import com.ruoyi.common.core.domain.entity.SysDept;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.web.modules.industryservice.domain.BizLivestockProductOutput;
import com.ruoyi.web.modules.industryservice.domain.vo.LivestockProductOutputImportResultVo;
import com.ruoyi.web.modules.industryservice.mapper.BizLivestockProductOutputMapper;
import com.ruoyi.web.modules.industryservice.service.IBizLivestockProductOutputService;
import com.ruoyi.web.modules.industryservice.support.LivestockProductOutputExcelExporter;
import com.ruoyi.web.modules.industryservice.support.LivestockProductOutputExcelParser;
import com.ruoyi.web.modules.industryservice.support.LivestockProductOutputExcelParser.ParsedRow;
import com.ruoyi.web.modules.industryservice.support.LivestockProductOutputValidation;
import com.ruoyi.web.modules.industryservice.support.ReportDeptDataScopeSupport;
import com.ruoyi.web.modules.industryservice.support.TownDeptSupport;
import com.ruoyi.web.modules.industryservice.support.YakHerdInventoryMonthSupport;
import com.ruoyi.web.modules.industryservice.support.YakHerdInventoryYearSupport;

/**
 * 畜牧产品产量登记服务实现
 */
@Service
public class BizLivestockProductOutputServiceImpl implements IBizLivestockProductOutputService
{
    @Autowired
    private BizLivestockProductOutputMapper bizLivestockProductOutputMapper;

    @Autowired
    private TownDeptSupport townDeptSupport;

    @Autowired
    private LivestockProductOutputExcelExporter livestockProductOutputExcelExporter;

    @Override
    public BizLivestockProductOutput selectBizLivestockProductOutputById(Long id)
    {
        return bizLivestockProductOutputMapper.selectBizLivestockProductOutputById(id);
    }

    @Override
    public List<BizLivestockProductOutput> selectBizLivestockProductOutputList(BizLivestockProductOutput query)
    {
        ReportDeptDataScopeSupport.apply(query, "r");
        return bizLivestockProductOutputMapper.selectBizLivestockProductOutputList(query);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int insertBizLivestockProductOutput(BizLivestockProductOutput row)
    {
        LivestockProductOutputValidation.validateRowForSave(row, townDeptSupport);
        assertUniqueYearMonthVillage(row.getStatYear(), row.getStatMonth(), row.getVillageDeptId(), null);
        return bizLivestockProductOutputMapper.insertBizLivestockProductOutput(row);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int updateBizLivestockProductOutput(BizLivestockProductOutput row)
    {
        if (row.getId() == null)
        {
            throw new ServiceException("主键不能为空");
        }
        LivestockProductOutputValidation.validateRowForSave(row, townDeptSupport);
        requireExistingRow(row.getId());
        assertUniqueYearMonthVillage(row.getStatYear(), row.getStatMonth(), row.getVillageDeptId(), row.getId());
        return bizLivestockProductOutputMapper.updateBizLivestockProductOutput(row);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int deleteBizLivestockProductOutputByIds(Long[] ids, String updateBy)
    {
        if (ids == null || ids.length == 0)
        {
            return 0;
        }
        List<Long> cleaned = new ArrayList<>();
        for (Long id : ids)
        {
            if (id != null)
            {
                cleaned.add(id);
            }
        }
        if (cleaned.isEmpty())
        {
            return 0;
        }
        return bizLivestockProductOutputMapper.deleteBizLivestockProductOutputByIds(cleaned.toArray(new Long[0]));
    }

    /**
     * 导入：按 (年, 月, 村) 存在则更新、不存在则插入；任一行失败则整批不写。
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public LivestockProductOutputImportResultVo importData(MultipartFile file, Integer statYear, Integer statMonth,
            String username)
    {
        if (file == null || file.isEmpty())
        {
            throw new ServiceException("请上传 Excel 文件");
        }
        String filename = file.getOriginalFilename();
        if (filename == null || !filename.toLowerCase().endsWith(".xlsx"))
        {
            throw new ServiceException("仅支持 .xlsx 格式");
        }
        YakHerdInventoryMonthSupport.validateStatPeriod(statYear, statMonth);
        LivestockProductOutputImportResultVo result = new LivestockProductOutputImportResultVo();
        List<ParsedRow> parsedRows;
        try
        {
            parsedRows = LivestockProductOutputExcelParser.parse(file.getInputStream());
        }
        catch (ServiceException ex)
        {
            throw ex;
        }
        catch (Exception ex)
        {
            throw new ServiceException("读取 Excel 失败：" + ex.getMessage());
        }

        List<BizLivestockProductOutput> toSave = new ArrayList<>();
        java.util.Set<String> batchKeys = new java.util.HashSet<>();
        for (ParsedRow parsed : parsedRows)
        {
            try
            {
                SysDept town = townDeptSupport.resolveTownByName(parsed.getTownName());
                if (town == null)
                {
                    result.addFail("第" + parsed.getExcelRowNum() + "行[" + parsed.getTownName() + "]: 乡镇未识别");
                    continue;
                }
                if (StringUtils.isEmpty(parsed.getVillageName()))
                {
                    result.addFail("第" + parsed.getExcelRowNum() + "行[" + parsed.getTownName() + "]: 所属村不能为空");
                    continue;
                }
                SysDept village = townDeptSupport.resolveVillageByTownAndName(town.getDeptId(), parsed.getVillageName());
                if (village == null)
                {
                    result.addFail("第" + parsed.getExcelRowNum() + "行[" + parsed.getTownName() + "/"
                            + parsed.getVillageName() + "]: 村未识别");
                    continue;
                }
                BizLivestockProductOutput row = LivestockProductOutputExcelParser.toEntity(parsed, statYear, statMonth,
                        town.getDeptId(), town.getDeptName(), village.getDeptId(), village.getDeptName());
                LivestockProductOutputValidation.validateRowForSave(row, townDeptSupport);
                String batchKey = row.getStatYear() + "|" + row.getStatMonth() + "|" + row.getVillageDeptId();
                if (!batchKeys.add(batchKey))
                {
                    result.addFail("第" + parsed.getExcelRowNum() + "行[" + row.getTownName() + "/"
                            + row.getVillageName() + "]: 文件内存在重复唯一键");
                    continue;
                }
                toSave.add(row);
            }
            catch (ServiceException ex)
            {
                result.addFail("第" + parsed.getExcelRowNum() + "行[" + parsed.getTownName() + "/"
                        + parsed.getVillageName() + "]: " + ex.getMessage());
            }
        }

        if (result.getFailCount() > 0)
        {
            throw new ServiceException("导入失败，未写入数据：" + String.join("；", result.getFailMessages()));
        }
        if (toSave.isEmpty())
        {
            throw new ServiceException("未解析到有效数据行");
        }

        for (BizLivestockProductOutput row : toSave)
        {
            BizLivestockProductOutput existing = bizLivestockProductOutputMapper.selectByYearMonthAndVillage(
                    row.getStatYear(), row.getStatMonth(), row.getVillageDeptId());
            if (existing != null)
            {
                row.setId(existing.getId());
                row.setUpdateBy(username);
                bizLivestockProductOutputMapper.updateBizLivestockProductOutput(row);
                result.setUpdateCount(result.getUpdateCount() + 1);
            }
            else
            {
                row.setCreateBy(username);
                bizLivestockProductOutputMapper.insertBizLivestockProductOutput(row);
                result.setInsertCount(result.getInsertCount() + 1);
            }
        }
        return result;
    }

    @Override
    public List<Integer> listYearOptions()
    {
        return YakHerdInventoryYearSupport.resolveYearOptions();
    }

    @Override
    public List<Integer> listMonthOptions()
    {
        return YakHerdInventoryMonthSupport.resolveMonthOptions();
    }

    @Override
    public List<java.util.Map<String, Object>> listTownOptions()
    {
        return townDeptSupport.toOptionList();
    }

    @Override
    public List<com.ruoyi.common.core.domain.TreeSelect> listVillageTree()
    {
        return townDeptSupport.buildVillageTree();
    }

    @Override
    public byte[] exportByYearMonth(Integer statYear, Integer statMonth, Long townDeptId)
    {
        return exportByYearMonth(statYear, statMonth, townDeptId, null);
    }

    @Override
    public byte[] exportByYearMonth(Integer statYear, Integer statMonth, Long townDeptId, String fillerName)
    {
        YakHerdInventoryMonthSupport.validateStatPeriod(statYear, statMonth);
        if (townDeptId != null && townDeptSupport.requireReportTown(townDeptId) == null)
        {
            throw new ServiceException("导出乡镇不合法");
        }
        BizLivestockProductOutput query = new BizLivestockProductOutput();
        query.setStatYear(statYear);
        query.setStatMonth(statMonth);
        query.setTownDeptId(townDeptId);
        ReportDeptDataScopeSupport.apply(query, "r");
        List<BizLivestockProductOutput> list = bizLivestockProductOutputMapper.selectBizLivestockProductOutputList(query);
        return livestockProductOutputExcelExporter.export(statYear, statMonth, list, fillerName);
    }

    private void assertUniqueYearMonthVillage(Integer statYear, Integer statMonth, Long villageDeptId, Long excludeId)
    {
        if (bizLivestockProductOutputMapper.countByYearMonthAndVillageExcludeId(statYear, statMonth, villageDeptId,
                excludeId) > 0)
        {
            throw new ServiceException("该年该月该村已填报");
        }
    }

    private BizLivestockProductOutput requireExistingRow(Long id)
    {
        BizLivestockProductOutput row = bizLivestockProductOutputMapper.selectBizLivestockProductOutputById(id);
        if (row == null)
        {
            throw new ServiceException("记录不存在");
        }
        return row;
    }
}
