using Azure.Core; using FlexJobApi.Core; using Furion.DatabaseAccessor; using Furion.DataEncryption; using Furion.FriendlyException; using Mapster; using MediatR; using Microsoft.EntityFrameworkCore; using System; using System.Collections.Generic; using System.Linq; using System.Security.Cryptography; using System.Text; using System.Threading.Tasks; namespace FlexJobApi.UserServer.Application { /// /// 登录命令处理器 /// public class LoginCommandHandler( IRepository rep, SmsUtils smsUtils, WxmpUtils wxmpUtils) : IRequestHandler, IRequestHandler, IRequestHandler, IRequestHandler, IRequestHandler, IRequestHandler, IRequestHandler { private readonly IRepository rep = rep; private readonly SmsUtils smsUtils = smsUtils; private readonly WxmpUtils wxmpUtils = wxmpUtils; /// /// 密码登录 /// /// /// /// public async Task Handle(PasswordLoginCommand request, CancellationToken cancellationToken) { var user = await rep.AsQueryable().AsNoTracking() .Where(it => it.UserName == request.UserName && it.Type == request.Type) .FirstOrDefaultAsync(cancellationToken); if (user == null || !PBKDF2Encryption.Compare(request.Password, user.Password)) throw Oops.Oh(EnumUserErrorCodeType.u1000); if (user.Status == EnumUserStatus.Disabled) throw Oops.Oh(EnumUserErrorCodeType.u1001); return GetCurrentLogier(user, request.ClientType); } /// /// 短信登录 /// /// /// /// public async Task Handle(SmsLoginCommand request, CancellationToken cancellationToken) { await smsUtils.CheckVerifyCode(new CheckVerifyCodeModel { PhoneNumber = request.PhoneNumber, VerifyCode = request.VerifyCode, TemplateCode = EnumSmsTemplateCode.LoginOrRegister }); var user = await rep.AsQueryable().AsNoTracking() .Where(it => it.UserName == request.PhoneNumber && it.Type == request.Type) .FirstOrDefaultAsync(cancellationToken); if (user == null) throw Oops.Oh(EnumUserErrorCodeType.u1000); if (user.Status == EnumUserStatus.Disabled) throw Oops.Oh(EnumUserErrorCodeType.u1001); return GetCurrentLogier(user, request.ClientType); } /// /// 微信小程序登录 /// /// /// /// public async Task Handle(WxmpLoginCommand request, CancellationToken cancellationToken) { var snsJscode2session = await wxmpUtils.SnsJscode2sessionAsync(request.Type, request.Code, cancellationToken); var user = await rep.AsQueryable().AsNoTracking() .Where(it => it.WxmpOpenId == snsJscode2session.openid && it.Type == request.Type) .FirstOrDefaultAsync(cancellationToken); if (user == null && request.Type == EnumUserType.Personal) { user = new User { Type = request.Type, UserName = snsJscode2session.openid, WxmpOpenId = snsJscode2session.openid, Level = 1, Status = EnumUserStatus.Normal }; await rep.InsertAsync(user); } if (user == null) throw Oops.Oh(EnumUserErrorCodeType.u1000); if (user.Status == EnumUserStatus.Disabled) throw Oops.Oh(EnumUserErrorCodeType.u1001); return GetCurrentLogier(user, EnumClientType.Wxmp, snsJscode2session.session_key); } /// /// 更换手机号 /// /// /// /// public async Task Handle(ChangePhoneNumberCommand request, CancellationToken cancellationToken) { await smsUtils.CheckVerifyCode(new CheckVerifyCodeModel { PhoneNumber = request.PhoneNumber, VerifyCode = request.VerifyCode, TemplateCode = EnumSmsTemplateCode.UpdateUserPhoneNumber }); var logier = JwtUtils.GetCurrentLogier(); var user = await rep.AsQueryable() .FirstOrDefaultAsync(it => it.Id == logier.Id); if (user == null) throw Oops.Oh(EnumErrorCodeType.s404, "当前账号不存在"); var exist = await rep.AsQueryable() .Where(it => it.Type == logier.Type && it.PhoneNumber == request.PhoneNumber && it.Id != user.Id) .AnyAsync(cancellationToken); if (exist) throw Oops.Oh(EnumUserErrorCodeType.u1010); user.UserName = request.PhoneNumber; user.PhoneNumber = request.PhoneNumber; user.IsCheckPhoneNumber = true; return true; } /// /// 绑定微信小程序用户信息 /// /// /// /// /// public async Task Handle(BindWxmpUserInfoCommand request, CancellationToken cancellationToken) { var logier = JwtUtils.GetCurrentLogier(); var user = await rep.AsQueryable() .Where(it => it.Type == logier.Type && it.Id == logier.Id) .FirstOrDefaultAsync(cancellationToken); if (user == null) throw Oops.Oh(EnumErrorCodeType.s404, "当前账号"); if (user.WxmpOpenId.IsNull()) throw Oops.Oh(EnumUserErrorCodeType.u1100); if (user.PhoneNumber.IsNull()) { var result = string.Empty; RijndaelManaged rijalg = new RijndaelManaged(); rijalg.KeySize = 128; rijalg.Padding = PaddingMode.PKCS7; rijalg.Mode = CipherMode.CBC; rijalg.Key = Convert.FromBase64String(request.SessionKey); rijalg.IV = Convert.FromBase64String(request.Iv); byte[] encryptedData = Convert.FromBase64String(request.EncryptedData); ICryptoTransform decryptor = rijalg.CreateDecryptor(rijalg.Key, rijalg.IV); using (MemoryStream msDecrypt = new MemoryStream(encryptedData)) { using (CryptoStream csDecrypt = new CryptoStream(msDecrypt, decryptor, CryptoStreamMode.Read)) { using (StreamReader srDecrypt = new StreamReader(csDecrypt)) { result = srDecrypt.ReadToEnd(); } } } var info = result.JsonTo(); if (info == null || info.PhoneNumber.IsNull()) { throw Oops.Oh(EnumUserErrorCodeType.u1110); } var bindPhoneNumberUser = await rep.AsQueryable() .Where(it => it.Type == logier.Type && it.PhoneNumber == info.PhoneNumber && it.Id != user.Id) .FirstOrDefaultAsync(cancellationToken); if (bindPhoneNumberUser == null) { user.PhoneNumber = info.PhoneNumber; user.ContactPhoneNumber = info.PhoneNumber; user.IsCheckPhoneNumber = true; if (user.UserName == user.WxmpOpenId) { user.UserName = info.PhoneNumber; } } else if (bindPhoneNumberUser.WxmpOpenId.IsNull()) { bindPhoneNumberUser.WxmpOpenId = user.WxmpOpenId; await rep.DeleteAsync(user); return GetCurrentLogier(bindPhoneNumberUser, EnumClientType.Wxmp); } else if (bindPhoneNumberUser.WxmpOpenId != user.WxmpOpenId) { throw Oops.Oh(EnumUserErrorCodeType.u1120); } } return GetCurrentLogier(user, EnumClientType.Wxmp); } /// /// 注册个人账号 /// /// /// /// public async Task Handle(RegisterPersonalUserCommand request, CancellationToken cancellationToken) { await smsUtils.CheckVerifyCode(new CheckVerifyCodeModel { PhoneNumber = request.PhoneNumber, VerifyCode = request.VerifyCode, TemplateCode = EnumSmsTemplateCode.LoginOrRegister }); var user = await rep.AsQueryable() .FirstOrDefaultAsync(it => it.Type == EnumUserType.Personal && it.UserName == request.PhoneNumber); if (user != null) throw Oops.Oh(EnumUserErrorCodeType.u1010); user = new User { Type = EnumUserType.Personal, UserName = request.PhoneNumber, PhoneNumber = request.PhoneNumber, IsCheckPhoneNumber = true, Level = 1, Status = EnumUserStatus.Normal }; await rep.InsertAsync(user); return true; } /// /// 查询个人用户登录信息 /// /// /// /// public async Task Handle(GetPersonalLoginInfoQuery request, CancellationToken cancellationToken) { var model = await rep.GetPersonalQueryable(true) .GetDetail(cancellationToken); var taskUserQueryable = rep.Change().AsQueryable().AsNoTracking() .Where(it => it.UserId == model.Id); model.TaskCount = await taskUserQueryable.CountAsync(); model.HirePassTaskCount = await taskUserQueryable.CountAsync(it => it.HireStatus == EnumTaskUserHireStatus.Pass); model.HireRefuseTaskCount = await taskUserQueryable.CountAsync(it => it.HireStatus == EnumTaskUserHireStatus.Refuse); return model; } /// /// 获取当前登录用户 /// /// /// /// /// private LoginCommandCallback GetCurrentLogier(User user, EnumClientType clientType, string? session_key = null) { var logier = new CurrentLogier { Id = user.Id, Avatar = user.Avatar, Name = user.Name, UserName = user.UserName, PhoneNumber = user.PhoneNumber, Level = user.Level, Type = user.Type, ClientType = clientType, EnterpriseId = user.EnterpriseId, }; JwtUtils.GenerateToken(logier); return new LoginCommandCallback { Id = logier.Id, AccessToken = logier.AccessToken, RefreshToken = logier.RefreshToken, SessionKey = session_key, IsBindPhoneNumber = logier.PhoneNumber.IsNotNull() }; } } }