58 lines
1.4 KiB
C#
58 lines
1.4 KiB
C#
using System;
|
|
using System.Linq;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.AspNetCore.Mvc.Filters;
|
|
using SimApi.Communications;
|
|
using SimApi.Exceptions;
|
|
|
|
namespace SimApi.Attributes;
|
|
|
|
/// <summary>
|
|
/// 检测登录中间件
|
|
/// </summary>
|
|
[AttributeUsage(AttributeTargets.Class | AttributeTargets.Method)]
|
|
public class SimApiAuthAttribute : ActionFilterAttribute
|
|
{
|
|
private string[] Types { get; }
|
|
|
|
|
|
//默认是user登录类型
|
|
public SimApiAuthAttribute()
|
|
{
|
|
Types = new[] { "user" };
|
|
}
|
|
|
|
//只检测一种用户类型的快捷方式
|
|
public SimApiAuthAttribute(string type)
|
|
{
|
|
Types = type.Split(",");
|
|
}
|
|
|
|
//设定特定类型的检测
|
|
public SimApiAuthAttribute(string[] types)
|
|
{
|
|
Types = types;
|
|
}
|
|
|
|
//只检测一种用户类型的快捷方式
|
|
public SimApiAuthAttribute(string type, string url)
|
|
{
|
|
Types = new[] { type };
|
|
new HttpPostAttribute(url);
|
|
}
|
|
|
|
public override void OnActionExecuting(ActionExecutingContext context)
|
|
{
|
|
var loginInfo = (SimApiLoginItem)context.HttpContext.Items["LoginInfo"];
|
|
//检测是否登录
|
|
if (loginInfo == null)
|
|
{
|
|
throw new SimApiException(401);
|
|
}
|
|
//检测用户类型
|
|
if (!Types.Intersect(loginInfo.Type).Any())
|
|
{
|
|
throw new SimApiException(403);
|
|
}
|
|
}
|
|
} |