dengjunjie
2025-02-11 8467804705615f7614c29e7ef7ac3e99f5c13a54
ÏîÄ¿´úÂë/WMS/WIDESEA_WMSServer/WIDESEA_Core/Middlewares/SwaggerAuthMiddleware.cs
¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,80 @@
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Http;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Text;
using System.Threading.Tasks;
namespace WIDESEA_Core.Middlewares
{
    public class SwaggerAuthMiddleware
    {
        private readonly RequestDelegate next;
        public SwaggerAuthMiddleware(RequestDelegate next)
        {
            this.next = next;
        }
        public async Task InvokeAsync(HttpContext context)
        {
            // ä¹Ÿå¯ä»¥æ ¹æ®æ˜¯å¦æ˜¯æœ¬åœ°åšåˆ¤æ–­ IsLocalRequest
            if (context.Request.Path.Value.ToLower().Contains("index.html"))
            {
                // åˆ¤æ–­æƒé™æ˜¯å¦æ­£ç¡®
                if (IsAuthorized(context))
                {
                    await next.Invoke(context);
                    return;
                }
                // æ— æƒé™ï¼Œè·³è½¬swagger登录页
                context.Response.Redirect("/swg-login.html");
            }
            else
            {
                await next.Invoke(context);
            }
        }
        public bool IsAuthorized(HttpContext context)
        {
            // ä½¿ç”¨session模式
            // å¯ä»¥ä½¿ç”¨å…¶ä»–çš„
            return context.Session.GetString("swagger-code") == "success";
        }
        /// <summary>
        /// åˆ¤æ–­æ˜¯ä¸æ˜¯æœ¬åœ°è®¿é—®
        /// æœ¬åœ°ä¸ç”¨swagger拦截
        /// </summary>
        /// <param name="context"></param>
        /// <returns></returns>
        public bool IsLocalRequest(HttpContext context)
        {
            if (context.Connection.RemoteIpAddress == null && context.Connection.LocalIpAddress == null)
            {
                return true;
            }
            if (context.Connection.RemoteIpAddress.Equals(context.Connection.LocalIpAddress))
            {
                return true;
            }
            if (IPAddress.IsLoopback(context.Connection.RemoteIpAddress))
            {
                return true;
            }
            return false;
        }
    }
    public static class SwaggerAuthorizeExtensions
    {
        public static IApplicationBuilder UseSwaggerAuthorized(this IApplicationBuilder builder)
        {
            return builder.UseMiddleware<SwaggerAuthMiddleware>();
        }
    }
}