using ArcaneLibs.Extensions.Streams; using Microsoft.AspNetCore.Mvc; using Spacebar.AdminApi.TestClient.Services.Services; using Spacebar.Cdn.Extensions; using Spacebar.Interop.Cdn.Abstractions; namespace Spacebar.Cdn.Controllers; [ApiController] public class GetImageController(LruFileCache lfc, IFileSource fs, DiscordImageResizeService dirs) : ImageController { // [HttpGet("/avatars/{_:required}")] [HttpGet("/emojis/{emoji_id:required}.{ext:required}")] [HttpGet("/stickers/{sticker_id:required}.{ext:required}")] // [HttpGet("/avatars/{user_id:required}/{avatar_hash:required}.{ext:required}")] [HttpGet("/banners/{user_id:required}/{user_banner:required}.{ext:required}")] public async Task GetImage(string? ext) { var originalKey = fs.BaseUrl + Request.Path; var cacheKey = Request.Path + Request.QueryString; DiscordImageResizeParams resizeParams = GetResizeParams(); var entry = await lfc.GetOrAdd(cacheKey, async () => { var original = await fs.GetFile(Request.Path); if (Request.Query.Any()) { using var img = await original.ToMagickImageCollectionAsync(); dirs.Apply(img, resizeParams); var outStream = new MemoryStream(); await img.WriteAsync(outStream, img.First().Format); outStream.Position = 0; return new LruFileCache.Entry() { Data = outStream.ReadToEnd().ToArray(), MimeType = original.MimeType }; } return new LruFileCache.Entry() { Data = original.Stream.ReadToEnd().ToArray(), MimeType = original.MimeType }; }); // byte array with mime type result return new FileContentResult(entry.Data, entry.MimeType); } // TODO: is message_id required? Can't tell from discord.food: /attachments/{channel_id}/[{message_id]/attachment_id}/{attachment_filename} [HttpGet("/attachments/{channel_id:required}/{message_id:required}/{attachment_id:required}/{filename:required}")] [HttpGet("/ephemeral-attachments/{application_id:required}/{attachment_id:required}/{attachment_filename:required}")] public async Task GetAttachmentImage() { // TODO: url signing, file type checks return await GetImage(""); } }