From ed5c5a715644e222e9d4bbf89aa09b33636cd3d0 Mon Sep 17 00:00:00 2001
From: Laura Hausmann <laura@hausmann.dev>
Date: Tue, 18 Jul 2023 17:08:13 +0200
Subject: [PATCH] [mastodon-client] fix mentions of remote local users without
 host

---
 packages/megalodon/src/misskey.ts | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/packages/megalodon/src/misskey.ts b/packages/megalodon/src/misskey.ts
index 491375ad84..a3afff7282 100644
--- a/packages/megalodon/src/misskey.ts
+++ b/packages/megalodon/src/misskey.ts
@@ -1514,19 +1514,35 @@ export default class Misskey implements MegalodonInterface {
 		if (status.quote != null)
 			status.quote = await this.addMentionsToStatus(status.quote, cache);
 
+		const idx = status.account.acct.indexOf('@');
+		const origin = idx < 0 ? null : status.account.acct.substring(idx + 1);
+
 		status.mentions = (
-			await this.getMentions(status.plain_content!, cache)
+			await this.getMentions(status.plain_content!, origin, cache)
 		).filter((p) => p != null);
 		for (const m of status.mentions.filter(
 			(value, index, array) => array.indexOf(value) === index,
 		)) {
-			if (m.acct == m.username)
+			const regexFull = new RegExp(`(?<=^|\\s|>)@${m.acct}(?=[^a-zA-Z0-9]|$)`, 'gi');
+			const regexLocalUser = 	new RegExp(`(?<=^|\\s|>)@${m.acct}@${this.baseUrlToHost(this.baseUrl)}(?=[^a-zA-Z0-9]|$)`, 'gi');
+			const regexRemoteUser = 	new RegExp(`(?<=^|\\s|>)@${m.username}(?=[^a-zA-Z0-9@]|$)`, 'gi');
+
+			if (m.acct == m.username) {
 				status.content = status.content.replace(
-					new RegExp(`(?<=^|\\s|>)@${m.acct}@${this.baseUrlToHost(this.baseUrl)}(?=[^a-zA-Z0-9]|$)`, 'g'),
+					regexLocalUser,
 					`@${m.acct}`,
 				);
+			}
+
+			else if (!status.content.match(regexFull)) {
+				status.content = status.content.replace(
+					regexRemoteUser,
+					`@${m.acct}`,
+				);
+			}
+
 			status.content = status.content.replace(
-				new RegExp(`(?<=^|\\s|>)@${m.acct}(?=[^a-zA-Z0-9]|$)`, 'g'),
+				regexFull,
 				`<a href="${m.url}" class="u-url mention" rel="nofollow noopener noreferrer" target="_blank">@${m.acct}</a>`,
 			);
 		}
@@ -1535,6 +1551,7 @@ export default class Misskey implements MegalodonInterface {
 
 	public async getMentions(
 		text: string,
+		origin: string | null,
 		cache: AccountCache,
 	): Promise<Entity.Mention[]> {
 		const mentions: Entity.Mention[] = [];
@@ -1551,7 +1568,7 @@ export default class Misskey implements MegalodonInterface {
 
 				const account = await this.getAccountByNameCached(
 					m.groups.user,
-					m.groups.host,
+					m.groups.host ?? origin,
 					cache,
 				);