python-xbox 0.1.0__py3-none-any.whl

pythonxbox/api/provider/smartglass/__init__.py

@@ -0,0 +1,402 @@
+"""
+SmartGlass - Control Registered Devices
+"""
+
+from typing import TYPE_CHECKING
+from uuid import uuid4
+
+from httpx import Response
+
+from pythonxbox.api.provider.baseprovider import BaseProvider
+from pythonxbox.api.provider.smartglass.models import (
+    CommandResponse,
+    GuideTab,
+    InputKeyType,
+    InstalledPackagesList,
+    OperationStatusResponse,
+    SmartglassConsoleList,
+    SmartglassConsoleStatus,
+    StorageDevicesList,
+    VolumeDirection,
+)
+
+if TYPE_CHECKING:
+    from pythonxbox.api.client import XboxLiveClient
+
+
+class SmartglassProvider(BaseProvider):
+    SG_URL = "https://xccs.xboxlive.com"
+    HEADERS_SG = {
+        "x-xbl-contract-version": "4",
+        "skillplatform": "RemoteManagement",
+    }
+
+    def __init__(self, client: "XboxLiveClient") -> None:
+        """
+        Initialize Baseclass, create smartglass session id
+
+        Args: Instance of XBL client
+        """
+        super().__init__(client)
+        self._smartglass_session_id = str(uuid4())
+
+    async def get_console_list(
+        self, include_storage_devices: bool = True, **kwargs
+    ) -> SmartglassConsoleList:
+        """
+        Get Console list
+
+        Args:
+            include_storage_devices: Include a list of storage devices in the response
+
+        Returns: Console List
+        """
+        params = {
+            "queryCurrentDevice": "false",
+            "includeStorageDevices": str(include_storage_devices).lower(),
+        }
+        resp = await self._fetch_list("devices", params, **kwargs)
+        return SmartglassConsoleList(**resp.json())
+
+    async def get_installed_apps(
+        self, device_id: str | None = None, **kwargs
+    ) -> InstalledPackagesList:
+        """
+        Get Installed Apps
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Installed Apps
+        """
+        params = {}
+        if device_id:
+            params["deviceId"] = device_id
+        resp = await self._fetch_list("installedApps", params, **kwargs)
+        return InstalledPackagesList(**resp.json())
+
+    async def get_storage_devices(self, device_id: str, **kwargs) -> StorageDevicesList:
+        """
+        Get Installed Apps
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Storage Devices list
+        """
+        params = {"deviceId": device_id}
+        resp = await self._fetch_list("storageDevices", params, **kwargs)
+        return StorageDevicesList(**resp.json())
+
+    async def get_console_status(
+        self, device_id: str, **kwargs
+    ) -> SmartglassConsoleStatus:
+        """
+        Get Console Status
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Console Status
+        """
+        url = f"{self.SG_URL}/consoles/{device_id}"
+        resp = await self.client.session.get(url, headers=self.HEADERS_SG, **kwargs)
+        resp.raise_for_status()
+        return SmartglassConsoleStatus(**resp.json())
+
+    async def get_op_status(
+        self, device_id: str, op_id: str, **kwargs
+    ) -> OperationStatusResponse:
+        """
+        Get Operation Status
+
+        Args:
+            device_id: ID of console (from console list)
+            op_id: Operation ID (from previous command)
+
+        Returns: Operation Status
+        """
+        url = f"{self.SG_URL}/opStatus"
+        headers = {
+            "x-xbl-contract-version": "3",
+            "x-xbl-opId": op_id,
+            "x-xbl-deviceId": device_id,
+        }
+        resp = await self.client.session.get(url, headers=headers, **kwargs)
+        resp.raise_for_status()
+        return OperationStatusResponse(**resp.json())
+
+    async def wake_up(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Wake Up Console
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Power", "WakeUp", **kwargs)
+
+    async def turn_off(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Turn Off Console
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(
+            device_id, "Power", "TurnOff", **kwargs
+        )
+
+    async def reboot(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Reboot Console
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Power", "Reboot", **kwargs)
+
+    async def mute(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Mute
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Audio", "Mute", **kwargs)
+
+    async def unmute(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Unmute
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Audio", "Unmute", **kwargs)
+
+    async def volume(
+        self, device_id: str, direction: VolumeDirection, amount: int = 1, **kwargs
+    ) -> CommandResponse:
+        """
+        Adjust Volume
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        params = [{"direction": direction.value, "amount": str(amount)}]
+        return await self._send_one_shot_command(
+            device_id, "Audio", "Volume", params, **kwargs
+        )
+
+    async def play(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Play (media controls)
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Media", "Play", **kwargs)
+
+    async def pause(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Pause (media controls)
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Media", "Pause", **kwargs)
+
+    async def previous(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Previous (media controls)
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(
+            device_id, "Media", "Previous", **kwargs
+        )
+
+    async def next(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Next (media controls)
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Media", "Next", **kwargs)
+
+    async def go_home(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Go Home
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Shell", "GoHome", **kwargs)
+
+    async def go_back(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Go Back
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns:
+            :class:`SmartglassConsoleStatus`: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "Shell", "GoBack", **kwargs)
+
+    async def show_guide_tab(
+        self, device_id: str, tab: GuideTab = GuideTab.Guide, **kwargs
+    ) -> CommandResponse:
+        """
+        Show Guide Tab
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        params = [{"tabName": tab.value}]
+        return await self._send_one_shot_command(
+            device_id, "Shell", "ShowGuideTab", params, **kwargs
+        )
+
+    async def press_button(
+        self, device_id: str, button: InputKeyType, **kwargs
+    ) -> CommandResponse:
+        """
+        Press Button
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        params = [{"keyType": button.value}]
+        return await self._send_one_shot_command(
+            device_id, "Shell", "InjectKey", params, **kwargs
+        )
+
+    async def insert_text(self, device_id: str, text: str, **kwargs) -> CommandResponse:
+        """
+        Insert Text
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        params = [{"replacementString": text}]
+        return await self._send_one_shot_command(
+            device_id, "Shell", "InjectString", params, **kwargs
+        )
+
+    async def launch_app(
+        self, device_id: str, one_store_product_id: str, **kwargs
+    ) -> CommandResponse:
+        """
+        Launch Application
+
+        Args:
+            device_id: ID of console (from console list)
+            one_store_product_id: OneStoreProductID for the app to launch
+
+        Returns: Command Response
+        """
+        params = [{"oneStoreProductId": one_store_product_id}]
+        return await self._send_one_shot_command(
+            device_id,
+            "Shell",
+            "ActivateApplicationWithOneStoreProductId",
+            params,
+            **kwargs,
+        )
+
+    async def show_tv_guide(self, device_id: str, **kwargs) -> CommandResponse:
+        """
+        Show TV Guide
+
+        Args:
+            device_id: ID of console (from console list)
+
+        Returns: Command Response
+        """
+        return await self._send_one_shot_command(device_id, "TV", "ShowGuide", **kwargs)
+
+    async def _fetch_list(
+        self, list_name: str, params: dict | None = None, **kwargs
+    ) -> Response:
+        """
+        Fetch arbitrary list
+
+        Args:
+            list_name: name of list
+            params: query params
+
+        Returns:
+            :class:`httpx.Response`: HTTP Response
+        """
+        url = f"{self.SG_URL}/lists/{list_name}"
+        resp = await self.client.session.get(
+            url, params=params, headers=self.HEADERS_SG, **kwargs
+        )
+        resp.raise_for_status()
+        return resp
+
+    async def _send_one_shot_command(
+        self,
+        device_id: str,
+        command_type: str,
+        command: str,
+        params: list[dict] | None = None,
+        **kwargs,
+    ) -> CommandResponse:
+        """
+        Send One Shot command to console
+
+        Args:
+            device_id: ID of console (from console list)
+            type: type of command
+            command: name of command
+            params: command parameters
+
+        Returns: Command Response
+        """
+        url = f"{self.SG_URL}/commands"
+        body = {
+            "destination": "Xbox",
+            "type": command_type,
+            "command": command,
+            "sessionId": self._smartglass_session_id,
+            "sourceId": "com.microsoft.smartglass",
+            "parameters": params or [{}],
+            "linkedXboxId": device_id,
+        }
+        resp = await self.client.session.post(
+            url, json=body, headers=self.HEADERS_SG, **kwargs
+        )
+        resp.raise_for_status()
+        return CommandResponse(**resp.json())

pythonxbox/api/provider/smartglass/models.py

@@ -0,0 +1,186 @@
+from datetime import datetime
+from enum import Enum
+
+from pythonxbox.common.models import CamelCaseModel
+
+# Responses
+
+
+class ConsoleType(str, Enum):
+    XboxOne = "XboxOne"
+    XboxOneS = "XboxOneS"
+    XboxOneSDigital = "XboxOneSDigital"
+    XboxOneX = "XboxOneX"
+    XboxSeriesS = "XboxSeriesS"
+    XboxSeriesX = "XboxSeriesX"
+
+
+class PowerState(str, Enum):
+    Unknown = "Unknown"
+    On = "On"
+    Off = "Off"
+    ConnectedStandby = "ConnectedStandby"
+    SystemUpdate = "SystemUpdate"
+
+
+class PlaybackState(str, Enum):
+    Unknown = "Unknown"
+    Playing = "Playing"
+    Paused = "Paused"
+    Stopped = "Stopped"
+
+
+class ErrorCode(str, Enum):
+    OK = "OK"
+    CurrentConsoleNotFound = "CurrentConsoleNotFound"
+    RemoteManagementDisabled = "RemoteManagementDisabled"
+    XboxDataNotFound = "XboxDataNotFound"
+    XboxNotPaired = "XboxNotPaired"
+
+
+class OpStatus(str, Enum):
+    Paused = "Paused"
+    OffConsoleError = "OffConsoleError"
+    Pending = "Pending"
+    TimedOut = "TimedOut"
+    Error = "Error"
+    Succeeded = "Succeeded"
+
+
+class SmartglassApiStatus(CamelCaseModel):
+    error_code: str
+    error_message: str | None = None
+
+
+class StorageDevice(CamelCaseModel):
+    storage_device_id: str
+    storage_device_name: str
+    is_default: bool
+    total_space_bytes: float
+    free_space_bytes: float
+
+
+class SmartglassConsole(CamelCaseModel):
+    id: str
+    name: str
+    console_type: ConsoleType
+    power_state: PowerState
+    console_streaming_enabled: bool
+    digital_assistant_remote_control_enabled: bool
+    remote_management_enabled: bool
+    storage_devices: list[StorageDevice] | None = None
+
+
+class SmartglassConsoleList(CamelCaseModel):
+    agent_user_id: str | None = None
+    result: list[SmartglassConsole]
+    status: SmartglassApiStatus
+
+
+class SmartglassConsoleStatus(CamelCaseModel):
+    power_state: PowerState
+    console_streaming_enabled: bool
+    digital_assistant_remote_control_enabled: bool
+    remote_management_enabled: bool
+    focus_app_aumid: str
+    is_tv_configured: bool
+    login_state: str | None = None
+    playback_state: PlaybackState
+    power_state: PowerState
+    storage_devices: list[StorageDevice] | None = None
+    status: SmartglassApiStatus
+
+
+class InstalledPackage(CamelCaseModel):
+    one_store_product_id: str | None = None
+    title_id: int
+    aumid: str | None = None
+    last_active_time: datetime | None = None
+    is_game: bool
+    name: str | None = None
+    content_type: str
+    instance_id: str
+    storage_device_id: str
+    unique_id: str
+    legacy_product_id: str | None = None
+    version: int
+    size_in_bytes: int
+    install_time: datetime
+    update_time: datetime | None = None
+    parent_id: str | None = None
+
+
+class InstalledPackagesList(CamelCaseModel):
+    result: list[InstalledPackage]
+    status: SmartglassApiStatus
+    agent_user_id: str | None = None
+
+
+class StorageDevicesList(CamelCaseModel):
+    device_id: str
+    result: list[StorageDevice]
+    status: SmartglassApiStatus
+
+
+class OpStatusNode(CamelCaseModel):
+    operation_status: OpStatus
+    op_id: str
+    originating_session_id: str
+    command: str
+    succeeded: bool
+    console_status_code: int | None = None
+    xccs_error_code: ErrorCode | None = None
+    h_result: int | None = None
+    message: str | None = None
+
+
+class OperationStatusResponse(CamelCaseModel):
+    op_status_list: list[OpStatusNode]
+    status: SmartglassApiStatus
+
+
+class CommandDestination(CamelCaseModel):
+    id: str
+    name: str
+    power_state: PowerState
+    remote_management_enabled: bool
+    console_streaming_enabled: bool
+    console_type: ConsoleType
+    wireless_warning: str | None = None
+    out_of_home_warning: str | None = None
+
+
+class CommandResponse(CamelCaseModel):
+    result: str | None = None
+    ui_text: str | None = None
+    destination: CommandDestination
+    user_info: str | None = None
+    op_id: str
+    status: SmartglassApiStatus
+
+
+# Requests
+
+
+class VolumeDirection(str, Enum):
+    Up = "Up"
+    Down = "Down"
+
+
+class InputKeyType(str, Enum):
+    Guide = "Guide"
+    Menu = "Menu"
+    View = "View"
+    A = "A"
+    B = "B"
+    X = "X"
+    Y = "Y"
+    Up = "Up"
+    Down = "Down"
+    Left = "Left"
+    Right = "Right"
+    Nexus = "Nexus"
+
+
+class GuideTab(str, Enum):
+    Guide = "Guide"