Initial commit: Media server and Home Assistant integration
- FastAPI server for Windows media control via WinRT/SMTC - Home Assistant custom integration with media player entity - Script button entities for system commands - Position tracking with grace period for track skip handling - Server availability detection in HA entity Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
131
custom_components/remote_media_player/button.py
Normal file
131
custom_components/remote_media_player/button.py
Normal file
@@ -0,0 +1,131 @@
|
||||
"""Button platform for Remote Media Player integration."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
from homeassistant.components.button import ButtonEntity
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.entity import DeviceInfo
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from .api_client import MediaServerClient, MediaServerError
|
||||
from .const import DOMAIN
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
entry: ConfigEntry,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Set up script buttons from a config entry."""
|
||||
client: MediaServerClient = hass.data[DOMAIN][entry.entry_id]["client"]
|
||||
|
||||
try:
|
||||
scripts = await client.list_scripts()
|
||||
except MediaServerError as err:
|
||||
_LOGGER.error("Failed to fetch scripts list: %s", err)
|
||||
return
|
||||
|
||||
entities = [
|
||||
ScriptButtonEntity(
|
||||
client=client,
|
||||
entry=entry,
|
||||
script_name=script["name"],
|
||||
script_label=script["label"],
|
||||
script_description=script.get("description", ""),
|
||||
)
|
||||
for script in scripts
|
||||
]
|
||||
|
||||
if entities:
|
||||
async_add_entities(entities)
|
||||
_LOGGER.info("Added %d script button entities", len(entities))
|
||||
|
||||
|
||||
class ScriptButtonEntity(ButtonEntity):
|
||||
"""Button entity for executing a script on the media server."""
|
||||
|
||||
_attr_has_entity_name = True
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
client: MediaServerClient,
|
||||
entry: ConfigEntry,
|
||||
script_name: str,
|
||||
script_label: str,
|
||||
script_description: str,
|
||||
) -> None:
|
||||
"""Initialize the script button."""
|
||||
self._client = client
|
||||
self._entry = entry
|
||||
self._script_name = script_name
|
||||
self._script_label = script_label
|
||||
self._script_description = script_description
|
||||
|
||||
# Entity attributes
|
||||
self._attr_unique_id = f"{entry.entry_id}_script_{script_name}"
|
||||
self._attr_name = script_label
|
||||
self._attr_icon = self._get_icon_for_script(script_name)
|
||||
|
||||
def _get_icon_for_script(self, script_name: str) -> str:
|
||||
"""Get an appropriate icon based on script name."""
|
||||
icon_map = {
|
||||
"lock": "mdi:lock",
|
||||
"unlock": "mdi:lock-open",
|
||||
"shutdown": "mdi:power",
|
||||
"restart": "mdi:restart",
|
||||
"sleep": "mdi:sleep",
|
||||
"hibernate": "mdi:power-sleep",
|
||||
"cancel": "mdi:cancel",
|
||||
}
|
||||
|
||||
script_lower = script_name.lower()
|
||||
for keyword, icon in icon_map.items():
|
||||
if keyword in script_lower:
|
||||
return icon
|
||||
|
||||
return "mdi:script-text"
|
||||
|
||||
@property
|
||||
def device_info(self) -> DeviceInfo:
|
||||
"""Return device info."""
|
||||
return DeviceInfo(
|
||||
identifiers={(DOMAIN, self._entry.entry_id)},
|
||||
name=self._entry.title,
|
||||
manufacturer="Remote Media Player",
|
||||
model="Media Server",
|
||||
)
|
||||
|
||||
@property
|
||||
def extra_state_attributes(self) -> dict[str, Any]:
|
||||
"""Return extra state attributes."""
|
||||
return {
|
||||
"script_name": self._script_name,
|
||||
"description": self._script_description,
|
||||
}
|
||||
|
||||
async def async_press(self) -> None:
|
||||
"""Handle button press - execute the script."""
|
||||
_LOGGER.info("Executing script: %s", self._script_name)
|
||||
try:
|
||||
result = await self._client.execute_script(self._script_name)
|
||||
if result.get("success"):
|
||||
_LOGGER.info(
|
||||
"Script '%s' executed successfully (exit_code=%s)",
|
||||
self._script_name,
|
||||
result.get("exit_code"),
|
||||
)
|
||||
else:
|
||||
_LOGGER.warning(
|
||||
"Script '%s' failed: %s",
|
||||
self._script_name,
|
||||
result.get("stderr") or result.get("error"),
|
||||
)
|
||||
except MediaServerError as err:
|
||||
_LOGGER.error("Failed to execute script '%s': %s", self._script_name, err)
|
||||
raise
|
||||
Reference in New Issue
Block a user