feat: check polls and media for muted keywords

packages/frontend/src/scripts/check-word-mute.ts

@@ -3,16 +3,22 @@
  * SPDX-License-Identifier: AGPL-3.0-only
  */
 
-export function checkWordMute(note: Record<string, any>, me: Record<string, any> | null | undefined, mutedWords: Array<string | string[]>): boolean {
+import type { Note, MeDetailed } from "misskey-js/entities.js";
+
+// TODO: this implementation is horribly inefficient.
+// Each regex is validated (using a regex, ironically), transformed, and then parsed - for each note being checked.
+// These regex objects should be cached somewhere.
+
+export function checkWordMute(note: Note, me: MeDetailed | null | undefined, mutedWords: Array<string | string[]>): boolean {
 	// 自分自身
 	if (me && (note.userId === me.id)) return false;
 
-	if (mutedWords.length > 0) {
+	if (mutedWords.length < 1) return false;
-		const text = ((note.cw ?? '') + '\n' + (note.text ?? '')).trim();
 
+	const text = getNoteText(note);
 	if (text === '') return false;
 
-		const matched = mutedWords.some(filter => {
+	return mutedWords.some(filter => {
 		if (Array.isArray(filter)) {
 			// Clean up
 			const filteredFilter = filter.filter(keyword => keyword !== '');
@@ -34,9 +40,26 @@ export function checkWordMute(note: Record<string, any>, me: Record<string, any>
 			}
 		}
 	});
-
-		if (matched) return true;
 }
 
-	return false;
+function getNoteText(note: Note): string {
+	const textParts: string[] = [];
+
+	if (note.cw)
+		textParts.push(note.cw);
+
+	if (note.text)
+		textParts.push(note.text);
+
+	if (note.files)
+		for (const file of note.files)
+			if (file.comment)
+				textParts.push(file.comment);
+
+	if (note.poll)
+		for (const choice of note.poll.choices)
+			if (choice.text)
+				textParts.push(choice.text);
+
+	return textParts.join('\n').trim();
 }