#!/usr/bin/env python3
"""Host probe v3 — socket-id variants via the library business API.

Historical sweep that established the socket-transport command space
(``1101/1307/1705/1707`` instead of the ``off_8`` ids). Superseded by
``host_probe6.py`` for routine use; kept as a variant-list driver over
``LocalDeviceClient.send_probe``.

Run from the repo root on the SoftAP-joined host:
    python3 scripts/host_probe3.py --sn A03X2516013210060005 --ports 9000
"""

import argparse
import json
import os
import socket
import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(
    os.path.abspath(__file__))))

from franklinwh_agate_client.battery import extract_soc  # noqa: E402
from franklinwh_agate_client.device_local import (  # noqa: E402
    LocalDeviceClient,
)
from franklinwh_agate_client.protocol import (  # noqa: E402
    CommunicationCmd,
    generate_secret_key,
)

VARIANTS = (
    # (label, cmd, data, equip_no override or None for the client SN)
    ("queryInfo/1101+SN", CommunicationCmd.QUERY_DEVICE_INFO,
     {"minProtocolVer": "V1.00.00"}, None),
    ("queryInfo/1101+empty", CommunicationCmd.QUERY_DEVICE_INFO,
     {"minProtocolVer": "V1.00.00"}, ""),
    ("apower/1307+update2", CommunicationCmd.DEVICE_APOWER_INFO,
     {"update": 2}, None),
    ("apower/1307+none", CommunicationCmd.DEVICE_APOWER_INFO, None, None),
    ("bms/1705", CommunicationCmd.BMS, None, None),
    ("ems/1707", CommunicationCmd.EMS, None, None),
)


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("--sn", required=True)
    ap.add_argument("--host", default="10.100.1.1")
    ap.add_argument("--ports", default="9000")
    ap.add_argument("--timeout", type=float, default=5.0)
    args = ap.parse_args()
    print("secret key for %s: %s (radix 16)"
          % (args.sn, generate_secret_key(args.sn)))
    for port in [int(p) for p in args.ports.split(",") if p.strip()]:
        client = LocalDeviceClient(args.sn, host=args.host,
                                   socket_port=port, timeout=args.timeout)
        for label, cmd, data, equip in VARIANTS:
            try:
                out = client.send_probe(cmd, data=data, timeout=args.timeout,
                                        equip_no=equip)
            except (ConnectionError, socket.error, TimeoutError) as exc:
                print("--- %s: error: %s: %s"
                      % (label, type(exc).__name__, exc))
                continue
            rx = out.get("reply_raw") or b""
            print("--- %s tx=%dB rx=%dB"
                  % (label, len(out["tx"]), len(rx)))
            if out.get("reply") is not None:
                print("  reply: "
                      + json.dumps(out["reply"], default=str)[:1500])
                soc = extract_soc(out["reply"])
                print("  soc: %s (source=%s)" % (soc["value"],
                                                 soc["source"]))


if __name__ == "__main__":
    main()