mirror of
https://github.com/natekspencer/hacs-oasis_mini.git
synced 2025-12-06 18:44:14 -05:00
89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
"""Helpers for the Oasis devices integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
from typing import Any
|
|
|
|
import async_timeout
|
|
|
|
from homeassistant.const import CONF_ACCESS_TOKEN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
|
|
from .pyoasiscontrol import OasisCloudClient, OasisDevice
|
|
from .pyoasiscontrol.const import TRACKS
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def create_client(hass: HomeAssistant, data: dict[str, Any]) -> OasisCloudClient:
|
|
"""
|
|
Create an Oasis cloud client configured with the Home Assistant HTTP session and access token.
|
|
|
|
Parameters:
|
|
hass: Home Assistant instance used to obtain the shared HTTP client session.
|
|
data: Configuration mapping; the function reads the `CONF_ACCESS_TOKEN` key for the cloud access token.
|
|
|
|
Returns:
|
|
An `OasisCloudClient` initialized with the Home Assistant HTTP session and the configured access token.
|
|
"""
|
|
session = async_get_clientsession(hass)
|
|
return OasisCloudClient(session=session, access_token=data.get(CONF_ACCESS_TOKEN))
|
|
|
|
|
|
async def add_and_play_track(device: OasisDevice, track: int) -> None:
|
|
"""
|
|
Ensure a track is present in the device playlist, position it as the next item, select it, and start playback if necessary.
|
|
|
|
Adds the specified track to the device playlist if missing, waits up to 10 seconds for the track to appear, moves it to be the next item after the current playlist index if needed, selects that track, and starts playback when the device is not already playing.
|
|
|
|
Parameters:
|
|
device (OasisDevice): The target Oasis device.
|
|
track (int): The track id to add and play.
|
|
|
|
Raises:
|
|
async_timeout.TimeoutError: If the operation does not complete within 10 seconds.
|
|
"""
|
|
async with async_timeout.timeout(10):
|
|
if track not in device.playlist:
|
|
await device.async_add_track_to_playlist(track)
|
|
|
|
while track not in device.playlist:
|
|
await asyncio.sleep(0.1)
|
|
|
|
# Move track to next item in the playlist and then select it
|
|
if (index := device.playlist.index(track)) != device.playlist_index:
|
|
if index != (
|
|
_next := min(device.playlist_index + 1, len(device.playlist) - 1)
|
|
):
|
|
await device.async_move_track(index, _next)
|
|
await device.async_change_track(_next)
|
|
|
|
if device.status_code != 4:
|
|
await device.async_play()
|
|
|
|
|
|
def get_track_id(track: str) -> int | None:
|
|
"""
|
|
Convert a track identifier or title to its integer track id.
|
|
|
|
Parameters:
|
|
track: A track reference, either a numeric id as a string or a track title.
|
|
|
|
Returns:
|
|
The integer track id if the input is a valid id or matches a known title, `None` if the input is invalid.
|
|
"""
|
|
track = track.lower().strip()
|
|
if track not in map(str, TRACKS):
|
|
track = next(
|
|
(id for id, info in TRACKS.items() if info["name"].lower() == track), track
|
|
)
|
|
|
|
try:
|
|
return int(track)
|
|
except ValueError:
|
|
_LOGGER.warning("Invalid track: %s", track)
|
|
return None
|