78 lines
2.7 KiB
C#
78 lines
2.7 KiB
C#
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);
|
|
}
|
|
}
|