sunpengfei
2025-08-12 c582e460986388dec3c337f3fbc2b8cf616c6e4a
FlexJobApi.UserServer.Application/Auths/Commands/LoginCommandHandler.cs
@@ -9,6 +9,8 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Security.Claims;
using System.Security.Cryptography;
using System.Text;
using System.Threading.Tasks;
@@ -24,7 +26,10 @@
        IRequestHandler<PasswordLoginCommand, LoginCommandCallback>,
        IRequestHandler<SmsLoginCommand, LoginCommandCallback>,
        IRequestHandler<WxmpLoginCommand, LoginCommandCallback>,
        IRequestHandler<BindPhoneNumberCommand, bool>
        IRequestHandler<BindWxmpUserInfoCommand, LoginCommandCallback>,
        IRequestHandler<ChangePhoneNumberCommand, bool>,
        IRequestHandler<RegisterPersonalUserCommand, bool>,
        IRequestHandler<GetPersonalLoginInfoQuery, GetPersonalLoginInfoQueryResult>
    {
        private readonly IRepository<User> rep = rep;
        private readonly SmsUtils smsUtils = smsUtils;
@@ -66,19 +71,6 @@
            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)
@@ -94,11 +86,15 @@
        /// <returns></returns>
        public async Task<LoginCommandCallback> Handle(WxmpLoginCommand request, CancellationToken cancellationToken)
        {
            if (request.Type == EnumUserType.Operation)
            {
                throw Oops.Oh(EnumErrorCodeType.s400, "不支持该类型账号登录");
            }
            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)
            if (user == null)
            {
                user = new User
                {
@@ -114,28 +110,34 @@
                throw Oops.Oh(EnumUserErrorCodeType.u1000);
            if (user.Status == EnumUserStatus.Disabled)
                throw Oops.Oh(EnumUserErrorCodeType.u1001);
            return GetCurrentLogier(user, EnumClientType.Wxmp);
            return GetCurrentLogier(user, EnumClientType.Wxmp, snsJscode2session.session_key);
        }
        /// <summary>
        /// 绑定手机号
        /// 更换手机号
        /// </summary>
        /// <param name="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public async Task<bool> Handle(BindPhoneNumberCommand request, CancellationToken cancellationToken)
        public async Task<bool> Handle(ChangePhoneNumberCommand request, CancellationToken cancellationToken)
        {
            await smsUtils.CheckVerifyCode(new CheckVerifyCodeModel
            {
                PhoneNumber = request.PhoneNumber,
                VerifyCode = request.VerifyCode,
                TemplateCode = EnumSmsTemplateCode.LoginOrRegister
                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;
@@ -143,12 +145,151 @@
        }
        /// <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 claims = JWTEncryption.ReadJwtToken(request.AccessToken).Claims;
            var claimIdentity = new ClaimsIdentity("AuthenticationTypes.Federation");
            claimIdentity.AddClaims(claims);
            var claimsPrincipal = new ClaimsPrincipal(claimIdentity);
            var id = claimsPrincipal.FindFirstValue("Id").ToGuid();
            var type = claimsPrincipal.FindFirstValue("Type").ToEnum<EnumUserType>();
            var user = await rep.AsQueryable()
                .Where(it => it.Type == type && it.Id == 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 == type && it.PhoneNumber == info.PhoneNumber && it.Id != user.Id)
                    .FirstOrDefaultAsync(cancellationToken);
                if (bindPhoneNumberUser == null)
                {
                    if (type == EnumUserType.Enterprise)
                    {
                        throw Oops.Oh(EnumErrorCodeType.s404, "该账号");
                    }
                    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="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public async Task<bool> 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;
        }
        /// <summary>
        /// 查询个人用户登录信息
        /// </summary>
        /// <param name="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public async Task<GetPersonalLoginInfoQueryResult> Handle(GetPersonalLoginInfoQuery request, CancellationToken cancellationToken)
        {
            var model = await rep.GetPersonalQueryable(true)
                .GetDetail<User, GetPersonalLoginInfoQueryResult>(cancellationToken);
            var taskUserQueryable = rep.Change<TaskInfoUser>().AsQueryable().AsNoTracking()
                .Where(it => it.EnterpriseEmployee.UserId == model.Id);
            model.TaskCount = await taskUserQueryable.CountAsync(tu =>
                tu.EnterpriseEmployee.HireStatus == EnumTaskUserHireStatus.Wait
                || tu.EnterpriseEmployee.UserSignContractStatus == EnumTaskUserSignContractStatus.Wait);
            model.HirePassTaskCount = await taskUserQueryable.CountAsync(tu =>
                tu.ArrangeStatus == EnumTaskUserArrangeStatus.Complete);
            model.HireRefuseTaskCount = await taskUserQueryable.CountAsync(tu =>
                tu.EnterpriseEmployee.HireStatus == EnumTaskUserHireStatus.Refuse
                || tu.EnterpriseEmployee.UserSignContractStatus == EnumTaskUserSignContractStatus.Refuse
                || tu.EnterpriseEmployee.EnterpriseSignContractStatus == EnumTaskUserSignContractStatus.Refuse);
            return model;
        }
        /// <summary>
        /// 获取当前登录用户
        /// </summary>
        /// <param name="user"></param>
        /// <param name="clientType"></param>
        /// <param name="session_key"></param>
        /// <returns></returns>
        private LoginCommandCallback GetCurrentLogier(User user, EnumClientType clientType)
        private LoginCommandCallback GetCurrentLogier(User user, EnumClientType clientType, string? session_key = null)
        {
            var logier = new CurrentLogier
            {
@@ -169,6 +310,8 @@
                Id = logier.Id,
                AccessToken = logier.AccessToken,
                RefreshToken = logier.RefreshToken,
                SessionKey = session_key,
                IsBindPhoneNumber = logier.PhoneNumber.IsNotNull()
            };
        }
    }