From 5976735af20185cc1b20305ffc61c78485494c67 Mon Sep 17 00:00:00 2001
From: pancakes <trypancakes@tutanota.com>
Date: Wed, 26 Feb 2025 18:14:29 +1000
Subject: [PATCH] [frontend/components] Fallback to detecting attachment type
 by filename extension

---
 .../Components/AttachmentComponent.razor              |  7 ++++---
 Iceshrimp.Frontend/Core/Miscellaneous/Constants.cs    | 11 +++++++++++
 2 files changed, 15 insertions(+), 3 deletions(-)
 create mode 100644 Iceshrimp.Frontend/Core/Miscellaneous/Constants.cs

diff --git a/Iceshrimp.Frontend/Components/AttachmentComponent.razor b/Iceshrimp.Frontend/Components/AttachmentComponent.razor
index 57c056f6..f035f3e8 100644
--- a/Iceshrimp.Frontend/Components/AttachmentComponent.razor
+++ b/Iceshrimp.Frontend/Components/AttachmentComponent.razor
@@ -1,4 +1,5 @@
 @using Iceshrimp.Assets.PhosphorIcons
+@using Iceshrimp.Frontend.Core.Miscellaneous
 @using Iceshrimp.Frontend.Core.Services
 @using Iceshrimp.Frontend.Localization
 @using Iceshrimp.Shared.Schemas.Web
@@ -7,7 +8,7 @@
 @inject IStringLocalizer<Localization> Loc;
 
 <div class="wrapper" @onclick="Open" @onclick:stopPropagation="true">
-    @if (Attachment.ContentType.StartsWith("image"))
+    @if (Attachment.ContentType.StartsWith("image") || Constants.CommonImageExtensions.Any(e => Attachment.FileName.EndsWith(e)))
     {
         <img class="attachment @(BlurImage ? "blur-image" : "")" src="@Attachment.Url" alt="@Attachment.AltText"
              fetchpriority="low"/>
@@ -24,11 +25,11 @@
     //  {
     //      TODO: Module player
     //  }
-    else if (Attachment.ContentType.StartsWith("audio"))
+    else if (Attachment.ContentType.StartsWith("audio") || Constants.CommonAudioExtensions.Any(e => Attachment.FileName.EndsWith(e)))
     {
         <audio controls class="attachment" src="@Attachment.Url"></audio>
     }
-    else if (Attachment.ContentType.StartsWith("video"))
+    else if (Attachment.ContentType.StartsWith("video") || Constants.CommonVideoExtensions.Any(e => Attachment.FileName.EndsWith(e)))
     {
         <video controls class="attachment">
             <source src="@Attachment.Url"/>
diff --git a/Iceshrimp.Frontend/Core/Miscellaneous/Constants.cs b/Iceshrimp.Frontend/Core/Miscellaneous/Constants.cs
new file mode 100644
index 00000000..71bd7883
--- /dev/null
+++ b/Iceshrimp.Frontend/Core/Miscellaneous/Constants.cs
@@ -0,0 +1,11 @@
+namespace Iceshrimp.Frontend.Core.Miscellaneous;
+
+public static class Constants
+{
+    public static readonly string[]
+        CommonImageExtensions = [".png", ".gif", ".jpeg", ".jpg", ".apng", ".bmp", ".tiff", ".avif", ".webp", ".jxl"];
+
+    public static readonly string[] CommonAudioExtensions = [".opus", ".ogg", ".m4a", ".mp3", ".flac", ".wav"];
+
+    public static readonly string[] CommonVideoExtensions = [".mov", ".mp4", ".webm", ".mkv"];
+}