# SSAS - Simple Smart Automotive Software
# Copyright (C) 2021 Parai Wang <parai@foxmail.com>

import os
import json
from .helper import *
from .DoIp import Gen_DoIp
from .SomeIp import Gen_SomeIp
from .SoAd import Gen_SoAd

__all__ = ["Gen"]

__GENNET__ = {
    "DoIp": Gen_DoIp,
    "SomeIp": Gen_SomeIp,
    "SoAd": Gen_SoAd,
}


def Gen_Net(cfg, dir):
    source = {}
    for mod in cfg["Modules"]:
        if mod["class"] in __GENNET__:
            ret = __GENNET__[mod["class"]](mod, dir)
            source.update(ret)
        else:
            raise Exception("unknown class %s" % (mod["class"]))
    return source


def ProcDoIp(cfg, mod):
    sock = {
        "name": "DOIP_UDP",
        "server": mod["discovery"],
        "protocol": "UDP",
        "multicast": True,
        "up": "DoIP",
        "RxPduId": "DOIP_RX_PID_UDP",
    }
    cfg["sockets"].append(sock)
    sock = {
        "name": "DOIP_TCP",
        "server": "NULL:13400",
        "protocol": "TCP",
        "up": "DoIP",
        "listen": mod["max_connections"],
        "RxPduId": "DOIP_RX_PID_TCP",
    }
    if mod.get("EnableTLS", False):
        sock["up"] = "TLS"
        sock["RxPduId"] = "TLS_RX_PID_DOIP_TCP"
    cfg["sockets"].append(sock)


def ProcSomeIp(cfg, mod):
    sock = {
        "name": "SD_MULTICAST",
        "server": mod["SD"]["multicast"] + ":30490",
        "multicast": True,
        "protocol": "UDP",
        "up": "SD",
        "RxPduId": "SD_RX_PID_MULTICAST",
    }
    cfg["sockets"].append(sock)
    sock = {"name": "SD_UNICAST", "server": "NULL:30490", "protocol": "UDP", "up": "SD", "RxPduId": "SD_RX_PID_UNICAST"}
    cfg["sockets"].append(sock)
    for service in mod.get("servers", []):
        name = "SOMEIP_%s" % (toMacro(service["name"]))
        if "reliable" in service:
            protocol = "TCP"
            port = service.get("reliable", 0)
        else:
            protocol = "UDP"
            port = service.get("unreliable", 0)
        sock = {
            "name": name,
            "server": "NULL:%s" % (port),
            "protocol": protocol,
            "up": "SOMEIP",
            "RxPduId": "SOMEIP_RX_PID_%s" % (name),
        }
        if "reliable" in service:
            sock["listen"] = service["listen"] if "listen" in service else 1
        cfg["sockets"].append(sock)
        for eg in service.get("event-groups", []):
            if "multicast" in eg:
                sock = {
                    "name": "_".join([name, eg["name"]]),
                    "server": "%s" % (eg["multicast"].get("addr", "NULL:0")),
                    "multicast": True,
                    "protocol": "UDP",
                    "up": "SOMEIP",
                    "ModeChg": "Sd",
                    "RxPduId": "SOMEIP_RX_PID_%s" % (name),
                }
                cfg["sockets"].append(sock)
    for service in mod.get("clients", []):
        name = "SOMEIP_%s" % (toMacro(service["name"]))
        if "protocol" in service:
            protocol = service["protocol"]
            if protocol == "TCP":
                port = service.get("reliable", 0)
            else:
                port = service.get("unreliable", 0)
        else:
            if "reliable" in service:
                protocol = "TCP"
                port = service.get("reliable", 0)
            else:
                protocol = "UDP"
                port = service.get("unreliable", 0)
        sock = {
            "name": name,
            "client": "NULL:%s" % (port),
            "protocol": protocol,
            "up": "SOMEIP",
            "RxPduId": "SOMEIP_RX_PID_%s" % (name),
        }
        cfg["sockets"].append(sock)
        for eg in service.get("event-groups", []):
            if "multicast" in eg:
                sock = {
                    "name": "_".join([name, eg["name"]]),
                    "client": "%s" % (eg["multicast"].get("addr", "NULL:0")),
                    "multicast": True,
                    "protocol": "UDP",
                    "up": "SOMEIP",
                    "ModeChg": "Sd",
                    "RxPduId": "SOMEIP_RX_PID_%s" % (name),
                }
                cfg["sockets"].append(sock)


def ProcSoAd(netCfg, dir):
    cfg = None
    for mod in netCfg["Modules"]:
        if mod["class"] == "SoAd":
            cfg = mod
    if cfg == None:
        cfg = {"class": "SoAd", "sockets": []}
        netCfg["Modules"].append(cfg)
    for mod in netCfg["Modules"]:
        if mod["class"] == "DoIp":
            ProcDoIp(cfg, mod)
        elif mod["class"] == "SomeIp":
            ProcSomeIp(cfg, mod)
    with open("%s/Network.json" % (dir), "w") as f:
        json.dump(netCfg, f, indent=2)


RandomPort = 60000


def GenAntiTestForSomeIp(cfgj):
    global RandomPort
    dir = os.path.join(os.path.dirname(cfgj), "GENT")
    os.makedirs(dir, exist_ok=True)
    with open(cfgj) as f:
        cfg = json.load(f)
    someIpCfg = None
    for mod in cfg["Modules"]:
        if mod["class"] == "SomeIp":
            someIpCfg = mod
            servers = []
            clients = []
            if "servers" in mod:
                servers = mod["servers"]
                del mod["servers"]
            if "clients" in mod:
                clients = mod["clients"]
                del mod["clients"]
            for ser in servers:
                if "unreliable" in ser:
                    ser["unreliable"] = RandomPort
                    RandomPort += 1
            mod["clients"] = servers
            mod["servers"] = clients
    if someIpCfg:
        cfg["Modules"] = [someIpCfg]
    ProcSoAd(cfg, dir)
    return Gen_Net(cfg, dir)


def Gen(cfgj):
    dir = os.path.join(os.path.dirname(cfgj), "GEN")
    os.makedirs(dir, exist_ok=True)
    with open(cfgj) as f:
        cfg = json.load(f)
    ProcSoAd(cfg, dir)
    source = Gen_Net(cfg, dir)
    if cfg.get("AntiTest", False):
        sourceAnti = GenAntiTestForSomeIp(cfgj)
        for name, src in sourceAnti.items():
            source[name + "_T"] = src
    return source