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
|
{
|
/// <summary>
|
/// 登录命令处理器
|
/// </summary>
|
public class LoginCommandHandler(
|
IRepository<User> rep,
|
SmsUtils smsUtils,
|
WxmpUtils wxmpUtils) :
|
IRequestHandler<PasswordLoginCommand, LoginCommandCallback>,
|
IRequestHandler<SmsLoginCommand, LoginCommandCallback>,
|
IRequestHandler<WxmpLoginCommand, LoginCommandCallback>,
|
IRequestHandler<BindWxmpUserInfoCommand, LoginCommandCallback>,
|
IRequestHandler<BindPhoneNumberCommand, bool>
|
{
|
private readonly IRepository<User> rep = rep;
|
private readonly SmsUtils smsUtils = smsUtils;
|
private readonly WxmpUtils wxmpUtils = wxmpUtils;
|
|
/// <summary>
|
/// 密码登录
|
/// </summary>
|
/// <param name="request"></param>
|
/// <param name="cancellationToken"></param>
|
/// <returns></returns>
|
public async Task<LoginCommandCallback> 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);
|
}
|
|
/// <summary>
|
/// 短信登录
|
/// </summary>
|
/// <param name="request"></param>
|
/// <param name="cancellationToken"></param>
|
/// <returns></returns>
|
public async Task<LoginCommandCallback> 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);
|
}
|
|
/// <summary>
|
/// 微信小程序登录
|
/// </summary>
|
/// <param name="request"></param>
|
/// <param name="cancellationToken"></param>
|
/// <returns></returns>
|
public async Task<LoginCommandCallback> 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);
|
}
|
|
/// <summary>
|
/// 绑定手机号
|
/// </summary>
|
/// <param name="request"></param>
|
/// <param name="cancellationToken"></param>
|
/// <returns></returns>
|
public async Task<bool> 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;
|
}
|
|
/// <summary>
|
/// 绑定微信小程序用户信息
|
/// </summary>
|
/// <param name="request"></param>
|
/// <param name="cancellationToken"></param>
|
/// <returns></returns>
|
/// <exception cref="NotImplementedException"></exception>
|
public async Task<LoginCommandCallback> 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<BindWxmpUserInfoCommandEncryptedData>();
|
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);
|
}
|
|
/// <summary>
|
/// 获取当前登录用户
|
/// </summary>
|
/// <param name="user"></param>
|
/// <param name="clientType"></param>
|
/// <returns></returns>
|
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,
|
};
|
}
|
}
|
}
|
