versia-pub/server
2
0
Fork 0
mirror of https://github.com/versia-pub/server.git synced 2025-12-06 16:38:19 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Timeline refactors

Browse source
This commit is contained in:
Jesse Wierzbinski 2024-04-07 16:24:18 -10:00
parent 69ffd5fafc
commit e0335c33a9
No known key found for this signature in database
4 changed files with 190 additions and 218 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

120
server/api/api/v1/accounts/[id]/statuses.ts
View file

@ -1,8 +1,13 @@
import { apiRoute, applyConfig } from "@api";
import type { Prisma, Status, User } from "@prisma/client";
/* eslint-disable @typescript-eslint/no-unsafe-member-access */
import { errorResponse, jsonResponse } from "@response";
import { fetchTimeline } from "@timelines";
import { client } from "~database/datasource";
import { statusToAPI } from "~database/entities/Status";
import {
statusToAPI,
type StatusWithRelations,
} from "~database/entities/Status";
import {
statusAndUserRelations,
userRelations,
@ -56,7 +61,9 @@ export default apiRoute<{
if (!user) return errorResponse("User not found", 404);
if (pinned) {
const objects = await client.status.findMany({
const { objects, link } = await fetchTimeline<StatusWithRelations>(
client.status,
{
where: {
authorId: id,
isReblog: false,
@ -76,64 +83,9 @@ export default apiRoute<{
orderBy: {
id: "desc",
},
});
// Constuct HTTP Link header (next and prev) only if there are more statuses
const linkHeader = [];
if (objects.length > 0) {
// Check if there are statuses before the first one
const objectsBefore = await client.status.findMany({
where: {
authorId: id,
isReblog: false,
pinnedBy: {
some: {
id: user.id,
},
},
id: {
gt: objects[0].id,
},
},
take: 1,
});
if (objectsBefore.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
// Add prev link
linkHeader.push(
`<${urlWithoutQuery}?min_id=${objects[0].id}>; rel="prev"`,
req,
);
}
// Check if there are statuses after the last one
const objectsAfter = await client.status.findMany({
where: {
authorId: id,
isReblog: false,
pinnedBy: {
some: {
id: user.id,
},
},
id: {
lt: objects.at(-1)?.id,
},
},
take: 1,
});
if (objectsAfter.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
// Add next link
linkHeader.push(
`<${urlWithoutQuery}?max_id=${
objects.at(-1)?.id
}>; rel="next"`,
);
}
}
return jsonResponse(
await Promise.all(
@ -141,12 +93,14 @@ export default apiRoute<{
),
200,
{
Link: linkHeader.join(", "),
Link: link,
},
);
}
const objects = await client.status.findMany({
const { objects, link } = await fetchTimeline<StatusWithRelations>(
client.status,
{
where: {
authorId: id,
isReblog: exclude_reblogs ? true : undefined,
@ -161,57 +115,15 @@ export default apiRoute<{
orderBy: {
id: "desc",
},
});
// Constuct HTTP Link header (next and prev) only if there are more statuses
const linkHeader = [];
if (objects.length > 0) {
// Check if there are statuses before the first one
const objectsBefore = await client.status.findMany({
where: {
authorId: id,
isReblog: exclude_reblogs ? true : undefined,
id: {
gt: objects[0].id,
},
},
take: 1,
});
if (objectsBefore.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
// Add prev link
linkHeader.push(
`<${urlWithoutQuery}?min_id=${objects[0].id}>; rel="prev"`,
req,
);
}
// Check if there are statuses after the last one
const objectsAfter = await client.status.findMany({
where: {
authorId: id,
isReblog: exclude_reblogs ? true : undefined,
id: {
lt: objects.at(-1)?.id,
},
},
take: 1,
});
if (objectsAfter.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
// Add next link
linkHeader.push(
`<${urlWithoutQuery}?max_id=${objects.at(-1)?.id}>; rel="next"`,
);
}
}
return jsonResponse(
await Promise.all(objects.map((status) => statusToAPI(status, user))),
200,
{
Link: linkHeader.join(", "),
Link: link,
},
);
});

26
server/api/api/v1/timelines/home.ts
View file

@ -1,7 +1,11 @@
import { apiRoute, applyConfig } from "@api";
import { errorResponse, jsonResponse } from "@response";
import { fetchTimeline } from "@timelines";
import { client } from "~database/datasource";
import { statusToAPI } from "~database/entities/Status";
import {
type StatusWithRelations,
statusToAPI,
} from "~database/entities/Status";
import { statusAndUserRelations } from "~database/entities/relations";
export const meta = applyConfig({
@ -29,13 +33,15 @@ export default apiRoute<{
const { limit = 20, max_id, min_id, since_id } = extraData.parsedRequest;
if (limit < 1 || limit > 40) {
if (limit < 1 || limit > 80) {
return errorResponse("Limit must be between 1 and 40", 400);
}
if (!user) return errorResponse("Unauthorized", 401);
const objects = await client.status.findMany({
const { objects, link } = await fetchTimeline<StatusWithRelations>(
client.status,
{
where: {
id: {
lt: max_id ?? undefined,
@ -75,17 +81,9 @@ export default apiRoute<{
orderBy: {
id: "desc",
},
});
// Constuct HTTP Link header (next and prev)
const linkHeader = [];
if (objects.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
linkHeader.push(
`<${urlWithoutQuery}?max_id=${objects.at(-1)?.id}>; rel="next"`,
`<${urlWithoutQuery}?min_id=${objects[0].id}>; rel="prev"`,
},
req,
);
}
return jsonResponse(
await Promise.all(
@ -93,7 +91,7 @@ export default apiRoute<{
),
200,
{
Link: linkHeader.join(", "),
Link: link,
},
);
});

24
server/api/api/v1/timelines/public.ts
View file

@ -1,7 +1,11 @@
import { apiRoute, applyConfig } from "@api";
import { errorResponse, jsonResponse } from "@response";
import { fetchTimeline } from "@timelines";
import { client } from "~database/datasource";
import { statusToAPI } from "~database/entities/Status";
import {
statusToAPI,
type StatusWithRelations,
} from "~database/entities/Status";
import { statusAndUserRelations } from "~database/entities/relations";
export const meta = applyConfig({
@ -44,7 +48,9 @@ export default apiRoute<{
return errorResponse("Cannot use both local and remote", 400);
}
const objects = await client.status.findMany({
const { objects, link } = await fetchTimeline<StatusWithRelations>(
client.status,
{
where: {
id: {
lt: max_id ?? undefined,
@ -64,17 +70,9 @@ export default apiRoute<{
orderBy: {
id: "desc",
},
});
// Constuct HTTP Link header (next and prev)
const linkHeader = [];
if (objects.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
linkHeader.push(
`<${urlWithoutQuery}?max_id=${objects.at(-1)?.id}>; rel="next"`,
`<${urlWithoutQuery}?min_id=${objects[0].id}>; rel="prev"`,
},
req,
);
}
return jsonResponse(
await Promise.all(
@ -84,7 +82,7 @@ export default apiRoute<{
),
200,
{
Link: linkHeader.join(", "),
Link: link,
},
);
});

64
utils/timelines.ts Normal file
View file

@ -0,0 +1,64 @@
import type { Status, User, Prisma } from "@prisma/client";
export async function fetchTimeline<T extends User | Status>(
model: Prisma.StatusDelegate | Prisma.UserDelegate,
args: Prisma.StatusFindManyArgs | Prisma.UserFindManyArgs,
req: Request,
) {
// @ts-expect-error This is a hack to get around the fact that Prisma doesn't have a common base type for all models
const objects = (await model.findMany(args)) as T[];
// Constuct HTTP Link header (next and prev) only if there are more statuses
const linkHeader = [];
if (objects.length > 0) {
// Check if there are statuses before the first one
// @ts-expect-error This is a hack to get around the fact that Prisma doesn't have a common base type for all models
const objectsBefore = await model.findMany({
where: {
id: {
gt: objects[0].id,
},
...args.where,
},
take: 1,
});
if (objectsBefore.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
// Add prev link
linkHeader.push(
`<${urlWithoutQuery}?min_id=${objects[0].id}>; rel="prev"`,
);
}
if (objects.length < (args.take ?? Number.POSITIVE_INFINITY)) {
// Check if there are statuses after the last one
// @ts-expect-error This is a hack to get around the fact that Prisma doesn't have a common base type for all models
const objectsAfter = await model.findMany({
where: {
id: {
lt: objects.at(-1)?.id,
},
...args.where,
},
take: 1,
});
if (objectsAfter.length > 0) {
const urlWithoutQuery = req.url.split("?")[0];
// Add next link
linkHeader.push(
`<${urlWithoutQuery}?max_id=${
objects.at(-1)?.id
}>; rel="next"`,
);
}
}
}
return {
link: linkHeader.join(", "),
objects,
};
}