resource "aws_s3_bucket" "function_bucket" {
  bucket_prefix = "function-bucket"
}

resource "aws_s3_bucket_server_side_encryption_configuration" "aes256" {
  bucket = aws_s3_bucket.function_bucket.id

  rule {
    apply_server_side_encryption_by_default {
      sse_algorithm = "AES256"
    }
  }
}

resource "aws_s3_bucket_policy" "function_bucket_policy" {
  count  = var.pipeline.inputs.environment_account_ids != "" ? 1 : 0
  policy = data.aws_iam_policy_document.function_bucket_policy_document.json
  bucket = aws_s3_bucket.function_bucket.id
}

resource "aws_codebuild_project" "build_project" {
  name         = "${var.service.name}-build-project"
  service_role = aws_iam_role.publish_role.arn

  artifacts {
    type = "CODEPIPELINE"
  }

  environment {
    compute_type                = "BUILD_GENERAL1_SMALL"
    image                       = "aws/codebuild/amazonlinux2-x86_64-standard:3.0"
    type                        = "LINUX_CONTAINER"
    image_pull_credentials_type = "CODEBUILD"

    environment_variable {
      name  = "bucket_name"
      value = aws_s3_bucket.function_bucket.bucket
    }

    environment_variable {
      name  = "service_name"
      value = var.service.name
    }
  }

  source {
    buildspec = <<EOF
                {
                  "version": "0.2",
                  "phases": {
                    "install": {
                      "runtime-versions":
                      ${tomap({
    "ruby2.7"       = jsonencode({ "ruby" = "2.7" })
    "nodejs12.x"    = jsonencode({ "nodejs" = "12.x" })
    "python3.8"     = jsonencode({ "python" = "3.8" })
    "java11"        = jsonencode({ "java" = "openjdk11.x" })
    "dotnetcore3.1" = jsonencode({ "dotnet" : "3.1" })
})[var.service_instances[0].outputs.LambdaRuntime]},
                      "commands": [
                        "pip3 install --upgrade --user awscli",
                        "echo 'f6bd1536a743ab170b35c94ed4c7c4479763356bd543af5d391122f4af852460  yq_linux_amd64' > yq_linux_amd64.sha",
                        "wget https://github.com/mikefarah/yq/releases/download/3.4.0/yq_linux_amd64",
                        "sha256sum -c yq_linux_amd64.sha",
                        "mv yq_linux_amd64 /usr/bin/yq",
                        "chmod +x /usr/bin/yq"
                      ]
                    },
                    "pre_build": {
                      "commands": [
                        "cd $CODEBUILD_SRC_DIR/${var.pipeline.inputs.code_dir}",
                        "${var.pipeline.inputs.unit_test_command}"
                      ]
                    },
                    "build": {
                      "commands": [
                        "${var.pipeline.inputs.packaging_command}",
                        "FUNCTION_KEY=$CODEBUILD_BUILD_NUMBER/function.zip",
                        "aws s3 cp function.zip s3://$bucket_name/$FUNCTION_KEY"
                      ]
                    },
                    "post_build": {
                      "commands": [
                        "aws proton --region $AWS_DEFAULT_REGION get-service --name $service_name | jq -r .service.spec > service.yaml",
                        "yq w service.yaml 'instances[*].spec.lambda_bucket' \"$bucket_name\" > rendered_service_tmp.yaml",
                        "yq w rendered_service_tmp.yaml 'instances[*].spec.lambda_key' \"$FUNCTION_KEY\" > rendered_service.yaml"
                      ]
                    }
                  },
                  "artifacts": {
                    "files": [
                      "${var.pipeline.inputs.code_dir}/rendered_service.yaml"
                    ]
                  }
                }
EOF

type = "CODEPIPELINE"
}

encryption_key = aws_kms_key.pipeline_artifacts_bucket_key.arn
}


