packer {
  required_plugins {
    hcloud = {
      version = ">= 1.6.0"
      source  = "github.com/hetznercloud/hcloud"
    }
  }
}

variable "hcloud_token" {
  type      = string
  sensitive = true
}

variable "agent_name" {
  type = string
}

variable "cloud_init_tier" {
  type    = string
  default = "minimal"
}

variable "install_commands" {
  type    = list(string)
  default = []
}

locals {
  timestamp  = formatdate("YYYYMMDD-hhmm", timestamp())
  image_name = "spawn-${var.agent_name}-${local.timestamp}"
}

source "hcloud" "spawn" {
  token        = var.hcloud_token
  image        = "ubuntu-24.04"
  location     = "fsn1"
  # 4 GB RAM — Claude's native installer and zeroclaw's Rust build
  # get OOM-killed on smaller instances. Snapshots built here work on all sizes.
  server_type  = "cx23"
  ssh_username = "root"

  snapshot_name = local.image_name
  snapshot_labels = {
    managed-by = "packer"
    project    = "spawn"
    agent      = var.agent_name
  }
}

build {
  sources = ["source.hcloud.spawn"]

  # Wait for cloud-init to finish (Hetzner base images run it on first boot)
  provisioner "shell" {
    inline = [
      "cloud-init status --wait || true",
    ]
  }

  # Wait for any apt locks to be released (cloud-init may hold them)
  provisioner "shell" {
    inline = [
      "for i in $(seq 1 30); do fuser /var/lib/dpkg/lock-frontend >/dev/null 2>&1 || break; echo 'Waiting for apt lock...'; sleep 2; done",
    ]
  }

  # Run the tier script (installs base packages: curl, git, node, bun, etc.)
  provisioner "shell" {
    script = "packer/scripts/tier-${var.cloud_init_tier}.sh"
  }

  # Install the agent
  provisioner "shell" {
    inline = var.install_commands
    environment_vars = [
      "HOME=/root",
      "DEBIAN_FRONTEND=noninteractive",
      "PATH=/root/.local/bin:/root/.bun/bin:/root/.npm-global/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
    ]
  }

  # Leave a marker so the CLI knows this is a pre-baked snapshot
  provisioner "shell" {
    inline = [
      "echo 'spawn-${var.agent_name}' > /root/.spawn-snapshot",
      "date -u '+%Y-%m-%dT%H:%M:%SZ' >> /root/.spawn-snapshot",
      "touch /root/.cloud-init-complete",
    ]
    environment_vars = [
      "HOME=/root",
      "PATH=/root/.local/bin:/root/.bun/bin:/root/.npm-global/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
    ]
  }

  # Install security updates and clean up
  provisioner "shell" {
    inline = [
      "apt-get update -y",
      "apt-get -o Dpkg::Options::='--force-confold' dist-upgrade -y",
      "apt-get -y autoremove",
      "apt-get -y autoclean",
    ]
    environment_vars = [
      "DEBIAN_FRONTEND=noninteractive",
    ]
  }

  # Cleanup — clear secrets, keys, history, logs so each server gets a fresh identity.
  # cloud-init re-runs on first boot to re-inject SSH keys.
  provisioner "shell" {
    inline = [
      # Ensure /tmp exists with correct permissions
      "mkdir -p /tmp",
      "chmod 1777 /tmp",

      # Remove SSH authorized keys (cloud-init re-injects on first boot)
      "rm -f /root/.ssh/authorized_keys",
      "find /home -name authorized_keys -delete",

      # Remove SSH host keys (regenerated on first boot)
      "rm -f /etc/ssh/ssh_host_*",
      "touch /etc/ssh/revoked_keys",
      "chmod 600 /etc/ssh/revoked_keys",

      # Clear bash history
      "rm -f /root/.bash_history",
      "find /home -name .bash_history -delete",

      # Truncate recent log files and remove archived logs
      "find /var/log -mtime -1 -type f -exec truncate -s 0 {} \\;",
      "rm -rf /var/log/*.gz /var/log/*.[0-9] /var/log/*-????????",

      # Clear apt cache
      "apt-get clean",
      "rm -rf /var/lib/apt/lists/*",

      # Clear tmp
      "rm -rf /tmp/* /var/tmp/*",

      # Remove cloud-init instance data so it re-runs on first boot
      "rm -rf /var/lib/cloud/instances/*",

      # Remove machine-id so each server gets a unique one
      "truncate -s 0 /etc/machine-id",
      "rm -f /var/lib/dbus/machine-id",
      "ln -sf /etc/machine-id /var/lib/dbus/machine-id",

      # Reset cloud-init so it runs again on first boot
      "cloud-init clean --logs",

      # Zero-fill free disk space to reduce snapshot size
      "dd if=/dev/zero of=/zerofile bs=4096 || true",
      "rm -f /zerofile",

      "sync",
    ]
  }

  # Write Packer manifest for CI
  post-processor "manifest" {
    output     = "packer/manifest.json"
    strip_path = true
  }
}