core/homeassistant/components/switchbot_cloud/service.py

"""SwitchBot Cloud Custom Service."""

from logging import getLogger

from switchbot_api import ArtFrameCommands
import voluptuous as vol

from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv, device_registry as dr

from .const import AI_ART_FRAME_UPLOAD_IMAGE_SERVICE, DOMAIN

_LOGGER = getLogger(__name__)


UPLOAD_IMAGE_SCHEMA = vol.Schema(
    {
        vol.Required("device_id"): vol.All(
            cv.ensure_list, [cv.string], vol.Length(min=1)
        ),
        vol.Required("image_url"): cv.url,
    }
)


async def handle_upload_image(call: ServiceCall) -> None:
    """Handle Ai Art Frame Upload Image."""
    hass = call.hass
    image_url = call.data["image_url"]
    device_ids = call.data.get("device_id", [])
    dev_reg = dr.async_get(hass)

    for ha_device_id in device_ids:
        device = dev_reg.async_get(ha_device_id)
        if device is None:
            raise ServiceValidationError(f"Device {ha_device_id} not found")

        device_mac = next(
            (iid[1] for iid in device.identifiers if iid[0] == DOMAIN), None
        )
        if device_mac:
            for config_entry_id in device.config_entries:
                entry = hass.config_entries.async_get_entry(config_entry_id)
                if entry is not None and entry.domain == DOMAIN:
                    break
            else:
                raise ServiceValidationError(
                    f"Device {ha_device_id} is not a SwitchBot Cloud device"
                )
            await entry.runtime_data.api.send_command(
                device_id=device_mac,
                command=ArtFrameCommands.UPLOAD.value,
                command_type="command",
                parameters={"imageUrl": image_url},
            )
        else:
            raise ServiceValidationError("No valid MAC address obtained.")


def async_register_services(hass: HomeAssistant) -> None:
    """Async register services."""
    hass.services.async_register(
        DOMAIN,
        AI_ART_FRAME_UPLOAD_IMAGE_SERVICE,
        handle_upload_image,
        schema=UPLOAD_IMAGE_SCHEMA,
    )