package com.example.demo.repository;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Optional;

import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import com.example.demo.model.ImageToVideoTask;

/**
 * 图生视频任务数据访问层
 */
@Repository
public interface ImageToVideoTaskRepository extends JpaRepository<ImageToVideoTask, Long> {

    /**
     * 根据任务ID查找任务
     */
    Optional<ImageToVideoTask> findByTaskId(String taskId);

    /**
     * 根据用户名查找任务列表（分页）
     */
    Page<ImageToVideoTask> findByUsernameOrderByCreatedAtDesc(String username, Pageable pageable);

    /**
     * 根据用户名查找任务列表
     */
    List<ImageToVideoTask> findByUsernameOrderByCreatedAtDesc(String username);

    /**
     * 统计用户任务数量
     */
    long countByUsername(String username);
    
    /**
     * 统计用户进行中的任务数量（PENDING 或 PROCESSING）
     */
    @Query("SELECT COUNT(t) FROM ImageToVideoTask t WHERE t.username = :username AND t.status IN ('PENDING', 'PROCESSING')")
    long countProcessingTasksByUsername(@Param("username") String username);

    /**
     * 根据状态查找任务列表
     */
    List<ImageToVideoTask> findByStatus(ImageToVideoTask.TaskStatus status);

    /**
     * 根据用户名和状态查找任务列表
     */
    List<ImageToVideoTask> findByUsernameAndStatus(String username, ImageToVideoTask.TaskStatus status);

    /**
     * 查找需要处理的任务（状态为PENDING或PROCESSING）
     */
    @Query("SELECT t FROM ImageToVideoTask t WHERE t.status IN ('PENDING', 'PROCESSING') ORDER BY t.createdAt ASC")
    List<ImageToVideoTask> findPendingTasks();
    
    /**
     * 查找指定状态的任务列表
     */
    @Query("SELECT t FROM ImageToVideoTask t WHERE t.status = :status ORDER BY t.createdAt DESC")
    List<ImageToVideoTask> findByStatusOrderByCreatedAtDesc(@Param("status") ImageToVideoTask.TaskStatus status);

    /**
     * 统计用户各状态任务数量
     */
    @Query("SELECT t.status, COUNT(t) FROM ImageToVideoTask t WHERE t.username = :username GROUP BY t.status")
    List<Object[]> countTasksByStatus(@Param("username") String username);

    /**
     * 查找用户最近的任务
     */
    @Query("SELECT t FROM ImageToVideoTask t WHERE t.username = :username ORDER BY t.createdAt DESC")
    List<ImageToVideoTask> findRecentTasksByUsername(@Param("username") String username, Pageable pageable);

    /**
     * 删除过期的任务（超过30天且已完成或失败）
     */
    @Modifying
    @Query("DELETE FROM ImageToVideoTask t WHERE t.createdAt < :expiredDate AND t.status IN ('COMPLETED', 'FAILED', 'CANCELLED')")
    int deleteExpiredTasks(@Param("expiredDate") java.time.LocalDateTime expiredDate);

    /**
     * 根据状态删除任务
     */
    @Modifying
    @Query("DELETE FROM ImageToVideoTask t WHERE t.status = :status")
    int deleteByStatus(@Param("status") String status);
    
    /**
     * 查找超时的图生视频任务
     * 条件：状态为PROCESSING，且创建时间超过10分钟
     */
    @Query("SELECT t FROM ImageToVideoTask t WHERE t.status = :status " +
           "AND t.createdAt < :timeoutTime")
    List<ImageToVideoTask> findTimeoutTasks(
        @Param("status") ImageToVideoTask.TaskStatus status,
        @Param("timeoutTime") LocalDateTime timeoutTime
    );
}
