From 67939860b231068f033800a8a58ae5393392a493 Mon Sep 17 00:00:00 2001 From: Cadence Ember Date: Wed, 14 Feb 2024 11:32:07 +1300 Subject: [PATCH] Consolidate convertEmoji function --- m2d/converters/event-to-message.js | 86 ++++++++++++++----------- m2d/converters/event-to-message.test.js | 51 ++++++++++++++- 2 files changed, 97 insertions(+), 40 deletions(-) diff --git a/m2d/converters/event-to-message.js b/m2d/converters/event-to-message.js index 8b601a4..6ddbd26 100644 --- a/m2d/converters/event-to-message.js +++ b/m2d/converters/event-to-message.js @@ -162,32 +162,8 @@ turndownService.addRule("emoji", { replacement: function (content, node) { const mxcUrl = node.getAttribute("src") - // Get the known emoji from the database. (We may not be able to actually use this if it was from another server.) - const row = select("emoji", ["emoji_id", "name", "animated"], {mxc_url: mxcUrl}).get() - // Also guess a suitable emoji based on the ID (if available) or name - let guess = null const guessedName = node.getAttribute("title").replace(/^:|:$/g, "") - for (const guild of discord.guilds.values()) { - /** @type {{name: string, id: string, animated: number}[]} */ - // @ts-ignore - const emojis = guild.emojis - const match = emojis.find(e => e.id === row?.emoji_id) || emojis.find(e => e.name === guessedName) || emojis.find(e => e.name?.toLowerCase() === guessedName.toLowerCase()) - if (match) { - guess = match - break - } - } - if (guess) { - // We know an emoji, and we can use it - const animatedChar = guess.animated ? "a" : "" - return `<${animatedChar}:${guess.name}:${guess.id}>` - } else if (endOfMessageEmojis.includes(mxcUrl)) { - // We can't locate or use a suitable emoji. After control returns, it will rewind over this, delete this section, and upload the emojis as a sprite sheet. - return `<::>` - } else { - // We prefer not to upload this as a sprite sheet because the emoji is not at the end of the message, it is in the middle. - return `[${node.getAttribute("title")}](${mxUtils.getPublicUrlForMxc(mxcUrl)})` - } + return convertEmoji(mxcUrl, guessedName, true, true) } }) @@ -216,6 +192,52 @@ turndownService.addRule("fencedCodeBlock", { } }) +/** + * @param {string | null} mxcUrl + * @param {string | null} nameForGuess without colons + * @param {boolean} allowSpriteSheetIndicator + * @param {boolean} allowLink + * @returns {string} discord markdown that represents the custom emoji in some form + */ +function convertEmoji(mxcUrl, nameForGuess, allowSpriteSheetIndicator, allowLink) { + // Get the known emoji from the database. + let row + if (mxcUrl) row = select("emoji", ["emoji_id", "name", "animated"], {mxc_url: mxcUrl}).get() + if (!row && nameForGuess) { + // We don't know the emoji, but we could guess a suitable emoji based on the name + const nameForGuessLower = nameForGuess.toLowerCase() + for (const guild of discord.guilds.values()) { + /** @type {{name: string, id: string, animated: number}[]} */ + // @ts-ignore + const emojis = guild.emojis + const found = emojis.find(e => e.name?.toLowerCase() === nameForGuessLower) + if (found) { + row = { + animated: found.animated, + emoji_id: found.id, + name: found.name + } + break + } + } + } + if (row) { + // We know an emoji, and we can use it + const animatedChar = row.animated ? "a" : "" + return `<${animatedChar}:${row.name}:${row.emoji_id}>` + } else if (allowSpriteSheetIndicator && mxcUrl && endOfMessageEmojis.includes(mxcUrl)) { + // We can't locate or use a suitable emoji. After control returns, it will rewind over this, delete this section, and upload the emojis as a sprite sheet. + return `<::>` + } else if (allowLink && mxcUrl && nameForGuess) { + // We prefer not to upload this as a sprite sheet because the emoji is not at the end of the message, it is in the middle. + return `[:${nameForGuess}:](${mxUtils.getPublicUrlForMxc(mxcUrl)})` + } else if (nameForGuess) { + return `:${nameForGuess}:` + } else { + return "" + } +} + /** * @param {string} roomID * @param {string} mxid @@ -530,21 +552,9 @@ async function eventToMessage(event, guild, di) { repliedToContent = repliedToContent.replace(/(?:\n|
)+/g, " ") // Should all be on one line repliedToContent = repliedToContent.replace(/]*data-mx-spoiler\b[^>]*>.*?<\/span>/g, "[spoiler]") // Good enough method of removing spoiler content. (I don't want to break out the HTML parser unless I have to.) repliedToContent = repliedToContent.replace(/]*)>/g, (_, att) => { // Convert Matrix emoji images into Discord emoji markdown - if (!att.includes("data-mx-emoticon")) return "" - // Try to get the equivalent Discord emoji, if there is a src and if we know about it const mxcUrlMatch = att.match(/\bsrc="(mxc:\/\/[^"]+)"/) - if (mxcUrlMatch) { - const row = select("emoji", ["emoji_id", "name", "animated"], {mxc_url: mxcUrlMatch[1]}).get() - if (row) { - const animatedChar = row.animated ? "a" : "" - return `<${animatedChar}:${row.name}:${row.emoji_id}>` - } - } - // Emoji is unknown or inaccessible, try substituting the title text instead const titleTextMatch = att.match(/\btitle=":?([^:"]+)/) - if (titleTextMatch) return `:${titleTextMatch[1]}:` - // Otherwise we can't use the emoji. - return "" + return convertEmoji(mxcUrlMatch?.[1], titleTextMatch?.[1], false, false) }) repliedToContent = repliedToContent.replace(/<[^:>][^>]*>/g, "") // Completely strip all HTML tags and formatting. repliedToContent = entities.decodeHTML5Strict(repliedToContent) // Remove entities like & " diff --git a/m2d/converters/event-to-message.test.js b/m2d/converters/event-to-message.test.js index 2c42c3e..7cc0fdd 100644 --- a/m2d/converters/event-to-message.test.js +++ b/m2d/converters/event-to-message.test.js @@ -1339,7 +1339,7 @@ test("event2message: reply preview converts emoji formatting when replying to a ) }) -test("event2message: reply preview uses emoji title text when replying to an unknown custom emoji", async t => { +test("event2message: reply preview can guess custom emoji based on the name if it is unknown", async t => { t.deepEqual( await eventToMessage({ type: "m.room.message", @@ -1378,7 +1378,54 @@ test("event2message: reply preview uses emoji title text when replying to an unk messagesToSend: [{ username: "cadence [they]", content: "> <:L1:1144820033948762203><:L2:1144820084079087647>Ⓜ️**cadence [they]**:" - + "\n> :hippo:" + + "\n> <:hippo:230201364309868544>" + + "\nreply", + avatar_url: undefined + }] + } + ) +}) + +test("event2message: reply preview uses emoji title text when replying to an unknown custom emoji", async t => { + t.deepEqual( + await eventToMessage({ + type: "m.room.message", + sender: "@cadence:cadence.moe", + content: { + msgtype: "m.text", + body: "> <@cadence:cadence.moe> :svkftngur_gkdne:\n\nreply", + format: "org.matrix.custom.html", + formatted_body: "
In reply to @cadence:cadence.moe
\":svkftngur_gkdne:\"
reply", + "m.relates_to": { + "m.in_reply_to": { + event_id: "$zmO-dtPO6FubBkDxJZ5YmutPIsG1RgV5JJku-9LeGWs" + } + } + }, + event_id: "$bCMLaLiMfoRajaGTgzaxAci-g8hJfkspVJIKwYktnvc", + room_id: "!TqlyQmifxGUggEmdBN:cadence.moe" + }, data.guild.general, { + api: { + getEvent: mockGetEvent(t, "!TqlyQmifxGUggEmdBN:cadence.moe", "$zmO-dtPO6FubBkDxJZ5YmutPIsG1RgV5JJku-9LeGWs", { + type: "m.room.message", + sender: "@cadence:cadence.moe", + content: { + msgtype: "m.text", + body: ":svkftngur_gkdne:", + format: "org.matrix.custom.html", + formatted_body: "\":svkftngur_gkdne:\"" + } + }) + } + }), + { + ensureJoined: [], + messagesToDelete: [], + messagesToEdit: [], + messagesToSend: [{ + username: "cadence [they]", + content: "> <:L1:1144820033948762203><:L2:1144820084079087647>Ⓜ️**cadence [they]**:" + + "\n> :svkftngur_gkdne:" + "\nreply", avatar_url: undefined }]