mirror of
https://github.com/esphome/esphome.git
synced 2025-03-26 04:28:17 +00:00
144 lines
4.3 KiB
Python
144 lines
4.3 KiB
Python
from esphome import automation
|
|
from esphome.automation import maybe_simple_id
|
|
import esphome.codegen as cg
|
|
from esphome.components import audio_dac
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_DATA, CONF_ID, CONF_VOLUME
|
|
from esphome.core import CORE
|
|
from esphome.coroutine import coroutine_with_priority
|
|
|
|
AUTO_LOAD = ["audio"]
|
|
CODEOWNERS = ["@jesserockz", "@kahrendt"]
|
|
|
|
IS_PLATFORM_COMPONENT = True
|
|
|
|
CONF_AUDIO_DAC = "audio_dac"
|
|
|
|
speaker_ns = cg.esphome_ns.namespace("speaker")
|
|
|
|
Speaker = speaker_ns.class_("Speaker")
|
|
|
|
PlayAction = speaker_ns.class_(
|
|
"PlayAction", automation.Action, cg.Parented.template(Speaker)
|
|
)
|
|
StopAction = speaker_ns.class_(
|
|
"StopAction", automation.Action, cg.Parented.template(Speaker)
|
|
)
|
|
FinishAction = speaker_ns.class_(
|
|
"FinishAction", automation.Action, cg.Parented.template(Speaker)
|
|
)
|
|
VolumeSetAction = speaker_ns.class_(
|
|
"VolumeSetAction", automation.Action, cg.Parented.template(Speaker)
|
|
)
|
|
MuteOnAction = speaker_ns.class_(
|
|
"MuteOnAction", automation.Action, cg.Parented.template(Speaker)
|
|
)
|
|
MuteOffAction = speaker_ns.class_(
|
|
"MuteOffAction", automation.Action, cg.Parented.template(Speaker)
|
|
)
|
|
|
|
|
|
IsPlayingCondition = speaker_ns.class_("IsPlayingCondition", automation.Condition)
|
|
IsStoppedCondition = speaker_ns.class_("IsStoppedCondition", automation.Condition)
|
|
|
|
|
|
async def setup_speaker_core_(var, config):
|
|
if audio_dac_config := config.get(CONF_AUDIO_DAC):
|
|
aud_dac = await cg.get_variable(audio_dac_config)
|
|
cg.add(var.set_audio_dac(aud_dac))
|
|
|
|
|
|
async def register_speaker(var, config):
|
|
if not CORE.has_id(config[CONF_ID]):
|
|
var = cg.Pvariable(config[CONF_ID], var)
|
|
await setup_speaker_core_(var, config)
|
|
|
|
|
|
SPEAKER_SCHEMA = cv.Schema(
|
|
{
|
|
cv.Optional(CONF_AUDIO_DAC): cv.use_id(audio_dac.AudioDac),
|
|
}
|
|
)
|
|
|
|
SPEAKER_AUTOMATION_SCHEMA = maybe_simple_id({cv.GenerateID(): cv.use_id(Speaker)})
|
|
|
|
|
|
async def speaker_action(config, action_id, template_arg, args):
|
|
var = cg.new_Pvariable(action_id, template_arg)
|
|
await cg.register_parented(var, config[CONF_ID])
|
|
return var
|
|
|
|
|
|
@automation.register_action(
|
|
"speaker.play",
|
|
PlayAction,
|
|
cv.maybe_simple_value(
|
|
{
|
|
cv.GenerateID(): cv.use_id(Speaker),
|
|
cv.Required(CONF_DATA): cv.templatable(cv.ensure_list(cv.hex_uint8_t)),
|
|
},
|
|
key=CONF_DATA,
|
|
),
|
|
)
|
|
async def speaker_play_action(config, action_id, template_arg, args):
|
|
var = cg.new_Pvariable(action_id, template_arg)
|
|
await cg.register_parented(var, config[CONF_ID])
|
|
data = config[CONF_DATA]
|
|
|
|
if cg.is_template(data):
|
|
templ = await cg.templatable(data, args, cg.std_vector.template(cg.uint8))
|
|
cg.add(var.set_data_template(templ))
|
|
else:
|
|
cg.add(var.set_data_static(data))
|
|
return var
|
|
|
|
|
|
automation.register_action("speaker.stop", StopAction, SPEAKER_AUTOMATION_SCHEMA)(
|
|
speaker_action
|
|
)
|
|
automation.register_action("speaker.finish", FinishAction, SPEAKER_AUTOMATION_SCHEMA)(
|
|
speaker_action
|
|
)
|
|
|
|
automation.register_condition(
|
|
"speaker.is_playing", IsPlayingCondition, SPEAKER_AUTOMATION_SCHEMA
|
|
)(speaker_action)
|
|
|
|
automation.register_condition(
|
|
"speaker.is_stopped", IsStoppedCondition, SPEAKER_AUTOMATION_SCHEMA
|
|
)(speaker_action)
|
|
|
|
|
|
@automation.register_action(
|
|
"speaker.volume_set",
|
|
VolumeSetAction,
|
|
cv.maybe_simple_value(
|
|
{
|
|
cv.GenerateID(): cv.use_id(Speaker),
|
|
cv.Required(CONF_VOLUME): cv.templatable(cv.percentage),
|
|
},
|
|
key=CONF_VOLUME,
|
|
),
|
|
)
|
|
async def speaker_volume_set_action(config, action_id, template_arg, args):
|
|
var = cg.new_Pvariable(action_id, template_arg)
|
|
await cg.register_parented(var, config[CONF_ID])
|
|
volume = await cg.templatable(config[CONF_VOLUME], args, float)
|
|
cg.add(var.set_volume(volume))
|
|
return var
|
|
|
|
|
|
@automation.register_action(
|
|
"speaker.mute_off", MuteOffAction, SPEAKER_AUTOMATION_SCHEMA
|
|
)
|
|
@automation.register_action("speaker.mute_on", MuteOnAction, SPEAKER_AUTOMATION_SCHEMA)
|
|
async def speaker_mute_action_to_code(config, action_id, template_arg, args):
|
|
paren = await cg.get_variable(config[CONF_ID])
|
|
return cg.new_Pvariable(action_id, template_arg, paren)
|
|
|
|
|
|
@coroutine_with_priority(100.0)
|
|
async def to_code(config):
|
|
cg.add_global(speaker_ns.using)
|
|
cg.add_define("USE_SPEAKER")
|