Hazelnoot
73 lines
2.1 KiB
TypeScript
73 lines
2.1 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
import * as Misskey from 'misskey-js';
|
|
|
|
export function checkWordMute(note: string | Misskey.entities.Note, me: Misskey.entities.UserLite | null | undefined, mutedWords: Array<string | string[]>): Array<string | string[]> | false {
|
|
// 自分自身
|
|
if (me && typeof(note) === 'object' && (note.userId === me.id)) return false;
|
|
|
|
if (mutedWords.length > 0) {
|
|
const text = typeof(note) === 'object' ? getNoteText(note) : note;
|
|
|
|
if (text === '') return false;
|
|
|
|
const matched = mutedWords.reduce((matchedWords, filter) => {
|
|
if (Array.isArray(filter)) {
|
|
// Clean up
|
|
const filteredFilter = filter.filter(keyword => keyword !== '');
|
|
if (filteredFilter.length > 0 && filteredFilter.every(keyword => text.includes(keyword))) {
|
|
const fullFilter = filteredFilter.join(' ');
|
|
matchedWords.add(fullFilter);
|
|
}
|
|
} else {
|
|
// represents RegExp
|
|
const regexp = filter.match(/^\/(.+)\/(.*)$/);
|
|
|
|
// This should never happen due to input sanitisation.
|
|
if (regexp) {
|
|
try {
|
|
const flags = regexp[2].includes('g') ? regexp[2] : (regexp[2] + 'g');
|
|
const matches = text.matchAll(new RegExp(regexp[1], flags));
|
|
for (const match of matches) {
|
|
matchedWords.add(match[0]);
|
|
}
|
|
} catch {
|
|
// This should never happen due to input sanitisation.
|
|
}
|
|
}
|
|
}
|
|
|
|
return matchedWords;
|
|
}, new Set<string>());
|
|
|
|
// Nested arrays are intentional, otherwise the note components will join with space (" ") and it's confusing.
|
|
if (matched.size > 0) return [[Array.from(matched).join(', ')]];
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
function getNoteText(note: Misskey.entities.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();
|
|
}
|