package com.mosquito.project.permission;

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;

/**
 * 用户角色关联Repository
 */
@Repository
public interface UserRoleRepository extends JpaRepository<SysUserRole, Long> {

    /**
     * 根据用户ID查询所有角色ID
     */
    List<SysUserRole> findByUserId(Long userId);

    /**
     * 根据用户ID查询所有角色代码（两步查询）
     */
    @Query("SELECT ur.roleId FROM SysUserRole ur WHERE ur.userId = :userId")
    List<Long> findRoleIdsByUserId(@Param("userId") Long userId);

    /**
     * 根据用户ID和角色ID查询
     */
    List<SysUserRole> findByUserIdAndRoleId(Long userId, Long roleId);

    /**
     * 检查用户是否拥有指定角色
     */
    boolean existsByUserIdAndRoleId(Long userId, Long roleId);

    /**
     * 删除用户角色关联
     */
    void deleteByUserIdAndRoleId(Long userId, Long roleId);

    /**
     * 删除用户所有角色关联
     */
    void deleteByUserId(Long userId);
}
