notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/asp] Fix BlazorSsrHandoffMiddleware not getting triggered

Browse source
This commit is contained in:
Laura Hausmann 2024-11-19 20:06:02 +01:00
parent 1cccdcb832
commit 55f7b48d80
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 16 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
Iceshrimp.Backend/Core/Middleware/BlazorSsrHandoffMiddleware.cs
View file

@ -1,30 +1,33 @@
using System.Collections.Concurrent;
using System.Reflection;
using Iceshrimp.Backend.Core.Extensions;
using Microsoft.AspNetCore.Components.Endpoints;
namespace Iceshrimp.Backend.Core.Middleware;
public class BlazorSsrHandoffMiddleware(RequestDelegate next) : ConditionalMiddleware<BlazorSsrAttribute>
public class BlazorSsrHandoffMiddleware(RequestDelegate next) : IConditionalMiddleware
{
private static readonly ConcurrentDictionary<Endpoint, bool> Cache = [];
public async Task InvokeAsync(HttpContext context)
{
var attribute = context.GetEndpoint()
?.Metadata.GetMetadata<RootComponentMetadata>()
?.Type.GetCustomAttributes<BlazorSsrAttribute>()
.FirstOrDefault();
if (attribute != null)
context.Response.OnStarting(() =>
{
context.Response.OnStarting(() =>
{
context.Response.Headers.Remove("blazor-enhanced-nav");
return Task.CompletedTask;
});
}
context.Response.Headers.Remove("blazor-enhanced-nav");
return Task.CompletedTask;
});
await next(context);
}
public static bool Predicate(HttpContext ctx)
=> ctx.GetEndpoint() is { } endpoint &&
Cache.GetOrAdd(endpoint, e => e.Metadata.GetMetadata<RootComponentMetadata>()
?.Type
.GetCustomAttributes<BlazorSsrAttribute>()
.Any() ??
false);
public static void DisableBlazorJsInitializers(RazorComponentsServiceOptions options)
{
var property =