resource "aws_codebuild_project" "deploy_project" {
  for_each     = { for instance in var.service_instances : instance.name => instance }
  name         = "deploy-${var.service.name}-${index(var.service_instances, each.value)}"
  service_role = aws_iam_role.deployment_role.arn

  artifacts {
    type = "CODEPIPELINE"
  }

  environment {
    compute_type                = "BUILD_GENERAL1_SMALL"
    image                       = "aws/codebuild/amazonlinux2-x86_64-standard:3.0"
    type                        = "LINUX_CONTAINER"
    privileged_mode             = false
    image_pull_credentials_type = "CODEBUILD"

    environment_variable {
      name  = "service_instance_name"
      value = each.value.name
    }

    environment_variable {
      name  = "service_name"
      value = var.service.name
    }
  }

  source {
    type      = "CODEPIPELINE"
    buildspec = <<EOF
          {
            "version": "0.2",
            "phases": {
              "build": {
                "commands": [
                  "pip3 install --upgrade --user awscli",
                  "echo 'rendered_service.yaml':",
                  "cat ${var.pipeline.inputs.code_dir}/rendered_service.yaml",
                  "aws proton --region $AWS_DEFAULT_REGION update-service-instance --deployment-type CURRENT_VERSION --name $service_instance_name --service-name $service_name --spec file://${var.pipeline.inputs.code_dir}/rendered_service.yaml",
                  "aws proton --region $AWS_DEFAULT_REGION wait service-instance-deployed --name $service_instance_name --service-name $service_name"
                ]
              }
            }
          }
EOF
  }

  encryption_key = aws_kms_key.pipeline_artifacts_bucket_key.arn
}

resource "aws_iam_role" "publish_role" {
  name_prefix = "publish-role"

  assume_role_policy = jsonencode({
    "Version" : "2012-10-17",
    "Statement" : [
      {
        "Effect" : "Allow",
        "Principal" : {
          "Service" : "codebuild.amazonaws.com"
        },
        "Action" : "sts:AssumeRole"
      }
    ]
  })
}

resource "aws_iam_policy" "publish_role_policy" {
  policy = data.aws_iam_policy_document.publish_role_policy_document.json
}

resource "aws_iam_role_policy_attachment" "publish_role_policy_attachment" {
  policy_arn = aws_iam_policy.publish_role_policy.arn
  role       = aws_iam_role.publish_role.name
}

resource "aws_iam_role" "deployment_role" {
  name_prefix = "deployment-role"

  assume_role_policy = jsonencode({
    "Version" : "2012-10-17",
    "Statement" : [
      {
        "Effect" : "Allow",
        "Principal" : {
          "Service" : "codebuild.amazonaws.com"
        },
        "Action" : "sts:AssumeRole"
      }
    ]
  })
}

resource "aws_iam_policy" "deployment_role_policy" {
  policy = data.aws_iam_policy_document.deployment_role_policy.json
}

resource "aws_iam_role_policy_attachment" "deployment_role_policy_attachment" {
  policy_arn = aws_iam_policy.deployment_role_policy.arn
  role       = aws_iam_role.deployment_role.name
}

resource "aws_s3_bucket" "pipeline_artifacts_bucket" {
  bucket_prefix = "pipeline-artifacts-bucket"
}

resource "aws_s3_bucket_public_access_block" "pipeline_artifacts_bucket_access_block" {
  bucket                  = aws_s3_bucket.pipeline_artifacts_bucket.id
  block_public_acls       = true
  block_public_policy     = true
  ignore_public_acls      = true
  restrict_public_buckets = true
}

resource "aws_s3_bucket_server_side_encryption_configuration" "pipeline_artifacts_bucket_encryption" {
  bucket = aws_s3_bucket.pipeline_artifacts_bucket.id

  rule {
    apply_server_side_encryption_by_default {
      sse_algorithm     = "aws:kms"
      kms_master_key_id = aws_kms_key.pipeline_artifacts_bucket_key.arn
    }
  }
}

resource "aws_kms_key" "pipeline_artifacts_bucket_key" {
  policy = data.aws_iam_policy_document.pipeline_artifacts_bucket_key_policy.json
}

resource "aws_kms_alias" "pipeline_artifacts_bucket_key_alias" {
  target_key_id = aws_kms_key.pipeline_artifacts_bucket_key.id
  name          = "alias/codepipeline-encryption-key-${var.service.name}"
}

resource "aws_iam_role" "pipeline_role" {
  name_prefix = "pipeline-role"

  assume_role_policy = jsonencode({
    "Version" : "2012-10-17",
    "Statement" : [
      {
        "Effect" : "Allow",
        "Principal" : {
          "Service" : "codepipeline.amazonaws.com"
        },
        "Action" : "sts:AssumeRole"
      }
    ]
  })
}

