From 654fcc29b5d2284d301df3f63e9db35c21ec63cf Mon Sep 17 00:00:00 2001 From: Rory& Date: Sun, 11 Jan 2026 16:31:22 +0100 Subject: CDN-CS stuff --- .../Controllers/Internal/GetImageController.cs | 149 +++++++++++++++++++++ 1 file changed, 149 insertions(+) create mode 100644 extra/admin-api/Spacebar.Cdn/Controllers/Internal/GetImageController.cs (limited to 'extra/admin-api/Spacebar.Cdn/Controllers/Internal/GetImageController.cs') diff --git a/extra/admin-api/Spacebar.Cdn/Controllers/Internal/GetImageController.cs b/extra/admin-api/Spacebar.Cdn/Controllers/Internal/GetImageController.cs new file mode 100644 index 00000000..7d4ef087 --- /dev/null +++ b/extra/admin-api/Spacebar.Cdn/Controllers/Internal/GetImageController.cs @@ -0,0 +1,149 @@ +using System.Configuration; +using System.Net.Http.Headers; +using ArcaneLibs.Extensions.Streams; +using ImageMagick; +using Microsoft.AspNetCore.Mvc; +using Microsoft.Build.Globbing; +using Spacebar.AdminApi.TestClient.Services.Services; + +namespace Spacebar.AdminApi.TestClient.Services.Controllers; + +[ApiController] +public class IsPixelArtController(LruFileCache lfc, IFileSource fs, PixelArtDetectionService pads, DiscordImageResizeService dirs) : ControllerBase { + private static readonly LruCache _isPixelArtCache = new(100_000); + private static readonly LruFileCache _edgeCache = new(100_000_000); + + [HttpGet("/isPixelArt/{*_:required}")] + public async Task IsPixelArt() { + return await _isPixelArtCache.GetOrAddAsync(Request.Path.ToString(), async () => { + await using var original = await fs.GetFile(Request.Path.ToString().Replace("/isPixelArt", "")); + using var img = await original.ToMagickImageCollectionAsync(); + return pads.IsPixelArt(img); + }); + } + + [HttpGet("/isCartoonArt/{*_:required}")] + public async Task IsCartoonArt() { + return await _isPixelArtCache.GetOrAddAsync(Request.Path.ToString(), async () => { + await using var original = await fs.GetFile(Request.Path.ToString().Replace("/isCartoonArt", "")); + using var img = await original.ToMagickImageCollectionAsync(); + return pads.IsCartoonArt(img); + }); + } + + [HttpGet("/edges/{*_:required}")] + public async Task GetEdges([FromQuery] string applyMode = "pre") { + return await _edgeCache.GetOrAdd(Request.Path.ToString() + Request.QueryString, async () => { + var original = await fs.GetFile(Request.Path.ToString().Replace("/edges", "")); + + DiscordImageResizeParams resizeParams = new() { + Size = Request.Query.ContainsKey("size") && uint.TryParse(Request.Query["size"], out uint size) ? size : null, + Quality = Request.Query.ContainsKey("quality") && Enum.TryParse(Request.Query["quality"], true, out var quality) ? quality : DiscordImageResizeQuality.High, + KeepAspectRatio = !Request.Query.ContainsKey("keepAspectRatio") || !bool.TryParse(Request.Query["keepAspectRatio"], out bool kar) || kar, + Passthrough = Request.Query.ContainsKey("passthrough") && bool.TryParse(Request.Query["passthrough"], out bool pt) && pt, + Animated = Request.Query.ContainsKey("animated") && bool.TryParse(Request.Query["animated"], out bool an) && an, + SpacebarAllowUpscale = Request.Query.ContainsKey("allowUpscale") && bool.TryParse(Request.Query["allowUpscale"], out bool au) && au, + SpacebarOptimiseGif = Request.Query.ContainsKey("optimiseGif") && bool.TryParse(Request.Query["optimiseGif"], out bool og) && og + }; + + + double radius = 1; + if (Request.Query.ContainsKey("radius")) double.TryParse(Request.Query["radius"], out radius); + + var img = await original.ToMagickImageCollectionAsync(); + + // if (applyMode == "pre") img = dirs.Apply(img,resizeParams); + + int inFrames = img.Count; + using var edged = pads.RenderEdges(img, radius); + // if (applyMode == "post") img = dirs.Apply(img,resizeParams); + + Console.WriteLine($"Generated edges for {Request.Path}, radius={radius}, inFrames={inFrames}, outFrames={edged.Count}"); + return new LruFileCache.Entry { + Data = edged.ToByteArray(), + // MimeType = Mimes.GetMime(img.First().Format), + MimeType = "image/apng" + }; + }).ContinueWith(t => File(t.Result.Data, t.Result.MimeType)); + } + + [HttpGet("/posterize/{*_:required}")] + public async Task Posterize() { + return await _edgeCache.GetOrAdd(Request.Path.ToString() + Request.QueryString, async () => { + var original = await fs.GetFile(Request.Path.ToString().Replace("/posterize", "")); + DiscordImageResizeParams resizeParams = new() { + Size = Request.Query.ContainsKey("size") && uint.TryParse(Request.Query["size"], out uint size) ? size : null, + Quality = Request.Query.ContainsKey("quality") && Enum.TryParse(Request.Query["quality"], true, out var quality) ? quality : DiscordImageResizeQuality.High, + KeepAspectRatio = !Request.Query.ContainsKey("keepAspectRatio") || !bool.TryParse(Request.Query["keepAspectRatio"], out bool kar) || kar, + Passthrough = Request.Query.ContainsKey("passthrough") && bool.TryParse(Request.Query["passthrough"], out bool pt) && pt, + Animated = Request.Query.ContainsKey("animated") && bool.TryParse(Request.Query["animated"], out bool an) && an, + SpacebarAllowUpscale = Request.Query.ContainsKey("allowUpscale") && bool.TryParse(Request.Query["allowUpscale"], out bool au) && au + }; + + + double radius = 1; + if (Request.Query.ContainsKey("radius")) double.TryParse(Request.Query["radius"], out radius); + + var img = await original.ToMagickImageCollectionAsync(); + + + int inFrames = img.Count; + // using var edged = pads.RenderEdges(img, radius); + foreach (var frame in img) { + // get major color count + frame.ColorFuzz = new Percentage(25); + frame.Posterize(16, DitherMethod.No); + frame.ColorFuzz = new Percentage(0); + } + + if(resizeParams.Size.HasValue) + img = dirs.Apply(img,resizeParams); + + Console.WriteLine($"Generated edges for {Request.Path}, radius={radius}, inFrames={inFrames}, outFrames={img.Count}"); + return new LruFileCache.Entry { + Data = img.ToByteArray(), + // MimeType = Mimes.GetMime(img.First().Format), + MimeType = "image/apng" + }; + }).ContinueWith(t => File(t.Result.Data, t.Result.MimeType)); + } + + [HttpGet("/colorFuzz/{*_:required}")] + public async Task ColorFuzz() { + return await _edgeCache.GetOrAdd(Request.Path.ToString() + Request.QueryString, async () => { + var original = await fs.GetFile(Request.Path.ToString().Replace("/colorFuzz", "")); + DiscordImageResizeParams resizeParams = new() { + Size = Request.Query.ContainsKey("size") && uint.TryParse(Request.Query["size"], out uint size) ? size : null, + Quality = Request.Query.ContainsKey("quality") && Enum.TryParse(Request.Query["quality"], true, out var quality) ? quality : DiscordImageResizeQuality.High, + KeepAspectRatio = !Request.Query.ContainsKey("keepAspectRatio") || !bool.TryParse(Request.Query["keepAspectRatio"], out bool kar) || kar, + Passthrough = Request.Query.ContainsKey("passthrough") && bool.TryParse(Request.Query["passthrough"], out bool pt) && pt, + Animated = Request.Query.ContainsKey("animated") && bool.TryParse(Request.Query["animated"], out bool an) && an, + SpacebarAllowUpscale = Request.Query.ContainsKey("allowUpscale") && bool.TryParse(Request.Query["allowUpscale"], out bool au) && au + }; + + + double colorFuzz = 1; + if (Request.Query.ContainsKey("colorFuzz")) double.TryParse(Request.Query["colorFuzz"], out colorFuzz); + + var img = await original.ToMagickImageCollectionAsync(); + + + int inFrames = img.Count; + // using var edged = pads.RenderEdges(img, radius); + foreach (var frame in img) { + // get major color count + frame.ColorFuzz = new Percentage(colorFuzz); + } + + if(resizeParams.Size.HasValue) + img = dirs.Apply(img,resizeParams); + + Console.WriteLine($"Generated colorFuzz for {Request.Path}, fuzz={colorFuzz}, inFrames={inFrames}, outFrames={img.Count}"); + return new LruFileCache.Entry { + Data = img.ToByteArray(), + // MimeType = Mimes.GetMime(img.First().Format), + MimeType = "image/apng" + }; + }).ContinueWith(t => File(t.Result.Data, t.Result.MimeType)); + } +} \ No newline at end of file -- cgit 1.5.1