from dataclasses import dataclass
from ssl import SSLContext

import httpx

from homematicip.connection.client_characteristics_builder import (
    ClientCharacteristicsBuilder,
)
from homematicip.connection.client_token_builder import ClientTokenBuilder
from homematicip.connection.connection_url_resolver import ConnectionUrlResolver


class ConnectionContextBuilder:

    @classmethod
    async def build_context_async(cls, accesspoint_id: str,
                                  lookup_url: str = "https://lookup.homematic.com:48335/getHost",
                                  auth_token: str | None = None,
                                  enforce_ssl: bool = True,
                                  httpx_client_session: httpx.AsyncClient | None = None,
                                  ssl_ctx=None,
                                  websocket_heartbeat_interval: int = 30,
                                  websocket_connect_timeout: int = 30,
                                  websocket_message_stale_timeout: int = 28800,
                                  websocket_initial_backoff: int = 8,
                                  websocket_max_backoff: int = 900):
        """
        Create a new connection context and lookup urls

        :param accesspoint_id: Access point id
        :param lookup_url: Url to lookup the connection urls
        :param auth_token: The Auth Token if exists. If no one is provided None will be used
        :param enforce_ssl: Disable ssl verification by setting enforce_ssl to False
        :param httpx_client_session: The httpx client session if you want to use a custom one
        :param ssl_ctx: ssl context to use
        :return: a new ConnectionContext
        """
        ctx = ConnectionContext()
        ctx.accesspoint_id = accesspoint_id
        ctx.client_auth_token = ClientTokenBuilder.build_client_token(accesspoint_id)
        ctx.ssl_ctx = ssl_ctx
        ctx.enforce_ssl = enforce_ssl
        ctx.websocket_heartbeat_interval = websocket_heartbeat_interval
        ctx.websocket_connect_timeout = websocket_connect_timeout
        ctx.websocket_message_stale_timeout = websocket_message_stale_timeout
        ctx.websocket_initial_backoff = websocket_initial_backoff
        ctx.websocket_max_backoff = websocket_max_backoff

        cc = ClientCharacteristicsBuilder.get(accesspoint_id)
        ctx.rest_url, ctx.websocket_url = await ConnectionUrlResolver().lookup_urls_async(cc, lookup_url, enforce_ssl,
                                                                                          ssl_ctx, httpx_client_session)

        if auth_token is not None:
            ctx.auth_token = auth_token

        return ctx

    @classmethod
    def build_context(cls, accesspoint_id: str,
                      lookup_url: str = "https://lookup.homematic.com:48335/getHost",
                      auth_token: str | None = None,
                      enforce_ssl: bool = True,
                      ssl_ctx: SSLContext | str | bool | None = None,
                      websocket_heartbeat_interval: int = 30,
                      websocket_connect_timeout: int = 30,
                      websocket_message_stale_timeout: int = 28800,
                      websocket_initial_backoff: int = 8,
                      websocket_max_backoff: int = 900):
        """
        Create a new connection context and lookup urls

        :param accesspoint_id: Access point id
        :param lookup_url: Url to lookup the connection urls
        :param auth_token: The Auth Token if exists. If no one is provided None will be used
        :param enforce_ssl: Disable ssl verification by setting enforce_ssl to False
        :param ssl_ctx: ssl context to use
        :return: a new ConnectionContext
        """
        ctx = ConnectionContext()
        ctx.accesspoint_id = accesspoint_id
        ctx.client_auth_token = ClientTokenBuilder.build_client_token(accesspoint_id)
        ctx.ssl_ctx = ssl_ctx
        ctx.enforce_ssl = enforce_ssl
        ctx.websocket_heartbeat_interval = websocket_heartbeat_interval
        ctx.websocket_connect_timeout = websocket_connect_timeout
        ctx.websocket_message_stale_timeout = websocket_message_stale_timeout
        ctx.websocket_initial_backoff = websocket_initial_backoff
        ctx.websocket_max_backoff = websocket_max_backoff

        cc = ClientCharacteristicsBuilder.get(accesspoint_id)
        ctx.rest_url, ctx.websocket_url = ConnectionUrlResolver().lookup_urls(cc, lookup_url, enforce_ssl, ssl_ctx)

        if auth_token is not None:
            ctx.auth_token = auth_token

        return ctx


@dataclass
class ConnectionContext:
    auth_token: str | None = None
    client_auth_token: str | None = None

    websocket_url: str = "ws://localhost:8765"
    rest_url: str | None = None
    accesspoint_id: str | None = None

    enforce_ssl: bool = True
    ssl_ctx: SSLContext | str | bool | None = None
    websocket_heartbeat_interval: int = 30
    websocket_connect_timeout: int = 30
    websocket_message_stale_timeout: int = 28800
    websocket_initial_backoff: int = 8
    websocket_max_backoff: int = 900