47 lines
1.7 KiB
C#
47 lines
1.7 KiB
C#
|
using Microsoft.OpenApi.Any;
|
|||
|
|
using Microsoft.OpenApi.Models;
|
|||
|
|
using Swashbuckle.AspNetCore.SwaggerGen;
|
|||
|
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
|
|||
|
|
namespace YBDevice.NApi
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// swagger ui 增加header
|
|||
|
|
/// </summary>
|
|||
|
|
public class AddAuthTokenHeaderParameter : IOperationFilter
|
|||
|
|
{
|
|||
|
|
public void Apply(OpenApiOperation operation, OperationFilterContext context)
|
|||
|
|
{
|
|||
|
|
var filterDescriptors = context.ApiDescription.ActionDescriptor.FilterDescriptors;
|
|||
|
|
//检查是否存在自定义的过滤器ApiAuthorizeFilter，如果有说明需要token
|
|||
|
|
var isAuthorized = filterDescriptors.Select(filterInfo => filterInfo.Filter).Any(filter => filter is ApiAuthorizeFilter);
|
|||
|
|
if (isAuthorized)
|
|||
|
|
{
|
|||
|
|
if (operation.Parameters == null)
|
|||
|
|
{
|
|||
|
|
operation.Parameters = new List<OpenApiParameter>();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
operation.Parameters.Add(new OpenApiParameter
|
|||
|
|
{
|
|||
|
|
Name = "token",
|
|||
|
|
In = ParameterLocation.Header,
|
|||
|
|
Description = "请输入token值",
|
|||
|
|
Required = true,
|
|||
|
|
Schema = new OpenApiSchema
|
|||
|
|
{
|
|||
|
|
Type = "string",
|
|||
|
|
Default = new OpenApiString(""),
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
//operation.Responses.Add("401", new OpenApiResponse { Description = "Unauthorized" });
|
|||
|
|
//operation.Responses.Add("403", new OpenApiResponse { Description = "Forbidden" });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|