package cn.iocoder.yudao.module.bi.service.decision;

import cn.hutool.core.date.DateUtil;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionForecastWorkOrderCreateReqVO;
import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiForecastDataDO;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiForecastDataMapper;
import cn.iocoder.yudao.module.mes.api.workorder.MesProWorkOrderApi;
import cn.iocoder.yudao.module.mes.api.workorder.dto.MesProWorkOrderCreateReqDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.List;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.bi.enums.ErrorCodeConstants.BI_FORECAST_NOT_EXISTS;

/**
 * 负荷预测 Service 实现
 *
 * 本期使用简单移动平均占位，预留 ML 模型接口
 *
 * @author 超级管理员
 */
@Slf4j
@Service
public class BiForecastServiceImpl implements BiForecastService {

    /**
     * 工单来源类型：负荷预测（与字典 mes_pro_work_order_source_type 的 value=4 一致）
     */
    private static final Integer WORK_ORDER_SOURCE_TYPE_FORECAST = 4;

    @Resource
    private BiForecastDataMapper forecastDataMapper;
    @Resource
    private MesProWorkOrderApi mesProWorkOrderApi;

    @Override
    public List<BiForecastDataDO> getForecastList(String forecastCode, LocalDateTime beginTime, LocalDateTime endTime) {
        return forecastDataMapper.selectListByCodeAndTimeRange(forecastCode, beginTime, endTime);
    }

    @Override
    public void generateForecast(String forecastCode, LocalDateTime pointTime) {
        // 本期占位：简单移动平均预测
        // 实际项目应集成 ML 模型（ARIMA/Prophet/LSTM）
        log.info("[generateForecast][生成预测] forecastCode={}, pointTime={}, 当前为占位实现", forecastCode, pointTime);

        BiForecastDataDO forecast = BiForecastDataDO.builder()
                .forecastCode(forecastCode)
                .forecastName("负荷预测(占位)")
                .pointTime(pointTime)
                .forecastValue(BigDecimal.ZERO)
                .lowerBound(BigDecimal.ZERO)
                .upperBound(BigDecimal.ZERO)
                .confidenceLevel(new BigDecimal("95.00"))
                .modelVersion("v1.0-placeholder")
                .build();
        forecastDataMapper.insert(forecast);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createWorkOrderFromForecast(DecisionForecastWorkOrderCreateReqVO reqVO) {
        // 1. 校验预测数据存在
        BiForecastDataDO forecast = forecastDataMapper.selectById(reqVO.getForecastId());
        if (forecast == null) {
            throw exception(BI_FORECAST_NOT_EXISTS);
        }
        // 2. 组装工单 DTO 并调用 MES API 创建（无产品工单，跳 BOM）
        MesProWorkOrderCreateReqDTO dto = new MesProWorkOrderCreateReqDTO();
        dto.setCode(generateWorkOrderCode());
        dto.setName(reqVO.getWorkOrderName() != null
                ? reqVO.getWorkOrderName()
                : ("负荷预测-" + forecast.getForecastName()));
        dto.setType(reqVO.getWorkOrderType());
        dto.setOrderSourceType(WORK_ORDER_SOURCE_TYPE_FORECAST);
        dto.setOrderSourceCode(forecast.getForecastCode());
        dto.setRequestDate(reqVO.getRequestDate());
        dto.setQuantity(reqVO.getQuantity());
        dto.setRemark(reqVO.getRemark());
        return mesProWorkOrderApi.createWorkOrder(dto);
    }

    private String generateWorkOrderCode() {
        return "GZ" + DateUtil.format(LocalDateTime.now(), "yyyyMMddHHmmss")
                + String.format("%03d", (int) (Math.random() * 1000));
    }

}