firehose-delivery-stream.ts

import { singletonEventRole } from './util';
import * as events from '../../aws-events';
import * as iam from '../../aws-iam';
import * as firehose from '../../aws-kinesisfirehose';
import { IResource } from '../../core';

/**
 * Customize the Amazon Data Firehose Stream Event Target
 *
 * @deprecated Use FirehoseDeliveryStreamProps
 */
export interface KinesisFirehoseStreamProps {
  /**
   * The message to send to the stream.
   *
   * Must be a valid JSON text passed to the target stream.
   *
   * @default - the entire Event Bridge event
   */
  readonly message?: events.RuleTargetInput;
}

/**
 * Customize the Amazon Data Firehose Stream Event Target
 */
export interface FirehoseDeliveryStreamProps {
  /**
   * The message to send to the stream.
   *
   * Must be a valid JSON text passed to the target stream.
   *
   * @default - the entire Event Bridge event
   */
  readonly message?: events.RuleTargetInput;
}

/**
 * Customize the Amazon Data Firehose Stream Event Target
 *
 * @deprecated Use FirehoseDeliveryStream
 */
export class KinesisFirehoseStream implements events.IRuleTarget {
  constructor(private readonly stream: firehose.CfnDeliveryStream, private readonly props: KinesisFirehoseStreamProps = {}) {
  }

  /**
   * Returns a RuleTarget that can be used to trigger this Firehose Stream as a
   * result from a Event Bridge event.
   */
  public bind(_rule: events.IRule, _id?: string): events.RuleTargetConfig {
    const role = singletonEventRole(this.stream);
    role.addToPrincipalPolicy(new iam.PolicyStatement({
      actions: ['firehose:PutRecord', 'firehose:PutRecordBatch'],
      resources: [this.stream.attrArn],
    }));

    return {
      arn: this.stream.attrArn,
      role,
      input: this.props.message,
      targetResource: this.stream,
    };
  }
}

/**
 * Represents an Amazon Data Firehose delivery stream.
 *
 * @deprecated Use aws_kinesisfirehose.IDeliveryStream
 */
export interface IDeliveryStream extends IResource {
  /**
   * The ARN of the delivery stream.
   *
   * @attribute
   */
  readonly deliveryStreamArn: string;

  /**
   * The name of the delivery stream.
   *
   * @attribute
   */
  readonly deliveryStreamName: string;
}

/**
 * Customize the Amazon Data Firehose Stream Event Target V2 to support L2 Amazon Data Firehose Delivery Stream
 * instead of L1 Cfn Firehose Delivery Stream.
 *
 * @deprecated Use FirehoseDeliveryStream
 */
export class KinesisFirehoseStreamV2 implements events.IRuleTarget {
  constructor(private readonly stream: IDeliveryStream, private readonly props: KinesisFirehoseStreamProps = {}) {
  }

  /**
   * Returns a RuleTarget that can be used to trigger this Firehose Stream as a
   * result from a Event Bridge event.
   */
  public bind(_rule: events.IRule, _id?: string): events.RuleTargetConfig {
    const role = singletonEventRole(this.stream);
    role.addToPrincipalPolicy(new iam.PolicyStatement({
      actions: ['firehose:PutRecord', 'firehose:PutRecordBatch'],
      resources: [this.stream.deliveryStreamArn],
    }));

    return {
      arn: this.stream.deliveryStreamArn,
      role,
      input: this.props.message,
      targetResource: this.stream,
    };
  }
}

/**
 * Customize the Amazon Data Firehose Stream Event Target.
 */
export class FirehoseDeliveryStream implements events.IRuleTarget {
  constructor(private readonly stream: firehose.IDeliveryStream, private readonly props: FirehoseDeliveryStreamProps = {}) {
  }

  /**
   * Returns a RuleTarget that can be used to trigger this Firehose Stream as a
   * result from a Event Bridge event.
   */
  public bind(_rule: events.IRule, _id?: string): events.RuleTargetConfig {
    const role = singletonEventRole(this.stream);
    this.stream.grantPutRecords(role);

    return {
      arn: this.stream.deliveryStreamArn,
      role,
      input: this.props.message,
      targetResource: this.stream,
    };
  }
}