iceshrimp-legacy/packages/backend/src/server/api/common/generate-muted-user-query.ts
ThatOneCalculator 6b00abf05c
refactor: 🎨 rome
2023-01-12 20:40:33 -08:00

82 lines
2.3 KiB
TypeScript

import type { SelectQueryBuilder } from "typeorm";
import { Brackets } from "typeorm";
import type { User } from "@/models/entities/user.js";
import { Mutings, UserProfiles } from "@/models/index.js";
export function generateMutedUserQuery(
q: SelectQueryBuilder<any>,
me: { id: User["id"] },
exclude?: User,
) {
const mutingQuery = Mutings.createQueryBuilder("muting")
.select("muting.muteeId")
.where("muting.muterId = :muterId", { muterId: me.id });
if (exclude) {
mutingQuery.andWhere("muting.muteeId != :excludeId", {
excludeId: exclude.id,
});
}
const mutingInstanceQuery = UserProfiles.createQueryBuilder("user_profile")
.select("user_profile.mutedInstances")
.where("user_profile.userId = :muterId", { muterId: me.id });
// 投稿の作者をミュートしていない かつ
// 投稿の返信先の作者をミュートしていない かつ
// 投稿の引用元の作者をミュートしていない
q.andWhere(`note.userId NOT IN (${mutingQuery.getQuery()})`)
.andWhere(
new Brackets((qb) => {
qb.where("note.replyUserId IS NULL").orWhere(
`note.replyUserId NOT IN (${mutingQuery.getQuery()})`,
);
}),
)
.andWhere(
new Brackets((qb) => {
qb.where("note.renoteUserId IS NULL").orWhere(
`note.renoteUserId NOT IN (${mutingQuery.getQuery()})`,
);
}),
)
// mute instances
.andWhere(
new Brackets((qb) => {
qb.andWhere("note.userHost IS NULL").orWhere(
`NOT ((${mutingInstanceQuery.getQuery()})::jsonb ? note.userHost)`,
);
}),
)
.andWhere(
new Brackets((qb) => {
qb.where("note.replyUserHost IS NULL").orWhere(
`NOT ((${mutingInstanceQuery.getQuery()})::jsonb ? note.replyUserHost)`,
);
}),
)
.andWhere(
new Brackets((qb) => {
qb.where("note.renoteUserHost IS NULL").orWhere(
`NOT ((${mutingInstanceQuery.getQuery()})::jsonb ? note.renoteUserHost)`,
);
}),
);
q.setParameters(mutingQuery.getParameters());
q.setParameters(mutingInstanceQuery.getParameters());
}
export function generateMutedUserQueryForUsers(
q: SelectQueryBuilder<any>,
me: { id: User["id"] },
) {
const mutingQuery = Mutings.createQueryBuilder("muting")
.select("muting.muteeId")
.where("muting.muterId = :muterId", { muterId: me.id });
q.andWhere(`user.id NOT IN (${mutingQuery.getQuery()})`);
q.setParameters(mutingQuery.getParameters());
}