// Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: MIT-0


resource "aws_lambda_function" "lambda_scheduler" {
  s3_bucket     = var.code_s3_bucket
  s3_key        = var.code_s3_file
  function_name = "scheduler"
  role          = var.iam_role
  handler       = "lambda_entry_points.handle_input_upload"
  runtime       = "python3.7"
  timeout       = 60
  memory_size   = 512 
  vpc_config {
    security_group_ids = [var.sg_ids]
    subnet_ids = var.subnet_ids
  }
  environment {
    variables = {
      SSM_PREFIX = var.ssm_prefix
    }
  }
}

resource "aws_lambda_permission" "allow_bucket" {
  statement_id   = "AllowExecutionFromS3Bucket"
  action         = "lambda:InvokeFunction"
  function_name  = aws_lambda_function.lambda_scheduler.arn
  principal      = "s3.amazonaws.com"
  source_arn     = var.s3bucket_arn
  source_account = var.account_id
}

resource "aws_s3_bucket_notification" "input_store_notification" {
  bucket = var.s3bucket
  lambda_function {
    lambda_function_arn = aws_lambda_function.lambda_scheduler.arn 
    events = ["s3:ObjectCreated:*"]
    filter_suffix = ".json"
    }
  depends_on = [aws_lambda_permission.allow_bucket]
}