Add Yandex Station player provider

music_assistant/providers/yandex_station/__init__.py

@@ -0,0 +1,170 @@
+"""Yandex Station Player Provider for Music Assistant.
+
+Play music on Yandex Station smart speakers via local Glagol WebSocket protocol.
+Adapted from AlexxIT/YandexStation (MIT license).
+"""
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING, cast
+
+from music_assistant_models.config_entries import ConfigEntry
+from music_assistant_models.enums import ConfigEntryType, ProviderFeature
+from music_assistant_models.errors import LoginFailed
+
+from .constants import (
+    CONF_ACTION_AUTH_COOKIES,
+    CONF_ACTION_AUTH_QR,
+    CONF_ACTION_CLEAR_AUTH,
+    CONF_COOKIES,
+    CONF_MUSIC_TOKEN,
+    CONF_X_TOKEN,
+)
+from .provider import YandexStationProvider
+from .yandex_auth import login_with_cookies, perform_qr_auth
+
+if TYPE_CHECKING:
+    from music_assistant_models.config_entries import ConfigValueType, ProviderConfig
+    from music_assistant_models.provider import ProviderManifest
+
+    from music_assistant.mass import MusicAssistant
+    from music_assistant.models import ProviderInstanceType
+
+_LOGGER = logging.getLogger(__name__)
+
+SUPPORTED_FEATURES: set[ProviderFeature] = set()
+
+
+async def get_config_entries(
+    mass: MusicAssistant,
+    instance_id: str | None = None,  # noqa: ARG001
+    action: str | None = None,
+    values: dict[str, ConfigValueType] | None = None,
+) -> tuple[ConfigEntry, ...]:
+    """Return Config entries to setup this provider."""
+    if values is None:
+        values = {}
+
+    # Handle QR auth action
+    if action == CONF_ACTION_AUTH_QR:
+        session_id = values.get("session_id")
+        if not session_id:
+            raise LoginFailed("Missing session_id for QR authentication")
+        x_token, music_token = await perform_qr_auth(mass, str(session_id))
+        values[CONF_X_TOKEN] = x_token
+        values[CONF_MUSIC_TOKEN] = music_token
+
+    # Handle cookies auth action
+    if action == CONF_ACTION_AUTH_COOKIES:
+        cookies_val = values.get(CONF_COOKIES)
+        if not cookies_val:
+            raise LoginFailed("Cookies field is empty")
+        x_token, music_token = await login_with_cookies(str(cookies_val))
+        values[CONF_X_TOKEN] = x_token
+        values[CONF_MUSIC_TOKEN] = music_token
+        values[CONF_COOKIES] = None  # don't persist raw cookies
+
+    # Handle clear auth action
+    if action == CONF_ACTION_CLEAR_AUTH:
+        values[CONF_X_TOKEN] = None
+        values[CONF_MUSIC_TOKEN] = None
+
+    is_authenticated = bool(values.get(CONF_X_TOKEN))
+
+    # Dynamic label text
+    if not is_authenticated:
+        label_text = (
+            "Scan a QR code with the Yandex app on your phone to authenticate.\n\n"
+            "Alternatively, you can enter tokens manually in the advanced settings."
+        )
+    elif action == CONF_ACTION_AUTH_QR:
+        label_text = "✅ Authenticated! Don't forget to save to complete setup."
+    else:
+        label_text = "✅ Authenticated to Yandex."
+
+    return (
+        # Status label
+        ConfigEntry(
+            key="label_text",
+            type=ConfigEntryType.LABEL,
+            label=label_text,
+        ),
+        # QR authentication (primary)
+        ConfigEntry(
+            key=CONF_ACTION_AUTH_QR,
+            type=ConfigEntryType.ACTION,
+            label="Login with QR code",
+            description="Opens a QR code page — scan it with the Yandex app on your phone.",
+            action=CONF_ACTION_AUTH_QR,
+            action_label="Login with QR code",
+            hidden=is_authenticated,
+        ),
+        # Cookies authentication (advanced fallback)
+        ConfigEntry(
+            key=CONF_COOKIES,
+            type=ConfigEntryType.SECURE_STRING,
+            label="Browser Cookies",
+            description=(
+                "Open passport.yandex.ru/profile in your browser, "
+                'use "Copy Cookies" extension to copy cookies, paste here. '
+                "Supports JSON array or raw cookie string."
+            ),
+            required=False,
+            hidden=is_authenticated,
+            advanced=True,
+            value="",
+        ),
+        ConfigEntry(
+            key=CONF_ACTION_AUTH_COOKIES,
+            type=ConfigEntryType.ACTION,
+            label="Login with Cookies",
+            description="Authenticate using browser cookies from passport.yandex.ru.",
+            action=CONF_ACTION_AUTH_COOKIES,
+            action_label="Login with Cookies",
+            hidden=is_authenticated,
+            advanced=True,
+        ),
+        # Clear auth
+        ConfigEntry(
+            key=CONF_ACTION_CLEAR_AUTH,
+            type=ConfigEntryType.ACTION,
+            label="Reset authentication",
+            description="Clear the current authentication details.",
+            action=CONF_ACTION_CLEAR_AUTH,
+            action_label="Reset authentication",
+            hidden=not is_authenticated,
+        ),
+        # x_token (internal storage)
+        ConfigEntry(
+            key=CONF_X_TOKEN,
+            type=ConfigEntryType.SECURE_STRING,
+            label="Yandex X-Token",
+            description="Long-lived auth token (~1 year). Auto-obtained via QR login.",
+            required=True,
+            hidden=is_authenticated,
+            advanced=True,
+            value=cast("str", values.get(CONF_X_TOKEN)) if values else None,
+        ),
+        # music_token (internal storage)
+        ConfigEntry(
+            key=CONF_MUSIC_TOKEN,
+            type=ConfigEntryType.SECURE_STRING,
+            label="Yandex Music Token",
+            description="Auto-obtained from X-Token. No manual entry needed.",
+            required=False,
+            hidden=True,
+            value=cast("str", values.get(CONF_MUSIC_TOKEN)) if values else None,
+        ),
+    )
+
+
+async def setup(
+    mass: MusicAssistant, manifest: ProviderManifest, config: ProviderConfig
+) -> ProviderInstanceType:
+    """Initialize provider(instance) with given configuration."""
+    x_token = config.get_value(CONF_X_TOKEN)
+    if not x_token:
+        msg = "Authentication required. Please login with your Yandex credentials."
+        raise LoginFailed(msg)
+    return YandexStationProvider(mass, manifest, config, SUPPORTED_FEATURES)

