##########################################################################################
# ensure we can get to the node first
resource "null_resource" "remote-config" {
  count = var.data-node-count
  provisioner "remote-exec" {
    connection {
      user        = "ubuntu"
      host        = element(aws_eip.re-eip.*.public_ip, count.index)
      private_key = file(local.ssh_key_path)
      agent       = true
    }
    inline = ["sudo apt update > /dev/null"]
  }
  depends_on = [aws_instance.re, aws_eip_association.re-eip-assoc, null_resource.inventory-setup, null_resource.ssh-setup]
}

###############################################################################
# Template Data
data "template_file" "ansible_inventory" {
  count    = var.data-node-count
  template = "${file("${path.module}/inventory.tpl")}"
  vars = {
    host_ip  = element(aws_eip.re-eip.*.public_ip, count.index)
    vpc_name = var.vpc-name
    ncount   = count.index
  }
  depends_on = [aws_instance.re, aws_eip_association.re-eip-assoc, aws_volume_attachment.re-ephemeral]
}
data "template_file" "ansible_tester_inventory" {
  count    = local.tester_count
  template = "${file("${path.module}/inventory.tpl")}"
  vars = {
    host_ip  = element(aws_eip.tester-eip.*.public_ip, count.index)
    vpc_name = var.vpc-name
    ncount   = count.index
  }
  depends_on = [aws_instance.tester, aws_eip_association.tester-eip-assoc]
}

data "template_file" "ssh_config" {
  template = "${file("${path.module}/ssh.tpl")}"
  vars = {
    vpc_name = var.vpc-name
  }
  depends_on = [aws_instance.re, aws_eip_association.re-eip-assoc, aws_volume_attachment.re-ephemeral]
}

###############################################################################
# Template Write
resource "null_resource" "inventory-setup" {
  count = var.data-node-count
  provisioner "local-exec" {
    command = "echo \"${element(data.template_file.ansible_inventory.*.rendered, count.index)}\" > /tmp/${var.vpc-name}_node_${count.index}.ini"
  }
  depends_on = [data.template_file.ansible_inventory]
}

resource "null_resource" "tester-inventory-setup" {
  count = local.tester_count
  provisioner "local-exec" {
    command = "echo \"${element(data.template_file.ansible_tester_inventory.*.rendered, count.index)}\" > /tmp/${var.vpc-name}_tester_node_${count.index}.ini"
  }
  depends_on = [data.template_file.ansible_inventory]
}

resource "null_resource" "ssh-setup" {
  provisioner "local-exec" {
    command = "echo \"${data.template_file.ssh_config.rendered}\" > /tmp/${var.vpc-name}_node.cfg"
  }
  depends_on = [data.template_file.ssh_config]
}

###############################################################################
# Run some ansible
resource "null_resource" "ansible-run" {
  count = var.data-node-count
  provisioner "local-exec" {
    command = "ansible-playbook ${path.module}/ansible/playbook.yml --private-key ${local.ssh_key_path} -i /tmp/${var.vpc-name}_node_${count.index}.ini --become -e 'ENABLE_VOLUMES=${var.enable-volumes}'"
  }
  depends_on = [null_resource.remote-config]
}

resource "null_resource" "ansible-tester-run" {
  count = var.data-node-count
  provisioner "local-exec" {
    command = "ansible-playbook ${path.module}/ansible/testernode.yml --private-key ${local.ssh_key_path} -i /tmp/${var.vpc-name}_tester_node_${count.index}.ini --become -e 'ENABLE_VOLUMES=${var.enable-volumes}'"
  }
  depends_on = [null_resource.remote-config]
}