server/utils/timelines.ts

import type { findManyStatuses, Status } from "~database/entities/Status";
import type { findManyUsers, User } from "~database/entities/User";
import type { Notification } from "~database/entities/Notification";
import type { db } from "~drizzle/db";

export async function fetchTimeline<T extends User | Status | Notification>(
    model:
        | typeof findManyStatuses
        | typeof findManyUsers
        | typeof db.query.notification.findMany,
    args:
        | Parameters<typeof findManyStatuses>[0]
        | Parameters<typeof findManyUsers>[0]
        | Parameters<typeof db.query.notification.findMany>[0],
    req: Request,
) {
    // BEFORE: Before in a top-to-bottom order, so the most recent posts
    // AFTER: After in a top-to-bottom order, so the oldest posts
    // @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(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({
            ...args,
            // @ts-expect-error this hack breaks typing :(
            where: (object, { gt }) => gt(object.id, objects[0].id),
            limit: 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 < (Number(args?.limit) ?? Number.POSITIVE_INFINITY)
        ) {
            // Check if there are statuses after the last one
            // @ts-expect-error hack again
            const objectsAfter = await model({
                ...args,
                // @ts-expect-error this hack breaks typing :(
                where: (object, { lt }) => lt(object.id, objects.at(-1).id),
                limit: 1,
            });

            if (objectsAfter.length > 0) {
                const urlWithoutQuery = req.url.split("?")[0];
                // Add next link
                linkHeader.push(
                    `<${urlWithoutQuery}?max_id=${objectsAfter[0].id}>; rel="next"`,
                );
            }
        }
    }

    return {
        link: linkHeader.join(", "),
        objects,
    };
}