server/api/api/v1/media/:id/index.ts

import { apiRoute, applyConfig, auth } from "@/api";
import { createRoute } from "@hono/zod-openapi";
import { z } from "zod";
import { Attachment } from "~/classes/database/attachment";
import { MediaManager } from "~/classes/media/media-manager";
import { RolePermissions } from "~/drizzle/schema";
import { config } from "~/packages/config-manager/index.ts";
import { ErrorSchema } from "~/types/api";

export const meta = applyConfig({
    ratelimits: {
        max: 10,
        duration: 60,
    },
    route: "/api/v1/media/:id",
    auth: {
        required: true,
        oauthPermissions: ["write:media"],
    },
    permissions: {
        required: [RolePermissions.ManageOwnMedia],
    },
});

export const schemas = {
    param: z.object({
        id: z.string().uuid(),
    }),
    form: z.object({
        thumbnail: z.instanceof(File).optional(),
        description: z
            .string()
            .max(config.validation.max_media_description_size)
            .optional(),
        focus: z.string().optional(),
    }),
};

const routePut = createRoute({
    method: "put",
    path: "/api/v1/media/{id}",
    summary: "Update media",
    middleware: [auth(meta.auth, meta.permissions)],
    request: {
        params: schemas.param,
        body: {
            content: {
                "multipart/form-data": {
                    schema: schemas.form,
                },
            },
        },
    },
    responses: {
        204: {
            description: "Media updated",
            content: {
                "application/json": {
                    schema: Attachment.schema,
                },
            },
        },
        401: {
            description: "Unauthorized",
            content: {
                "application/json": {
                    schema: ErrorSchema,
                },
            },
        },
        404: {
            description: "Media not found",
            content: {
                "application/json": {
                    schema: ErrorSchema,
                },
            },
        },
    },
});

const routeGet = createRoute({
    method: "get",
    path: "/api/v1/media/{id}",
    summary: "Get media",
    middleware: [auth(meta.auth, meta.permissions)],
    request: {
        params: schemas.param,
    },
    responses: {
        200: {
            description: "Media",
            content: {
                "application/json": {
                    schema: Attachment.schema,
                },
            },
        },
        404: {
            description: "Media not found",
            content: {
                "application/json": {
                    schema: ErrorSchema,
                },
            },
        },
        401: {
            description: "Unauthorized",
            content: {
                "application/json": {
                    schema: ErrorSchema,
                },
            },
        },
    },
});

export default apiRoute((app) => {
    app.openapi(routePut, async (context) => {
        const { id } = context.req.valid("param");

        const attachment = await Attachment.fromId(id);

        if (!attachment) {
            return context.json({ error: "Media not found" }, 404);
        }

        const { description, thumbnail } = context.req.valid("form");

        let thumbnailUrl = attachment.data.thumbnailUrl;

        const mediaManager = new MediaManager(config);

        if (thumbnail) {
            const { path } = await mediaManager.addFile(thumbnail);
            thumbnailUrl = Attachment.getUrl(path);
        }

        const descriptionText = description || attachment.data.description;

        if (
            descriptionText !== attachment.data.description ||
            thumbnailUrl !== attachment.data.thumbnailUrl
        ) {
            await attachment.update({
                description: descriptionText,
                thumbnailUrl,
            });

            return context.json(attachment.toApi(), 204);
        }

        return context.json(attachment.toApi(), 204);
    });

    app.openapi(routeGet, async (context) => {
        const { id } = context.req.valid("param");

        const attachment = await Attachment.fromId(id);

        if (!attachment) {
            return context.json({ error: "Media not found" }, 404);
        }

        return context.json(attachment.toApi(), 200);
    });
});