import { Handler } from '../../handler';
import { BaseTriggerEvent, StringMap } from './_common';

export type UserStatus =
  | 'UNCONFIRMED'
  | 'CONFIRMED'
  | 'ARCHIVED'
  | 'COMPROMISED'
  | 'UNKNOWN'
  | 'RESET_REQUIRED'
  | 'FORCE_CHANGE_PASSWORD';

export interface BaseUserMigrationTriggerEvent<T extends string> extends BaseTriggerEvent<T> {
  request: {
      password: string;
      validationData?: StringMap | undefined;
      clientMetadata?: StringMap | undefined;
  };
  response: {
      userAttributes: StringMap;
      finalUserStatus?: UserStatus | undefined;
      messageAction?: 'RESEND' | 'SUPPRESS' | undefined;
      desiredDeliveryMediums: Array<'SMS' | 'EMAIL'>;
      forceAliasCreation?: boolean | undefined;
  };
}

export type UserMigrationAuthenticationTriggerEvent = BaseUserMigrationTriggerEvent<'UserMigration_Authentication'>;

export type UserMigrationForgotPasswordTriggerEvent = BaseUserMigrationTriggerEvent<'UserMigration_ForgotPassword'>;

/**
 * @see https://docs.aws.amazon.com/cognito/latest/developerguide/user-pool-lambda-migrate-user.html
 */
export type UserMigrationTriggerEvent =
  | UserMigrationAuthenticationTriggerEvent
  | UserMigrationForgotPasswordTriggerEvent;

export type UserMigrationTriggerHandler = Handler<UserMigrationTriggerEvent>;