mirror of
				https://github.com/nsnail/IGeekFan.AspNetCore.Knife4jUI.git
				synced 2025-11-04 17:30:49 +08:00 
			
		
		
		
	@@ -0,0 +1,40 @@
 | 
			
		||||
using System.Linq;
 | 
			
		||||
using System.Collections.Generic;
 | 
			
		||||
using Microsoft.AspNetCore.Authorization;
 | 
			
		||||
using Microsoft.OpenApi.Models;
 | 
			
		||||
using Swashbuckle.AspNetCore.SwaggerGen;
 | 
			
		||||
 | 
			
		||||
namespace OAuth2Integration.ResourceServer.Swagger
 | 
			
		||||
{
 | 
			
		||||
    public class SecurityRequirementsOperationFilter : IOperationFilter
 | 
			
		||||
    {
 | 
			
		||||
        public void Apply(OpenApiOperation operation, OperationFilterContext context)
 | 
			
		||||
        {
 | 
			
		||||
            // Policy names map to scopes
 | 
			
		||||
            var requiredScopes = context.MethodInfo
 | 
			
		||||
                .GetCustomAttributes(true)
 | 
			
		||||
                .OfType<AuthorizeAttribute>()
 | 
			
		||||
                .Select(attr => attr.Policy)
 | 
			
		||||
                .Distinct();
 | 
			
		||||
 | 
			
		||||
            if (requiredScopes.Any())
 | 
			
		||||
            {
 | 
			
		||||
                operation.Responses.Add("401", new OpenApiResponse { Description = "Unauthorized" });
 | 
			
		||||
                operation.Responses.Add("403", new OpenApiResponse { Description = "Forbidden" });
 | 
			
		||||
 | 
			
		||||
                var oAuthScheme = new OpenApiSecurityScheme
 | 
			
		||||
                {
 | 
			
		||||
                    Reference = new OpenApiReference { Type = ReferenceType.SecurityScheme, Id = "oauth2" }
 | 
			
		||||
                };
 | 
			
		||||
 | 
			
		||||
                operation.Security = new List<OpenApiSecurityRequirement>
 | 
			
		||||
                {
 | 
			
		||||
                    new OpenApiSecurityRequirement
 | 
			
		||||
                    {
 | 
			
		||||
                        [ oAuthScheme ] = requiredScopes.ToList()
 | 
			
		||||
                    }
 | 
			
		||||
                };
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user