inventory

#!/usr/bin/env python3

import argparse
import json
import subprocess

# TODO: read from setenv
UNGROUPED = "ungrouped"
NETWORK = "podlab"
LABEL = f"org.podlab"
DNS_SUFFIX = "dns.podman"

USE_POD_IP = False


def get_pods():
    pr = subprocess.run(
        [
            "podman",
            "ps",
            "--format",
            "{{.ID}}",
            "--filter",
            f"label={LABEL}.group",
        ],
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        check=False,
    )

    if pr.returncode != 0:
        raise RuntimeError(pr.stdout)
    return pr.stdout.decode().splitlines()


def inspect_pod(pods):
    if not pods:
        return []
    pr = subprocess.run(
        ["podman", "inspect", "--format", "json"] + pods
        if isinstance(pods, list)
        else [pods],
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        check=False,
    )

    if pr.returncode != 0:
        raise RuntimeError(pr.stdout)
    pods = json.loads(pr.stdout)
    return pods


def get_list(pods):
    # "Labels" "org.ansible-lab.group"
    groups = {}
    for p in pods:
        g: str = p["Config"]["Labels"].get(
            f"{LABEL}.group", UNGROUPED
        )  # .replace("-", "_")
        if g not in groups:
            groups[g] = {"hosts": [], "vars": {}, "children": []}
        groups[g]["hosts"].append(
            # p["NetworkSettings"]["Networks"].get(NETWORK, {}).get("IPAddress", None)
            f'{p["Config"]["Hostname"]}'
        )
    return groups


def get_host(pod):
    i = inspect_pod(pod)[0]
    return json.dumps(
        {
            **(
                {
                    "ansible_host": i["NetworkSettings"]["Networks"]
                    .get(NETWORK, {})
                    .get("IPAddress", None)
                }
                if USE_POD_IP
                else {}
            ),
            # f'{i["Config"]["Hostname"]}.{DNS_SUFFIX}',
            **(lambda v: {"podlab_version": v} if v else {})(
                i["Config"]["Labels"].get("org.podlab.version", None)
            ),
        }
    )


def main():
    parser = argparse.ArgumentParser("Podman Inventory Module")
    parser.add_argument("--list", action="store_true", help="List active pods")
    parser.add_argument("--host", help="Details about a specific pod")
    args = parser.parse_args()

    if args.host:
        print(get_host([args.host]))
    if args.list:
        print(json.dumps(get_list(inspect_pod(get_pods()))))


if __name__ == "__main__":
    main()