feat: add mapping for cognito pool migration handler

cabiri-io · Feb 21, 2021 · 97564dd · 97564dd
1 parent cf9c0a9
commit 97564dd
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/packages/sls-aws/src/cognito/user-pool-handler.ts b/packages/sls-aws/src/cognito/user-pool-handler.ts
@@ -3,7 +3,8 @@ import type {
   Context,
   CustomMessageTriggerEvent,
   PreSignUpTriggerEvent,
-  PreTokenGenerationTriggerEvent
+  PreTokenGenerationTriggerEvent,
+  UserMigrationTriggerEvent
 } from 'aws-lambda'
 import { responseOrError } from '../reponse/response-or-error'
 
@@ -31,5 +32,17 @@ const cognitoUserPoolCustomMessage = <D, C = never>(
 ): SlsEnvironment<CustomMessageHandler, C, D> =>
   environment<CustomMessageHandler, C, D>(config).successHandler(responseOrError)
 
-export type { PreSignUpHandler, PreTokenGenerationHandler, CustomMessageHandler }
-export { cognitoUserPoolPreSignUp, cognitoUserPoolPreTokenGeneration, cognitoUserPoolCustomMessage }
+type UserMigrationHandler = Handler<UserMigrationTriggerEvent, Context, Promise<UserMigrationTriggerEvent>>
+
+const cognitoUserPoolMigration = <D, C = never>(
+  config?: EnvironmentConfig<UserMigrationHandler>
+): SlsEnvironment<UserMigrationHandler, C, D> =>
+  environment<UserMigrationHandler, C, D>(config).successHandler(responseOrError)
+
+export type { PreSignUpHandler, PreTokenGenerationHandler, CustomMessageHandler, UserMigrationHandler }
+export {
+  cognitoUserPoolPreSignUp,
+  cognitoUserPoolPreTokenGeneration,
+  cognitoUserPoolCustomMessage,
+  cognitoUserPoolMigration
+}