core/homeassistant/components/androidtv/services.py

"""Services for Android/Fire TV devices."""

from __future__ import annotations

import voluptuous as vol

from homeassistant.components.media_player import DOMAIN as MEDIA_PLAYER_DOMAIN
from homeassistant.const import ATTR_COMMAND
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, service

from .const import DOMAIN

ATTR_ADB_RESPONSE = "adb_response"
ATTR_DEVICE_PATH = "device_path"
ATTR_HDMI_INPUT = "hdmi_input"
ATTR_LOCAL_PATH = "local_path"


@callback
def async_setup_services(hass: HomeAssistant) -> None:
    """Register the Android TV / Fire TV services."""

    service.async_register_platform_entity_service(
        hass,
        DOMAIN,
        "adb_command",
        entity_domain=MEDIA_PLAYER_DOMAIN,
        schema={vol.Required(ATTR_COMMAND): cv.string},
        func="adb_command",
    )
    service.async_register_platform_entity_service(
        hass,
        DOMAIN,
        "learn_sendevent",
        entity_domain=MEDIA_PLAYER_DOMAIN,
        schema=None,
        func="learn_sendevent",
    )
    service.async_register_platform_entity_service(
        hass,
        DOMAIN,
        "download",
        entity_domain=MEDIA_PLAYER_DOMAIN,
        schema={
            vol.Required(ATTR_DEVICE_PATH): cv.string,
            vol.Required(ATTR_LOCAL_PATH): cv.string,
        },
        func="service_download",
    )
    service.async_register_platform_entity_service(
        hass,
        DOMAIN,
        "upload",
        entity_domain=MEDIA_PLAYER_DOMAIN,
        schema={
            vol.Required(ATTR_DEVICE_PATH): cv.string,
            vol.Required(ATTR_LOCAL_PATH): cv.string,
        },
        func="service_upload",
    )