trigger-function.ts

import { Construct } from 'constructs';
import { ITrigger, Trigger, TriggerOptions } from '.';
import * as lambda from '../../aws-lambda';

/**
 * Props for `InvokeFunction`.
 */
export interface TriggerFunctionProps extends lambda.FunctionProps, TriggerOptions {
}

/**
 * Invokes an AWS Lambda function during deployment.
 */
export class TriggerFunction extends lambda.Function implements ITrigger {

  /**
   * The underlying trigger resource.
   */
  public readonly trigger: Trigger;

  constructor(scope: Construct, id: string, props: TriggerFunctionProps) {
    super(scope, id, props);

    this.trigger = new Trigger(this, 'Trigger', {
      ...props,
      handler: this,
    });
  }

  public executeAfter(...scopes: Construct[]): void {
    this.trigger.executeAfter(...scopes);
  }

  public executeBefore(...scopes: Construct[]): void {
    this.trigger.executeBefore(...scopes);
  }
}