import { Stack, Construct, StackProps} from '@aws-cdk/core';
import { Function, Runtime, Code } from '@aws-cdk/aws-lambda';
import { LogGroup, SubscriptionFilter, FilterPattern } from '@aws-cdk/aws-logs';
import { LambdaDestination } from '@aws-cdk/aws-logs-destinations';

export class LoggingStack extends Stack {
  constructor(scope: Construct, id: string, props?: StackProps) {
    super(scope, id, props);

    const handler = new Function(this, 'LoggingLambda', {
      runtime: Runtime.PYTHON_2_7,
      code: Code.asset('lambda'),
      handler: 'logging.lambda_handler',
    });

    const logGroupIr = new LogGroup(this, 'IrLogGroup', {
      logGroupName: '/aws/lambda/logging-1'
    });

    const logGroupUpload = new LogGroup(this, 'UploadLogGroup', {
      logGroupName: '/aws/lambda/logging-2'
    });

    const lambdaLogDest = new LambdaDestination(handler);

    new SubscriptionFilter(this, 'IrSubscription', {
      logGroup: logGroupIr,
      destination: lambdaLogDest,
      filterPattern: FilterPattern.allEvents(),
    });

    new SubscriptionFilter(this, 'UploadSubscription', {
      logGroup: logGroupUpload,
      destination: lambdaLogDest,
      filterPattern: FilterPattern.allEvents(),
    });
  }
}