add response hook

fixes #49

README.md

@@ -39,7 +39,13 @@ installGlobals()
 export const handler = createRequestHandler({
     build,
     mode: process.env.NODE_ENV,
-    awsProxy: AWSProxy.APIGatewayV1
+    awsProxy: AWSProxy.APIGatewayV1,
+    responseHook: async (response: Response) => {
+
+        // do some transformations before sending the response
+
+        return response
+    }
 })
 ```
 
@@ -54,6 +60,10 @@ By default the `awsProxy` is set to `AWSProxy.APIGatewayV2`.
 - `AWSProxy.ALB`
 - `AWSProxy.FunctionURL`
 
+### `responseHook`
+
+This hook allows you to do transformations on the response before sending it to the client.
+
 ## Notes
 
 ### split from @remix/architect

src/server.ts

@@ -45,12 +45,14 @@ export function createRequestHandler({
   build,
   getLoadContext,
   mode = process.env.NODE_ENV,
-  awsProxy = AWSProxy.APIGatewayV2
+  awsProxy = AWSProxy.APIGatewayV2,
+  responseHook,
 }: {
   build: ServerBuild;
   getLoadContext?: GetLoadContextFunction;
   mode?: string;
   awsProxy?: AWSProxy;
+  responseHook?: (response: Response) => Promise<Response>
 }): RequestHandler {
   const handleRequest = createRemixRequestHandler(build, mode)
 
@@ -61,6 +63,12 @@ export function createRequestHandler({
 
     const response = (await handleRequest(request, loadContext)) as Response
 
+    if (responseHook) {
+      const modifiedResponse = await responseHook(response)
+
+      return awsAdapter.sendRemixResponse(modifiedResponse)
+    }
+
     return awsAdapter.sendRemixResponse(response)
   }
 }