package shop.alien.store.util.ali.ocr;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * OCR策略工厂
 * 根据OCR类型创建对应的策略实例
 * 
 * @author lyx
 * @date 2025/11/18
 */
@Slf4j
@Component
public class OcrStrategyFactory {
    
    @Autowired
    private List<OcrStrategy> ocrStrategies;
    
    private final Map<String, OcrStrategy> strategyMap = new HashMap<>();
    
    /**
     * 初始化策略映射
     */
    @PostConstruct
    public void init() {
        if (ocrStrategies != null && !ocrStrategies.isEmpty()) {
            for (OcrStrategy strategy : ocrStrategies) {
                strategyMap.put(strategy.getType(), strategy);
                log.info("注册OCR策略: {} -> {}", strategy.getType(), strategy.getClass().getSimpleName());
            }
        }
    }
    
    /**
     * 根据类型获取OCR策略
     * 
     * @param type OCR类型
     * @return OCR策略实例
     * @throws IllegalArgumentException 如果类型不存在
     */
    public OcrStrategy getStrategy(String type) {
        OcrStrategy strategy = strategyMap.get(type);
        if (strategy == null) {
            throw new IllegalArgumentException("不支持的OCR类型: " + type);
        }
        return strategy;
    }
    
    /**
     * 根据类型代码获取OCR策略
     * 
     * @param typeCode 类型代码
     * @return OCR策略实例
     * @throws IllegalArgumentException 如果类型不存在
     */
    public OcrStrategy getStrategyByCode(String typeCode) {
        return getStrategy(typeCode);
    }
    
    /**
     * 检查是否支持指定的OCR类型
     * 
     * @param type OCR类型
     * @return 是否支持
     */
    public boolean supports(String type) {
        return strategyMap.containsKey(type);
    }
}



