el.xwx.moe/pages/api/v1/avatar/[id].ts
2023-11-06 08:25:57 -05:00

53 lines
1.3 KiB
TypeScript

import type { NextApiRequest, NextApiResponse } from "next";
import { prisma } from "@/lib/api/db";
import readFile from "@/lib/api/storage/readFile";
import verifyUser from "@/lib/api/verifyUser";
export default async function Index(req: NextApiRequest, res: NextApiResponse) {
const queryId = Number(req.query.id);
const user = await verifyUser({ req, res });
if (!user) return;
if (!queryId)
return res
.setHeader("Content-Type", "text/plain")
.status(401)
.send("Invalid parameters.");
if (user.id !== queryId) {
const targetUser = await prisma.user.findUnique({
where: {
id: queryId,
},
include: {
whitelistedUsers: true,
},
});
const whitelistedUsernames = targetUser?.whitelistedUsers.map(
(whitelistedUsername) => whitelistedUsername.username
);
if (
targetUser?.isPrivate &&
user.username &&
!whitelistedUsernames?.includes(user.username)
) {
return res
.setHeader("Content-Type", "text/plain")
.status(400)
.send("File not found.");
}
}
const { file, contentType, status } = await readFile(
`uploads/avatar/${queryId}.jpg`
);
return res
.setHeader("Content-Type", contentType)
.status(status as number)
.send(file);
}