iceshrimp-legacy/packages/backend/src/misc/check-word-mute.ts

51 lines
1.1 KiB
TypeScript
Raw Normal View History

2023-01-13 05:40:33 +01:00
import RE2 from "re2";
import type { Note } from "@/models/entities/note.js";
import type { User } from "@/models/entities/user.js";
type NoteLike = {
2023-01-13 05:40:33 +01:00
userId: Note["userId"];
text: Note["text"];
};
type UserLike = {
2023-01-13 05:40:33 +01:00
id: User["id"];
};
2023-01-13 05:40:33 +01:00
export async function checkWordMute(
note: NoteLike,
me: UserLike | null | undefined,
mutedWords: Array<string | string[]>,
): Promise<boolean> {
// 自分自身
2023-01-13 05:40:33 +01:00
if (me && note.userId === me.id) return false;
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
2023-01-13 05:40:33 +01:00
if (text === "") return false;
2023-01-13 05:40:33 +01:00
const matched = mutedWords.some((filter) => {
if (Array.isArray(filter)) {
2023-01-13 05:40:33 +01:00
return filter.every((keyword) => text.includes(keyword));
} else {
// represents RegExp
const regexp = filter.match(/^\/(.+)\/(.*)$/);
// This should never happen due to input sanitisation.
if (!regexp) return false;
try {
2022-06-23 13:26:47 +02:00
return new RE2(regexp[1], regexp[2]).test(text);
} catch (err) {
// This should never happen due to input sanitisation.
return false;
}
}
});
if (matched) return true;
}
return false;
}