fix(post): some links to images could be embedded as videos

same as plebbit/seedit#385
plebbit · Oct 11, 2024 · 46e1189 · 46e1189
1 parent 561e395
commit 46e1189
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/src/lib/utils/media-utils.ts b/src/lib/utils/media-utils.ts
@@ -80,15 +80,17 @@ export const getLinkMediaInfo = memoize(
     }
 
     try {
-      const fileName = url.pathname.slice(url.pathname.lastIndexOf('/') + 1).toLowerCase();
-      mime = extName(fileName)[0]?.mime;
-      if (mime) {
-        if (mime.startsWith('image')) {
-          type = mime === 'image/gif' ? 'gif' : 'image';
-        } else if (mime.startsWith('video')) {
-          type = 'video';
-        } else if (mime.startsWith('audio')) {
-          type = 'audio';
+      const fileName = url.pathname.split('/').pop() || '';
+      if (fileName.includes('.')) {
+        mime = extName(fileName)[0]?.mime;
+        if (mime) {
+          if (mime.startsWith('image')) {
+            type = mime === 'image/gif' ? 'gif' : 'image';
+          } else if (mime.startsWith('video')) {
+            type = 'video';
+          } else if (mime.startsWith('audio')) {
+            type = 'audio';
+          }
         }
       }