2022-04-06 22:22:45 +08:00
|
|
|
|
using Microsoft.AspNetCore.Mvc.Filters;
|
2022-04-09 16:16:32 +08:00
|
|
|
|
using Microsoft.IdentityModel.JsonWebTokens;
|
2022-04-06 22:22:45 +08:00
|
|
|
|
using System;
|
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
|
using System.Linq;
|
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
|
|
namespace Yi.Framework.WebCore.AttributeExtend
|
|
|
|
|
|
{
|
|
|
|
|
|
[AttributeUsage(AttributeTargets.Method)]
|
|
|
|
|
|
public class PermissionAttribute : ActionFilterAttribute
|
|
|
|
|
|
{
|
|
|
|
|
|
private string permission { get; set; }
|
|
|
|
|
|
|
|
|
|
|
|
public PermissionAttribute(string permission)
|
|
|
|
|
|
{
|
|
|
|
|
|
this.permission = permission;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
|
/// 动作鉴权
|
|
|
|
|
|
/// </summary>
|
|
|
|
|
|
/// <param name="context"></param>
|
|
|
|
|
|
/// <exception cref="Exception"></exception>
|
|
|
|
|
|
public override void OnActionExecuting(ActionExecutingContext context)
|
|
|
|
|
|
{
|
|
|
|
|
|
base.OnActionExecuting(context);
|
|
|
|
|
|
|
|
|
|
|
|
if (string.IsNullOrEmpty(permission))
|
|
|
|
|
|
{
|
|
|
|
|
|
throw new Exception("权限不能为空!");
|
|
|
|
|
|
}
|
2022-04-09 16:16:32 +08:00
|
|
|
|
var result = false;
|
|
|
|
|
|
|
2022-04-06 22:22:45 +08:00
|
|
|
|
|
|
|
|
|
|
//可以从Redis得到用户菜单列表,或者直接从jwt中获取
|
2022-04-09 16:16:32 +08:00
|
|
|
|
var sid = context.HttpContext.User.Claims.FirstOrDefault(u => u.Type == JwtRegisteredClaimNames.Sid);
|
2022-04-06 22:22:45 +08:00
|
|
|
|
|
2022-04-09 16:16:32 +08:00
|
|
|
|
//jwt存在的权限列表
|
|
|
|
|
|
var perList = context.HttpContext.User.Claims.Where(u => u.Type == "permission").Select(u=> u.Value.ToString().ToLower()). ToList();
|
|
|
|
|
|
//判断权限是否存在Redis中,或者jwt中
|
2022-04-06 22:22:45 +08:00
|
|
|
|
|
2022-04-09 16:16:32 +08:00
|
|
|
|
//if (perList.Contains(permission.ToLower()))
|
|
|
|
|
|
//{
|
|
|
|
|
|
// result = true;
|
|
|
|
|
|
//}
|
|
|
|
|
|
result = true;
|
2022-04-06 22:22:45 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (!result)
|
|
|
|
|
|
{
|
|
|
|
|
|
throw new Exception("拦截未授权请求!");
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|