import argparse
import os
import requests
import json
from collections import namedtuple

Config = namedtuple("Config", ["port", "outfile", "outlink", "env"])


def gateway_targets(entry):
    targets = [
        # Config(
        #     None,
        #     "/tmp/temp_targets_gateway.json",
        #     f"/tmp/prom_targets_gateway_{entry['env']}.json",
        #     entry["env"],
        # ),
        Config(
            9100,
            "/tmp/temp_targets_gateway_node.json",
            f"/tmp/prom_targets_gateway_node_{entry['env']}.json",
            entry["env"],
        ),
    ]

    gateways = requests.get(f"{entry['nym_api']}/api/v1/gateways").json()

    for config in targets:
        config_to_targets(config, gateways, {"kind": "gateway"})


def mixnode_targets(entry):
    targets = [
        Config(
            None,
            "/tmp/temp_targets_mix.json",
            f"/tmp/prom_targets_mix_{entry['env']}.json",
            entry["env"],
        ),
        Config(
            9100,
            "/tmp/temp_targets_node.json",
            f"/tmp/prom_targets_node_{entry['env']}.json",
            entry["env"],
        ),
    ]

    mixnodes = requests.get(f"{entry['nym_api']}/api/v1/mixnodes").json()

    for config in targets:
        config_to_targets(config, mixnodes, {"kind": "mixnode"})


def validate_config_entry(entry):
    return entry.get("nym_api") and entry.get("env")


def config_to_targets(config, mixnodes, labels=None):
    prom_targets = [
        make_prom_target(config.env, mixnode, config.port, labels)
        for mixnode in mixnodes
    ]
    with open(config.outfile, "w") as f:
        json.dump(prom_targets, f)

    os.chmod(config.outfile, 0o777)
    os.rename(config.outfile, config.outlink)
    os.chmod(config.outlink, 0o777)

    print(f"Prometheus -> {len(prom_targets)} targets written to {config.outlink}")


def make_prom_target(env, mixnode, port=None, labels=None):
    bond_info = mixnode.get("bond_information", {})
    mix_node = bond_info.get("mix_node")
    if mix_node is None:
        mix_node = mixnode.get("gateway")
    host = mix_node.get("host", None)
    port = port if port else mix_node.get("http_api_port", None)
    if host is None or port is None:
        return None

    target = {
        "targets": [f"{host}:{port}"],
        "labels": {
            "mix_node_host": host,
            "identity_key": mix_node.get("identity_key", None),
            "sphinx_key": mix_node.get("sphinx_key", None),
            "mix_node_version": mix_node.get("version", None),
            "mixnet_env": env,
        },
    }

    for k, v in labels.items():
        target["labels"][k] = v

    return target


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Create prometheus targets for rewarded set mixnodes."
    )
    parser.add_argument(
        "config",
        type=str,
        help="Config file, see scripts/prom_targets_config.json",
    )

    args = parser.parse_args()
    config_file = args.config

    with open(config_file, "r") as f:
        config = json.load(f)

    for entry in config:
        mixnode_targets(entry)

        gateway_targets(entry)