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.BizFamilyRanch;
import com.ruoyi.web.modules.industryservice.domain.BizGrassLivestockImbalanceWarning;
import com.ruoyi.web.modules.industryservice.domain.vo.FamilyRanchImportResultVo;
import com.ruoyi.web.modules.industryservice.mapper.BizFamilyRanchMapper;
import com.ruoyi.web.modules.industryservice.service.IBizFamilyRanchService;
import com.ruoyi.web.modules.industryservice.support.FamilyRanchDerivedSupport;
import com.ruoyi.web.modules.industryservice.support.FamilyRanchExcelExporter;
import com.ruoyi.web.modules.industryservice.support.FamilyRanchExcelParser;
import com.ruoyi.web.modules.industryservice.support.FamilyRanchExcelParser.ParsedRow;
import com.ruoyi.web.modules.industryservice.support.FamilyRanchValidation;
import com.ruoyi.web.modules.industryservice.support.GrassLivestockImbalanceWarningRules;
import com.ruoyi.web.modules.industryservice.support.GrassLivestockImbalanceWarningSyncSupport;
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 BizFamilyRanchServiceImpl implements IBizFamilyRanchService
{
    @Autowired
    private BizFamilyRanchMapper bizFamilyRanchMapper;

    @Autowired
    private GrassLivestockImbalanceWarningSyncSupport grassLivestockImbalanceWarningSyncSupport;

    @Autowired
    private TownDeptSupport townDeptSupport;

    @Autowired
    private FamilyRanchExcelExporter familyRanchExcelExporter;

    @Override
    public BizFamilyRanch selectBizFamilyRanchById(Long id)
    {
        return bizFamilyRanchMapper.selectBizFamilyRanchById(id);
    }

    @Override
    public List<BizFamilyRanch> selectBizFamilyRanchList(BizFamilyRanch query)
    {
        ReportDeptDataScopeSupport.apply(query, "r");
        return bizFamilyRanchMapper.selectBizFamilyRanchList(query);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int insertBizFamilyRanch(BizFamilyRanch row)
    {
        FamilyRanchValidation.validateRowForSave(row, townDeptSupport);
        assertUnique(row.getStatYear(), row.getStatMonth(), row.getVillageDeptId(), row.getEnterpriseName(), null);
        int rows = bizFamilyRanchMapper.insertBizFamilyRanch(row);
        syncGrassImbalanceWarning(row);
        return rows;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int updateBizFamilyRanch(BizFamilyRanch row)
    {
        if (row.getId() == null)
        {
            throw new ServiceException("主键不能为空");
        }
        FamilyRanchValidation.validateRowForSave(row, townDeptSupport);
        requireExistingRow(row.getId());
        assertUnique(row.getStatYear(), row.getStatMonth(), row.getVillageDeptId(), row.getEnterpriseName(),
                row.getId());
        int rows = bizFamilyRanchMapper.updateBizFamilyRanch(row);
        syncGrassImbalanceWarning(row);
        return rows;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public int deleteBizFamilyRanchByIds(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;
        }
        Long[] ranchIds = cleaned.toArray(new Long[0]);
        List<String> sourceKeys = new ArrayList<>();
        for (Long ranchId : ranchIds)
        {
            sourceKeys.add(GrassLivestockImbalanceWarningRules.familyRanchSourceKey(ranchId));
        }
        grassLivestockImbalanceWarningSyncSupport.deleteBySourceKeys(sourceKeys);
        return bizFamilyRanchMapper.deleteBizFamilyRanchByIds(ranchIds);
    }

    /**
     * 导入：按 (年, 月, 村, 企业名称) 存在则更新、不存在则插入；任一行失败则整批不写。
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public FamilyRanchImportResultVo 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);
        FamilyRanchImportResultVo result = new FamilyRanchImportResultVo();
        List<ParsedRow> parsedRows;
        try
        {
            parsedRows = FamilyRanchExcelParser.parse(file.getInputStream());
        }
        catch (ServiceException ex)
        {
            throw ex;
        }
        catch (Exception ex)
        {
            throw new ServiceException("读取 Excel 失败：" + ex.getMessage());
        }

        List<BizFamilyRanch> 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;
                }
                BizFamilyRanch row = FamilyRanchExcelParser.toEntity(parsed, statYear, statMonth, town.getDeptId(),
                        town.getDeptName(), village.getDeptId(), village.getDeptName());
                FamilyRanchValidation.validateRowForSave(row, townDeptSupport);
                String batchKey = row.getStatYear() + "|" + row.getStatMonth() + "|" + row.getVillageDeptId() + "|"
                        + row.getEnterpriseName();
                if (!batchKeys.add(batchKey))
                {
                    result.addFail("第" + parsed.getExcelRowNum() + "行[" + row.getTownName() + "/"
                            + row.getVillageName() + "/" + row.getEnterpriseName() + "]: 文件内存在重复唯一键");
                    continue;
                }
                toSave.add(row);
            }
            catch (ServiceException ex)
            {
                result.addFail("第" + parsed.getExcelRowNum() + "行[" + parsed.getTownName() + "/"
                        + parsed.getVillageName() + "/" + parsed.getEnterpriseName() + "]: " + ex.getMessage());
            }
        }

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

        for (BizFamilyRanch row : toSave)
        {
            BizFamilyRanch existing = bizFamilyRanchMapper.selectByUnique(row.getStatYear(), row.getStatMonth(),
                    row.getVillageDeptId(), row.getEnterpriseName());
            if (existing != null)
            {
                row.setId(existing.getId());
                row.setUpdateBy(username);
                bizFamilyRanchMapper.updateBizFamilyRanch(row);
                result.setUpdateCount(result.getUpdateCount() + 1);
            }
            else
            {
                row.setCreateBy(username);
                bizFamilyRanchMapper.insertBizFamilyRanch(row);
                result.setInsertCount(result.getInsertCount() + 1);
            }
            syncGrassImbalanceWarning(row);
        }
        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("导出乡镇不合法");
        }
        BizFamilyRanch query = new BizFamilyRanch();
        query.setStatYear(statYear);
        query.setStatMonth(statMonth);
        query.setTownDeptId(townDeptId);
        ReportDeptDataScopeSupport.apply(query, "r");
        List<BizFamilyRanch> list = bizFamilyRanchMapper.selectBizFamilyRanchList(query);
        return familyRanchExcelExporter.export(statYear, statMonth, list, fillerName);
    }

    /**
     * 草畜失衡预警同步（不走「草畜平衡计算」模块）。
     * <ul>
     *   <li>新增 / 导入 / 修改后：核定年末草畜平衡载畜量 &lt; 年底牲畜存栏 → 写入/更新预警</li>
     *   <li>修改后核定载畜量 &gt;= 年底存栏（或字段不全）→ 删除对应预警</li>
     *   <li>定位键：source_key=FR-{id}，粒度=年+月+村+家庭牧场（企业）</li>
     * </ul>
     */
    private void syncGrassImbalanceWarning(BizFamilyRanch row)
    {
        if (row == null || row.getId() == null)
        {
            return;
        }
        BizGrassLivestockImbalanceWarning warning = new BizGrassLivestockImbalanceWarning();
        warning.setSourceKey(GrassLivestockImbalanceWarningRules.familyRanchSourceKey(row.getId()));
        warning.setDataSource(GrassLivestockImbalanceWarningRules.DATA_SOURCE_FAMILY_RANCH);
        warning.setSourceRecordId(row.getId());
        warning.setStatYear(row.getStatYear());
        warning.setStatMonth(row.getStatMonth());
        warning.setTownDeptId(row.getTownDeptId());
        warning.setTownName(row.getTownName());
        warning.setVillageDeptId(row.getVillageDeptId());
        warning.setVillageName(row.getVillageName());
        warning.setSubjectName(row.getEnterpriseName());
        warning.setApprovedCarryingCapacity(row.getApprovedCarryingCapacity());
        warning.setYearEndStockSheepUnit(row.getYearEndStockSheepUnit());
        warning.setAlertMessage(GrassLivestockImbalanceWarningSyncSupport.buildMessage(row.getStatYear(),
                row.getStatMonth(), row.getTownName(), row.getVillageName(), row.getEnterpriseName(),
                row.getApprovedCarryingCapacity(), row.getYearEndStockSheepUnit()));
        grassLivestockImbalanceWarningSyncSupport.sync(FamilyRanchDerivedSupport.isImbalanced(row), warning);
    }

    private void assertUnique(Integer statYear, Integer statMonth, Long villageDeptId, String enterpriseName,
            Long excludeId)
    {
        if (bizFamilyRanchMapper.countByUniqueExcludeId(statYear, statMonth, villageDeptId, enterpriseName,
                excludeId) > 0)
        {
            throw new ServiceException("该年该月该村该企业已填报");
        }
    }

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