Fix return types

Author: baileytincher

package.json

@@ -18,6 +18,7 @@
   },
   "devDependencies": {
     "@types/aws-lambda": "^8.10.76",
+    "@types/http-errors": "^1.8.0",
     "@types/jest": "^26.0.23",
     "@types/node": "^15.0.1",
     "@types/webpack": "^5.28.0",

src/index.ts

@@ -109,7 +109,7 @@ export function ShallotAWS<
     event: TEvent,
     context: Context,
     callback: Callback<TResult>
-  ) => {
+  ): Promise<TResult | void> => {
     const request: ShallotAWSRequest<TEvent, TResult> = {
       event,
       context,
@@ -140,7 +140,9 @@ export function ShallotAWS<
    * @param middleware the middleware to apply
    * @return the handler with middleware applied
    */
-  ShallotAWSHandler.use = (middleware: ShallotAWSMiddleware) => {
+  ShallotAWSHandler.use = (
+    middleware: ShallotAWSMiddleware
+  ): ShallotAWSHandler<TEvent, TResult> => {
     if (middleware.before != null) {
       middlewares.before.push(middleware.before);
     }
@@ -158,7 +160,7 @@ export function ShallotAWS<
       middlewares.finally.push(middleware.finally);
     }
 
-    return ShallotAWSHandler;
+    return ShallotAWSHandler as ShallotAWSHandler<TEvent, TResult>;
   };
 
   return ShallotAWSHandler as ShallotAWSHandler<TEvent, TResult>;

yarn.lock

@@ -678,6 +678,11 @@
   dependencies:
     "@types/node" "*"
 
+"@types/http-errors@^1.8.0":
+  version "1.8.0"
+  resolved "https://registry.yarnpkg.com/@types/http-errors/-/http-errors-1.8.0.tgz#682477dbbbd07cd032731cb3b0e7eaee3d026b69"
+  integrity sha512-2aoSC4UUbHDj2uCsCxcG/vRMXey/m17bC7UwitVm5hn22nI8O8Y9iDpA76Orc+DWkQ4zZrOKEshCqR/jSuXAHA==
+
 "@types/istanbul-lib-coverage@*", "@types/istanbul-lib-coverage@^2.0.0", "@types/istanbul-lib-coverage@^2.0.1":
   version "2.0.3"
   resolved "https://registry.yarnpkg.com/@types/istanbul-lib-coverage/-/istanbul-lib-coverage-2.0.3.tgz#4ba8ddb720221f432e443bd5f9117fd22cfd4762"