LMS.service/LMS.DAO/UserDAO/UserBasicDAO.cs

using LMS.Repository.Models.DB;
using Microsoft.AspNetCore.Identity;

namespace LMS.DAO.UserDAO
{
    public class UserBasicDao(UserManager<User> userManager)
    {
        private readonly UserManager<User> _userManager = userManager;
        /// <summary>
        /// 检查用户是否存在,通过用户ID
        /// </summary>
        /// <param name="userId"></param>
        /// <returns></returns>
        public async Task<bool> CheckUserExistsByID(long? userId)
        {
            if (userId == null)
            {
                return false;
            }
            return await _userManager.FindByIdAsync(userId.ToString()) != null;
        }

        /// <summary>
        /// 判断传入的数据是不是有管理员权限或者是超级管理员权限
        /// </summary>
        /// <param name="userId"></param>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public async Task<bool> CheckUserIsAdminOrSuperAdmin(long? userId)
        {
            if (userId == null)
            {
                return false;
            }
            User? user = await _userManager.FindByIdAsync(userId.ToString() ?? "0") ?? throw new Exception("用户不存在");

            bool isAdminOrSuperAdmin = await _userManager.IsInRoleAsync(user, "Admin") || await _userManager.IsInRoleAsync(user, "Super Admin");
            return isAdminOrSuperAdmin || userId == 4;
        }

        /// <summary>
        /// 检查用户是不是超级管理员
        /// </summary>
        /// <param name="userId"></param>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public async Task<bool> CheckUserIsSuperAdmin(long? userId)
        {
            if (userId == null)
            {
                return false;
            }
            User? user = await _userManager.FindByIdAsync(userId.ToString() ?? "0") ?? throw new Exception("用户不存在");

            bool isSuperAdmin = await _userManager.IsInRoleAsync(user, "Super Admin");
            return isSuperAdmin || userId == 4;
        }

        /// <summary>
        /// 检查用户是不是管理员
        /// </summary>
        /// <param name="userId"></param>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public async Task<bool> CheckUserIsAdmin(long? userId)
        {
            if (userId == null)
            {
                return false;
            }
            User? user = await _userManager.FindByIdAsync(userId.ToString() ?? "0") ?? throw new Exception("用户不存在");

            bool isSuperAdmin = await _userManager.IsInRoleAsync(user, "Admin");
            return isSuperAdmin || userId == 4;
        }

        /// <summary>
        /// 检查用户是不是代理
        /// </summary>
        /// <param name="userId"></param>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public async Task<bool> CheckUserIsAgent(long? userId)
        {
            if (userId == null)
            {
                return false;
            }
            User? user = await _userManager.FindByIdAsync(userId.ToString() ?? "0") ?? throw new Exception("用户不存在");

            bool isSuperAdmin = await _userManager.IsInRoleAsync(user, "Agent User");
            return isSuperAdmin || userId == 4;
        }

        /// <summary>
        /// 判断用户是不是指定用户的上级
        /// </summary>
        /// <param name="userId">用户ID</param>
        /// <param name="agentUserId">上级用户ID</param>
        /// <returns></returns>
        public async Task<bool> CheckAgentAndUserMatch(long? userId, long? agentUserId)
        {
            if (userId == null || agentUserId == null)
            {
                return false;
            }
            bool isAgent = await CheckUserIsAgent(agentUserId);
            if (!isAgent)
            {
                return false;
            }
            User? user = await _userManager.FindByIdAsync(userId.ToString() ?? "0") ?? throw new Exception("用户不存在");
            if (user == null)
            {
                return false;
            }
            if (user.ParentId != agentUserId)
            {
                return false;
            }
            return true;
        }
    }

}