add back responses

ShallotJS · May 26, 2021 · 0594121 · 0594121
1 parent 940e2e2
commit 0594121
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -47,15 +47,18 @@ export type TShallotSocketEvent<
 
 const ShallotSocketWrapper = <TEvent extends TShallotSocketEvent = TShallotSocketEvent>(
   handler: ShallotRawHandler<TEvent>,
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  _successStatusCode = 200,
+  successStatusCode = 200,
   middlewareOpts: {
     HttpErrorHandlerOpts?: TShallotErrorHandlerOptions;
     HttpJsonBodyParserOpts?: TShallotJSONBodyParserOptions;
   } = {}
 ): ShallotAWSHandler<TEvent, APIGatewayProxyResult> => {
   const wrappedResponseHandler: Handler = async (...args) => {
-    await handler(...args);
+    const res = await handler(...args);
+    return {
+      statusCode: successStatusCode,
+      body: res == null ? undefined : JSON.stringify(res),
+    };
   };
 
   const wrapper = ShallotAWS(wrappedResponseHandler)