chore: use dynamic return types

stormkit-io · Apr 17, 2023 · 6aefe1a · 6aefe1a
1 parent 40ff774
commit 6aefe1a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/serverless.ts b/src/serverless.ts
@@ -11,17 +11,15 @@ export type App = (
   context?: NodeContext
 ) => void;
 
-type HandlerType = "stormkit" | "stormkit:api";
-
 type ReturnTypes = {
   stormkit: StormkitHandler;
   "stormkit:api": typeof handleApi;
 };
 
 export default (
   app?: App,
-  handler: HandlerType = "stormkit"
-): ReturnTypes[typeof handler] => {
+  handler?: keyof ReturnTypes
+): ReturnTypes[keyof ReturnTypes] => {
   switch (handler) {
     case "stormkit:api":
       return handleApi;