cluster.tf

variable instance_count {
	description = "Defines the number of VMs to be provisioned."
	default     = "1"
}
variable app_name {
	description = "Application Name"
	default = "RAMDAM"
}

variable location {
	description = "Location of the infrastructure"
	default = "East US"
}

variable instance_size {
	description = "Size of the instance"
	#default = "Standard_B2ms"
	default = "Standard_F64s_v2"
	#default = "Standard_F8s_v2"
	#default = "Standard_H16r"
	#default = "Standard_Hc44rs"
	#default = "Standard_Hb60rs"
}

variable accelerated {
	description = "List of accelerated instance sizes"
	default=[ "Standard_F4s_v2",
						"Standard_F8s_v2",
						"Standard_F16s_v2",
						"Standard_F32s_v2",
						"Standard_F64s_v2",
						#"Standard_Hc44rs",
						#"Standard_Hb60rs"
					]
}

resource "azurerm_resource_group" "RG" {  
	name     = "HPC-${upper(var.app_name)}-RG"
	location = var.location
}

# Use existing HPC-Network resource group
data "azurerm_resource_group" "network-rg" {  
	name     = "HPC-Network"
}

# Use existing HPC-VNET
data "azurerm_virtual_network" "vnet" {
	name                = "HPC-VNET"
	resource_group_name = data.azurerm_resource_group.network-rg.name
}

data "azurerm_subnet" "subnet" {
	name                 = "default"
	virtual_network_name = data.azurerm_virtual_network.vnet.name
	resource_group_name 	= data.azurerm_resource_group.network-rg.name
}

resource "azurerm_public_ip" "pip" {
	name                = "${lower(var.app_name)}-vm${count.index+1}-pip"
	location            = azurerm_resource_group.RG.location
	resource_group_name = azurerm_resource_group.RG.name
	allocation_method   = "Static"
	count               = var.instance_count
}

resource "azurerm_network_interface" "vnic" {
	count               = var.instance_count
	name                = "hpc-${lower(var.app_name)}-nic${count.index+1}"
	location            = azurerm_resource_group.RG.location
	resource_group_name = azurerm_resource_group.RG.name
	enable_accelerated_networking = "${contains(var.accelerated, var.instance_size) ? true : false}"

	ip_configuration {
		name                          = "testConfiguration"
		subnet_id                     = data.azurerm_subnet.subnet.id
		private_ip_address_allocation = "dynamic"
		public_ip_address_id          = element(azurerm_public_ip.pip.*.id, count.index)
	}
}

resource "azurerm_availability_set" "avset" {
	name                         = "${lower(var.app_name)}-avset"
	location                     = azurerm_resource_group.RG.location
	resource_group_name          = azurerm_resource_group.RG.name
	platform_fault_domain_count  = 1
	platform_update_domain_count = 1
	managed                      = true
}

resource "azurerm_virtual_machine" "vm" {
	count                 = var.instance_count
	name                  = "hpc-${lower(var.app_name)}-vm${count.index+1}"
	location              = azurerm_resource_group.RG.location
	availability_set_id   = azurerm_availability_set.avset.id
	resource_group_name   = azurerm_resource_group.RG.name
	network_interface_ids = ["${element(azurerm_network_interface.vnic.*.id, count.index)}"]
	vm_size               = var.instance_size

	

	# Uncomment this line to delete the OS disk automatically when deleting the VM
	delete_os_disk_on_termination = true

	# Uncomment this line to delete the data disks automatically when deleting the VM
	delete_data_disks_on_termination = true

  storage_image_reference {
    publisher = "Canonical"
    offer     = "UbuntuServer"
    sku       = "18.04-LTS"
    version   = "latest"
  }

	storage_os_disk {
		name              = "datadisk${count.index+1}"
		caching           = "ReadWrite"
		create_option     = "FromImage"
		managed_disk_type = "StandardSSD_LRS"
		#managed_disk_type = "Premium_LRS"
		#disk_size_gb		= 200
	}

	# Optional data disks
	storage_data_disk {
	name              = "datadisk_${count.index}"
	managed_disk_type = "Premium_LRS"
	create_option     = "Empty"
	lun               = 0
	disk_size_gb      = "256"
	}

	os_profile {
		computer_name  = "hpc-${lower(var.app_name)}-vm${count.index+1}"
		admin_username = "ansible"
	}
	os_profile_linux_config {
		disable_password_authentication = "true"

		ssh_keys {
			path     = "/home/ansible/.ssh/authorized_keys"
			key_data = "${file("../ansible.key.pub")}"
		}
	}
	
}

resource "null_resource" "prep_ansible" {
	triggers = {
		build_number = "${timestamp()}"
	}
	depends_on = ["azurerm_virtual_machine.vm"]

	provisioner "local-exec" {
		command = "echo [default] ${join(" ", azurerm_public_ip.pip.*.ip_address)} | tr \" \" \"\n\" > ansible.hosts"
	}
}

output "pips_for_ansible_hosts" {
	value = "${azurerm_public_ip.pip.*.ip_address}"
}

output "ime" {
	value = "${formatlist("%s", azurerm_public_ip.pip.*.ip_address)}"
}