package com.ruoyi.system.utils;

import java.text.SimpleDateFormat;
import java.util.Date;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.system.mapper.SysTaskMapper;

/**
 * 任务编号生成器
 * 
 * @author ruoyi
 */
@Component
public class TaskCodeGenerator {
    
    @Autowired
    private SysTaskMapper sysTaskMapper;
    
    /**
     * 生成任务编号（使用默认前缀）
     * 格式：T + YYYYMMDD + 4位序号
     * 
     * @return 任务编号
     * @deprecated 建议使用 {@link #generateTaskCode(String)} 指定单据类型前缀
     */
    @Deprecated
    public String generateTaskCode() {
        return generateTaskCode("T");
    }
    
    /**
     * 生成任务编号
     * 格式：单据类型前缀 + YYYYMMDD + 4位序号
     * 
     * @param serviceOrdClass 单据类型（如：BF、JZ等）
     * @return 任务编号
     */
    public String generateTaskCode(String serviceOrdClass) {
        String dateStr = DateUtils.dateTimeNow("yyyyMMdd");
        return generateTaskCode(serviceOrdClass, dateStr);
    }
    
    /**
     * 根据日期生成任务编号
     * 
     * @param serviceOrdClass 单据类型（如：BF、JZ等）
     * @param date 日期
     * @return 任务编号
     */
    public String generateTaskCode(String serviceOrdClass, Date date) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        String dateStr = sdf.format(date);
        return generateTaskCode(serviceOrdClass, dateStr);
    }
    
    /**
     * 根据日期字符串生成任务编号
     * 
     * @param serviceOrdClass 单据类型（如：BF、JZ等）
     * @param dateStr 日期字符串（格式：yyyyMMdd）
     * @return 任务编号
     */
    private String generateTaskCode(String serviceOrdClass, String dateStr) {
        // 确保单据类型前缀不为空，默认使用"T"
        String prefix = (serviceOrdClass != null && !serviceOrdClass.trim().isEmpty()) 
                        ? serviceOrdClass.trim() : "T";
        
        String datePrefix = prefix + dateStr;
        
        // 查询当日最大编号
        String maxTaskCode = sysTaskMapper.selectMaxTaskCodeByDatePrefix(datePrefix);
        
        int sequence = 1;
        if (maxTaskCode != null && maxTaskCode.startsWith(datePrefix)) {
            // 提取序号部分
            String sequenceStr = maxTaskCode.substring(datePrefix.length());
            try {
                sequence = Integer.parseInt(sequenceStr) + 1;
            } catch (NumberFormatException e) {
                sequence = 1;
            }
        }
        
        // 格式化为4位序号
        return datePrefix + String.format("%04d", sequence);
    }
}