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

65 lines
1.4 KiB
TypeScript
Raw Normal View History

export type Muted = {
muted: boolean;
matched: string[];
};
const NotMuted = { muted: false, matched: [] };
function escapeRegExp(x: string) {
return x.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"); // $& means the whole matched string
}
export function getWordMute(
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-01-13 05:40:33 +01:00
const text = ((note.cw ?? "") + "\n" + (note.text ?? "")).trim();
2022-06-23 13:26:47 +02:00
if (text === "") {
return NotMuted;
}
for (const mutePattern of mutedWords) {
let mute: RegExp;
let matched: string[];
if (Array.isArray(mutePattern)) {
matched = mutePattern.filter((keyword) => keyword !== "");
if (matched.length === 0) {
continue;
}
mute = new RegExp(
`\\b${matched.map(escapeRegExp).join("\\b.*\\b")}\\b`,
"g",
);
} else {
const regexp = mutePattern.match(/^\/(.+)\/(.*)$/);
// This should never happen due to input sanitisation.
if (!regexp) {
console.warn(`Found invalid regex in word mutes: ${mutePattern}`);
continue;
}
mute = new RegExp(regexp[1], regexp[2]);
matched = [mutePattern];
}
try {
if (mute.test(text)) {
return { muted: true, matched };
}
} catch (err) {
// This should never happen due to input sanitisation.
}
}
}
return NotMuted;
}