rudeshark.net/packages/client/src/scripts/check-word-mute.ts

91 lines
2.0 KiB
TypeScript
Raw Normal View History

export type Muted = {
muted: boolean;
matched: string[];
2023-05-04 07:41:18 +02:00
what?: string; // "note" || "reply" || "renote" || "quote"
};
const NotMuted = { muted: false, matched: [] };
2023-05-04 22:17:16 +02:00
function checkWordMute(
note: NoteLike,
mutedWords: Array<string | string[]>,
): Muted {
2023-05-04 07:13:13 +02:00
const text = ((note.cw ?? "") + " " + (note.text ?? "")).trim();
if (text === "") return NotMuted;
2023-05-05 05:23:44 +02:00
let result = { muted: false, matched: [] };
2023-05-04 07:13:13 +02:00
for (const mutePattern of mutedWords) {
if (Array.isArray(mutePattern)) {
2023-05-05 05:23:44 +02:00
// Clean up
const keywords = mutePattern.filter((keyword) => keyword !== "");
if (
keywords.length > 0 &&
keywords.every((keyword) => text.includes(keyword))
) {
result.muted = true;
result.matched.push(...keywords);
2023-05-04 07:13:13 +02:00
}
} else {
2023-05-05 05:23:44 +02:00
// represents RegExp
2023-05-04 07:13:13 +02:00
const regexp = mutePattern.match(/^\/(.+)\/(.*)$/);
2023-05-05 05:23:44 +02:00
2023-05-04 07:13:13 +02:00
// This should never happen due to input sanitisation.
if (!regexp) {
console.warn(`Found invalid regex in word mutes: ${mutePattern}`);
continue;
}
2023-05-05 05:23:44 +02:00
try {
if (new RegExp(regexp[1], regexp[2]).test(text)) {
result.muted = true;
result.matched.push(mutePattern);
}
} catch (err) {
// This should never happen due to input sanitisation.
2023-05-04 07:13:13 +02:00
}
}
}
2023-05-04 22:22:32 +02:00
2023-05-05 05:23:44 +02:00
result.matched = [...new Set(result.matched)];
return result;
2023-05-04 07:13:13 +02:00
}
export function getWordSoftMute(
2023-01-13 05:40:33 +01:00
note: Record<string, any>,
me: Record<string, any> | null | undefined,
mutedWords: Array<string | string[]>,
): Muted {
// 自分自身
if (me && note.userId === me.id) {
return NotMuted;
}
if (mutedWords.length > 0) {
2023-05-04 22:17:16 +02:00
let noteMuted = checkWordMute(note, mutedWords);
2023-05-04 07:13:13 +02:00
if (noteMuted.muted) {
noteMuted.what = "note";
return noteMuted;
}
2023-05-04 07:13:13 +02:00
if (note.renote) {
2023-05-04 22:17:16 +02:00
let renoteMuted = checkWordMute(note.renote, mutedWords);
2023-05-04 07:13:13 +02:00
if (renoteMuted.muted) {
2023-05-04 07:41:18 +02:00
renoteMuted.what = note.text == null ? "renote" : "quote";
2023-05-04 07:13:13 +02:00
return renoteMuted;
}
}
2023-05-05 01:17:45 +02:00
if (note.reply) {
let replyMuted = checkWordMute(note.reply, mutedWords);
if (replyMuted.muted) {
replyMuted.what = "reply";
return replyMuted;
}
}
}
return NotMuted;
}