mirror of
https://iceshrimp.dev/limepotato/jormungandr-bite.git
synced 2024-11-12 04:57:30 -07:00
Handle deleted antenna results correctly
This commit is contained in:
parent
aeddce2a60
commit
79471cf450
1 changed files with 64 additions and 44 deletions
|
@ -2,12 +2,12 @@ import define from "../../define.js";
|
|||
import readNote from "@/services/note/read.js";
|
||||
import { Antennas, Notes } from "@/models/index.js";
|
||||
import { redisClient } from "@/db/redis.js";
|
||||
import { genId } from "@/misc/gen-id.js";
|
||||
import { makePaginationQuery } from "../../common/make-pagination-query.js";
|
||||
import { generateVisibilityQuery } from "../../common/generate-visibility-query.js";
|
||||
import { generateMutedUserQuery } from "../../common/generate-muted-user-query.js";
|
||||
import { ApiError } from "../../error.js";
|
||||
import { generateBlockedUserQuery } from "../../common/generate-block-query.js";
|
||||
import { Note } from "@/models/entities/note.js";
|
||||
|
||||
export const meta = {
|
||||
tags: ["antennas", "account", "notes"],
|
||||
|
@ -44,8 +44,8 @@ export const meta = {
|
|||
nullable: false,
|
||||
ref: "Note",
|
||||
},
|
||||
}
|
||||
}
|
||||
},
|
||||
},
|
||||
},
|
||||
} as const;
|
||||
|
||||
|
@ -65,58 +65,78 @@ export default define(meta, paramDef, async (ps, user) => {
|
|||
userId: user.id,
|
||||
});
|
||||
|
||||
let pagination = ps.pagination || "+";
|
||||
|
||||
if (antenna == null) {
|
||||
throw new ApiError(meta.errors.noSuchAntenna);
|
||||
}
|
||||
|
||||
const noteIdsRes = await redisClient.xrevrange(
|
||||
`antennaTimeline:${antenna.id}`,
|
||||
ps.pagination || "+",
|
||||
"-",
|
||||
"COUNT",
|
||||
ps.limit + 1,
|
||||
);
|
||||
let notes: Note[] = [];
|
||||
let paginationMap: string[][] = [];
|
||||
|
||||
if (noteIdsRes.length === 0) {
|
||||
return [];
|
||||
while (notes.length < ps.limit && pagination !== "-1") {
|
||||
// exclusive range
|
||||
if (pagination != "+" && !pagination.startsWith("("))
|
||||
pagination = `(${pagination}`;
|
||||
|
||||
const noteIdsRes = await redisClient.xrevrange(
|
||||
`antennaTimeline:${antenna.id}`,
|
||||
pagination,
|
||||
"-",
|
||||
"COUNT",
|
||||
ps.limit - notes.length,
|
||||
);
|
||||
|
||||
const noteIds = noteIdsRes.map((x) => x[1][1]);
|
||||
|
||||
if (noteIds.length === 0) {
|
||||
pagination = "-1";
|
||||
break;
|
||||
}
|
||||
|
||||
const query = makePaginationQuery(Notes.createQueryBuilder("note"))
|
||||
.where("note.id IN (:...noteIds)", { noteIds: noteIds })
|
||||
.innerJoinAndSelect("note.user", "user")
|
||||
.leftJoinAndSelect("user.avatar", "avatar")
|
||||
.leftJoinAndSelect("user.banner", "banner")
|
||||
.leftJoinAndSelect("note.reply", "reply")
|
||||
.leftJoinAndSelect("note.renote", "renote")
|
||||
.leftJoinAndSelect("reply.user", "replyUser")
|
||||
.leftJoinAndSelect("replyUser.avatar", "replyUserAvatar")
|
||||
.leftJoinAndSelect("replyUser.banner", "replyUserBanner")
|
||||
.leftJoinAndSelect("renote.user", "renoteUser")
|
||||
.leftJoinAndSelect("renoteUser.avatar", "renoteUserAvatar")
|
||||
.leftJoinAndSelect("renoteUser.banner", "renoteUserBanner")
|
||||
.andWhere("note.visibility != 'home'");
|
||||
|
||||
generateVisibilityQuery(query, user);
|
||||
generateMutedUserQuery(query, user);
|
||||
generateBlockedUserQuery(query, user);
|
||||
|
||||
pagination = noteIdsRes[noteIdsRes.length - 1][0];
|
||||
paginationMap = paginationMap.concat(
|
||||
noteIdsRes.map((x) => [x[1][1], x[0]]),
|
||||
);
|
||||
notes = notes.concat(await query.take(ps.limit - notes.length).getMany());
|
||||
}
|
||||
|
||||
const noteIds = noteIdsRes
|
||||
.map((x) => x[1][1]);
|
||||
|
||||
if (noteIds.length === 0) {
|
||||
return {pagination: "", notes: []};
|
||||
}
|
||||
|
||||
const query = makePaginationQuery(
|
||||
Notes.createQueryBuilder("note")
|
||||
)
|
||||
.where("note.id IN (:...noteIds)", { noteIds: noteIds })
|
||||
.innerJoinAndSelect("note.user", "user")
|
||||
.leftJoinAndSelect("user.avatar", "avatar")
|
||||
.leftJoinAndSelect("user.banner", "banner")
|
||||
.leftJoinAndSelect("note.reply", "reply")
|
||||
.leftJoinAndSelect("note.renote", "renote")
|
||||
.leftJoinAndSelect("reply.user", "replyUser")
|
||||
.leftJoinAndSelect("replyUser.avatar", "replyUserAvatar")
|
||||
.leftJoinAndSelect("replyUser.banner", "replyUserBanner")
|
||||
.leftJoinAndSelect("renote.user", "renoteUser")
|
||||
.leftJoinAndSelect("renoteUser.avatar", "renoteUserAvatar")
|
||||
.leftJoinAndSelect("renoteUser.banner", "renoteUserBanner")
|
||||
.andWhere("note.visibility != 'home'");
|
||||
|
||||
generateVisibilityQuery(query, user);
|
||||
generateMutedUserQuery(query, user);
|
||||
generateBlockedUserQuery(query, user);
|
||||
|
||||
const notes = await query.take(ps.limit).getMany();
|
||||
|
||||
if (notes.length > 0) {
|
||||
if (notes.length === 0) {
|
||||
return { pagination: "-1", notes: [] };
|
||||
} else {
|
||||
readNote(user.id, notes);
|
||||
}
|
||||
|
||||
if (notes.length < ps.limit) {
|
||||
pagination = "-1";
|
||||
} else {
|
||||
// I'm so sorry, FIXME: rewrite pagination system
|
||||
pagination = paginationMap.find(
|
||||
(p) => p[0] == notes[notes.length - (notes.length > 1 ? 2 : 1)].id,
|
||||
)[1];
|
||||
}
|
||||
|
||||
return {
|
||||
pagination: noteIdsRes[noteIdsRes.length - 1][0],
|
||||
pagination: pagination,
|
||||
notes: await Notes.packMany(notes, user),
|
||||
};
|
||||
});
|
||||
|
|
Loading…
Reference in a new issue