2023-05-05 19:49:34 +09:00
|
|
|
import { AhoCorasick } from 'slacc';
|
2022-02-27 11:07:39 +09:00
|
|
|
import RE2 from 're2';
|
2022-09-18 03:27:08 +09:00
|
|
|
import type { Note } from '@/models/entities/Note.js';
|
|
|
|
import type { User } from '@/models/entities/User.js';
|
2020-07-27 13:34:20 +09:00
|
|
|
|
|
|
|
type NoteLike = {
|
|
|
|
userId: Note['userId'];
|
|
|
|
text: Note['text'];
|
2022-09-24 14:45:44 +09:00
|
|
|
cw?: Note['cw'];
|
2020-07-27 13:34:20 +09:00
|
|
|
};
|
|
|
|
|
|
|
|
type UserLike = {
|
|
|
|
id: User['id'];
|
|
|
|
};
|
|
|
|
|
2023-05-05 19:49:34 +09:00
|
|
|
const acCache = new Map<string, AhoCorasick>();
|
|
|
|
|
2022-02-10 19:47:46 +09:00
|
|
|
export async function checkWordMute(note: NoteLike, me: UserLike | null | undefined, mutedWords: Array<string | string[]>): Promise<boolean> {
|
2020-07-27 13:34:20 +09:00
|
|
|
// 自分自身
|
|
|
|
if (me && (note.userId === me.id)) return false;
|
|
|
|
|
2022-02-10 19:47:46 +09:00
|
|
|
if (mutedWords.length > 0) {
|
2022-06-23 20:26:47 +09:00
|
|
|
const text = ((note.cw ?? '') + '\n' + (note.text ?? '')).trim();
|
|
|
|
|
2022-06-24 19:44:22 +09:00
|
|
|
if (text === '') return false;
|
2020-07-27 13:34:20 +09:00
|
|
|
|
2023-05-05 19:49:34 +09:00
|
|
|
const acable = mutedWords.filter(filter => Array.isArray(filter) && filter.length === 1).map(filter => filter[0]).sort();
|
|
|
|
const unacable = mutedWords.filter(filter => !Array.isArray(filter) || filter.length !== 1);
|
|
|
|
const acCacheKey = acable.join('\n');
|
|
|
|
const ac = acCache.get(acCacheKey) ?? AhoCorasick.withPatterns(acable);
|
|
|
|
acCache.delete(acCacheKey);
|
|
|
|
for (const obsoleteKeys of acCache.keys()) {
|
|
|
|
if (acCache.size > 1000) {
|
|
|
|
acCache.delete(obsoleteKeys);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
acCache.set(acCacheKey, ac);
|
|
|
|
if (ac.isMatch(text)) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
const matched = unacable.some(filter => {
|
2022-02-10 19:47:46 +09:00
|
|
|
if (Array.isArray(filter)) {
|
2022-06-23 20:26:47 +09:00
|
|
|
return filter.every(keyword => text.includes(keyword));
|
2022-02-10 19:47:46 +09:00
|
|
|
} else {
|
|
|
|
// represents RegExp
|
|
|
|
const regexp = filter.match(/^\/(.+)\/(.*)$/);
|
|
|
|
|
|
|
|
// This should never happen due to input sanitisation.
|
|
|
|
if (!regexp) return false;
|
|
|
|
|
|
|
|
try {
|
2022-06-23 20:26:47 +09:00
|
|
|
return new RE2(regexp[1], regexp[2]).test(text);
|
2022-02-10 19:47:46 +09:00
|
|
|
} catch (err) {
|
|
|
|
// This should never happen due to input sanitisation.
|
|
|
|
return false;
|
2020-07-27 13:34:20 +09:00
|
|
|
}
|
2022-02-10 19:47:46 +09:00
|
|
|
}
|
|
|
|
});
|
2020-07-27 13:34:20 +09:00
|
|
|
|
|
|
|
if (matched) return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|