resource "aws_iam_policy" "pipeline_role_policy" {
  policy = data.aws_iam_policy_document.pipeline_role_policy.json
}

resource "aws_iam_role_policy_attachment" "pipeline_role_policy_attachment" {
  policy_arn = aws_iam_policy.pipeline_role_policy.arn
  role       = aws_iam_role.pipeline_role.name
}

resource "aws_codepipeline" "pipeline" {
  name     = "${var.service.name}-pipeline"
  role_arn = aws_iam_role.pipeline_role.arn

  stage {
    name = "Source"
    action {
      category  = "Source"
      name      = "Checkout"
      owner     = "AWS"
      provider  = "CodeStarSourceConnection"
      version   = "1"
      run_order = 1

      configuration = {
        ConnectionArn : var.service.repository_connection_arn
        FullRepositoryId : var.service.repository_id
        BranchName : var.service.branch_name
      }
      output_artifacts = ["Artifact_Source_Checkout"]
    }
  }

  stage {
    name = "Build"
    action {
      category  = "Build"
      name      = "Build"
      owner     = "AWS"
      provider  = "CodeBuild"
      version   = "1"
      run_order = 1

      configuration = {
        ProjectName = aws_codebuild_project.build_project.name
      }
      input_artifacts  = ["Artifact_Source_Checkout"]
      output_artifacts = ["BuildOutput"]
      role_arn         = aws_iam_role.pipeline_build_codepipeline_action_role.arn
    }
  }

  dynamic "stage" {
    for_each = toset(var.service_instances)

    content {
      name = "deploy-${var.service.name}-${index(var.service_instances, stage.value)}"

      action {
        category  = "Build"
        name      = "deploy-${var.service.name}-${index(var.service_instances, stage.value)}"
        owner     = "AWS"
        provider  = "CodeBuild"
        version   = "1"
        run_order = 1

        configuration = {
          ProjectName = "deploy-${var.service.name}-${index(var.service_instances, stage.value)}"
        }
        input_artifacts = ["BuildOutput"]
        role_arn        = aws_iam_role.pipeline_deploy_codepipeline_action_role.arn
      }
    }
  }
  artifact_store {
    encryption_key {
      id   = aws_kms_key.pipeline_artifacts_bucket_key.arn
      type = "KMS"
    }
    location = aws_s3_bucket.pipeline_artifacts_bucket.bucket
    type     = "S3"
  }
  depends_on = [
    aws_iam_role.pipeline_role,
    data.aws_iam_policy_document.pipeline_role_policy
  ]

}

resource "aws_iam_role" "pipeline_build_codepipeline_action_role" {
  name_prefix = "pipeline-build-action-role"

  assume_role_policy = jsonencode({
    "Version" : "2012-10-17",
    "Statement" : [
      {
        "Effect" : "Allow",
        "Principal" : {
          "AWS" : "arn:aws:iam::${local.account_id}:root"
        },
        "Action" : "sts:AssumeRole"
      }
    ]
  })
}

resource "aws_iam_policy" "pipeline_build_action_role_policy" {
  policy = data.aws_iam_policy_document.pipeline_build_codepipeline_action_role_policy.json
}

resource "aws_iam_role_policy_attachment" "pipeline_build_codepipeline_action_role_attachment" {
  policy_arn = aws_iam_policy.pipeline_build_action_role_policy.arn
  role       = aws_iam_role.pipeline_build_codepipeline_action_role.name
}

resource "aws_iam_role" "pipeline_deploy_codepipeline_action_role" {
  name_prefix = "pipeline-deploy-action-role"

  assume_role_policy = jsonencode({
    "Version" : "2012-10-17",
    "Statement" : [
      {
        "Effect" : "Allow",
        "Principal" : {
          "AWS" : "arn:aws:iam::${local.account_id}:root"
        },
        "Action" : "sts:AssumeRole"
      }
    ]
  })
}

resource "aws_iam_policy" "pipeline_deploy_action_role_policy" {
  policy = data.aws_iam_policy_document.pipeline_deploy_codepipeline_action_role_policy.json
}

resource "aws_iam_role_policy_attachment" "pipeline_deploy_codepipeline_action_role_attachment" {
  policy_arn = aws_iam_policy.pipeline_deploy_action_role_policy.arn
  role       = aws_iam_role.pipeline_deploy_codepipeline_action_role.name
}