Tiobon.Web.Core/Tiobon.Core.Common/HttpContextUser/AspNetUser.cs

using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Logging;
using System.IdentityModel.Tokens.Jwt;
using System.Security.Claims;
using Tiobon.Core.Common.Swagger;
using Tiobon.Core.Model.Entity;

namespace Tiobon.Core.Common.HttpContextUser;

public class AspNetUser : IUser
{
    private readonly IHttpContextAccessor _accessor;
    private readonly ILogger<AspNetUser> _logger;

    public AspNetUser(IHttpContextAccessor accessor, ILogger<AspNetUser> logger)
    {
        _accessor = accessor;
        _logger = logger;
    }

    public string Name => GetName();

    private string GetName()
    {
        if (IsAuthenticated() && _accessor.HttpContext.User.Identity.Name.IsNotEmptyOrNull())
        {
            return _accessor.HttpContext.User.Identity.Name;
        }
        else
        {
            var token = GetToken();
            if (!string.IsNullOrEmpty(token))
            {
                var getNameType = Permissions.IsUseIds4
                    ? "name"
                    : "http://schemas.xmlsoap.org/ws/2005/05/identity/claims/name";
                return GetUserInfoFromToken(getNameType, token).FirstOrDefault().ObjToString();
            }
        }

        return "";
    }

    public int ID => GetClaimValueByType("jti").FirstOrDefault().ObjToInt();
    public long TenantId => GetClaimValueByType("TenantId").FirstOrDefault().ObjToLong();

    public bool IsAuthenticated()
    {
        return _accessor.HttpContext?.User?.Identity?.IsAuthenticated ?? false;
    }


    public string GetToken()
    {
        var token = _accessor.HttpContext?.Request?.Headers["Authorization"].ObjToString().Replace("Bearer ", "");
        if (!token.IsNullOrEmpty())
            return token;

        //var request = _accessor.HttpContext?.Request;
        //if (request != null && request?.Body != null)
        //{
        //    var sr = new StreamReader(request?.Body);
        //    var BodyData = sr.ReadToEnd();
        //    var jsonParam = JsonConvert.DeserializeObject<JsonParam>(BodyData);

        //    if (jsonParam != null && !jsonParam.token.IsNullOrEmpty())
        //    {
        //        if (_accessor.HttpContext?.Request?.Headers.Any(x => x.Key == "Authorization") == true)
        //            _accessor.HttpContext.Request.Headers["Authorization"] = jsonParam.token;
        //        else
        //            _accessor.HttpContext.Request.Headers.Add("Authorization", "Bearer " + jsonParam.token);
        //        return jsonParam.token;
        //    }
        //}

        if (_accessor.HttpContext?.IsSuccessSwagger() == true)
        {
            token = _accessor.HttpContext.GetSuccessSwaggerJwt();
            if (token.IsNotEmptyOrNull())
            {
                if (_accessor.HttpContext.User.Claims.Any(s => s.Type == JwtRegisteredClaimNames.Jti))
                    return token;

                var claims = new ClaimsIdentity(GetClaimsIdentity(token));
                _accessor.HttpContext.User.AddIdentity(claims);
                return token;
            }
        }

        return token;
    }

    public List<string> GetUserInfoFromToken(string ClaimType, string token)
    {
        var jwtHandler = new JwtSecurityTokenHandler();

        // token校验
        if (token.IsNotEmptyOrNull() && jwtHandler.CanReadToken(token))
        {
            JwtSecurityToken jwtToken = jwtHandler.ReadJwtToken(token);

            return (from item in jwtToken.Claims
                    where item.Type == ClaimType
                    select item.Value).ToList();
        }

        return new List<string>() { };
    }

    public ServiceResult<string> MessageModel { get; set; }

    public IEnumerable<Claim> GetClaimsIdentity()
    {
        if (_accessor.HttpContext == null) return ArraySegment<Claim>.Empty;

        if (!IsAuthenticated()) return GetClaimsIdentity(GetToken());

        var claims = _accessor.HttpContext.User.Claims.ToList();
        var headers = _accessor.HttpContext.Request.Headers;
        foreach (var header in headers)
        {
            claims.Add(new Claim(header.Key, header.Value));
        }

        return claims;
    }

    public IEnumerable<Claim> GetClaimsIdentity(string token)
    {
        var jwtHandler = new JwtSecurityTokenHandler();
        // token校验
        if (token.IsNotEmptyOrNull() && jwtHandler.CanReadToken(token))
        {
            var jwtToken = jwtHandler.ReadJwtToken(token);

            return jwtToken.Claims;
        }

        return new List<Claim>();
    }

    public List<string> GetClaimValueByType(string ClaimType)
    {
        return (from item in GetClaimsIdentity()
                where item.Type == ClaimType
                select item.Value).ToList();
    }

    public string StaffNo => GetStaffNo();

    private string GetStaffNo()
    {
        return GetClaimValueByType("StaffNo").FirstOrDefault(); ;
    }

    public string StaffName => GetStaffName();

    private string GetStaffName()
    {
        return GetClaimValueByType("StaffName").FirstOrDefault();
    }
    public int StaffId => GetStaffId();

    private int GetStaffId()
    {
        return GetClaimValueByType("StaffId").FirstOrDefault().ObjToInt();
    }

    public int? GetLangId()
    {
        var langId = _accessor.HttpContext?.Request?.Headers["Langid"].ObjToInt();
        if (langId.IsNullOrEmpty())
            return 1;

        return langId;
    }
}