resource "nutanix_volume_group_disk_v2" "disks" {
  for_each = merge([
    for vm in var.virtualmachine : {
      for idx, disk in vm.additional_disks :
      "${vm.hostname}-${idx}" => {
        hostname = vm.hostname
        idx      = idx
        size     = disk[1]
      }
      if vm.hypervisor == "ahv" && length(vm.additional_disks) > 0
    }
  ]...)

  volume_group_ext_id = resource.nutanix_volume_group_v2.volume_groups[each.value.hostname].ext_id
  disk_size_bytes = each.value.size * 1024 * 1024 * 1024
  disk_data_source_reference {
    name        = "disk${each.value.idx+1}"
    ext_id      = data.nutanix_storage_containers_v2.sg[each.value.hostname].storage_containers[0].ext_id
    entity_type = "STORAGE_CONTAINER"
  }
  lifecycle {
    ignore_changes = [
      disk_data_source_reference
    ]
  }
}

data "nutanix_storage_containers_v2" "sg" {
  for_each = { for vm in var.virtualmachine : vm.hostname => vm 
    if vm.hypervisor == "ahv"
  }
  filter = "clusterExtId eq '${data.nutanix_cluster.cluster[each.value.hostname].cluster_id}'"
  limit  = 1
}

resource "nutanix_volume_group_v2" "volume_groups" {
  for_each = { for vm in var.virtualmachine : vm.hostname => vm 
    if vm.hypervisor == "ahv"
  }
  depends_on = [ resource.nutanix_virtual_machine.vm ]
  name        = "${each.value.hostname}-disk"
  description = "Volume group for ${each.value.hostname}"
  cluster_reference         = data.nutanix_cluster.cluster[each.value.hostname].cluster_id
}

resource "nutanix_volume_group_vm_v2" "attach_disk" {
  for_each = { for vm in var.virtualmachine : vm.hostname => vm if vm.hypervisor == "ahv" }
  depends_on = [ resource.nutanix_volume_group_vm_v2.attach_disk ]

  volume_group_ext_id = nutanix_volume_group_v2.volume_groups[each.key].id
  vm_ext_id           = nutanix_virtual_machine.vm[each.key].id
  index               = 1
}