feat: check polls and media for muted keywords

This commit is contained in:
Hazel Koehler 2024-05-02 20:31:34 -04:00
parent d0a2708f91
commit 26f8919432

View file

@ -3,16 +3,22 @@
* SPDX-License-Identifier: AGPL-3.0-only * 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 (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; if (text === '') return false;
const matched = mutedWords.some(filter => { return mutedWords.some(filter => {
if (Array.isArray(filter)) { if (Array.isArray(filter)) {
// Clean up // Clean up
const filteredFilter = filter.filter(keyword => keyword !== ''); 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();
} }