mirror of
https://github.com/Threnklyn/esphome-dev.git
synced 2026-05-31 10:08:27 +02:00
Jesse Hills
95 lines
3.0 KiB
C++
95 lines
3.0 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/defines.h"
|
|
|
|
#ifdef USE_VOICE_ASSISTANT
|
|
|
|
#include "esphome/core/automation.h"
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/defines.h"
|
|
#include "esphome/core/helpers.h"
|
|
|
|
#include "esphome/components/api/api_pb2.h"
|
|
#include "esphome/components/api/api_server.h"
|
|
#include "esphome/components/microphone/microphone.h"
|
|
#ifdef USE_SPEAKER
|
|
#include "esphome/components/speaker/speaker.h"
|
|
#endif
|
|
#include "esphome/components/socket/socket.h"
|
|
|
|
namespace esphome {
|
|
namespace voice_assistant {
|
|
|
|
// Version 1: Initial version
|
|
// Version 2: Adds raw speaker support
|
|
static const uint32_t INITIAL_VERSION = 1;
|
|
static const uint32_t SPEAKER_SUPPORT = 2;
|
|
|
|
class VoiceAssistant : public Component {
|
|
public:
|
|
void setup() override;
|
|
void loop() override;
|
|
float get_setup_priority() const override;
|
|
void start(struct sockaddr_storage *addr, uint16_t port);
|
|
|
|
void set_microphone(microphone::Microphone *mic) { this->mic_ = mic; }
|
|
#ifdef USE_SPEAKER
|
|
void set_speaker(speaker::Speaker *speaker) { this->speaker_ = speaker; }
|
|
#endif
|
|
|
|
uint32_t get_version() const {
|
|
#ifdef USE_SPEAKER
|
|
if (this->speaker_ != nullptr)
|
|
return SPEAKER_SUPPORT;
|
|
#endif
|
|
return INITIAL_VERSION;
|
|
}
|
|
|
|
void request_start();
|
|
void signal_stop();
|
|
|
|
void on_event(const api::VoiceAssistantEventResponse &msg);
|
|
|
|
Trigger<> *get_start_trigger() const { return this->start_trigger_; }
|
|
Trigger<std::string> *get_stt_end_trigger() const { return this->stt_end_trigger_; }
|
|
Trigger<std::string> *get_tts_start_trigger() const { return this->tts_start_trigger_; }
|
|
Trigger<std::string> *get_tts_end_trigger() const { return this->tts_end_trigger_; }
|
|
Trigger<> *get_end_trigger() const { return this->end_trigger_; }
|
|
Trigger<std::string, std::string> *get_error_trigger() const { return this->error_trigger_; }
|
|
|
|
protected:
|
|
std::unique_ptr<socket::Socket> socket_ = nullptr;
|
|
struct sockaddr_storage dest_addr_;
|
|
|
|
Trigger<> *start_trigger_ = new Trigger<>();
|
|
Trigger<std::string> *stt_end_trigger_ = new Trigger<std::string>();
|
|
Trigger<std::string> *tts_start_trigger_ = new Trigger<std::string>();
|
|
Trigger<std::string> *tts_end_trigger_ = new Trigger<std::string>();
|
|
Trigger<> *end_trigger_ = new Trigger<>();
|
|
Trigger<std::string, std::string> *error_trigger_ = new Trigger<std::string, std::string>();
|
|
|
|
microphone::Microphone *mic_{nullptr};
|
|
#ifdef USE_SPEAKER
|
|
speaker::Speaker *speaker_{nullptr};
|
|
#endif
|
|
|
|
bool running_{false};
|
|
};
|
|
|
|
template<typename... Ts> class StartAction : public Action<Ts...>, public Parented<VoiceAssistant> {
|
|
public:
|
|
void play(Ts... x) override { this->parent_->request_start(); }
|
|
};
|
|
|
|
template<typename... Ts> class StopAction : public Action<Ts...>, public Parented<VoiceAssistant> {
|
|
public:
|
|
void play(Ts... x) override { this->parent_->signal_stop(); }
|
|
};
|
|
|
|
extern VoiceAssistant *global_voice_assistant; // NOLINT(cppcoreguidelines-avoid-non-const-global-variables)
|
|
|
|
} // namespace voice_assistant
|
|
} // namespace esphome
|
|
|
|
#endif // USE_VOICE_ASSISTANT
|