feat: adapt validation in OTelBin next.js app to changed lambda contract

dash0hq · Jan 11, 2024 · 8e45fe1 · 8e45fe1
1 parent bd78cba
commit 8e45fe1
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 7 deletions.
diff --git a/packages/otelbin/package-lock.json b/packages/otelbin/package-lock.json
diff --git a/packages/otelbin/package.json b/packages/otelbin/package.json
@@ -55,7 +55,8 @@
 		"tailwindcss-animate": "^1.0.6",
 		"ts-jest": "^29.1.1",
 		"yaml": "^2.3.4",
-		"yaml-language-server": "^1.14.0"
+		"yaml-language-server": "^1.14.0",
+		"zod": "^3.22.4"
 	},
 	"devDependencies": {
 		"@svgr/webpack": "^8.1.0",

diff --git a/packages/otelbin/src/app/validation/route.ts b/packages/otelbin/src/app/validation/route.ts
@@ -1,6 +1,7 @@
 // SPDX-FileCopyrightText: 2023 Dash0 Inc.
 // SPDX-License-Identifier: Apache-2.0
 
+import z from "zod";
 import retryFetch from "fetch-retry";
 import { Redis } from "@upstash/redis";
 import { Ratelimit } from "@upstash/ratelimit";
@@ -19,6 +20,12 @@ const rateLimit = new Ratelimit({
 	prefix: "rate-limit-validate",
 });
 
+const validationPayloadSchema = z.object({
+	config: z.string(),
+	env: z.record(z.string()).optional(),
+});
+type ValidationPayload = z.infer<typeof validationPayloadSchema>;
+
 export async function POST(request: NextRequest): Promise<NextResponse> {
 	const distro = request.nextUrl.searchParams.get("distro");
 	const version = request.nextUrl.searchParams.get("version");
@@ -36,7 +43,21 @@ export async function POST(request: NextRequest): Promise<NextResponse> {
 		);
 	}
 
-	const config = await request.text();
+	const reqBody = await request.json();
+	let validationPayload: ValidationPayload;
+	try {
+		validationPayload = validationPayloadSchema.parse(reqBody);
+	} catch (e: unknown) {
+		console.error('Failed to parse request body as JSON: "%s"', reqBody, e);
+		return NextResponse.json(
+			{
+				error: `Could not parse request body.`,
+			},
+			{
+				status: 400,
+			}
+		);
+	}
 
 	const userIdentifier = getUserIdentifier(request);
 	const { success } = await rateLimit.blockUntilReady(userIdentifier, 1000 * 60);
@@ -66,9 +87,7 @@ export async function POST(request: NextRequest): Promise<NextResponse> {
 					"COLLECTOR_CONFIGURATION_VALIDATION_API_KEY env var is not configured"
 				),
 			},
-			body: JSON.stringify({
-				config,
-			}),
+			body: JSON.stringify(validationPayload),
 			retries: 3,
 			retryDelay: 1000,
 			retryOn: [500, 503],

diff --git a/packages/otelbin/src/components/validation/useServerSideValidation.ts b/packages/otelbin/src/components/validation/useServerSideValidation.ts
@@ -51,7 +51,10 @@ export function useServerSideValidation(): ValidationState {
 									"Content-Type": "application/json",
 									Accept: "application/json",
 								},
-								body: config,
+								body: JSON.stringify({
+									config,
+									env: {},
+								}),
 							}
 						);