97 lines
2.8 KiB
TypeScript
97 lines
2.8 KiB
TypeScript
/**
|
|
* The avatar routes.
|
|
*
|
|
* - `POST /api/avatars/me` — signed in. Replaces the caller's avatar (multipart
|
|
* `file`, at most 5 MB), announces it, and answers the caller's user.
|
|
* - `GET /api/avatars/:userId` — public. Serves a user's avatar; 404 without
|
|
* one.
|
|
*
|
|
* @module
|
|
*/
|
|
|
|
import { Router } from "@oak/oak";
|
|
import { authMiddleware } from "../middleware/auth.ts";
|
|
import { getUserById, updateUserAvatar } from "../services/user-service.ts";
|
|
import {
|
|
broadcastUserUpdated,
|
|
updateClientAvatar,
|
|
} from "../services/ws-service.ts";
|
|
import { APIErrorCode, type APIResponse } from "../../shared/api.ts";
|
|
import { APIException } from "../lib/api-exception.ts";
|
|
import type { SelfUser } from "../../shared/user.ts";
|
|
import { toPublicUser, toSelfUser } from "../model/user.ts";
|
|
import {
|
|
AVATARS_DIR,
|
|
serveUploadedFile,
|
|
validateImageUpload,
|
|
} from "../lib/upload.ts";
|
|
|
|
/** The router serving the endpoints the module doc lists. */
|
|
const router: Router = new Router();
|
|
|
|
router.post("/api/avatars/me", authMiddleware, async (ctx) => {
|
|
const authPayload = ctx.state.user;
|
|
const contentType = ctx.request.headers.get("content-type") ?? "";
|
|
if (!contentType.includes("multipart/form-data")) {
|
|
throw new APIException(
|
|
APIErrorCode.BAD_REQUEST,
|
|
400,
|
|
"Expected multipart/form-data",
|
|
);
|
|
}
|
|
|
|
const body = await ctx.request.body.formData();
|
|
const file = body.get("file");
|
|
|
|
if (!(file instanceof File)) {
|
|
throw new APIException(APIErrorCode.BAD_REQUEST, 400, "Missing file field");
|
|
}
|
|
|
|
const data = new Uint8Array(await file.arrayBuffer());
|
|
const mime = validateImageUpload(data);
|
|
|
|
const filePath = `${AVATARS_DIR}/${authPayload.userId}`;
|
|
await Deno.mkdir(AVATARS_DIR, { recursive: true });
|
|
await Deno.writeFile(filePath, data);
|
|
try {
|
|
updateUserAvatar(authPayload.userId, mime);
|
|
} catch (err) {
|
|
// DB write failed — clean up the orphaned file
|
|
await Deno.remove(filePath).catch(() => {});
|
|
throw err;
|
|
}
|
|
updateClientAvatar(authPayload.userId, mime);
|
|
|
|
const user = getUserById(authPayload.userId);
|
|
// Let every client refresh this user's avatar wherever it's shown (feeds,
|
|
// comments, OP, profiles) — not just the presence row. updated_at, bumped on
|
|
// the avatar write, doubles as the cache-busting version.
|
|
broadcastUserUpdated(toPublicUser(user));
|
|
const responseBody: APIResponse<SelfUser> = {
|
|
success: true,
|
|
data: toSelfUser(user),
|
|
};
|
|
ctx.response.body = responseBody;
|
|
});
|
|
|
|
router.get("/api/avatars/:userId", async (ctx) => {
|
|
const { userId } = ctx.params;
|
|
|
|
let user;
|
|
try {
|
|
user = getUserById(userId);
|
|
} catch {
|
|
ctx.response.status = 404;
|
|
return;
|
|
}
|
|
|
|
if (!user.avatarMime) {
|
|
ctx.response.status = 404;
|
|
return;
|
|
}
|
|
|
|
await serveUploadedFile(ctx, `${AVATARS_DIR}/${userId}`, user.avatarMime);
|
|
});
|
|
|
|
export default router;
|