"""
Port Forwarding API.
https://dev.freebox.fr/sdk/os/nat/#port-forwarding
"""

from freebox_api.access import Access


class Fw:
    """
    Port Forwarding
    """

    def __init__(self, access: Access):
        self._access = access

    ip_proto = ["tcp", "udp"]

    port_forwarding_config_schema = {
        "comment": "",
        "enabled": True,
        "ip_proto": ip_proto[0],
        "lan_ip": "",
        "lan_port": 0,
        "src_ip": "",
        "wan_port_end": 0,
        "wan_port_start": 0,
    }

    incoming_port_configuration_data_schema = {"enabled": True, "in_port": 0}

    dmz_configuration_schema = {"enabled": False, "ip": ""}

    async def create_port_forwarding_configuration(self, port_forwarding_config):
        """
        Create port forwarding configuration

        port_forwarding_config : `dict`
        """
        return await self._access.post("fw/redir/", port_forwarding_config)

    async def delete_port_forwarding_configuration(
        self, config_id: int
    ) -> dict[str, bool]:
        """
        Delete port forwarding configuration

        config_id : `int`
        """
        return await self._access.delete(f"fw/redir/{config_id}")  # type: ignore

    async def get_port_forwarding_configuration(self, redir_id):
        """
        Get a specific port forwarding
        """
        return await self._access.get(f"fw/redir/{redir_id}")

    async def get_all_port_forwarding_configuration(self):
        """
        Get port forwarding configuration
        """
        return await self._access.get("fw/redir/")

    async def edit_port_forwarding_configuration(
        self, redir_id, port_forwarding_config
    ):
        """
        Update a port forwarding
        """
        return await self._access.put(f"fw/redir/{redir_id}", port_forwarding_config)

    async def get_incoming_port_configuration(self, port_id):
        """
        Get incoming ports configuration
        """
        return await self._access.get(f"fw/incoming/{port_id}")

    async def get_all_incoming_port_configuration(self):
        """
        Get incoming ports configuration
        """
        return await self._access.get("fw/incoming/")

    async def edit_incoming_port_configuration(
        self, port_id, incoming_port_configuration_data
    ):
        """
        Edit incoming port configuration

        port_id : `int`
        incoming_port_configuration_data : `dict`
        """
        return await self._access.put(
            f"fw/incoming/{port_id}", incoming_port_configuration_data
        )

    async def get_dmz_configuration(self):
        """
        Get dmz configuration
        """
        return await self._access.get("fw/dmz/")

    async def set_dmz_configuration(self, dmz_configuration=None):
        """
        Set dmz configuration

        dmz_configuration : `dict`
        """
        if dmz_configuration is None:
            dmz_configuration = self.dmz_configuration_schema
        return await self._access.put("fw/dmz/", dmz_configuration)