import os
import platform
import socket
import time
from typing import Any

REGISTRY_SET_KEY = "james:nodes"
REGISTRY_LAST_SEEN_KEY = "james:nodes:last_seen"


def node_name() -> str:
    return os.environ.get("JAMES_NODE") or socket.gethostname()


def node_role() -> str:
    return os.environ.get("JAMES_ROLE", "unknown")


def registry_payload(interval: float, docker_available: bool, gpu_available: bool) -> dict[str, Any]:
    return {
        "node": node_name(),
        "role": node_role(),
        "hostname": socket.gethostname(),
        "fqdn": socket.getfqdn(),
        "os": platform.system().lower(),
        "arch": platform.machine().lower(),
        "watcher_interval": interval,
        "docker_available": docker_available,
        "gpu_available": gpu_available,
        "last_seen": int(time.time()),
    }


def register_node(r, payload: dict[str, Any], ttl_seconds: int) -> None:
    node = payload.get("node", node_name())
    key = f"{REGISTRY_SET_KEY}:{node}"
    pipeline = r.pipeline()
    pipeline.hset(key, mapping=payload)
    if ttl_seconds > 0:
        pipeline.expire(key, ttl_seconds)
    pipeline.sadd(REGISTRY_SET_KEY, node)
    pipeline.zadd(REGISTRY_LAST_SEEN_KEY, {node: payload.get("last_seen", int(time.time()))})
    pipeline.execute()