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 { 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 && request.Type == EnumUserType.Personal) { user = new User { Type = request.Type, UserName = request.PhoneNumber, PhoneNumber = request.PhoneNumber, IsCheckPhoneNumber = true, 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, 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); } /// /// 绑定手机号 /// /// /// /// public async Task Handle(BindPhoneNumberCommand request, CancellationToken cancellationToken) { await smsUtils.CheckVerifyCode(new CheckVerifyCodeModel { PhoneNumber = request.PhoneNumber, VerifyCode = request.VerifyCode, TemplateCode = EnumSmsTemplateCode.LoginOrRegister }); var logier = JwtUtils.GetCurrentLogier(); var user = await rep.AsQueryable() .FirstOrDefaultAsync(it => it.Id == logier.Id); if (user == null) throw Oops.Oh(EnumErrorCodeType.s404, "当前账号不存在"); 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); } /// /// 获取当前登录用户 /// /// /// /// private LoginCommandCallback GetCurrentLogier(User user, EnumClientType clientType) { 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, }; } } }