iceshrimp/packages/backend/src/server/api/stream/channels/hybrid-timeline.ts
Laura Hausmann ef3463e8dc
[backend] Rework note hard mutes
It's been shown that the current approach doesn't scale. This implementation should scale perfectly fine.
2023-11-27 19:43:45 +01:00

104 lines
3.8 KiB
TypeScript

import Channel from "../channel.js";
import { fetchMeta } from "@/misc/fetch-meta.js";
import { isUserRelated } from "@/misc/is-user-related.js";
import { isInstanceMuted } from "@/misc/is-instance-muted.js";
import type { Packed } from "@/misc/schema.js";
import { isFiltered } from "@/misc/is-filtered.js";
import { Note } from "@/models/entities/note.js";
export default class extends Channel {
public readonly chName = "hybridTimeline";
public static shouldShare = true;
public static requireCredential = true;
private withReplies: boolean;
constructor(id: string, connection: Channel["connection"]) {
super(id, connection);
this.onNote = this.withPackedNote(this.onNote.bind(this));
}
public async init(params: any) {
const meta = await fetchMeta();
if (
meta.disableLocalTimeline &&
!this.user!.isAdmin &&
!this.user!.isModerator
)
return;
this.withReplies = params.withReplies as boolean;
// Subscribe events
this.subscriber.on("notesStream", this.onNote);
}
private async onNote(note: Packed<"Note">) {
if (note.visibility === "hidden") return;
// チャンネルの投稿ではなく、自分自身の投稿 または
// チャンネルの投稿ではなく、その投稿のユーザーをフォローしている または
// チャンネルの投稿ではなく、全体公開のローカルの投稿 または
// フォローしているチャンネルの投稿 の場合だけ
if (
!(
(note.channelId == null && this.user!.id === note.userId) ||
(note.channelId == null && this.following.has(note.userId)) ||
(note.channelId == null &&
note.user.host == null &&
note.visibility === "public") ||
(note.channelId != null && this.followingChannels.has(note.channelId))
)
)
return;
// Ignore notes from instances the user has muted
if (
isInstanceMuted(
note,
new Set<string>(this.userProfile?.mutedInstances ?? []),
)
)
return;
// 関係ない返信は除外
if (note.reply && !this.withReplies) {
const reply = note.reply;
// 「チャンネル接続主への返信」でもなければ、「チャンネル接続主が行った返信」でもなければ、「投稿者の投稿者自身への返信」でもない場合
if (
reply.userId !== this.user!.id &&
note.userId !== this.user!.id &&
reply.userId !== note.userId
)
return;
}
// 流れてきたNoteがミュートしているユーザーが関わるものだったら無視する
if (isUserRelated(note, this.muting)) return;
// 流れてきたNoteがブロックされているユーザーが関わるものだったら無視する
if (isUserRelated(note, this.blocking)) return;
// Members of lists with hideFromHome set
if (note.userId !== this.user!.id && isUserRelated(note, this.hidden)) return;
if (note.renote && !note.text && this.renoteMuting.has(note.userId)) return;
// 流れてきたNoteがミュートすべきNoteだったら無視する
// TODO: 将来的には、単にMutedNoteテーブルにレコードがあるかどうかで判定したい(以下の理由により難しそうではある)
// 現状では、ワードミュートにおけるMutedNoteレコードの追加処理はストリーミングに流す処理と並列で行われるため、
// レコードが追加されるNoteでも追加されるより先にここのストリーミングの処理に到達することが起こる。
// そのためレコードが存在するかのチェックでは不十分なので、改めてgetWordHardMuteを呼んでいる
if (
this.userProfile &&
(await isFiltered(note as unknown as Note, this.user, this.userProfile))
)
return;
this.connection.cacheNote(note);
this.send("note", note);
}
public dispose() {
// Unsubscribe events
this.subscriber.off("notesStream", this.onNote);
}
}