mirror of
https://github.com/esphome/esphome.git
synced 2025-03-20 09:38:18 +00:00
109 lines
3.5 KiB
C++
109 lines
3.5 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/automation.h"
|
|
#include "esphome/components/esp32_ble_tracker/esp32_ble_tracker.h"
|
|
|
|
#ifdef USE_ESP32
|
|
|
|
namespace esphome {
|
|
namespace esp32_ble_tracker {
|
|
class ESPBTAdvertiseTrigger : public Trigger<const ESPBTDevice &>, public ESPBTDeviceListener {
|
|
public:
|
|
explicit ESPBTAdvertiseTrigger(ESP32BLETracker *parent) { parent->register_listener(this); }
|
|
void set_address(uint64_t address) { this->address_ = address; }
|
|
|
|
bool parse_device(const ESPBTDevice &device) override {
|
|
if (this->address_ && device.address_uint64() != this->address_) {
|
|
return false;
|
|
}
|
|
this->trigger(device);
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
uint64_t address_ = 0;
|
|
};
|
|
|
|
class BLEServiceDataAdvertiseTrigger : public Trigger<const adv_data_t &>, public ESPBTDeviceListener {
|
|
public:
|
|
explicit BLEServiceDataAdvertiseTrigger(ESP32BLETracker *parent) { parent->register_listener(this); }
|
|
void set_address(uint64_t address) { this->address_ = address; }
|
|
void set_service_uuid16(uint16_t uuid) { this->uuid_ = ESPBTUUID::from_uint16(uuid); }
|
|
void set_service_uuid32(uint32_t uuid) { this->uuid_ = ESPBTUUID::from_uint32(uuid); }
|
|
void set_service_uuid128(uint8_t *uuid) { this->uuid_ = ESPBTUUID::from_raw(uuid); }
|
|
|
|
bool parse_device(const ESPBTDevice &device) override {
|
|
if (this->address_ && device.address_uint64() != this->address_) {
|
|
return false;
|
|
}
|
|
for (auto &service_data : device.get_service_datas()) {
|
|
if (service_data.uuid == this->uuid_) {
|
|
this->trigger(service_data.data);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
protected:
|
|
uint64_t address_ = 0;
|
|
ESPBTUUID uuid_;
|
|
};
|
|
|
|
class BLEManufacturerDataAdvertiseTrigger : public Trigger<const adv_data_t &>, public ESPBTDeviceListener {
|
|
public:
|
|
explicit BLEManufacturerDataAdvertiseTrigger(ESP32BLETracker *parent) { parent->register_listener(this); }
|
|
void set_address(uint64_t address) { this->address_ = address; }
|
|
void set_manufacturer_uuid16(uint16_t uuid) { this->uuid_ = ESPBTUUID::from_uint16(uuid); }
|
|
void set_manufacturer_uuid32(uint32_t uuid) { this->uuid_ = ESPBTUUID::from_uint32(uuid); }
|
|
void set_manufacturer_uuid128(uint8_t *uuid) { this->uuid_ = ESPBTUUID::from_raw(uuid); }
|
|
|
|
bool parse_device(const ESPBTDevice &device) override {
|
|
if (this->address_ && device.address_uint64() != this->address_) {
|
|
return false;
|
|
}
|
|
for (auto &manufacturer_data : device.get_manufacturer_datas()) {
|
|
if (manufacturer_data.uuid == this->uuid_) {
|
|
this->trigger(manufacturer_data.data);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
protected:
|
|
uint64_t address_ = 0;
|
|
ESPBTUUID uuid_;
|
|
};
|
|
|
|
class BLEEndOfScanTrigger : public Trigger<>, public ESPBTDeviceListener {
|
|
public:
|
|
explicit BLEEndOfScanTrigger(ESP32BLETracker *parent) { parent->register_listener(this); }
|
|
|
|
bool parse_device(const ESPBTDevice &device) override { return false; }
|
|
void on_scan_end() override { this->trigger(); }
|
|
};
|
|
|
|
template<typename... Ts> class ESP32BLEStartScanAction : public Action<Ts...> {
|
|
public:
|
|
ESP32BLEStartScanAction(ESP32BLETracker *parent) : parent_(parent) {}
|
|
TEMPLATABLE_VALUE(bool, continuous)
|
|
void play(Ts... x) override {
|
|
this->parent_->set_scan_continuous(this->continuous_.value(x...));
|
|
this->parent_->start_scan();
|
|
}
|
|
|
|
protected:
|
|
ESP32BLETracker *parent_;
|
|
};
|
|
|
|
template<typename... Ts> class ESP32BLEStopScanAction : public Action<Ts...>, public Parented<ESP32BLETracker> {
|
|
public:
|
|
void play(Ts... x) override { this->parent_->stop_scan(); }
|
|
};
|
|
|
|
} // namespace esp32_ble_tracker
|
|
} // namespace esphome
|
|
|
|
#endif
|