#!/usr/bin/env python3
from json import dumps, loads
from os import environ


def get_tfstate(path = 'terraform.tfstate'):
    with open(path, 'r') as f:
        return loads(f.read())


def gen_kube_inv(tf = get_tfstate()):
    kube_ids = []
    hosts = []
    inv = {}

    for res in tf.get('resources', []):
        if res.get('type') != 'proxmox_virtual_environment_container':
            continue

        name = res.get('name', 'unknown').replace('-', '_')
        if inv.get('name') == None:
            inv[name] = {'hosts':[]}

        for ins in res.get('instances', []):
            attrs = ins.get('attributes', {})
            host = (attrs
                .get('initialization', [{}])[0]
                .get('ip_config', [{}])[0]
                .get('ipv4', [{}])[0]
                .get('address', 'unknown')
                .split('/')[0]
            )
            inv[name]['hosts'].append(host)
            kube_ids.append(attrs['id'])
            hosts.append(host)

    inv['kubes'] = {'hosts': hosts}
    return inv, kube_ids


if __name__ == '__main__':
    try:
        inv, kube_ids = gen_kube_inv()
    except KeyError as exc:
        print('Malformed terraform.tfstate detected, try running terraform import')
        raise exc

    proxmox_endpoint = environ.get('TF_VAR_endpoint')
    if proxmox_endpoint:
        proxmox = proxmox_endpoint.split('/')[2].split(':')[0]
        inv['_meta'] = {'hostvars': {proxmox: {'kube_ids': kube_ids}}}
        inv['proxmox'] = {'hosts': [proxmox]}

    print(dumps(inv, indent=2))