gitbruv/app/api/avatar/[filename]/route.ts

import { NextRequest, NextResponse } from "next/server";
import { r2Get } from "@/lib/r2";

export async function GET(request: NextRequest, { params }: { params: Promise<{ filename: string }> }) {
  const { filename } = await params;

  const key = `avatars/${filename}`;
  const data = await r2Get(key);

  if (!data) {
    return new NextResponse(null, { status: 404 });
  }

  const ext = filename.split(".").pop()?.toLowerCase();
  let contentType = "image/png";

  if (ext === "jpg" || ext === "jpeg") {
    contentType = "image/jpeg";
  } else if (ext === "gif") {
    contentType = "image/gif";
  } else if (ext === "webp") {
    contentType = "image/webp";
  }

  return new NextResponse(new Uint8Array(data), {
    headers: {
      "Content-Type": contentType,
      "Cache-Control": "public, max-age=31536000, immutable",
    },
  });
}