###########################
# Extract Lambda Function #
###########################

data "archive_file" "extract_lambda_zip" {
  type        = "zip"
  source_file = "${path.module}/../src/extract_lambda.py"
  output_path = "${path.module}/../extract_function.zip"
}
resource "aws_s3_object" "extract_lambda_code" {
  bucket = aws_s3_bucket.lambda_code_bucket.bucket
  key    = "${var.extract_lambda_name}/extract_function.zip"
  source = data.archive_file.extract_lambda_zip.output_path
  etag   = filemd5(data.archive_file.extract_lambda_zip.output_path)
}

resource "aws_lambda_function" "extract_lambda" {
  function_name    = var.extract_lambda_name
  s3_bucket        = aws_s3_bucket.lambda_code_bucket.bucket
  s3_key           = aws_s3_object.extract_lambda_code.key
  layers           = [aws_lambda_layer_version.lambda_layer.arn]
  role             = aws_iam_role.multi_service_role.arn
  handler          = "extract_lambda.lambda_handler"
  runtime          = "python3.11"
  source_code_hash = data.archive_file.extract_lambda_zip.output_base64sha256

  lifecycle {
    create_before_destroy = true
  }

  depends_on = [aws_s3_object.extract_lambda_code]
}

#############################
# Transform Lambda Function #
#############################

data "archive_file" "transform_lambda_zip" {
  type        = "zip"
  source_file = "${path.module}/../src/transform_lambda.py"
  output_path = "${path.module}/../transform_function.zip"
}
resource "aws_s3_object" "transform_lambda_code" {
  bucket = aws_s3_bucket.lambda_code_bucket.bucket
  key    = "${var.transform_lambda_name}/transform_function.zip"
  source = data.archive_file.transform_lambda_zip.output_path
  etag   = filemd5(data.archive_file.transform_lambda_zip.output_path)
}

resource "aws_lambda_function" "transform_lambda" {
  function_name    = var.transform_lambda_name
  s3_bucket        = aws_s3_bucket.lambda_code_bucket.bucket
  s3_key           = aws_s3_object.transform_lambda_code.key
  layers           = [aws_lambda_layer_version.lambda_layer.arn]
  role             = aws_iam_role.multi_service_role.arn
  handler          = "transform_lambda.lambda_handler"
  runtime          = "python3.11"
  source_code_hash = data.archive_file.transform_lambda_zip.output_base64sha256

  lifecycle {
    create_before_destroy = true
  }

  depends_on = [aws_s3_object.transform_lambda_code]
}

########################
# Load Lambda Function #
########################

data "archive_file" "load_lambda_zip" {
  type        = "zip"
  source_file = "${path.module}/../src/load_lambda.py"
  output_path = "${path.module}/../load_function.zip"
}
resource "aws_s3_object" "load_lambda_code" {
  bucket = aws_s3_bucket.lambda_code_bucket.bucket
  key    = "${var.load_lambda_name}/load_function.zip"
  source = data.archive_file.load_lambda_zip.output_path
  etag   = filemd5(data.archive_file.load_lambda_zip.output_path)
}

resource "aws_lambda_function" "load_lambda" {
  function_name    = var.load_lambda_name
  s3_bucket        = aws_s3_bucket.lambda_code_bucket.bucket
  s3_key           = aws_s3_object.load_lambda_code.key
  layers           = [aws_lambda_layer_version.lambda_layer.arn]
  role             = aws_iam_role.multi_service_role.arn
  handler          = "load_lambda.lambda_handler"
  runtime          = "python3.11"
  source_code_hash = data.archive_file.load_lambda_zip.output_base64sha256

  lifecycle {
    create_before_destroy = true
  }

  depends_on = [aws_s3_object.load_lambda_code]
}

######################
# Lambda Layer Setup #
######################

locals {
  layer_dir  = "../"
  layer_zip  = "layer.zip"
  layer_name = "lambda_layer"
  script_dir = "../scripts"
}

resource "null_resource" "prepare_layer" {
  provisioner "local-exec" {
    command = "bash ${local.script_dir}/make_layer_zip.sh"
  }
}

resource "aws_s3_object" "lambda_layer_zip" {
  bucket     = aws_s3_bucket.lambda_code_bucket.id #bucket instead of id
  key        = "${local.layer_name}/${local.layer_zip}"
  source     = "${local.layer_dir}/${local.layer_zip}"
  depends_on = [null_resource.prepare_layer]
}

resource "aws_lambda_layer_version" "lambda_layer" {
  layer_name          = local.layer_name
  compatible_runtimes = ["python3.11"]
  s3_bucket           = aws_s3_bucket.lambda_code_bucket.bucket
  s3_key              = aws_s3_object.lambda_layer_zip.key
  skip_destroy        = true
  depends_on          = [aws_s3_object.lambda_layer_zip]
}