resource "google_composer_environment" "airflow" {
  name    = var.project_id
  project = var.project_id
  region  = var.region
  config {
    private_environment_config {
      enable_private_endpoint          = true
      enable_privately_used_public_ips = false
      # ip_allocation_policy {
      #   cluster_ipv4_cidr_block  = var.cluster_ipv4_cidr_block
      #   services_ipv4_cidr_block = var.services_ipv4_cidr_block
      # }
    }

    software_config {
      env_variables = var.env_variables
      image_version = "composer-2-airflow-2"
    }
    workloads_config {
      scheduler {
        cpu        = 0.5
        memory_gb  = 1.875
        storage_gb = 1
        count      = 1
      }
      web_server {
        cpu        = 0.5
        memory_gb  = 1.875
        storage_gb = 1
      }
      worker {
        cpu        = 0.5
        memory_gb  = 1.875
        storage_gb = 1
        min_count  = 1
        max_count  = 3
      }
    }
    environment_size = "ENVIRONMENT_SIZE_SMALL"
    node_config {
      network         = google_compute_network.airflow.id
      subnetwork      = google_compute_subnetwork.airflow.id
      service_account = google_service_account.airflow.name
    }
  }
}