Make avatar imager resilient with upstream fallback everywhere

src/app/api/imaging/avatar/route.ts

@@ -1,99 +1,32 @@
 import { type NextRequest, NextResponse } from "next/server";
-import { resolveImagerBase } from "@/lib/runtime-asset-config";
-
-const FIGURE_RE = /^[a-z]{2}-\d+(?:-\d+)*(?:\.[a-z]{2}-\d+(?:-\d+)*)*$/i;
-const FIGURE_MAX_LEN = 512;
-const FIGURE_MAX_PARTS = 24;
-const UPSTREAM_TIMEOUT_MS = 10_000;
+import { fetchAvatarImage, parseAvatarQuery } from "@/lib/imager-upstream";
 
 export async function GET(request: NextRequest) {
 	const { searchParams } = new URL(request.url);
 
-	const figure = (
-		searchParams.get("figure") ??
-		searchParams.get("look") ??
-		""
-	).trim();
-	if (!figure) {
-		return NextResponse.json(
-			{ error: "Missing figure parameter" },
-			{ status: 400 },
-		);
+	const parsed = parseAvatarQuery(searchParams);
+	if (!parsed.ok) {
+		return NextResponse.json({ error: parsed.message }, { status: 400 });
 	}
-	if (figure.length > FIGURE_MAX_LEN || !FIGURE_RE.test(figure)) {
-		return NextResponse.json(
-			{ error: "Invalid figure format" },
-			{ status: 400 },
-		);
-	}
-	if (figure.split(".").length > FIGURE_MAX_PARTS) {
-		return NextResponse.json(
-			{ error: "Figure has too many parts" },
-			{ status: 400 },
-		);
-	}
-
-	const sizeParam = searchParams.get("size");
-	const size: "s" | "m" | "l" =
-		sizeParam === "s" || sizeParam === "l" ? sizeParam : "m";
-
-	const direction = clampInt(searchParams.get("direction"), 0, 7, 2);
-	const headDirection = clampInt(
-		searchParams.get("head_direction") ?? searchParams.get("headDirection"),
-		0,
-		7,
-		3,
-	);
-	const headOnly =
-		searchParams.get("headonly") === "1" ||
-		searchParams.get("headOnly") === "1";
-	const gesture = searchParams.get("gesture") ?? undefined;
-	const action = searchParams.get("action") ?? undefined;
-	const effect = searchParams.get("effect") ?? undefined;
-
-	const params = new URLSearchParams({
-		figure,
-		direction: String(direction),
-		head_direction: String(headDirection),
-		size,
-		img_format: "png",
-	});
-	if (headOnly) params.set("headonly", "1");
-	if (gesture) params.set("gesture", gesture);
-	if (action) params.set("action", action);
-	if (effect) params.set("effect", effect);
-
-	const imgFormat = searchParams.get("img_format");
-	if (imgFormat) params.set("img_format", imgFormat);
-
-	const upstream = resolveImagerBase(new URL(request.url).origin);
-	const upstreamUrl = new URL(upstream);
-	for (const [key, value] of params) upstreamUrl.searchParams.set(key, value);
 
 	try {
-		const res = await fetch(upstreamUrl, {
-			signal: AbortSignal.timeout(UPSTREAM_TIMEOUT_MS),
-			headers: {
-				"User-Agent":
-					"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
-			},
-		});
+		const { body, contentType, source } = await fetchAvatarImage(
+			new URL(request.url).origin,
+			parsed.params,
+		);
 
-		if (!res.ok) {
-			return NextResponse.json(
-				{ error: "Upstream imager returned an error" },
-				{ status: res.status },
-			);
-		}
-
-		const buffer = await res.arrayBuffer();
-		const contentType = res.headers.get("content-type") || "image/png";
-
-		return new NextResponse(buffer, {
+		return new NextResponse(body as unknown as BodyInit, {
 			status: 200,
 			headers: {
 				"Content-Type": contentType,
-				"Cache-Control": "public, max-age=3600",
+				"Content-Length": String(body.length),
+				// Fallback renders are served with a short TTL so the configured
+				// imager is retried soon instead of being masked for hours.
+				"Cache-Control":
+					source === "fallback"
+						? "public, max-age=300"
+						: "public, max-age=3600",
+				"X-Imager-Source": source,
 				"Access-Control-Allow-Origin": "*",
 			},
 		});
@@ -115,15 +48,3 @@ export async function OPTIONS() {
 		},
 	});
 }
-
-function clampInt(
-	raw: string | null,
-	min: number,
-	max: number,
-	fallback: number,
-): number {
-	if (raw == null) return fallback;
-	const n = Number.parseInt(raw, 10);
-	if (Number.isNaN(n) || n < min || n > max) return fallback;
-	return n;
-}