From 44572badccb2207f24b654a0634a8c6dd338ef71 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Tue, 17 Dec 2024 22:22:04 +0800
Subject: [PATCH] :sparkles: Support more type of attachments

---
 src/components/AttachmentRenderer.astro | 29 +++++++++++++++++++
 src/pages/posts/[...slug].astro         | 38 +++++++++++++------------
 2 files changed, 49 insertions(+), 18 deletions(-)
 create mode 100644 src/components/AttachmentRenderer.astro

diff --git a/src/components/AttachmentRenderer.astro b/src/components/AttachmentRenderer.astro
new file mode 100644
index 0000000..c136c17
--- /dev/null
+++ b/src/components/AttachmentRenderer.astro
@@ -0,0 +1,29 @@
+---
+import { getAttachmentUrl } from '../scripts/attachment'
+
+interface Props {
+  data: any
+}
+
+const { data: attachment } = Astro.props
+---
+
+{
+  attachment.mimetype.startsWith('image') ? (
+    <a href={getAttachmentUrl(attachment.rid)} target="_blank">
+      <img
+        src={getAttachmentUrl(attachment.rid)}
+        alt={attachment.alt}
+        class="rounded-lg"
+      />
+    </a>
+  ) : attachment.mimetype.startsWith('video') ? (
+    <video src={getAttachmentUrl(attachment.rid)} controls class="rounded-lg" />
+  ) : attachment.mimetype.startsWith('audio') ? (
+    <audio src={getAttachmentUrl(attachment.rid)} controls class="rounded-lg" />
+  ) : (
+    <a href={getAttachmentUrl(attachment.rid)} target="_blank">
+      Unable to preview, but you can open it in your broswer.
+    </a>
+  )
+}
diff --git a/src/pages/posts/[...slug].astro b/src/pages/posts/[...slug].astro
index ae60666..d721e21 100644
--- a/src/pages/posts/[...slug].astro
+++ b/src/pages/posts/[...slug].astro
@@ -6,6 +6,7 @@ import { Icon } from 'astro-icon/components'
 import { marked } from 'marked'
 
 import Layout from '../../layouts/Layout.astro'
+import AttachmentRenderer from '../../components/AttachmentRenderer.astro'
 import { getAttachmentUrl, fetchAttachmentMeta } from '../../scripts/attachment'
 
 const { slug } = Astro.params
@@ -85,24 +86,25 @@ const attachments = await fetchAttachmentMeta(data.body.attachments)
       )
     }
 
-    <article class="prose max-w-none max-md:prose-lg" set:html={content} />
+    <article>
+      <div class="prose max-w-none max-md:prose-lg" set:html={content} />
 
-    {
-      attachments && (
-        <div class="attachment-list mt-5 gap-4 grid grid-cols-1 md:grid-cols-2">
-          {attachments.map((attachment) => (
-            <div class="attachment">
-              <a href={getAttachmentUrl(attachment.rid)} target="_blank">
-                <img
-                  src={getAttachmentUrl(attachment.rid)}
-                  alt={attachment.alt}
-                  class="rounded-lg"
-                />
-              </a>
-            </div>
-          ))}
-        </div>
-      )
-    }
+      {
+        attachments && (
+          <div
+            class="attachment-list mt-5 gap-4 grid grid-cols-1"
+            class:list={
+              attachments.length >= 2 ? 'md:grid-cols-2' : 'md:grid-cols-1'
+            }
+          >
+            {attachments.map((attachment) => (
+              <div class="attachment">
+                <AttachmentRenderer data={attachment} />
+              </div>
+            ))}
+          </div>
+        )
+      }
+    </article>
   </div>
 </Layout>