__init__.py

Go to the documentation of this file.

"""Support for Joaoapps Join services."""
 
import logging
 
from pyjoin import (
    get_devices,
    ring_device,
    send_file,
    send_notification,
    send_sms,
    send_url,
    set_wallpaper,
)
import voluptuous as vol
 
from homeassistant.const import CONF_API_KEY, CONF_DEVICE_ID, CONF_NAME
from homeassistant.core import HomeAssistant, ServiceCall
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.typing import ConfigType
 
_LOGGER = logging.getLogger(__name__)
 
DOMAIN = "joaoapps_join"
 
CONF_DEVICE_IDS = "device_ids"
CONF_DEVICE_NAMES = "device_names"
 
CONFIG_SCHEMA = vol.Schema(
    {
        DOMAIN: vol.All(
            cv.ensure_list,
            [
                {
                    vol.Required(CONF_API_KEY): cv.string,
                    vol.Optional(CONF_DEVICE_ID): cv.string,
                    vol.Optional(CONF_DEVICE_IDS): cv.string,
                    vol.Optional(CONF_DEVICE_NAMES): cv.string,
                    vol.Optional(CONF_NAME): cv.string,
                }
            ],
        )
    },
    extra=vol.ALLOW_EXTRA,
)
 
 
def register_device(hass, api_key, name, device_id, device_ids, device_names):
    """Register services for each join device listed."""
 
    def ring_service(service: ServiceCall) -> None:
        """Service to ring devices."""
        ring_device(
            api_key=api_key,
            device_id=device_id,
            device_ids=device_ids,
            device_names=device_names,
        )
 
    def set_wallpaper_service(service: ServiceCall) -> None:
        """Service to set wallpaper on devices."""
        set_wallpaper(
            api_key=api_key,
            device_id=device_id,
            device_ids=device_ids,
            device_names=device_names,
            url=service.data.get("url"),
        )
 
    def send_file_service(service: ServiceCall) -> None:
        """Service to send files to devices."""
        send_file(
            api_key=api_key,
            device_id=device_id,
            device_ids=device_ids,
            device_names=device_names,
            url=service.data.get("url"),
        )
 
    def send_url_service(service: ServiceCall) -> None:
        """Service to open url on devices."""
        send_url(
            api_key=api_key,
            device_id=device_id,
            device_ids=device_ids,
            device_names=device_names,
            url=service.data.get("url"),
        )
 
    def send_tasker_service(service: ServiceCall) -> None:
        """Service to open url on devices."""
        send_notification(
            api_key=api_key,
            device_id=device_id,
            device_ids=device_ids,
            device_names=device_names,
            text=service.data.get("command"),
        )
 
    def send_sms_service(service: ServiceCall) -> None:
        """Service to send sms from devices."""
        send_sms(
            device_id=device_id,
            device_ids=device_ids,
            device_names=device_names,
            sms_number=service.data.get("number"),
            sms_text=service.data.get("message"),
            api_key=api_key,
        )
 
    hass.services.register(DOMAIN, f"{name}ring", ring_service)
    hass.services.register(DOMAIN, f"{name}set_wallpaper", set_wallpaper_service)
    hass.services.register(DOMAIN, f"{name}send_sms", send_sms_service)
    hass.services.register(DOMAIN, f"{name}send_file", send_file_service)
    hass.services.register(DOMAIN, f"{name}send_url", send_url_service)
    hass.services.register(DOMAIN, f"{name}send_tasker", send_tasker_service)
 
 
def setup(hass: HomeAssistant, config: ConfigType) -> bool:
    """Set up the Join services."""
    for device in config[DOMAIN]:
        api_key = device.get(CONF_API_KEY)
        device_id = device.get(CONF_DEVICE_ID)
        device_ids = device.get(CONF_DEVICE_IDS)
        device_names = device.get(CONF_DEVICE_NAMES)
        name = device.get(CONF_NAME)
        name = f"{name.lower().replace(' ', '_')}_" if name else ""
        if api_key and not get_devices(api_key):
            _LOGGER.error("Error connecting to Join, check API key")
            return False
        if device_id is None and device_ids is None and device_names is None:
            _LOGGER.error(
                "No device was provided. Please specify device_id"
                ", device_ids, or device_names"
            )
            return False
 
        register_device(hass, api_key, name, device_id, device_ids, device_names)
    return True