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
from librespot.proto.Player import ContextPlayerOptions
from librespot.proto.Player import PlayerState
from librespot.proto.Player import Restrictions
from librespot.proto.Player 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