locals {
  worker_prefix = "worker"
  worker_labels = "project.io/node-pool=worker"

  workers = { for k in flatten([
    for zone in local.zones : [
      for inx in range(lookup(try(var.instances[zone], {}), "worker_count", 0)) : {
        id : lookup(try(var.instances[zone], {}), "worker_id", 9000) + inx
        name : "${local.worker_prefix}-${lower(substr(zone, -1, -1))}${1 + inx}"
        zone : zone
        node_name : zone
        cpu : lookup(try(var.instances[zone], {}), "worker_cpu", 1)
        mem : lookup(try(var.instances[zone], {}), "worker_mem", 2048)
        ip0 : lookup(try(var.instances[zone], {}), "worke_ip0", "ip6=auto")
        ipv4 : "${cidrhost(local.subnets[zone], 4 + inx)}/24"
        gwv4 : local.gwv4
      }
    ]
  ]) : k.name => k }
}

resource "null_resource" "worker_machineconfig" {
  for_each = { for k, v in var.instances : k => v if lookup(try(var.instances[k], {}), "worker_count", 0) > 0 }
  connection {
    type = "ssh"
    user = "root"
    host = "${each.key}.${var.proxmox_domain}"
  }

  provisioner "file" {
    content = templatefile("${path.module}/templates/worker.yaml.tpl",
      merge(var.kubernetes, {
        lbv4        = local.ipv4_vip
        nodeSubnets = var.vpc_main_cidr
        labels      = local.worker_labels
    }))
    destination = "/var/lib/vz/snippets/${local.worker_prefix}.yaml"
  }

  triggers = {
    params = filemd5("${path.module}/templates/worker.yaml.tpl")
  }
}

resource "null_resource" "worker_metadata" {
  for_each = local.workers
  connection {
    type = "ssh"
    user = "root"
    host = "${each.value.node_name}.${var.proxmox_domain}"
  }

  provisioner "file" {
    content = templatefile("${path.module}/templates/metadata.yaml", {
      hostname : each.value.name,
      id : each.value.id,
      providerID : "proxmox://${var.region}/${each.value.id}",
      type : "${each.value.cpu}VCPU-${floor(each.value.mem / 1024)}GB",
      zone : each.value.zone,
      region : var.region,
    })
    destination = "/var/lib/vz/snippets/${each.value.name}.metadata.yaml"
  }

  triggers = {
    params = join(",", [for k, v in local.workers[each.key] : "${k}-${v}"])
  }
}

# resource "proxmox_virtual_environment_vm" "talos" {
#   for_each = local.workers
#   name     = each.value.name
#   tags     = ["talos"]

#   node_name = each.value.node_name
#   vm_id     = each.value.id

#   initialization {
#     datastore_id = "local"
#     ip_config {
#       ipv6 {
#         address = "slaac"
#         # gateway = ""
#       }
#     }
#     ip_config {
#       ipv4 {
#         address = "2.3.4.5/24"
#       }
#     }
#     user_data_file_id = ""
#   }
#   clone {
#     vm_id        = 102
#     datastore_id = var.proxmox_storage
#   }
#   disk {
#     datastore_id = var.proxmox_storage
#     interface    = "scsi0"
#     ssd          = true
#     size         = 32
#     file_format  = "raw"
#   }
#   cpu {
#     cores   = each.value.cpu
#     sockets = 1
#     type    = "host"
#     flags   = ["+aes"]
#   }
#   memory {
#     dedicated = each.value.mem
#   }

#   network_device {
#     model  = "virtio"
#     bridge = "vmbr0"
#     # firewall = true
#   }
#   network_device {
#     model  = "virtio"
#     bridge = "vmbr1"
#   }

#   operating_system {
#     type = "l26"
#   }
#   agent {
#     enabled = false
#   }

#   serial_device {}
#   lifecycle {
#     ignore_changes = [
#       tags,
#       cpu,
#       memory,
#       network_device,
#     ]
#   }

#   depends_on = [null_resource.worker_machineconfig, null_resource.worker_metadata]
# }

resource "proxmox_vm_qemu" "worker" {
  for_each    = local.workers
  name        = each.value.name
  vmid        = each.value.id
  target_node = each.value.node_name
  clone       = var.proxmox_image

  agent                   = 0
  define_connection_info  = false
  os_type                 = "ubuntu"
  qemu_os                 = "l26"
  ipconfig0               = each.value.ip0
  ipconfig1               = "ip=${each.value.ipv4},gw=${each.value.gwv4}"
  cicustom                = "user=local:snippets/${local.worker_prefix}.yaml,meta=local:snippets/${each.value.name}.metadata.yaml"
  cloudinit_cdrom_storage = var.proxmox_storage

  onboot  = false
  cpu     = "host,flags=+aes"
  sockets = 1
  cores   = each.value.cpu
  memory  = each.value.mem
  scsihw  = "virtio-scsi-single"

  vga {
    memory = 0
    type   = "serial0"
  }
  serial {
    id   = 0
    type = "socket"
  }

  network {
    model    = "virtio"
    bridge   = "vmbr0"
    firewall = true
  }
  network {
    model  = "virtio"
    bridge = "vmbr1"
  }

  boot = "order=scsi0"
  disk {
    type    = "scsi"
    storage = var.proxmox_storage
    size    = "32G"
    cache   = "writethrough"
    ssd     = 1
    backup  = false
  }
  disk {
    type     = "scsi"
    storage  = var.proxmox_storage
    size     = "128G"
    cache    = "none"
    iothread = 1
    ssd      = 1
    backup   = false
  }

  lifecycle {
    ignore_changes = [
      boot,
      disk,
      network,
      desc,
      numa,
      agent,
      ipconfig0,
      ipconfig1,
      define_connection_info,
    ]
  }

  depends_on = [null_resource.worker_machineconfig, null_resource.worker_metadata]
}