From bb4e418dde2176f3d829e5f19859832bbf248707 Mon Sep 17 00:00:00 2001
From: Julian Lam <julian@nodebb.org>
Date: Fri, 12 Apr 2024 12:56:18 -0400
Subject: [PATCH] fix: change default attachment type to Link, or Image if
 mediaType matches

---
 src/activitypub/mocks.js | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/src/activitypub/mocks.js b/src/activitypub/mocks.js
index f9bcad3f2b..f0bca1f134 100644
--- a/src/activitypub/mocks.js
+++ b/src/activitypub/mocks.js
@@ -282,11 +282,27 @@ Mocks.note = async (post) => {
 	}
 
 	let attachment = await posts.attachments.get(post.pid) || [];
-	attachment = attachment.map(({ mediaType, url }) => ({
-		type: 'Document',
-		mediaType,
-		url,
-	}));
+	attachment = attachment.map(({ mediaType, url }) => {
+		let type;
+
+		switch (true) {
+			case mediaType.startsWith('image'): {
+				type = 'Image';
+				break;
+			}
+
+			default: {
+				type = 'Link';
+				break;
+			}
+		}
+
+		return {
+			type,
+			mediaType,
+			url,
+		};
+	});
 
 	const object = {
 		'@context': 'https://www.w3.org/ns/activitystreams',