Enhance Jibo integration: Add service filtering by robot name, improve configuration flow, and update service documentation.
This commit is contained in:
@@ -1,52 +1,92 @@
|
|||||||
import asyncio
|
|
||||||
import aiohttp
|
import aiohttp
|
||||||
import voluptuous as vol
|
import voluptuous as vol
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant, ServiceCall
|
||||||
import logging
|
import logging
|
||||||
_LOGGER = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
|
||||||
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
_SAY_SCHEMA = vol.Schema({
|
||||||
|
vol.Required("message"): str,
|
||||||
|
vol.Optional("robot"): str,
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
async def async_setup(hass: HomeAssistant, config: dict):
|
async def async_setup(hass: HomeAssistant, config: dict):
|
||||||
"""Set up the Jibo integration."""
|
hass.data.setdefault(DOMAIN, {})
|
||||||
|
|
||||||
async def handle_say_service(call):
|
|
||||||
message = call.data.get("message")
|
|
||||||
ip = hass.data[DOMAIN]["jibo_ip"]
|
|
||||||
url = f"http://{ip}:8089/tts_speak"
|
|
||||||
payload = {
|
|
||||||
"prompt": message,
|
|
||||||
"locale": "en-us",
|
|
||||||
"voice": "griffin",
|
|
||||||
"duration_stretch": 1,
|
|
||||||
"pitch": 3,
|
|
||||||
"pitchBandwidth": 0.4,
|
|
||||||
"mode": "text",
|
|
||||||
"outputMode": "stream",
|
|
||||||
"timeout": None,
|
|
||||||
"volume": 0,
|
|
||||||
"whisper": "FALSE",
|
|
||||||
"samplerate": 48000,
|
|
||||||
"postfilter": 0.4,
|
|
||||||
"framerate": 240,
|
|
||||||
"unvoicedvoiced": 0.35,
|
|
||||||
"allPass": 0.76,
|
|
||||||
"gvMCEP": 0.9,
|
|
||||||
"cached": "TRUE"
|
|
||||||
}
|
|
||||||
|
|
||||||
async with aiohttp.ClientSession() as session:
|
|
||||||
try:
|
|
||||||
async with session.post(url, json=payload) as response:
|
|
||||||
if response.status != 200:
|
|
||||||
_LOGGER.error("Failed to speak: %s", await response.text())
|
|
||||||
except aiohttp.ClientError as e:
|
|
||||||
_LOGGER.error("Error communicating with Jibo: %s", e)
|
|
||||||
|
|
||||||
hass.services.async_register(DOMAIN, "say", handle_say_service)
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
async def async_setup_entry(hass, entry):
|
|
||||||
"""Store IP from config flow."""
|
async def async_setup_entry(hass: HomeAssistant, entry):
|
||||||
hass.data[DOMAIN] = {"jibo_ip": entry.data["jibo_ip"]}
|
hass.data.setdefault(DOMAIN, {})
|
||||||
return True
|
hass.data[DOMAIN][entry.entry_id] = {
|
||||||
|
"jibo_ip": entry.data["jibo_ip"],
|
||||||
|
"name": entry.data.get("name", entry.title),
|
||||||
|
}
|
||||||
|
|
||||||
|
if not hass.services.has_service(DOMAIN, "say"):
|
||||||
|
async def handle_say(call: ServiceCall):
|
||||||
|
message = call.data["message"]
|
||||||
|
robot_filter = call.data.get("robot")
|
||||||
|
|
||||||
|
targets = [
|
||||||
|
data["jibo_ip"]
|
||||||
|
for data in hass.data[DOMAIN].values()
|
||||||
|
if robot_filter is None or data["name"] == robot_filter
|
||||||
|
]
|
||||||
|
|
||||||
|
if not targets:
|
||||||
|
_LOGGER.warning(
|
||||||
|
"No Jibo robot matched filter %r. Configured robots: %s",
|
||||||
|
robot_filter,
|
||||||
|
[d["name"] for d in hass.data[DOMAIN].values()],
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
payload = {
|
||||||
|
"prompt": message,
|
||||||
|
"locale": "en-us",
|
||||||
|
"voice": "griffin",
|
||||||
|
"duration_stretch": 1,
|
||||||
|
"pitch": 3,
|
||||||
|
"pitchBandwidth": 0.4,
|
||||||
|
"mode": "text",
|
||||||
|
"outputMode": "stream",
|
||||||
|
"timeout": None,
|
||||||
|
"volume": 0,
|
||||||
|
"whisper": "FALSE",
|
||||||
|
"samplerate": 48000,
|
||||||
|
"postfilter": 0.4,
|
||||||
|
"framerate": 240,
|
||||||
|
"unvoicedvoiced": 0.35,
|
||||||
|
"allPass": 0.76,
|
||||||
|
"gvMCEP": 0.9,
|
||||||
|
"cached": "TRUE",
|
||||||
|
}
|
||||||
|
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
for ip in targets:
|
||||||
|
url = f"http://{ip}:8089/tts_speak"
|
||||||
|
try:
|
||||||
|
async with session.post(url, json=payload) as response:
|
||||||
|
if response.status != 200:
|
||||||
|
_LOGGER.error(
|
||||||
|
"Jibo at %s returned %s: %s",
|
||||||
|
ip, response.status, await response.text(),
|
||||||
|
)
|
||||||
|
except aiohttp.ClientError as e:
|
||||||
|
_LOGGER.error("Error communicating with Jibo at %s: %s", ip, e)
|
||||||
|
|
||||||
|
hass.services.async_register(DOMAIN, "say", handle_say, schema=_SAY_SCHEMA)
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
async def async_unload_entry(hass: HomeAssistant, entry):
|
||||||
|
hass.data[DOMAIN].pop(entry.entry_id, None)
|
||||||
|
|
||||||
|
if not hass.data[DOMAIN]:
|
||||||
|
hass.services.async_remove(DOMAIN, "say")
|
||||||
|
|
||||||
|
return True
|
||||||
|
|||||||
@@ -1,17 +1,36 @@
|
|||||||
import voluptuous as vol
|
import voluptuous as vol
|
||||||
from homeassistant import config_entries
|
from homeassistant import config_entries
|
||||||
|
from homeassistant.exceptions import HomeAssistantError
|
||||||
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
|
||||||
|
|
||||||
class JiboConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
class JiboConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
||||||
VERSION = 1
|
VERSION = 1
|
||||||
CONNECTION_CLASS = config_entries.CONN_CLASS_LOCAL_POLL
|
|
||||||
|
|
||||||
async def async_step_user(self, user_input=None):
|
async def async_step_user(self, user_input=None):
|
||||||
|
errors = {}
|
||||||
|
|
||||||
if user_input is not None:
|
if user_input is not None:
|
||||||
return self.async_create_entry(title="Jibo", data=user_input)
|
ip = user_input["jibo_ip"].strip()
|
||||||
|
|
||||||
|
await self.async_set_unique_id(ip)
|
||||||
|
self._abort_if_unique_id_configured()
|
||||||
|
|
||||||
|
name = user_input.get("name", "").strip() or f"Jibo ({ip})"
|
||||||
|
|
||||||
|
return self.async_create_entry(
|
||||||
|
title=name,
|
||||||
|
data={"jibo_ip": ip, "name": name},
|
||||||
|
)
|
||||||
|
|
||||||
data_schema = vol.Schema({
|
data_schema = vol.Schema({
|
||||||
vol.Required("jibo_ip"): str
|
vol.Required("name"): str,
|
||||||
|
vol.Required("jibo_ip"): str,
|
||||||
})
|
})
|
||||||
return self.async_show_form(step_id="user", data_schema=data_schema)
|
|
||||||
|
return self.async_show_form(
|
||||||
|
step_id="user",
|
||||||
|
data_schema=data_schema,
|
||||||
|
errors=errors,
|
||||||
|
)
|
||||||
|
|||||||
@@ -1,10 +1,11 @@
|
|||||||
{
|
{
|
||||||
"domain": "jibo",
|
"domain": "jibo",
|
||||||
"name": "OpenJibo",
|
"name": "OpenJibo",
|
||||||
"version": "0.1.0.a",
|
"version": "0.1.0.alpha.2",
|
||||||
"documentation": "https://jibohacks.zane.org/homeassistant/int",
|
"documentation": "https://jibohacks.zane.org/homeassistant/int",
|
||||||
"requirements": [],
|
"requirements": [],
|
||||||
"dependencies": [],
|
"dependencies": [],
|
||||||
"codeowners": ["@ZaneThePython"],
|
"codeowners": ["@ZaneThePython"],
|
||||||
"config_flow": true
|
"config_flow": true,
|
||||||
|
"iot_class": "local_polling"
|
||||||
}
|
}
|
||||||
@@ -1,7 +1,17 @@
|
|||||||
say:
|
say:
|
||||||
description: Make Jibo Say Something
|
description: Make a Jibo robot say something.
|
||||||
fields:
|
fields:
|
||||||
message:
|
message:
|
||||||
description: Text to speak
|
description: The text to speak.
|
||||||
example: "Hello Home Assistant"
|
example: "Hello, Home Assistant!"
|
||||||
required: true
|
required: true
|
||||||
|
selector:
|
||||||
|
text:
|
||||||
|
robot:
|
||||||
|
description: >
|
||||||
|
Name of the robot to speak on. Leave empty to speak on all configured
|
||||||
|
Jibo robots.
|
||||||
|
example: "Living Room Jibo"
|
||||||
|
required: false
|
||||||
|
selector:
|
||||||
|
text:
|
||||||
|
|||||||
Reference in New Issue
Block a user