103 lines
3.1 KiB
TypeScript
103 lines
3.1 KiB
TypeScript
import * as Acct from "@/misc/acct.js";
|
|
import { Cache } from "@/misc/cache.js";
|
|
import { getWordHardMute } from "@/misc/check-word-mute.js";
|
|
import { getFullApAccount } from "@/misc/convert-host.js";
|
|
import type { Packed } from "@/misc/schema.js";
|
|
import type { Antenna } from "@/models/entities/antenna.js";
|
|
import type { Note } from "@/models/entities/note.js";
|
|
import type { User } from "@/models/entities/user.js";
|
|
import { Blockings, UserProfiles } from "@/models/index.js";
|
|
|
|
const blockingCache = new Cache<User["id"][]>("blocking", 60 * 5);
|
|
const mutedWordsCache = new Cache<string[][] | undefined>("mutedWords", 60 * 5);
|
|
|
|
export async function checkHitAntenna(
|
|
antenna: Antenna,
|
|
note: Note | Packed<"Note">,
|
|
noteUser: { id: User["id"]; username: string; host: string | null },
|
|
): Promise<boolean> {
|
|
if (note.visibility === "specified") return false;
|
|
if (note.visibility === "home") return false;
|
|
if (!antenna.withReplies && note.replyId != null) return false;
|
|
if (antenna.withFile) {
|
|
if (note.fileIds && note.fileIds.length === 0) return false;
|
|
}
|
|
|
|
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;
|
|
} else if (antenna.src === "instances") {
|
|
const instances = antenna.instances
|
|
.filter((x) => x !== "")
|
|
.map((host) => {
|
|
return host.toLowerCase();
|
|
});
|
|
if (!instances.includes(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;
|
|
}
|
|
|
|
// アンテナ作成者がノート作成者にブロックされていたらスキップ
|
|
const blockings = await blockingCache.fetch(noteUser.id, () =>
|
|
Blockings.findBy({ blockerId: noteUser.id }).then((res) =>
|
|
res.map((x) => x.blockeeId),
|
|
),
|
|
);
|
|
if (blockings.includes(antenna.userId)) return false;
|
|
|
|
const mutedWords = await mutedWordsCache.fetch(antenna.userId, () =>
|
|
UserProfiles.findOneBy({ userId: antenna.userId }).then(
|
|
(profile) => profile?.mutedWords,
|
|
),
|
|
);
|
|
if (await getWordHardMute(note, antenna.userId, mutedWords)) return false;
|
|
|
|
// TODO: eval expression
|
|
|
|
return true;
|
|
}
|