feat: check polls and media for muted keywords
This commit is contained in:
parent
d0a2708f91
commit
26f8919432
1 changed files with 50 additions and 27 deletions
|
@ -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();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue