From 283b9f9d24f9785eb51a2762db9784071f85d252 Mon Sep 17 00:00:00 2001
From: Brandon Liu <bdon@bdon.org>
Date: Wed, 21 Dec 2022 16:30:56 +0800
Subject: [PATCH] refactor AWS handler with modular handlerRaw

---
 serverless/aws/src/index.ts | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/serverless/aws/src/index.ts b/serverless/aws/src/index.ts
index fbc7881..46bebc0 100644
--- a/serverless/aws/src/index.ts
+++ b/serverless/aws/src/index.ts
@@ -136,9 +136,10 @@ const apiResp = (
 // Assumes event is a API Gateway V2 or Lambda Function URL formatted dict
 // and returns API Gateway V2 / Lambda Function dict responses
 // Does not work with CloudFront events/Lambda@Edge; see README
-export const handler = async (
+export const handlerRaw = async (
 	event: APIGatewayProxyEventV2,
-	context: Context
+	context: Context,
+	tilePostprocess?: (a: ArrayBuffer, t: TileType) => ArrayBuffer
 ): Promise<APIGatewayProxyResult> => {
 	var path;
 	var is_api_gateway;
@@ -216,10 +217,16 @@ export const handler = async (
 					break;
 			}
 
+			let data = tile_result.data;
+
+			if (tilePostprocess) {
+				data = tilePostprocess(data, header.tileType);
+			}
+
 			if (is_api_gateway) {
 				// this is wasted work, but we need to force API Gateway to interpret the Lambda response as binary
 				// without depending on clients sending matching Accept: headers in the request.
-				const recompressed_data = zlib.gzipSync(tile_result.data);
+				const recompressed_data = zlib.gzipSync(data);
 				headers["Content-Encoding"] = "gzip";
 				return apiResp(
 					200,
@@ -231,7 +238,7 @@ export const handler = async (
 				// returns uncompressed response
 				return apiResp(
 					200,
-					Buffer.from(tile_result.data).toString("base64"),
+					Buffer.from(data).toString("base64"),
 					true,
 					headers
 				);
@@ -247,3 +254,10 @@ export const handler = async (
 	}
 	return apiResp(404, "Invalid URL", false, headers);
 };
+
+export const handler = async (
+	event: APIGatewayProxyEventV2,
+	context: Context
+): Promise<APIGatewayProxyResult> => {
+	return handlerRaw(event, context);
+};