#!/usr/bin/env python3
import json, yaml, subprocess, os, sys, pathlib

def load_servers():
    with open(os.path.join(os.path.dirname(__file__), '..', 'inventory', 'servers.yaml')) as f:
        data = yaml.safe_load(f)
    servers = data.get('servers', [])
    return servers

def load_tf_outputs():
    # Try to read terraform outputs to attach IPs, if available.
    try:
        out = subprocess.check_output(['terraform', '-chdir=../terraform/hetzner', 'output', '-json'], stderr=subprocess.DEVNULL, text=True)
        j = json.loads(out)
        servers = j.get('servers', {}).get('value', {})
        return servers  # {name: ip}
    except Exception:
        return {}

def main():
    server_list = load_servers()
    tf = load_tf_outputs()
    host_names = []
    hostvars = {}
    for s in server_list:
        name = s['name']
        host_names.append(name)
        hostvars[name] = {
            "ansible_host": tf.get(name) or s.get('ip'),
            "ansible_user": s.get('ssh_user', 'admin')
        }
    inv = {
        "all": {"hosts": host_names},
        "_meta": {"hostvars": hostvars}
    }
    print(json.dumps(inv))

if __name__ == "__main__":
    main()