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

[backend/middleware] Move ContentEncodingNegotiator into WebApplicationBlazorFrameworkExtensions.cs

Browse source
This commit is contained in:
Laura Hausmann 2024-04-16 23:54:29 +02:00
parent d2e3a89046
commit 40296b02f0
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
2 changed files with 120 additions and 122 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

122
Iceshrimp.Backend/Core/Extensions/WebApplicationBlazorFrameworkExtensions.cs
View file

@ -1,8 +1,9 @@
using AngleSharp.Io;
using Iceshrimp.Backend.Core.Middleware;
using System.IO.Compression;
using Microsoft.AspNetCore.StaticFiles;
using Microsoft.Extensions.FileProviders;
using Microsoft.Extensions.Primitives;
using Microsoft.Net.Http.Headers;
using HeaderNames = AngleSharp.Io.HeaderNames;
namespace Iceshrimp.Backend.Core.Extensions;
@ -91,4 +92,121 @@ public static class WebApplicationBlazorFrameworkExtensions
!remaining.StartsWithSegments((PathString)"/_framework/blazor.server.js") &&
!remaining.StartsWithSegments((PathString)"/_framework/blazor.web.js");
}
private sealed class ContentEncodingNegotiator(RequestDelegate next, IWebHostEnvironment webHostEnvironment)
{
private static readonly StringSegment[] PreferredEncodings = ["br", "gzip"];
private static readonly Dictionary<StringSegment, string> EncodingExtensionMap =
new(StringSegmentComparer.OrdinalIgnoreCase) { ["br"] = ".br", ["gzip"] = ".gz" };
public Task InvokeAsync(HttpContext context)
{
NegotiateEncoding(context);
return HookTransparentDecompression(context);
}
private async Task HookTransparentDecompression(HttpContext ctx)
{
if (ctx.Response.Headers.ContentEncoding.Count != 0 ||
!ResourceExists(ctx, ".br") ||
ResourceExists(ctx, ""))
{
await next(ctx);
return;
}
var responseStream = ctx.Response.Body;
using var tempStream = new MemoryStream();
await using var brotliStream = new BrotliStream(tempStream, CompressionMode.Decompress);
ctx.Response.Body = tempStream;
ctx.Request.Path = (PathString)(ctx.Request.Path + ".br");
ctx.Response.OnStarting(() =>
{
ctx.Response.Headers.ContentLength = null;
return Task.CompletedTask;
});
await next(ctx);
tempStream.Seek(0, SeekOrigin.Begin);
await brotliStream.CopyToAsync(responseStream);
}
private void NegotiateEncoding(HttpContext context)
{
var acceptEncoding = context.Request.Headers.AcceptEncoding;
if (StringValues.IsNullOrEmpty(acceptEncoding) ||
!StringWithQualityHeaderValue.TryParseList(acceptEncoding, out var parsedValues) ||
parsedValues.Count == 0)
return;
var stringSegment1 = StringSegment.Empty;
var num = 0.0;
foreach (var encoding in parsedValues)
{
var stringSegment2 = encoding.Value;
var valueOrDefault = encoding.Quality.GetValueOrDefault(1.0);
if (!(valueOrDefault >= double.Epsilon) || !(valueOrDefault >= num)) continue;
if (Math.Abs(valueOrDefault - num) < 0.001)
{
stringSegment1 = PickPreferredEncoding(context, stringSegment1, encoding);
}
else
{
if (EncodingExtensionMap.TryGetValue(stringSegment2, out var extension) &&
ResourceExists(context, extension))
{
stringSegment1 = stringSegment2;
num = valueOrDefault;
}
}
if (StringSegment.Equals("*", stringSegment2, StringComparison.Ordinal))
{
stringSegment1 = PickPreferredEncoding(context, new StringSegment(), encoding);
num = valueOrDefault;
}
if (!StringSegment.Equals("identity", stringSegment2, StringComparison.OrdinalIgnoreCase))
continue;
stringSegment1 = StringSegment.Empty;
num = valueOrDefault;
}
if (!EncodingExtensionMap.TryGetValue(stringSegment1, out var str))
return;
context.Request.Path += str;
context.Response.Headers.ContentEncoding = stringSegment1.Value;
context.Response.Headers.Append(HeaderNames.Vary, HeaderNames.ContentEncoding);
return;
StringSegment PickPreferredEncoding(
HttpContext innerContext,
StringSegment selectedEncoding,
StringWithQualityHeaderValue encoding
)
{
foreach (var preferredEncoding in PreferredEncodings)
{
if (preferredEncoding == selectedEncoding)
return selectedEncoding;
if ((preferredEncoding == encoding.Value || encoding.Value == "*") &&
ResourceExists(innerContext, EncodingExtensionMap[preferredEncoding]))
return preferredEncoding;
}
return StringSegment.Empty;
}
}
private bool ResourceExists(HttpContext context, string extension)
{
return webHostEnvironment.WebRootFileProvider.GetFileInfo(context.Request.Path + extension).Exists;
}
}
}

