77 lines
2.5 KiB
C++
77 lines
2.5 KiB
C++
#pragma once
|
|
|
|
#include <utility>
|
|
|
|
#include "esphome/core/automation.h"
|
|
#include "esphome/components/ble_client/ble_client.h"
|
|
|
|
#ifdef USE_ESP32
|
|
|
|
namespace esphome {
|
|
namespace ble_client {
|
|
class BLEClientConnectTrigger : public Trigger<>, public BLEClientNode {
|
|
public:
|
|
explicit BLEClientConnectTrigger(BLEClient *parent) { parent->register_ble_node(this); }
|
|
void loop() override {}
|
|
void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if,
|
|
esp_ble_gattc_cb_param_t *param) override {
|
|
if (event == ESP_GATTC_OPEN_EVT && param->open.status == ESP_GATT_OK)
|
|
this->trigger();
|
|
if (event == ESP_GATTC_SEARCH_CMPL_EVT)
|
|
this->node_state = espbt::ClientState::ESTABLISHED;
|
|
}
|
|
};
|
|
|
|
class BLEClientDisconnectTrigger : public Trigger<>, public BLEClientNode {
|
|
public:
|
|
explicit BLEClientDisconnectTrigger(BLEClient *parent) { parent->register_ble_node(this); }
|
|
void loop() override {}
|
|
void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if,
|
|
esp_ble_gattc_cb_param_t *param) override {
|
|
if (event == ESP_GATTC_DISCONNECT_EVT && memcmp(param->disconnect.remote_bda, this->parent_->remote_bda, 6) == 0)
|
|
this->trigger();
|
|
if (event == ESP_GATTC_SEARCH_CMPL_EVT)
|
|
this->node_state = espbt::ClientState::ESTABLISHED;
|
|
}
|
|
};
|
|
|
|
class BLEWriterClientNode : public BLEClientNode {
|
|
public:
|
|
BLEWriterClientNode(BLEClient *ble_client) {
|
|
ble_client->register_ble_node(this);
|
|
ble_client_ = ble_client;
|
|
}
|
|
|
|
void set_value(std::vector<uint8_t> value) { value_ = std::move(value); }
|
|
|
|
// Attempts to write the contents of value_ to char_uuid_.
|
|
void write();
|
|
|
|
void set_char_uuid128(uint8_t *uuid) { this->char_uuid_ = espbt::ESPBTUUID::from_raw(uuid); }
|
|
|
|
void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); }
|
|
|
|
void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if,
|
|
esp_ble_gattc_cb_param_t *param) override;
|
|
|
|
private:
|
|
BLEClient *ble_client_;
|
|
int ble_char_handle_ = 0;
|
|
esp_gatt_char_prop_t char_props_;
|
|
espbt::ESPBTUUID service_uuid_;
|
|
espbt::ESPBTUUID char_uuid_;
|
|
std::vector<uint8_t> value_;
|
|
};
|
|
|
|
template<typename... Ts> class BLEClientWriteAction : public Action<Ts...>, public BLEWriterClientNode {
|
|
public:
|
|
BLEClientWriteAction(BLEClient *ble_client) : BLEWriterClientNode(ble_client) {}
|
|
|
|
void play(Ts... x) override { return write(); }
|
|
};
|
|
|
|
} // namespace ble_client
|
|
} // namespace esphome
|
|
|
|
#endif
|