54 lines
1.7 KiB
C#
54 lines
1.7 KiB
C#
|
using Furion;
|
|||
|
|
using Furion.Authorization;
|
|||
|
|
using Furion.DataEncryption;
|
|||
|
|
using Microsoft.AspNetCore.Authorization;
|
|||
|
|
using Microsoft.AspNetCore.Http;
|
|||
|
|
using Nirvana.Common;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
using YBDevice.NApi.Application.BusinessClient.AccountInfo;
|
|||
|
|
|
|||
|
|
namespace YBDevice.NApi.Core
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// JWT 授权自定义处理程序
|
|||
|
|
/// </summary>
|
|||
|
|
public class JwtHandler : AppAuthorizeHandler
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 重写 Handler 添加自动刷新收取逻辑
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="context"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public override async Task HandleAsync(AuthorizationHandlerContext context)
|
|||
|
|
{
|
|||
|
|
// 自动刷新 token
|
|||
|
|
if (JWTEncryption.AutoRefreshToken(context, context.GetCurrentHttpContext()))
|
|||
|
|
{
|
|||
|
|
await AuthorizeHandleAsync(context);
|
|||
|
|
}
|
|||
|
|
else context.Fail(); // 授权失败
|
|||
|
|
}
|
|||
|
|
public override Task<bool> PipelineAsync(AuthorizationHandlerContext context, DefaultHttpContext httpContext)
|
|||
|
|
{
|
|||
|
|
// 这里写您的授权判断逻辑，授权通过返回 true，否则返回 false
|
|||
|
|
var securityDefineAttribute = httpContext.GetMetadata<SecurityDefineAttribute>();
|
|||
|
|
if (securityDefineAttribute == null)
|
|||
|
|
{
|
|||
|
|
var user = httpContext.User;
|
|||
|
|
if (user == null)
|
|||
|
|
{
|
|||
|
|
return Task.FromResult(false);
|
|||
|
|
}
|
|||
|
|
var userid = user.FindFirst("UserId").Value;
|
|||
|
|
if (string.IsNullOrEmpty(userid))
|
|||
|
|
{
|
|||
|
|
return Task.FromResult(false);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return Task.FromResult(true);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|