62 lines
1.3 KiB
TypeScript
62 lines
1.3 KiB
TypeScript
import RE2 from 're2';
|
|
import { Note } from '@/models/entities/note.js';
|
|
import { User } from '@/models/entities/user.js';
|
|
|
|
type NoteLike = {
|
|
userId: Note['userId'];
|
|
text: Note['text'];
|
|
cw: Note['cw'];
|
|
};
|
|
|
|
type UserLike = {
|
|
id: User['id'];
|
|
};
|
|
|
|
export async function checkWordMute(note: NoteLike, me: UserLike | null | undefined, mutedWords: Array<string | string[]>): Promise<boolean> {
|
|
// own posts
|
|
if (me && (note.userId === me.id)) return false;
|
|
|
|
if (mutedWords.length > 0) {
|
|
const text = [
|
|
note.cw,
|
|
note.text,
|
|
...note.files.map(file => file.comment)
|
|
]
|
|
.map(x => {
|
|
if (x == null || x.trim() == '') {
|
|
return null;
|
|
} else {
|
|
return x.trim();
|
|
}
|
|
})
|
|
.filter(x => x != null)
|
|
.join('\n');
|
|
|
|
if (text === '') return false;
|
|
const textLower = text.toLowerCase();
|
|
|
|
const matched = mutedWords.some(filter => {
|
|
if (Array.isArray(filter)) {
|
|
return filter.every(keyword => textLower.includes(keyword.toLowerCase()));
|
|
} else {
|
|
// represents RegExp
|
|
const regexp = filter.match(/^\/(.+)\/(.*)$/);
|
|
|
|
// This should never happen due to input sanitisation.
|
|
if (!regexp) return false;
|
|
|
|
try {
|
|
return new RE2(regexp[1], regexp[2]).test(text);
|
|
} catch (err) {
|
|
// This should never happen due to input sanitisation.
|
|
return false;
|
|
}
|
|
}
|
|
});
|
|
|
|
if (matched) return true;
|
|
}
|
|
|
|
return false;
|
|
}
|