forked from mirrors/iceshrimp
63 lines
1.7 KiB
TypeScript
63 lines
1.7 KiB
TypeScript
import type { User } from "@/models/entities/user.js";
|
|
import { Followings } from "@/models/index.js";
|
|
import type { SelectQueryBuilder } from "typeorm";
|
|
import { Brackets } from "typeorm";
|
|
|
|
export function generateVisibilityQuery(
|
|
q: SelectQueryBuilder<any>,
|
|
me?: { id: User["id"] } | null,
|
|
) {
|
|
// This code must always be synchronized with the checks in Notes.isVisibleForMe.
|
|
if (me == null) {
|
|
q.andWhere(
|
|
new Brackets((qb) => {
|
|
qb.where(`note.visibility = 'public'`).orWhere(
|
|
`note.visibility = 'home'`,
|
|
);
|
|
}),
|
|
)
|
|
.andWhere('note.localOnly = FALSE');
|
|
} else {
|
|
const followingQuery = Followings.createQueryBuilder("following")
|
|
.select("following.followeeId")
|
|
.where("following.followerId = :meId");
|
|
|
|
q.andWhere(
|
|
new Brackets((qb) => {
|
|
qb
|
|
// 公開投稿である
|
|
.where(
|
|
new Brackets((qb) => {
|
|
qb.where(`note.visibility = 'public'`).orWhere(
|
|
`note.visibility = 'home'`,
|
|
);
|
|
}),
|
|
)
|
|
// または 自分自身
|
|
.orWhere("note.userId = :meId")
|
|
// または 自分宛て
|
|
.orWhere(":meId = ANY(note.visibleUserIds)")
|
|
.orWhere(":meId = ANY(note.mentions)")
|
|
.orWhere(
|
|
new Brackets((qb) => {
|
|
qb
|
|
// または フォロワー宛ての投稿であり、
|
|
.where(`note.visibility = 'followers'`)
|
|
.andWhere(
|
|
new Brackets((qb) => {
|
|
qb
|
|
// 自分がフォロワーである
|
|
.where(`note.userId IN (${followingQuery.getQuery()})`)
|
|
// または 自分の投稿へのリプライ
|
|
.orWhere("note.replyUserId = :meId");
|
|
}),
|
|
);
|
|
}),
|
|
);
|
|
}),
|
|
);
|
|
|
|
q.setParameters({ meId: me.id });
|
|
}
|
|
}
|