/**
 * @ProjectName: 智慧评标-主观分助手
 * @Package: com.zhpb.gdyd_zhpb_zgf.repository.mysql
 * @ClassName: OcrResultRepository
 * @Description: OCR结果Repository接口
 * @Author: 张志锋
 * @Date: 2025-12-09
 * @Version: 1.0
 * @Copyright: © 2025 智慧评标-主观分助手
 */
package com.zhpb.gdyd_zhpb_zgf.repository.mysql;

import com.zhpb.gdyd_zhpb_zgf.entity.mysql.OcrResult;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

/**
 * OCR结果 Repository
 */
@Repository
public interface OcrResultRepository extends JpaRepository<OcrResult, Long> {

    /**
     * 根据文件ID查找OCR结果
     *
     * @param fileId 文件ID
     * @return OCR结果
     */
    Optional<OcrResult> findByFileId(Long fileId);

    /**
     * 根据项目ID查找所有OCR结果
     *
     * @param projectId 项目ID
     * @return OCR结果列表
     */
    List<OcrResult> findByProjectId(String projectId);

    /**
     * 根据项目ID和供应商ID查找OCR结果
     *
     * @param projectId 项目ID
     * @param supplierId 供应商ID
     * @return OCR结果列表
     */
    List<OcrResult> findByProjectIdAndSupplierId(String projectId, String supplierId);

    /**
     * 根据状态查找OCR结果
     *
     * @param status 状态
     * @return OCR结果列表
     */
    List<OcrResult> findByStatus(String status);

    /**
     * 查找失败的OCR任务（用于重试）
     *
     * @return OCR结果列表
     */
    @Query("SELECT o FROM OcrResult o WHERE o.status = 'failed' AND o.errorMessage IS NOT NULL")
    List<OcrResult> findFailedOcrTasks();

    /**
     * 统计项目OCR完成情况
     *
     * @param projectId 项目ID
     * @return 统计结果 [总文件数, 成功数, 失败数, 处理中数]
     */
    @Query("SELECT COUNT(f), " +
           "SUM(CASE WHEN o.status = 'success' THEN 1 ELSE 0 END), " +
           "SUM(CASE WHEN o.status = 'failed' THEN 1 ELSE 0 END), " +
           "SUM(CASE WHEN o.status = 'processing' THEN 1 ELSE 0 END) " +
           "FROM File f LEFT JOIN OcrResult o ON f.id = o.fileId " +
           "WHERE f.projectId = :projectId AND f.fileExtension = 'pdf'")
    List<Object[]> getProjectOcrStatistics(@Param("projectId") String projectId);
}
