import * as events from '@aws-cdk/aws-events';
import * as events_targets from '@aws-cdk/aws-events-targets';
import * as sqs from '@aws-cdk/aws-sqs';

interface EcsTaskProps extends events_targets.EcsTaskProps {
    deadLetterQueue?: sqs.Queue;
}

class EcsTask implements events.IRuleTarget {
    readonly deadLetterQueue?: sqs.Queue;
    readonly wrappedTask: events_targets.EcsTask
    constructor(private readonly props: EcsTaskProps) {
        this.deadLetterQueue = props.deadLetterQueue ;
        this.wrappedTask = new events_targets.EcsTask(props);
    }

    public bind(rule: events.IRule, _id?: string): events.RuleTargetConfig {
        const wrappedResult = this.wrappedTask.bind(rule, _id);
        let deadLetterConfig: events.CfnRule.DeadLetterConfigProperty | undefined = undefined
        if (this.deadLetterQueue) {
            events_targets.addToDeadLetterQueueResourcePolicy(rule, this.deadLetterQueue);
            deadLetterConfig = this.deadLetterQueue ? { arn: this.deadLetterQueue?.queueArn } : undefined
        }
        return { ...wrappedResult, deadLetterConfig }
    }
}