♻️ New error page

2025-11-21 23:30:43 +08:00
parent 2a35786204
commit 98b8d5f33b
7 changed files with 141 additions and 109 deletions
--- a/DysonNetwork.Zone/Publication/PublicationSiteMiddleware.cs
+++ b/DysonNetwork.Zone/Publication/PublicationSiteMiddleware.cs
@@ -0,0 +1,77 @@
+using System.Text.Json;
+using DysonNetwork.Shared.Models;
+using Microsoft.AspNetCore.StaticFiles;
+using Microsoft.EntityFrameworkCore;
+
+namespace DysonNetwork.Zone.Publication;
+
+public class PublicationSiteMiddleware(RequestDelegate next)
+{
+    public const string SiteContextKey = "PubSite";
+    
+    public async Task InvokeAsync(HttpContext context, AppDatabase db, PublicationSiteManager psm)
+    {
+        var siteNameValue = context.Request.Headers["X-SiteName"].ToString();
+        var currentPath = context.Request.Path.Value ?? "";
+
+        if (string.IsNullOrEmpty(siteNameValue))
+        {
+            await next(context);
+            return;
+        }
+
+        var site = await db.PublicationSites
+            .FirstOrDefaultAsync(s => EF.Functions.ILike(s.Name, siteNameValue));
+        if (site == null)
+        {
+            await next(context);
+            return;
+        }
+
+        context.Items[SiteContextKey] = site;
+        
+        var page = await db.PublicationPages
+            .FirstOrDefaultAsync(p => p.SiteId == site.Id && p.Path == currentPath);
+        if (page != null)
+        {
+            switch (page.Type)
+            {
+                case PublicationPageType.HtmlPage
+                    when page.Config.TryGetValue("html", out var html) && html is JsonElement content:
+                    context.Response.ContentType = "text/html";
+                    await context.Response.WriteAsync(content.ToString());
+                    return;
+                case PublicationPageType.Redirect
+                    when page.Config.TryGetValue("url", out var url) && url is JsonElement redirectUrl:
+                    context.Response.Redirect(redirectUrl.ToString());
+                    return;
+            }
+        }
+
+        // If the site is enabled the self-managed mode, try lookup the files then
+        if (site.Mode == PublicationSiteMode.SelfManaged)
+        {
+            var provider = new FileExtensionContentTypeProvider();
+            var hostedFilePath = psm.GetValidatedFullPath(site.Id, currentPath);
+            if (File.Exists(hostedFilePath))
+            {
+                if (!provider.TryGetContentType(hostedFilePath, out var mimeType))
+                    mimeType = "text/html";
+
+                context.Response.ContentType = mimeType;
+                await context.Response.SendFileAsync(hostedFilePath);
+                return;
+            }
+
+            var hostedNotFoundPath = psm.GetValidatedFullPath(site.Id, "404.html");
+            if (File.Exists(hostedNotFoundPath))
+            {
+                context.Response.ContentType = "text/html";
+                await context.Response.SendFileAsync(hostedNotFoundPath);
+                return;
+            }
+        }
+
+        await next(context);
+    }
+}