from typing import Dict, List

from keywords.ptp.setup.object.clock_setup import ClockSetup
from keywords.ptp.setup.object.grandmaster_settings import GrandmasterSettings
from keywords.ptp.setup.object.parent_data_set import ParentDataSet
from keywords.ptp.setup.object.phc2sys_setup import PHC2SysSetup
from keywords.ptp.setup.object.ptp4l_expected_dict import PTP4LExpectedDict
from keywords.ptp.setup.object.ptp4l_setup import PTP4LSetup
from keywords.ptp.setup.object.ptp_host_interface_setup import PTPHostInterfaceSetup
from keywords.ptp.setup.object.ts2phc_setup import TS2PHCSetup
from keywords.ptp.setup.time_properties_data_set import TimePropertiesDataSet


class PTPSetup:
    """
    This class represents a PTP Setup.
    """

    def __init__(self, setup_dict: Dict[str, Dict]):
        """
        Constructor.

        Args:
            setup_dict (Dict[str, Dict]): The dictionary read from the JSON config file associated with this ptp_setup.

        """
        self.ptp4l_setup_list: List[PTP4LSetup] = []
        self.phc2sys_setup_list: List[PHC2SysSetup] = []
        self.ts2phc_setup_list: List[TS2PHCSetup] = []
        self.clock_setup_list: List[ClockSetup] = []
        self.host_ptp_if_dict: Dict[str, PTPHostInterfaceSetup] = {}  # Name -> PTPHostInterfaceSetup
        self.ptp4l_expected_list: List[PTP4LExpectedDict] = []
        self.parent_data_set: Dict[str, ParentDataSet] = {}
        self.time_properties_data_set: Dict[str, TimePropertiesDataSet] = {}
        self.grandmaster_settings_tgm: Dict[str, GrandmasterSettings] = {}
        self.grandmaster_settings_tbc: Dict[str, GrandmasterSettings] = {}

        if "ptp_instances" not in setup_dict:
            raise Exception("You must define a ptp_instances section in your ptp setup_dict")

        if "ptp_host_ifs" not in setup_dict:
            raise Exception("You must define a ptp_host_ifs section in your ptp setup_dict")

        ptp_host_ifs = setup_dict["ptp_host_ifs"]
        for ptp_host_if in ptp_host_ifs:
            ptp_host_if_object = PTPHostInterfaceSetup(ptp_host_if)
            ptp_host_if_name = ptp_host_if_object.get_name()
            self.host_ptp_if_dict[ptp_host_if_name] = ptp_host_if_object

        ptp_instances_dict = setup_dict["ptp_instances"]
        if "ptp4l" in ptp_instances_dict:
            ptp4l_list = ptp_instances_dict["ptp4l"]
            for ptp4l_entry_dict in ptp4l_list:
                ptp4l_setup = PTP4LSetup(ptp4l_entry_dict, self.host_ptp_if_dict)
                self.ptp4l_setup_list.append(ptp4l_setup)

        if "phc2sys" in ptp_instances_dict:
            phc2sys_list = ptp_instances_dict["phc2sys"]
            for phc2sys_entry_dict in phc2sys_list:
                phc2sys_setup = PHC2SysSetup(phc2sys_entry_dict, self.host_ptp_if_dict)
                self.phc2sys_setup_list.append(phc2sys_setup)

        if "ts2phc" in ptp_instances_dict:
            ts2phc_list = ptp_instances_dict["ts2phc"]
            for ts2phc_entry_dict in ts2phc_list:
                ts2phc_setup = TS2PHCSetup(ts2phc_entry_dict, self.host_ptp_if_dict)
                self.ts2phc_setup_list.append(ts2phc_setup)

        if "clock" in ptp_instances_dict:
            clock_list = ptp_instances_dict["clock"]
            for clock_entry_dict in clock_list:
                clock_setup = ClockSetup(clock_entry_dict, self.host_ptp_if_dict)
                self.clock_setup_list.append(clock_setup)

        expected_dict = setup_dict["expected_dict"]
        if "ptp4l" in expected_dict:
            ptp4l_list = expected_dict["ptp4l"]
            for ptp4l_expected_dict in ptp4l_list:
                ptp4l_expected = PTP4LExpectedDict(ptp4l_expected_dict)
                self.ptp4l_expected_list.append(ptp4l_expected)

        if "parent_data_set" in expected_dict:
            self.parent_data_set = ParentDataSet(expected_dict["parent_data_set"])

        if "time_properties_data_set" in expected_dict:
            self.time_properties_data_set = TimePropertiesDataSet(expected_dict["time_properties_data_set"])

        if "grandmaster_settings_tgm" in expected_dict:
            self.grandmaster_settings_tgm = GrandmasterSettings(expected_dict["grandmaster_settings_tgm"])

        if "grandmaster_settings_tbc" in expected_dict:
            self.grandmaster_settings_tbc = GrandmasterSettings(expected_dict["grandmaster_settings_tbc"])

    def __str__(self) -> str:
        """
        String representation of this object.

        Returns:
            str: String representation of this object.

        """
        return "PTPSetup"

    def get_ptp4l_setup_list(self) -> List[PTP4LSetup]:
        """
        Getter for the list of ptp4l setups.

        Returns:
            List[PTP4LSetup]: list of ptp4l setups
        """
        return self.ptp4l_setup_list

    def get_ptp4l_setup(self, setup_name: str) -> PTP4LSetup:
        """
        Getter for the PTP4LSetup with the specified name.

        Args:
            setup_name (str): Name of the specified setup.

        Returns:
            PTP4LSetup: The setup matching that name.
        """
        for setup in self.ptp4l_setup_list:
            if setup.get_name() == setup_name:
                return setup
        raise Exception(f"There is no ptp4l setup named {setup_name}")

    def get_phc2sys_setup_list(self) -> List[PHC2SysSetup]:
        """
        Getter for the list of phc2sys setups.

        Returns:
            List[PHC2SysSetup]: list of phc2sys setups
        """
        return self.phc2sys_setup_list

    def get_phc2sys_setup(self, setup_name: str) -> PHC2SysSetup:
        """
        Getter for the PHC2SysSetup with the specified name.

        Args:
            setup_name (str): Name of the specified setup.

        Returns:
            PHC2SysSetup: The setup matching that name.
        """
        for setup in self.phc2sys_setup_list:
            if setup.get_name() == setup_name:
                return setup
        raise Exception(f"There is no phc2sys setup named {setup_name}")

    def get_ts2phc_setup_list(self) -> List[TS2PHCSetup]:
        """
        Getter for the list of ts2phc setups.

        Returns:
            List[TS2PHCSetup]: list of ts2phc setups
        """
        return self.ts2phc_setup_list

    def get_ts2phc_setup(self, setup_name: str) -> TS2PHCSetup:
        """
        Getter for the TS2PHCSetup with the specified name.

        Args:
            setup_name (str): Name of the specified setup.

        Returns:
            TS2PHCSetup: The setup matching that name.
        """
        for setup in self.ts2phc_setup_list:
            if setup.get_name() == setup_name:
                return setup
        raise Exception(f"There is no ts2phc setup named {setup_name}")

    def get_clock_setup_list(self) -> List[ClockSetup]:
        """
        Getter for the list of clock setups.

        Returns:
            List[ClockSetup]: list of clock setups
        """
        return self.clock_setup_list

    def get_clock_setup(self, setup_name: str) -> ClockSetup:
        """
        Getter for the ClockSetup with the specified name.

        Args:
            setup_name (str): Name of the specified setup.

        Returns:
            ClockSetup: The setup matching that name.
        """
        for setup in self.clock_setup_list:
            if setup.get_name() == setup_name:
                return setup
        raise Exception(f"There is no clock setup named {setup_name}")

    def get_expected_ptp4l_list(self) -> List[PTP4LExpectedDict]:
        """
        Getter for the list of expected ptp4l list.

        Returns:
            List[PTP4LExpectedDict]: list of ptp4l expected dict
        """
        return self.ptp4l_expected_list

    def get_parent_data_set(self) -> ParentDataSet:
        """
        Getter for the parent data set.

        Returns:
            ParentDataSet: The parent data set
        """
        return self.parent_data_set

    def get_time_properties_data_set(self) -> TimePropertiesDataSet:
        """
        Getter for the time properties data set.

        Returns:
            TimePropertiesDataSet: The time properties data set
        """
        return self.time_properties_data_set

    def get_grandmaster_settings_tgm(self) -> GrandmasterSettings:
        """
        Getter for the grandmaster settings tgm.

        Returns:
            GrandmasterSettings: The grandmaster settings tgm
        """
        return self.grandmaster_settings_tgm

    def get_grandmaster_settings_tbc(self) -> GrandmasterSettings:
        """
        Getter for the grandmaster settings tbc.

        Returns:
            GrandmasterSettings: The grandmaster settings tbc
        """
        return self.grandmaster_settings_tbc