perf: emoji lib performance fix

This commit is contained in:
ThatOneCalculator 2023-02-17 00:01:22 -08:00
parent 3c091643cc
commit ed3ef69b76

View file

@ -4,73 +4,68 @@ import { Emojis } from "@/models/index.js";
import { toPunyNullable } from "./convert-host.js"; import { toPunyNullable } from "./convert-host.js";
import { IsNull } from "typeorm"; import { IsNull } from "typeorm";
const legacies: Record<string, string> = { const legacies = new Map([
like: "👍", ['like', '👍'],
love: "❤️", // ここに記述する場合は異体字セレクタを入れない <- not that good because modern browsers just display it as the red heart so just convert it to it to not end up with two seperate reactions of "the same emoji" for the user ['love', '❤️'],
laugh: "😆", ['laugh', '😆'],
hmm: "🤔", ['hmm', '🤔'],
surprise: "😮", ['surprise', '😮'],
congrats: "🎉", ['congrats', '🎉'],
angry: "💢", ['angry', '💢'],
confused: "😥", ['confused', '😥'],
rip: "😇", ['rip', '😇'],
pudding: "🍮", ['pudding', '🍮'],
star: "⭐", ['star', '⭐'],
}; ]);
export async function getFallbackReaction(): Promise<string> { export async function getFallbackReaction() {
const meta = await fetchMeta(); const meta = await fetchMeta();
return meta.defaultReaction; return meta.defaultReaction;
} }
export function convertLegacyReactions(reactions: Record<string, number>) { export function convertLegacyReactions(reactions: Record<string, number>) {
const _reactions = {} as Record<string, number>; const _reactions = new Map();
const decodedReactions = new Map();
for (const reaction of Object.keys(reactions)) { for (const reaction in reactions) {
if (reactions[reaction] <= 0) continue; if (reactions[reaction] <= 0) continue;
if (Object.keys(legacies).includes(reaction)) { let decodedReaction;
if (_reactions[legacies[reaction]]) { if (decodedReactions.has(reaction)) {
_reactions[legacies[reaction]] += reactions[reaction]; decodedReaction = decodedReactions.get(reaction);
} else { } else {
_reactions[legacies[reaction]] = reactions[reaction]; decodedReaction = decodeReaction(reaction);
decodedReactions.set(reaction, decodedReaction);
} }
} else if (reaction === "♥️") {
if (_reactions["❤️"]) { let emoji = legacies.get(decodedReaction.reaction);
_reactions["❤️"] += reactions[reaction]; if (emoji) {
_reactions.set(emoji, (_reactions.get(emoji) || 0) + reactions[reaction]);
} else { } else {
_reactions["❤️"] = reactions[reaction]; _reactions.set(reaction, (_reactions.get(reaction) || 0) + reactions[reaction]);
}
} else {
if (_reactions[reaction]) {
_reactions[reaction] += reactions[reaction];
} else {
_reactions[reaction] = reactions[reaction];
}
} }
} }
const _reactions2 = {} as Record<string, number>; const _reactions2 = new Map();
for (const [reaction, count] of _reactions) {
for (const reaction of Object.keys(_reactions)) { const decodedReaction = decodedReactions.get(reaction);
_reactions2[decodeReaction(reaction).reaction] = _reactions[reaction]; _reactions2.set(decodedReaction.reaction, count);
} }
return _reactions2; return Object.fromEntries(_reactions2);
} }
export async function toDbReaction( export async function toDbReaction(
reaction?: string | null, reaction?: string | null,
reacterHost?: string | null, reacterHost?: string | null,
): Promise<string> { ): Promise<string> {
if (reaction == null) return await getFallbackReaction(); if (!reaction) return await getFallbackReaction();
reacterHost = toPunyNullable(reacterHost); reacterHost = toPunyNullable(reacterHost);
// Convert string-type reactions to unicode // Convert string-type reactions to unicode
if (Object.keys(legacies).includes(reaction)) return legacies[reaction]; const emoji = legacies.get(reaction) || (reaction === "♥️" ? "❤️" : null);
// Convert old heart to new if (emoji) return emoji;
if (reaction === "♥️") return "❤️";
// Allow unicode reactions // Allow unicode reactions
const match = emojiRegex.exec(reaction); const match = emojiRegex.exec(reaction);
@ -83,7 +78,7 @@ export async function toDbReaction(
if (custom) { if (custom) {
const name = custom[1]; const name = custom[1];
const emoji = await Emojis.findOneBy({ const emoji = await Emojis.findOneBy({
host: reacterHost ?? IsNull(), host: reacterHost || IsNull(),
name, name,
}); });
@ -132,7 +127,7 @@ export function decodeReaction(str: string): DecodedReaction {
} }
export function convertLegacyReaction(reaction: string): string { export function convertLegacyReaction(reaction: string): string {
reaction = decodeReaction(reaction).reaction; const decoded = decodeReaction(reaction).reaction;
if (Object.keys(legacies).includes(reaction)) return legacies[reaction]; if (legacies.has(decoded)) return legacies.get(decoded)!;
return reaction; return decoded;
} }