120
Iceshrimp.Backend/Core/Middleware/ContentEncodingNegotiatior.cs
View file

@ -1,120 +0,0 @@
using System.IO.Compression;
using Microsoft.Extensions.Primitives;
using Microsoft.Net.Http.Headers;
namespace Iceshrimp.Backend.Core.Middleware;
internal sealed class ContentEncodingNegotiator(RequestDelegate next, IWebHostEnvironment webHostEnvironment)
{
private static readonly StringSegment[] PreferredEncodings = ["br", "gzip"];
private static readonly Dictionary<StringSegment, string> EncodingExtensionMap =
new(StringSegmentComparer.OrdinalIgnoreCase) { ["br"] = ".br", ["gzip"] = ".gz" };
public Task InvokeAsync(HttpContext context)
{
NegotiateEncoding(context);
return HookTransparentDecompression(context);
}
private async Task HookTransparentDecompression(HttpContext ctx)
{
if (ctx.Response.Headers.ContentEncoding.Count != 0 || !ResourceExists(ctx, ".br") || ResourceExists(ctx, ""))
{
await next(ctx);
return;
}
var responseStream = ctx.Response.Body;
using var tempStream = new MemoryStream();
await using var brotliStream = new BrotliStream(tempStream, CompressionMode.Decompress);
ctx.Response.Body = tempStream;
ctx.Request.Path = (PathString)(ctx.Request.Path + ".br");
ctx.Response.OnStarting(() =>
{
ctx.Response.Headers.ContentLength = null;
return Task.CompletedTask;
});
await next(ctx);
tempStream.Seek(0, SeekOrigin.Begin);
await brotliStream.CopyToAsync(responseStream);
}
private void NegotiateEncoding(HttpContext context)
{
var acceptEncoding = context.Request.Headers.AcceptEncoding;
if (StringValues.IsNullOrEmpty(acceptEncoding) ||
!StringWithQualityHeaderValue.TryParseList(acceptEncoding, out var parsedValues) ||
parsedValues.Count == 0)
return;
var stringSegment1 = StringSegment.Empty;
var num = 0.0;
foreach (var encoding in parsedValues)
{
var stringSegment2 = encoding.Value;
var valueOrDefault = encoding.Quality.GetValueOrDefault(1.0);
if (!(valueOrDefault >= double.Epsilon) || !(valueOrDefault >= num)) continue;
if (Math.Abs(valueOrDefault - num) < 0.001)
{
stringSegment1 = PickPreferredEncoding(context, stringSegment1, encoding);
}
else
{
if (EncodingExtensionMap.TryGetValue(stringSegment2, out var extension) &&
ResourceExists(context, extension))
{
stringSegment1 = stringSegment2;
num = valueOrDefault;
}
}
if (StringSegment.Equals("*", stringSegment2, StringComparison.Ordinal))
{
stringSegment1 = PickPreferredEncoding(context, new StringSegment(), encoding);
num = valueOrDefault;
}
if (!StringSegment.Equals("identity", stringSegment2, StringComparison.OrdinalIgnoreCase))
continue;
stringSegment1 = StringSegment.Empty;
num = valueOrDefault;
}
if (!EncodingExtensionMap.TryGetValue(stringSegment1, out var str))
return;
context.Request.Path += str;
context.Response.Headers.ContentEncoding = stringSegment1.Value;
context.Response.Headers.Append(HeaderNames.Vary, HeaderNames.ContentEncoding);
return;
StringSegment PickPreferredEncoding(
HttpContext innerContext,
StringSegment selectedEncoding,
StringWithQualityHeaderValue encoding
)
{
foreach (var preferredEncoding in PreferredEncodings)
{
if (preferredEncoding == selectedEncoding)
return selectedEncoding;
if ((preferredEncoding == encoding.Value || encoding.Value == "*") &&
ResourceExists(innerContext, EncodingExtensionMap[preferredEncoding]))
return preferredEncoding;
}
return StringSegment.Empty;
}
}
private bool ResourceExists(HttpContext context, string extension)
{
return webHostEnvironment.WebRootFileProvider.GetFileInfo(context.Request.Path + extension).Exists;
}
}