from __future__ import annotations

import typing

from librespot.core import Session
from librespot.dealer import DealerClient
from librespot.player import Player
from librespot.player import PlayerConfiguration
from librespot.player.state import DeviceStateHandler
from librespot.proto import Connect_pb2 as Connect
from librespot.proto.Player_pb2 import ContextPlayerOptions
from librespot.proto.Player_pb2 import PlayerState
from librespot.proto.Player_pb2 import Restrictions
from librespot.proto.Player_pb2 import Suppressions


class StateWrapper(DeviceStateHandler.Listener, DealerClient.MessageListener):
    _state: PlayerState = None
    _session: Session = None
    _player: Player = None
    _device: DeviceStateHandler = None

    def __init__(self, session: Session, player: Player,
                 conf: PlayerConfiguration):
        self._session = session
        self._player = player
        self._device = DeviceStateHandler(session, self, conf)
        self._state = self._init_state()

        self._device.add_listener(self)
        self._session.dealer().add_message_listener(
            self,
            "spotify:user:attributes:update",
            "hm://playlist/",
            "hm://collection/collection/" + self._session.username() + "/json",
        )

    def _init_state(self) -> PlayerState:
        return PlayerState(
            playback_speed=1.0,
            suppressions=Suppressions(),
            context_restrictions=Restrictions(),
            options=ContextPlayerOptions(repeating_context=False,
                                         shuffling_context=False,
                                         repeating_track=False),
            position_as_of_timestamp=0,
            position=0,
            is_playing=False,
        )

    def add_listener(self, listener: DeviceStateHandler.Listener):
        self._device.add_listener(listener)

    def ready(self) -> None:
        self._device.update_state(Connect.PutStateReason.NEW_DEVICE, 0,
                                  self._state)

    def on_message(self, uri: str, headers: typing.Dict[str, str],
                   payload: bytes):
        pass