using Furion.SpecificationDocument; using Microsoft.Extensions.Caching.Memory; namespace Admin.NET.Core.Service; /// /// 系统登录授权服务 /// [ApiDescriptionSettings(Name = "登录授权", Order = 200)] public class AuthService : IDynamicApiController, ITransient { private readonly SqlSugarRepository _sysUserRep; private readonly RefreshTokenOptions _refreshTokenOptions; private readonly IHttpContextAccessor _httpContextAccessor; private readonly IUserManager _userManager; private readonly IEventPublisher _eventPublisher; private readonly SysUserService _sysUserService; private readonly SysUserRoleService _sysUserRoleService; private readonly IMemoryCache _cache; public AuthService(SqlSugarRepository sysUserRep, IOptions refreshTokenOptions, IHttpContextAccessor httpContextAccessor, IUserManager userManager, IEventPublisher eventPublisher, SysUserService sysUserService, SysUserRoleService sysUserRoleService, IMemoryCache cache) { _sysUserRep = sysUserRep; _httpContextAccessor = httpContextAccessor; _userManager = userManager; _refreshTokenOptions = refreshTokenOptions.Value; _eventPublisher = eventPublisher; _sysUserService = sysUserService; _sysUserRoleService = sysUserRoleService; _cache = cache; } /// /// 登录系统 /// /// /// 用户名/密码:vben/123456 /// [HttpPost("/login")] [AllowAnonymous] [SuppressMonitor] public async Task Login([Required] LoginInput input) { var encryptPasswod = MD5Encryption.Encrypt(input.Password); // 加密密码 // 判断用户名密码 var user = await _sysUserRep.AsQueryable().Includes(u => u.SysOrg) .FirstAsync(u => u.UserName.Equals(input.UserName) && u.Password.Equals(encryptPasswod)); _ = user ?? throw Oops.Oh(ErrorCodeEnum.D1000); // 验证账号是否被冻结 if (user.Status == StatusEnum.Disable) throw Oops.Oh(ErrorCodeEnum.D1017); // 生成Token令牌 var accessToken = JWTEncryption.Encrypt(new Dictionary { {ClaimConst.UserId, user.Id}, {ClaimConst.TenantId, user.TenantId}, {ClaimConst.UserName, user.UserName}, {ClaimConst.RealName, user.RealName}, {ClaimConst.SuperAdmin, user.UserType}, {ClaimConst.OrgId, user.OrgId}, {ClaimConst.OrgName, user.SysOrg?.Name}, {ClaimConst.OrgLevel, user.SysOrg?.Level}, }); // 设置Swagger自动登录 _httpContextAccessor.HttpContext.SigninToSwagger(accessToken); // 生成刷新Token令牌 var refreshToken = JWTEncryption.GenerateRefreshToken(accessToken, _refreshTokenOptions.ExpiredTime); // 设置刷新Token令牌 _httpContextAccessor.HttpContext.Response.Headers["x-access-token"] = refreshToken; return new LoginOutput { UserId = user.Id, Token = accessToken }; } /// /// 获取用户信息 /// /// [HttpGet("/getUserInfo")] public async Task GetUserInfo() { var user = _userManager.User; if (user == null) throw Oops.Oh(ErrorCodeEnum.D1011); // 角色信息 var roles = await _sysUserRoleService.GetUserRoleList(user.Id); // 数据范围 var dataScopes = await _sysUserService.GetUserOrgIdList(); // 增加登录日志 var client = Parser.GetDefault().Parse(_httpContextAccessor.HttpContext.Request.Headers["User-Agent"]); await _eventPublisher.PublishAsync("Add:VisLog", new SysLogVis { Success = YesNoEnum.Y, Message = "登录", Ip = _httpContextAccessor.HttpContext.GetRemoteIpAddressToIPv4(), Browser = client.UA.Family + client.UA.Major, Os = client.OS.Family + client.OS.Major, VisType = LoginTypeEnum.Login, UserName = user.UserName, RealName = user.RealName }); return new LoginUserInfoOutput { UserId = user.Id, Username = user.UserName, RealName = user.RealName, Avatar = user.Avatar, Desc = user.Introduction, OrgId = user.OrgId, OrgName = user.SysOrg != null ? user.SysOrg.Name : "", OrgLevel = user.SysOrg != null ? user.SysOrg.Level : "", Roles = roles.Select(u => new LoginRole { RoleName = u.Name, Value = u.Code }).ToList(), }; } /// /// 获取刷新Token /// /// /// [HttpPost("/getRefreshToken")] public string RefreshToken([Required] string accessToken) { return JWTEncryption.GenerateRefreshToken(accessToken, _refreshTokenOptions.ExpiredTime); } /// /// 退出系统 /// [HttpGet("/logout")] public async void Logout() { var user = _userManager.User; if (user == null) throw Oops.Oh(ErrorCodeEnum.D1011); // 退出Swagger _httpContextAccessor.HttpContext.SignoutToSwagger(); //_httpContextAccessor.HttpContext.Response.Headers["access-token"] = "invalid token"; // 增加退出日志 await _eventPublisher.PublishAsync("Add:VisLog", new SysLogVis { Success = YesNoEnum.Y, Message = "退出", VisType = LoginTypeEnum.Logout, Ip = _httpContextAccessor.HttpContext.GetRemoteIpAddressToIPv4(), UserName = user.UserName, RealName = user.RealName }); } /// /// Swagger登录检查 /// /// [HttpPost("/Swagger/CheckUrl"), NonUnify] [AllowAnonymous] public int SwaggerCheckUrl() { return _cache.Get(CacheConst.SwaggerLogin) ? 200 : 401; } /// /// Swagger登录 /// /// /// [HttpPost("/Swagger/SubmitUrl"), NonUnify] [AllowAnonymous] public int SwaggerSubmitUrl([FromForm] SpecificationAuth auth) { var userName = App.Configuration["SpecificationDocumentSettings:LoginInfo:UserName"]; var password = App.Configuration["SpecificationDocumentSettings:LoginInfo:Password"]; if (auth.UserName == userName && auth.Password == password) { _cache.Set(CacheConst.SwaggerLogin, true); return 200; } return 401; } }