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,40 +3,63 @@
* 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();
if (text === '') return false; const text = getNoteText(note);
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 !== '');
if (filteredFilter.length === 0) return false; if (filteredFilter.length === 0) return false;
return filteredFilter.every(keyword => text.includes(keyword)); return filteredFilter.every(keyword => text.includes(keyword));
} else { } else {
// represents RegExp // represents RegExp
const regexp = filter.match(/^\/(.+)\/(.*)$/); const regexp = filter.match(/^\/(.+)\/(.*)$/);
// This should never happen due to input sanitisation.
if (!regexp) return false;
try {
return new RegExp(regexp[1], regexp[2]).test(text);
} catch (err) {
// This should never happen due to input sanitisation. // This should never happen due to input sanitisation.
if (!regexp) return false; return false;
try {
return new RegExp(regexp[1], regexp[2]).test(text);
} catch (err) {
// This should never happen due to input sanitisation.
return false;
}
} }
}); }
});
if (matched) return true; }
}
function getNoteText(note: Note): string {
return false; 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();
} }