package com.qqchen.deploy.backend.deploy.repository;

import com.qqchen.deploy.backend.framework.repository.IBaseRepository;
import com.qqchen.deploy.backend.deploy.entity.RepositoryBranch;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.Collection;
import java.util.List;

@Repository
public interface IRepositoryBranchRepository extends IBaseRepository<RepositoryBranch, Long> {
    
    /**
     * 根据外部系统ID和项目ID查找分支列表
     *
     * @param externalSystemId 外部系统ID
     * @param projectId 项目ID
     * @return 分支列表
     */
    List<RepositoryBranch> findByExternalSystemIdAndProjectId(Long externalSystemId, Long projectId);

    /**
     * 统计指定外部系统下的分支数量
     *
     * @param externalSystemId 外部系统ID
     * @return 分支数量
     */
    Long countByExternalSystemId(Long externalSystemId);

    /**
     * 统计指定项目下的分支数量
     *
     * @param projectId 项目ID
     * @return 分支数量
     */
    Long countByProjectId(Long projectId);

    /**
     * 统计指定仓库项目下的分支数量（未删除）
     *
     * @param externalSystemId 外部系统ID
     * @param repoProjectId GitLab项目ID
     * @return 分支数量
     */
    Long countByExternalSystemIdAndRepoProjectIdAndDeletedFalse(Long externalSystemId, Long repoProjectId);

    List<RepositoryBranch> findByExternalSystemIdAndProjectIdAndDeletedFalse(Long externalSystemId, Long id);

    List<RepositoryBranch> findByExternalSystemIdAndDeletedFalse(Long externalSystemId);
    
    /**
     * 批量查询项目的分支数量（用于解决N+1查询问题）
     * @param projectIds 项目ID集合
     * @return Object数组列表，[0]=projectId, [1]=count
     */
    @Query("SELECT b.projectId as projectId, COUNT(b.id) as count " +
           "FROM RepositoryBranch b " +
           "WHERE b.projectId IN :projectIds AND b.deleted = false " +
           "GROUP BY b.projectId")
    List<Object[]> countByProjectIds(@Param("projectIds") Collection<Long> projectIds);
}