#!/usr/bin/env python3
"""
GlobalTalk Scraper

A script that uses netatalk's `getzones` and `nbplkup` to discover devices on
the GlobalTalk network, and create some nice visualisations of the network.

Created for MARCHintosh 2025
"""

import logging
import subprocess
import re
import json
import sys
import os
import argparse

NPBLKUP_RESULTS = re.compile(r"^(.*):(.*)\s(\d*\.\d*:\d*)$")


def getzones() -> list[str]:
    """Returns a list of AppleTalk zones using `getzones`"""
    result = subprocess.run(["getzones"], capture_output=True, text=True)
    return [x.strip() for x in result.stdout.split("\n") if x.strip() != ""]


def nbplkup(zone: str) -> list[(str, str)]:
    """Lookup members of a zone, and return a list of name and address tuples"""
    if zone not in getzones():
        logging.warning("%s not currently a known zone", zone)
        return []

    zone_results = []

    # Define the charset for the console as 'mac-roman' to avoid any
    # translation issues with some device names (AsanteTalk and such)
    environ = os.environ.copy()
    environ["ATALK_UNIX_CHARSET"] = "mac-roman"

    cmd = subprocess.run(
        ["nbplkup", "@{0}".format(zone)],
        capture_output=True,
        text=True,
        encoding="mac-roman",
        env=environ,
    )

    # Use regex to split the row nicely
    for node in [
        NPBLKUP_RESULTS.match(x.strip()).groups()
        for x in cmd.stdout.split("\n")
        if x.strip() != ""
    ]:
        object, endpoint_type, address = node
        address, socket = address.split(":")
        zone_results.append(
            {
                "object": object.strip(),
                "type": endpoint_type.strip(),
                "address": address,
                "socket": socket,
                "zone": zone,
            }
        )
    return zone_results


def main():
    parser = argparse.ArgumentParser("gtscraper")
    parser.add_argument("zone", default=None)
    parser.add_argument("--output", type=argparse.FileType("w"), default=sys.stdout)
    args = parser.parse_args()

    logging.basicConfig(level=logging.INFO, stream=sys.stderr)
    zone_results = {
        "format": "v1",
        "zones": getzones(),
        "nodes": [],
    }

    # Iterate the zones and scan them
    for zone in zone_results["zones"]:
        if args.zone and zone != args.zone:
            continue
        logging.info("Scanning %s", zone)
        node_data = nbplkup(zone)
        zone_results["nodes"].extend(node_data)

    nodes = len(zone_results["nodes"])
    zones = len(zone_results["zones"])

    print("{0} zones, {1} nodes".format(zones, nodes), file=sys.stderr)

    # Dump out the resulting JSON to stdout
    sys.stdout.write(json.dumps(zone_results))


if __name__ == "__main__":
    main()