more Free/Pro plan limitations

packages/backend/src/routes/user/upload-avatar.ts

@@ -1,13 +1,23 @@
 import { randomUUID } from "node:crypto";
-import type { BunRequest } from "bun";
 import sharp from "sharp";
+import type { AuthedRequest } from "../../auth/middleware";
+import { getSubscriptionByUserId } from "../../db/queries";
 import { s3Client, s3Endpoint, s3PublicUrl } from "../../s3";
 
 const MAX_FILE_SIZE = 5 * 1024 * 1024;
 const ALLOWED_TYPES = ["image/png", "image/jpeg", "image/webp", "image/gif"];
 const TARGET_SIZE = 256;
 
-export default async function uploadAvatar(req: BunRequest) {
+async function isAnimatedGIF(buffer: Buffer): Promise<boolean> {
+    try {
+        const metadata = await sharp(buffer).metadata();
+        return metadata.pages !== undefined && metadata.pages > 1;
+    } catch {
+        return false;
+    }
+}
+
+export default async function uploadAvatar(req: AuthedRequest) {
     if (req.method !== "POST") {
         return new Response("method not allowed", { status: 405 });
     }
@@ -29,14 +39,31 @@ export default async function uploadAvatar(req: BunRequest) {
         });
     }
 
+    const inputBuffer = Buffer.from(await file.arrayBuffer());
+
+    // check if user is pro
+    const subscription = await getSubscriptionByUserId(req.userId);
+    const isPro = subscription?.status === "active";
+
+    // block animated avatars for free users
+    if (!isPro && file.type === "image/gif") {
+        const animated = await isAnimatedGIF(inputBuffer);
+        if (animated) {
+            return new Response(
+                JSON.stringify({
+                    error: "Animated avatars are only available on Pro. Upgrade to upload animated avatars.",
+                }),
+                { status: 403, headers: { "Content-Type": "application/json" } },
+            );
+        }
+    }
+
     const isGIF = file.type === "image/gif";
     const outputExtension = isGIF ? "gif" : "png";
     const outputMimeType = isGIF ? "image/gif" : "image/png";
 
     let resizedBuffer: Buffer;
     try {
-        const inputBuffer = Buffer.from(await file.arrayBuffer());
-
         if (isGIF) {
             resizedBuffer = await sharp(inputBuffer, { animated: true })
                 .resize(TARGET_SIZE, TARGET_SIZE, { fit: "cover" })