mirror of https://github.com/esphome/esphome.git
89 lines
3.2 KiB
Python
89 lines
3.2 KiB
Python
import voluptuous as vol
|
|
|
|
from esphome.automation import ACTION_REGISTRY
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_DATA, CONF_DATA_TEMPLATE, CONF_ID, CONF_PASSWORD, CONF_PORT, \
|
|
CONF_REBOOT_TIMEOUT, CONF_SERVICE, CONF_VARIABLES
|
|
from esphome.core import CORE
|
|
from esphome.cpp_generator import Pvariable, add, get_variable, process_lambda
|
|
from esphome.cpp_helpers import setup_component
|
|
from esphome.cpp_types import Action, App, Component, StoringController, esphome_ns
|
|
|
|
api_ns = esphome_ns.namespace('api')
|
|
APIServer = api_ns.class_('APIServer', Component, StoringController)
|
|
HomeAssistantServiceCallAction = api_ns.class_('HomeAssistantServiceCallAction', Action)
|
|
KeyValuePair = api_ns.class_('KeyValuePair')
|
|
TemplatableKeyValuePair = api_ns.class_('TemplatableKeyValuePair')
|
|
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
cv.GenerateID(): cv.declare_variable_id(APIServer),
|
|
vol.Optional(CONF_PORT, default=6053): cv.port,
|
|
vol.Optional(CONF_PASSWORD, default=''): cv.string_strict,
|
|
vol.Optional(CONF_REBOOT_TIMEOUT): cv.positive_time_period_milliseconds,
|
|
}).extend(cv.COMPONENT_SCHEMA.schema)
|
|
|
|
|
|
def to_code(config):
|
|
rhs = App.init_api_server()
|
|
api = Pvariable(config[CONF_ID], rhs)
|
|
|
|
if config[CONF_PORT] != 6053:
|
|
add(api.set_port(config[CONF_PORT]))
|
|
if config.get(CONF_PASSWORD):
|
|
add(api.set_password(config[CONF_PASSWORD]))
|
|
if CONF_REBOOT_TIMEOUT in config:
|
|
add(api.set_reboot_timeout(config[CONF_REBOOT_TIMEOUT]))
|
|
|
|
setup_component(api, config)
|
|
|
|
|
|
BUILD_FLAGS = '-DUSE_API'
|
|
|
|
|
|
def lib_deps(config):
|
|
if CORE.is_esp32:
|
|
return 'AsyncTCP@1.0.1'
|
|
if CORE.is_esp8266:
|
|
return 'ESPAsyncTCP@1.1.3'
|
|
raise NotImplementedError
|
|
|
|
|
|
CONF_HOMEASSISTANT_SERVICE = 'homeassistant.service'
|
|
HOMEASSISTANT_SERVIC_ACTION_SCHEMA = vol.Schema({
|
|
cv.GenerateID(): cv.use_variable_id(APIServer),
|
|
vol.Required(CONF_SERVICE): cv.string,
|
|
vol.Optional(CONF_DATA): vol.Schema({
|
|
cv.string: cv.string,
|
|
}),
|
|
vol.Optional(CONF_DATA_TEMPLATE): vol.Schema({
|
|
cv.string: cv.string,
|
|
}),
|
|
vol.Optional(CONF_VARIABLES): vol.Schema({
|
|
cv.string: cv.lambda_,
|
|
}),
|
|
})
|
|
|
|
|
|
@ACTION_REGISTRY.register(CONF_HOMEASSISTANT_SERVICE, HOMEASSISTANT_SERVIC_ACTION_SCHEMA)
|
|
def homeassistant_service_to_code(config, action_id, arg_type, template_arg):
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_home_assistant_service_call_action(template_arg)
|
|
type = HomeAssistantServiceCallAction.template(arg_type)
|
|
act = Pvariable(action_id, rhs, type=type)
|
|
add(act.set_service(config[CONF_SERVICE]))
|
|
if CONF_DATA in config:
|
|
datas = [KeyValuePair(k, v) for k, v in config[CONF_DATA].items()]
|
|
add(act.set_data(datas))
|
|
if CONF_DATA_TEMPLATE in config:
|
|
datas = [KeyValuePair(k, v) for k, v in config[CONF_DATA_TEMPLATE].items()]
|
|
add(act.set_data_template(datas))
|
|
if CONF_VARIABLES in config:
|
|
datas = []
|
|
for key, value in config[CONF_VARIABLES].items():
|
|
for value_ in process_lambda(value, []):
|
|
yield None
|
|
datas.append(TemplatableKeyValuePair(key, value_))
|
|
add(act.set_variables(datas))
|
|
yield act
|