73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
import logging
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.websocket_api import websocket_command, result_message, event_message, async_register_command
|
|
|
|
from .mod_view import setup_view
|
|
|
|
DOMAIN = "browser_mod"
|
|
|
|
FRONTEND_SCRIPT_URL = "/browser_mod.js"
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
async def async_setup(hass, config):
|
|
_LOGGER.error(f"Setting up browser_mod")
|
|
|
|
setup_view(hass, FRONTEND_SCRIPT_URL)
|
|
_LOGGER.error(f"Registered frontend script")
|
|
|
|
aliases = {}
|
|
for d in config[DOMAIN].get("devices", {}):
|
|
name = config[DOMAIN]["devices"][d].get("name", None)
|
|
if name:
|
|
aliases[name] = d
|
|
|
|
hass.data[DOMAIN] = {
|
|
"devices": {},
|
|
"aliases": aliases,
|
|
"adders": [],
|
|
}
|
|
|
|
await hass.helpers.discovery.async_load_platform("media_player", DOMAIN, {}, config)
|
|
|
|
_LOGGER.error(f"Set up media_player")
|
|
_LOGGER.error(hass.data[DOMAIN]["adders"])
|
|
|
|
async_register_command(hass, handle_connect)
|
|
async_register_command(hass, handle_update)
|
|
_LOGGER.error(f"Registered connect ws command")
|
|
|
|
return True
|
|
|
|
|
|
|
|
@websocket_command({
|
|
vol.Required("type"): "browser_mod/connect",
|
|
vol.Required("deviceID"): str,
|
|
})
|
|
def handle_connect(hass, connection, msg):
|
|
_LOGGER.error(f"Got connection {msg}")
|
|
|
|
devices = hass.data[DOMAIN]["devices"]
|
|
deviceID = msg["deviceID"]
|
|
if deviceID in devices:
|
|
devices[deviceID].connect(connection, msg["id"])
|
|
else:
|
|
adder = hass.data[DOMAIN]["adders"][0]
|
|
devices[deviceID] = adder(hass, deviceID, connection, msg["id"])
|
|
connection.send_message(result_message(msg["id"]))
|
|
|
|
|
|
@websocket_command({
|
|
vol.Required("type"): "browser_mod/update",
|
|
vol.Required("deviceID"): str,
|
|
vol.Optional("browser"): dict,
|
|
vol.Optional("player"): dict,
|
|
})
|
|
def handle_update(hass, connection, msg):
|
|
devices = hass.data[DOMAIN]["devices"]
|
|
deviceID = msg["deviceID"]
|
|
if deviceID in devices:
|
|
devices[deviceID].update(msg.get("browser", None), msg.get("player", None))
|