61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
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
|