forked from FoundKeyGang/FoundKey
Johann150
d1ec058d5c
Instead of having to pass the fetcher every time you want to fetch something, the fetcher is stored in an attribute of the Cache.
104 lines
3.5 KiB
TypeScript
104 lines
3.5 KiB
TypeScript
import { Antenna } from '@/models/entities/antenna.js';
|
|
import { Note } from '@/models/entities/note.js';
|
|
import { User } from '@/models/entities/user.js';
|
|
import { UserListJoinings, UserGroupJoinings, Blockings } from '@/models/index.js';
|
|
import * as Acct from '@/misc/acct.js';
|
|
import { MINUTE } from '@/const.js';
|
|
import { getFullApAccount } from './convert-host.js';
|
|
import { Packed } from './schema.js';
|
|
import { Cache } from './cache.js';
|
|
|
|
const blockingCache = new Cache<User['id'][]>(
|
|
5 * MINUTE,
|
|
(blockerId) => Blockings.findBy({ blockerId }).then(res => res.map(x => x.blockeeId)),
|
|
);
|
|
|
|
// designation for users you follow, list users and groups is disabled for performance reasons
|
|
|
|
/**
|
|
* either noteUserFollowers or antennaUserFollowing must be specified
|
|
*/
|
|
export async function checkHitAntenna(antenna: Antenna, note: (Note | Packed<'Note'>), noteUser: { id: User['id']; username: string; host: string | null; }, noteUserFollowers?: User['id'][], antennaUserFollowing?: User['id'][]): Promise<boolean> {
|
|
if (note.visibility === 'specified') return false;
|
|
|
|
// skip if the antenna creator is blocked by the note author
|
|
const blockings = await blockingCache.fetch(noteUser.id);
|
|
if (blockings.some(blocking => blocking === antenna.userId)) return false;
|
|
|
|
if (note.visibility === 'followers') {
|
|
if (noteUserFollowers && !noteUserFollowers.includes(antenna.userId)) return false;
|
|
if (antennaUserFollowing && !antennaUserFollowing.includes(note.userId)) return false;
|
|
}
|
|
|
|
if (!antenna.withReplies && note.replyId != null) return false;
|
|
|
|
if (antenna.src === 'home') {
|
|
if (noteUserFollowers && !noteUserFollowers.includes(antenna.userId)) return false;
|
|
if (antennaUserFollowing && !antennaUserFollowing.includes(note.userId)) return false;
|
|
} else if (antenna.src === 'list') {
|
|
const listUsers = (await UserListJoinings.findBy({
|
|
userListId: antenna.userListId!,
|
|
})).map(x => x.userId);
|
|
|
|
if (!listUsers.includes(note.userId)) return false;
|
|
} else if (antenna.src === 'group') {
|
|
const joining = await UserGroupJoinings.findOneByOrFail({ id: antenna.userGroupJoiningId! });
|
|
|
|
const groupUsers = (await UserGroupJoinings.findBy({
|
|
userGroupId: joining.userGroupId,
|
|
})).map(x => x.userId);
|
|
|
|
if (!groupUsers.includes(note.userId)) return false;
|
|
} else if (antenna.src === 'users') {
|
|
const accts = antenna.users.map(x => {
|
|
const { username, host } = Acct.parse(x);
|
|
return getFullApAccount(username, host).toLowerCase();
|
|
});
|
|
if (!accts.includes(getFullApAccount(noteUser.username, noteUser.host).toLowerCase())) return false;
|
|
}
|
|
|
|
const keywords = antenna.keywords
|
|
// Clean up
|
|
.map(xs => xs.filter(x => x !== ''))
|
|
.filter(xs => xs.length > 0);
|
|
|
|
if (keywords.length > 0) {
|
|
if (note.text == null) return false;
|
|
|
|
const matched = keywords.some(and =>
|
|
and.every(keyword =>
|
|
antenna.caseSensitive
|
|
? note.text!.includes(keyword)
|
|
: note.text!.toLowerCase().includes(keyword.toLowerCase()),
|
|
));
|
|
|
|
if (!matched) return false;
|
|
}
|
|
|
|
const excludeKeywords = antenna.excludeKeywords
|
|
// Clean up
|
|
.map(xs => xs.filter(x => x !== ''))
|
|
.filter(xs => xs.length > 0);
|
|
|
|
if (excludeKeywords.length > 0) {
|
|
if (note.text == null) return false;
|
|
|
|
const matched = excludeKeywords.some(and =>
|
|
and.every(keyword =>
|
|
antenna.caseSensitive
|
|
? note.text!.includes(keyword)
|
|
: note.text!.toLowerCase().includes(keyword.toLowerCase()),
|
|
));
|
|
|
|
if (matched) return false;
|
|
}
|
|
|
|
if (antenna.withFile) {
|
|
if (note.fileIds && note.fileIds.length === 0) return false;
|
|
}
|
|
|
|
// TODO: eval expression
|
|
|
|
return true;
|
|
}
|