music_assistant/providers/yandex_station/constants.py

@@ -0,0 +1,40 @@
+"""Constants for Yandex Station Player provider."""
+
+from __future__ import annotations
+
+DOMAIN = "yandex_station"
+
+# Config entry keys
+CONF_X_TOKEN = "x_token"
+CONF_MUSIC_TOKEN = "music_token"
+CONF_COOKIE = "cookie"
+CONF_COOKIES = "cookies"
+
+# Config action keys
+CONF_ACTION_AUTH_QR = "action_auth_qr"
+CONF_ACTION_AUTH_COOKIES = "action_auth_cookies"
+CONF_ACTION_CLEAR_AUTH = "action_clear_auth"
+CONF_VOICE_CONTROL = "voice_control"
+
+# Yandex API endpoints
+GLAGOL_TOKEN_URL = "https://quasar.yandex.net/glagol/token"
+QUASAR_DEVICES_URL = "https://iot.quasar.yandex.ru/m/v3/user/devices"
+QUASAR_IOT_URL = "https://iot.quasar.yandex.ru"
+MUSIC_TOKEN_URL = "https://oauth.mobile.yandex.net/1/token"
+PASSPORT_API_URL = "https://mobileproxy.passport.yandex.net"
+
+# Music token OAuth credentials (from yandex-music-api)
+MUSIC_CLIENT_ID = "23cabbbdc6cd418abb4b39c32c41195d"
+MUSIC_CLIENT_SECRET = "53bc75238f0c4d08a118e51fe9203300"
+
+# mDNS service type for Yandex Station devices
+MDNS_TYPE = "_yandexio._tcp.local."
+
+# WebSocket settings
+WS_HEARTBEAT = 55
+WS_RECONNECT_BASE_DELAY = 30
+WS_RECONNECT_MAX_MULTIPLIER = 10
+WS_COMMAND_TIMEOUT = 5
+
+# DDoS protection: minimum interval between Yandex API requests (seconds)
+API_REQUEST_INTERVAL = 0.2