Zod for API Routes

app/api/sites/delete/route.ts

@@ -1,19 +1,25 @@
 import { NextRequest, NextResponse } from "next/server";
 import prisma from "@/app/prisma";
+import { z } from "zod/v4";
 
+const schema = z.object({
+    siteId: z.string(),
+});
 
 export async function DELETE(request: NextRequest) {
     const searchParams = request.nextUrl.searchParams;
-    const siteId = searchParams.get("siteId");
+    const siteId = schema.parse({ siteId: searchParams.get("siteId") });
 
     try {
         const site = await prisma.site.delete({
-            where: { id: Number(siteId) },
+            where: { id: Number(siteId.siteId) },
         });
 
         return NextResponse.json(site);
-    } catch (error) {
-        console.error(error);
-        return NextResponse.json({ error: "Failed to delete site" }, { status: 500 });
+    } catch (error: any) {
+        if(error instanceof z.ZodError) {
+            return NextResponse.json({ error: error.issues[0].message }, { status: 400 });
+        }
+        return NextResponse.json({ error: "Internal Server Error" }, { status: 500 });
     }
 }