package cn.iocoder.yudao.module.qcreport.service.aiimport.document;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import java.util.List;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.qcreport.enums.ErrorCodeConstants.AI_IMPORT_FILE_UNSUPPORTED;
/**
* 抽取门面:按固定顺序找第一个能处理该文件的适配器。
*
* 顺序由各适配器的 {@code @Order} 决定(图片 → PDF → Office),不是「谁更合适」的择优,
* 而是消除歧义的定序。真实的歧义是 PDF:它既可能被 PDF 适配器认领,
* 也可能因为 {@code contentType} 上报成 {@code application/pdf} 之外的怪值而落到 Office 的兜底判断上,
* 定序之后谁先谁后是确定的,不会随 Bean 注册顺序漂移。
*
* 只做分发,不做「读不出来就换个适配器再试」的兜底——每个适配器内部已经把该试的路径试过了
* (PDF 的渲染失败会回落文本通道),在这里再叠一层只会让失败原因变得说不清。
*/
@Service
public class QcReportDocumentExtractService {
@Resource
private List adapters;
/**
* 抽取文件内容。
*
* @param content 文件字节
* @param originalFilename 原始文件名(含扩展名)
* @param contentType 上传时记录的 MIME 类型,可能为 null
* @throws cn.iocoder.yudao.framework.common.exception.ServiceException 没有适配器认领该格式时
*/
public QcReportDocumentExtract extract(byte[] content, String originalFilename, String contentType) {
String extension = extensionOf(originalFilename);
for (QcReportImportAdapter adapter : adapters) {
if (adapter.supports(extension, contentType)) {
return adapter.extract(content, originalFilename);
}
}
throw exception(AI_IMPORT_FILE_UNSUPPORTED, originalFilename,
extension.isBlank() ? "无" : "." + extension);
}
private static String extensionOf(String filename) {
int dot = filename == null ? -1 : filename.lastIndexOf('.');
return dot < 0 ? "" : filename.substring(dot + 1).toLowerCase();
}
}