# Released under the MIT License. See LICENSE for details.
#
"""Functionality related to the cloud."""

import time
import logging
from typing import TYPE_CHECKING, overload

from efro.error import CommunicationError
from efro.call import CallbackSet
from efro.dataclassio import dataclass_from_dict, dataclass_to_dict
import bacommon.classic
import bacommon.cloud
import babase

if TYPE_CHECKING:
    from typing import Callable, Any

    from efro.message import Message, Response, BoolResponse
    from bacommon import securedata
    import bacommon.classic
    import bacommon.clouddialog as cdlg


# TODO: Should make it possible to define a protocol in bacommon.cloud and
# autogenerate this. That would give us type safety between this and
# internal protocols.


class CloudSubsystem(babase.AppSubsystem):
    """Manages communication with cloud components.

    Access the shared single instance of this class via the
    :attr:`~baplus.PlusAppSubsystem.cloud` attr on the
    :class:`~baplus.PlusAppSubsystem` class.
    """

    #: General engine config values provided by the cloud.
    #:
    #: :meta private:
    vals: bacommon.cloud.CloudVals

    def __init__(self) -> None:
        super().__init__()
        self.on_connectivity_changed_callbacks: CallbackSet[
            Callable[[bool], None]
        ] = CallbackSet()

        # Latest :class:`bacommon.securedata.Reader` bundled by basn
        # in the v2-transport handshake response. Stays the same
        # across sessions until a new handshake bundles a fresh
        # one. ``None`` until any session has connected.
        self._secure_data_reader: securedata.Reader | None = None

        # Restore saved cloud-vals (or init to default).
        try:
            cloudvals_data = babase.app.config.get('CloudVals')
            if isinstance(cloudvals_data, dict):
                self.vals = dataclass_from_dict(
                    bacommon.cloud.CloudVals, cloudvals_data
                )
            else:
                self.vals = bacommon.cloud.CloudVals()
        except Exception:
            babase.applog.warning(
                'Error loading CloudVals; resetting to default.', exc_info=True
            )
            self.vals = bacommon.cloud.CloudVals()

        # Set up to start updating cloud-vals once we've got
        # connectivity.
        self._vals_updated = False
        self._vals_update_timer: babase.AppTimer | None = None
        self._vals_last_request_time: float | None = None
        self._vals_update_conn_reg = (
            self.on_connectivity_changed_callbacks.register(
                self._update_vals_update_for_connectivity
            )
        )

    @property
    def connected(self) -> bool:
        """Whether a connection to the cloud is present.

        This is a good indicator (though not for certain) that sending
        messages will succeed.
        """
        return self.is_connected()

    @property
    def secure_data_reader(self) -> securedata.Reader:
        """The latest :class:`bacommon.securedata.Reader` from basn.

        Bundled into each v2-transport handshake response; valid
        for at least the connecting session's full lifetime. Use
        it to verify any :class:`bacommon.securedata.Archive` the
        client receives (``reader.read(archive)`` returns the
        signed bytes or raises :class:`bacommon.securedata.Invalid`).

        Raises :class:`RuntimeError` if no v2-transport session
        has connected yet — callers that need a Reader before any
        session is up should use the static-keys path
        (``_babase.verify_ed25519`` against
        :data:`bacommon.securedata.STATIC_DATA_PUBLIC_KEYS`)
        instead, which is what the InsecureDirective verification
        uses today.
        """
        if self._secure_data_reader is None:
            raise RuntimeError(
                'No secure-data Reader available;'
                ' no v2-transport session has connected yet.'
            )
        return self._secure_data_reader

    def _set_secure_data_reader(
        self, reader: 'securedata.Reader | None'
    ) -> None:
        """Stash the Reader from a v2-transport handshake response.

        :meta private:
        """
        # ``None`` arrives during a partial rollout where the
        # connected basn predates the secure_data_reader handshake
        # field. Keep whatever we already had — a slightly-stale
        # Reader from a prior handshake beats no Reader at all.
        if reader is not None:
            self._secure_data_reader = reader

    def is_connected(self) -> bool:
        """Implementation for connected attr.

        :meta private:
        """
        raise NotImplementedError()

    def get_connected_node_base_url(self) -> str | None:
        """Return a base url for fetches from our connected basn node.

        Used by the asset-download path to issue ``GET /casblob/{hash}``
        requests to the same node serving our transport session (the
        node's aiohttp app serves both the WebSocket transport and plain
        http(s)). The scheme mirrors the transport session's security:
        ``https://host`` normally, ``http://host`` when the session
        connected via insecure ws:// (insecure-connections mode means
        TLS can't be trusted on this network, so fetches must avoid it
        too). Returns ``None`` when not connected (or in implementations
        without a node-based transport).

        :meta private:
        """
        return None

    def on_connectivity_changed(self, connected: bool) -> None:
        """Called when cloud connectivity state changes.

        :meta private:
        """
        babase.balog.debug('Connectivity is now %s.', connected)

        plus = babase.app.plus
        assert plus is not None

        # Fire any registered callbacks for this.
        for call in self.on_connectivity_changed_callbacks.getcalls():
            try:
                call(connected)
            except Exception:
                logging.exception('Error in connectivity-changed callback.')

    def _update_vals_update_for_connectivity(self, connected: bool) -> None:

        # If we don't have vals yet and are connected, start asking.
        if connected and not self._vals_updated:
            # Ask immediately and set up a timer to keep doing so until
            # successful.
            self._possibly_send_vals_request()
            self._vals_update_timer = babase.AppTimer(
                61.23, self._possibly_send_vals_request, repeat=True
            )
        else:
            # Ok; we're disconnected or have vals - stop asking.
            self._vals_update_timer = None

    def _possibly_send_vals_request(self) -> None:
        now = time.monotonic()

        # Only send if we havn't already recently.
        if (
            self._vals_last_request_time is None
            or now - self._vals_last_request_time > 30.0
        ):
            self._vals_last_request_time = now
            self.send_message_cb(
                bacommon.cloud.CloudValsRequest(), self._on_cloud_vals_response
            )

    def _on_cloud_vals_response(
        self, response: bacommon.cloud.CloudValsResponse | Exception
    ) -> None:
        if isinstance(response, Exception):
            # Make noise for any non-communication errors
            if not isinstance(response, CommunicationError):
                babase.applog.exception(
                    'Unexpected error in _on_cloud_vals_response().'
                )
            return

        # If what we got differs from what we already had, store it.
        if response.vals != self.vals:
            cfg = babase.app.config
            cfg['CloudVals'] = dataclass_to_dict(response.vals)
            cfg.commit()
            self.vals = response.vals

        # We can stop asking now.
        self._vals_updated = True
        self._vals_update_timer = None

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.LoginProxyRequestMessage,
        on_response: Callable[
            [bacommon.cloud.LoginProxyRequestResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.CloudValsRequest,
        on_response: Callable[
            [bacommon.cloud.CloudValsResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.LoginProxyStateQueryMessage,
        on_response: Callable[
            [bacommon.cloud.LoginProxyStateQueryResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.LoginProxyCompleteMessage,
        on_response: Callable[[None | Exception], None],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.PingMessage,
        on_response: Callable[[bacommon.cloud.PingResponse | Exception], None],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.SignInMessage,
        on_response: Callable[
            [bacommon.cloud.SignInResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.ManageAccountMessage,
        on_response: Callable[
            [bacommon.cloud.ManageAccountResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.GetClassicPurchasesMessage,
        on_response: Callable[
            [bacommon.classic.GetClassicPurchasesResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.StoreQueryMessage,
        on_response: Callable[
            [bacommon.cloud.StoreQueryResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.PrivatePartyMessage,
        on_response: Callable[
            [bacommon.classic.PrivatePartyResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.InboxRequestMessage,
        on_response: Callable[
            [bacommon.classic.InboxRequestResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: cdlg.ActionMessage,
        on_response: Callable[[cdlg.ActionResponse | Exception], None],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.ChestInfoMessage,
        on_response: Callable[
            [bacommon.classic.ChestInfoResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.ChestActionMessage,
        on_response: Callable[
            [bacommon.cloud.ChestActionResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.GlobalProfileCheckMessage,
        on_response: Callable[[BoolResponse | Exception], None],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.ScoreSubmitMessage,
        on_response: Callable[
            [bacommon.classic.ScoreSubmitResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.SecureDataCheckerRequest,
        on_response: Callable[
            [bacommon.cloud.SecureDataCheckerResponse | Exception], None
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.classic.GetClassicLeaguePresidentButtonInfoMessage,
        on_response: Callable[
            [
                bacommon.classic.GetClassicLeaguePresidentButtonInfoResponse
                | Exception
            ],
            None,
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.AnalyticsEventMessage,
        on_response: Callable[
            [None | Exception],
            None,
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.AuthRequestMessage,
        on_response: Callable[
            [bacommon.cloud.AuthRequestResponse | Exception],
            None,
        ],
    ) -> None: ...

    @overload
    def send_message_cb(
        self,
        msg: bacommon.cloud.TransientAPIKeyRequest,
        on_response: Callable[
            [bacommon.cloud.TransientAPIKeyResponse | Exception],
            None,
        ],
    ) -> None: ...

    def send_message_cb(
        self,
        msg: Message,
        on_response: Callable[[Any], None],
    ) -> None:
        """Asynchronously send a message to the cloud from the logic thread.

        The provided ``on_response`` call will be run in the logic thread
        and passed either the response or the error that occurred.
        """
        raise NotImplementedError(
            'Cloud functionality is not present in this build.'
        )

    @overload
    def send_message(
        self, msg: bacommon.cloud.WorkspaceFetchMessage
    ) -> bacommon.cloud.WorkspaceFetchResponse: ...

    @overload
    def send_message(
        self, msg: bacommon.cloud.MerchAvailabilityMessage
    ) -> bacommon.cloud.MerchAvailabilityResponse: ...

    @overload
    def send_message(
        self, msg: bacommon.cloud.TestMessage
    ) -> bacommon.cloud.TestResponse: ...

    @overload
    def send_message(
        self, msg: bacommon.classic.LegacyRequest
    ) -> bacommon.classic.LegacyResponse: ...

    @overload
    def send_message(
        self, msg: bacommon.cloud.FulfillDocUIRequest
    ) -> bacommon.cloud.FulfillDocUIResponse: ...

    @overload
    def send_message(
        self, msg: bacommon.cloud.ResolveAssetPackageMessage
    ) -> bacommon.cloud.ResolveAssetPackageResponse: ...

    def send_message(self, msg: Message) -> Response | None:
        """Synchronously send a message to the cloud.

        Must be called from a background thread.
        """
        raise NotImplementedError(
            'Cloud functionality is not present in this build.'
        )

    @overload
    async def send_message_async(
        self, msg: bacommon.classic.SendInfoMessage
    ) -> bacommon.classic.SendInfoResponse: ...

    @overload
    async def send_message_async(
        self, msg: bacommon.cloud.TestMessage
    ) -> bacommon.cloud.TestResponse: ...

    @overload
    async def send_message_async(
        self, msg: bacommon.cloud.LoginProxyRequestMessage
    ) -> bacommon.cloud.LoginProxyRequestResponse: ...

    @overload
    async def send_message_async(
        self, msg: bacommon.cloud.LoginProxyStateQueryMessage
    ) -> bacommon.cloud.LoginProxyStateQueryResponse: ...

    @overload
    async def send_message_async(
        self, msg: bacommon.cloud.LoginProxyCompleteMessage
    ) -> None: ...

    async def send_message_async(self, msg: Message) -> Response | None:
        """Asynchronously send a message to the cloud.

        Must be called from the logic thread.
        """
        raise NotImplementedError(
            'Cloud functionality is not present in this build.'
        )

    def subscribe_test(
        self, updatecall: Callable[[int | None], None]
    ) -> babase.CloudSubscription:
        """Subscribe to some test data.

        :meta private:
        """
        raise NotImplementedError(
            'Cloud functionality is not present in this build.'
        )

    def subscribe_classic_account_data(
        self,
        updatecall: Callable[
            [bacommon.classic.ClassicLiveAccountClientData], None
        ],
    ) -> babase.CloudSubscription:
        """Subscribe to classic account data.

        :meta private:
        """
        raise NotImplementedError(
            'Cloud functionality is not present in this build.'
        )

    def unsubscribe(self, subscription_id: int) -> None:
        """Unsubscribe from some subscription.

        Do not call this manually; it is called by CloudSubscription.

        :meta private:
        """
        raise NotImplementedError(
            'Cloud functionality is not present in this build.'
        )


def cloud_console_exec(code: str) -> None:
    """Called by the cloud console to run code in the logic thread."""
    import sys
    import __main__

    try:
        # First try it as eval.
        try:
            evalcode = compile(code, '<console>', 'eval')
        except SyntaxError:
            evalcode = None
        except Exception:
            # hmm; when we can't compile it as eval will we always get
            # syntax error?
            logging.exception(
                'unexpected error compiling code for cloud-console eval.'
            )
            evalcode = None
        if evalcode is not None:
            # pylint: disable=eval-used
            value = eval(evalcode, vars(__main__), vars(__main__))
            # For eval-able statements, print the resulting value if
            # it is not None (just like standard Python interpreter).
            if value is not None:
                print(repr(value), file=sys.stderr)

        # Fall back to exec if we couldn't compile it as eval.
        else:
            execcode = compile(code, '<console>', 'exec')
            # pylint: disable=exec-used
            exec(execcode, vars(__main__), vars(__main__))

    except Exception:
        import traceback

        # Note to self: Seems like we should just use
        # logging.exception() here. Except currently that winds up
        # triggering our cloud logging stuff so we'd probably want a
        # specific logger or whatnot to avoid that.
        apptime = babase.apptime()
        print(f'Exec error at time {apptime:.2f}.', file=sys.stderr)
        traceback.print_exc()

        # This helps the logging system ship stderr back to the
        # cloud promptly.
        sys.stderr.flush()
