from dataclasses import dataclass

import yaml


@dataclass(frozen=True)
class MqttConfig:
    broker: str
    client_id: str
    mock_client_id: str
    sender_client_id: str
    subscriber_client_id: str
    username: str | None
    password: str | None
    pub_topic: str | None
    sub_topic: str | None
    ack_topic: str | None
    data_file: str


@dataclass(frozen=True)
class TmsConfig:
    device_code: str
    equipment_sn: str
    timeout: int
    keepalive: int
    server_ip: str | None
    server_port: int | None


@dataclass(frozen=True)
class AppConfig:
    mqtt: MqttConfig
    tms: TmsConfig
    raw: dict


def load_raw_config(path):
    with open(path, "r", encoding="utf-8") as f:
        return yaml.safe_load(f) or {}


def get_value(cfg, *paths, default=None):
    for path in paths:
        current = cfg
        found = True
        for key in path:
            if not isinstance(current, dict) or key not in current:
                found = False
                break
            current = current[key]
        if found and current is not None:
            return current
    return default


def load_app_config(path):
    raw = load_raw_config(path)
    base_client_id = get_value(raw, ("mqtt", "client-id"), ("client-id",), default="mqtt")
    device_code = get_value(
        raw,
        ("tms", "device-code"),
        ("tms", "equipment-sn"),
        ("device-code",),
        ("equipment-sn",),
        default="GJ-304-0088",
    )
    equipment_sn = get_value(
        raw,
        ("tms", "equipment-sn"),
        ("tms", "device-code"),
        ("equipment-sn",),
        ("device-code",),
        default=device_code,
    )

    return AppConfig(
        mqtt=MqttConfig(
            broker=get_value(raw, ("mqtt", "broker"), ("broker",), default=""),
            client_id=base_client_id,
            mock_client_id=get_value(raw, ("mqtt", "mock-client-id"), ("mock-client-id",), default=f"{base_client_id}-mock"),
            sender_client_id=get_value(raw, ("mqtt", "sender-client-id"), ("sender-client-id",), default=f"{base_client_id}-sender"),
            subscriber_client_id=get_value(raw, ("mqtt", "subscriber-client-id"), ("subscriber-client-id",), default=f"{base_client_id}-subscriber"),
            username=get_value(raw, ("mqtt", "username"), ("username",)),
            password=get_value(raw, ("mqtt", "password"), ("password",)),
            pub_topic=get_value(raw, ("mqtt", "pub-topic"), ("pub-topic",)),
            sub_topic=get_value(raw, ("mqtt", "sub-topic"), ("sub-topic",)),
            ack_topic=get_value(raw, ("mqtt", "ack-topic"), ("ack-topic",)),
            data_file=get_value(raw, ("mqtt", "data-file"), ("data-file",), default=""),
        ),
        tms=TmsConfig(
            device_code=device_code,
            equipment_sn=equipment_sn,
            timeout=int(get_value(raw, ("tms", "timeout"), ("timeout",), default=10)),
            keepalive=int(get_value(raw, ("tms", "keepalive"), ("keepalive",), default=20)),
            server_ip=get_value(raw, ("tms", "server-ip"), ("server-ip",)),
            server_port=get_value(raw, ("tms", "server-port"), ("server-port",)),
        ),
        raw=raw,
    )


load_config = load_app_config

__all__ = [
    "AppConfig",
    "MqttConfig",
    "TmsConfig",
    "get_value",
    "load_app_config",
    "load_config",
    "load_raw_config",
]