esphome/components/nimble_tracker/nimble_tracker.cpp
2023-12-10 09:08:23 +03:00

93 lines
3.1 KiB
C++

#include "nimble_tracker.h"
#include "esphome/core/log.h"
#include "esphome/core/hal.h"
// using namespace esphome;
namespace esphome
{
namespace nimble_tracker
{
static const char *const TAG = "nimble_tracker";
class MyAdvertisedDeviceCallbacks : public BLEAdvertisedDeviceCallbacks
{
public:
MyAdvertisedDeviceCallbacks(NimbleTracker *nimble_tracker)
{
nimble_tracker_ = nimble_tracker;
}
void onResult(BLEAdvertisedDevice *advertised_device)
{
// Because setMaxResults is set to 0 for the NimBLEScan, we need to make a copy
// of the data of the advertised device, because this is deleted immediately by NimBLESCan
// after this callback is called.
auto *tracker_event = new NimbleTrackerEvent(
advertised_device->getAddress(),
advertised_device->getAddressType(),
advertised_device->getRSSI(),
advertised_device->getTXPower());
nimble_tracker_->tracker_events_.push(tracker_event);
}
protected:
NimbleTracker *nimble_tracker_;
};
void NimbleTracker::setup()
{
// Set the name to empty string to not broadcast the name
NimBLEDevice::init("");
this->pBLEScan_ = NimBLEDevice::getScan();
this->pBLEScan_->setInterval(this->scan_interval_);
this->pBLEScan_->setWindow(this->scan_window_);
this->pBLEScan_->setAdvertisedDeviceCallbacks(new MyAdvertisedDeviceCallbacks(this), true);
this->pBLEScan_->setActiveScan(this->scan_active_);
this->pBLEScan_->setDuplicateFilter(false);
this->pBLEScan_->setMaxResults(0);
ESP_LOGV(TAG, "Trying to start the scan");
if (!pBLEScan_->start(0, nullptr, false))
{
ESP_LOGE(TAG, "Error starting continuous ble scan");
// this->mark_failed();
return;
}
// TODO: It takes some time to setup bluetooth? Why not just move this to loop?
esphome::delay(200);
}
void NimbleTracker::loop()
{
if (!this->pBLEScan_->isScanning())
{
if (!this->pBLEScan_->start(0, nullptr, false))
{
ESP_LOGE(TAG, "Error starting continuous ble scan");
return;
}
// TODO: we shouldn't block the main thread here, instead work with a setTimeout callback?
esphome::delay(200);
}
NimbleTrackerEvent *tracker_event = this->tracker_events_.pop();
while (tracker_event != nullptr)
{
for (NimbleDeviceListener *listener : this->listeners_)
{
listener->parse_event(tracker_event);
}
delete tracker_event;
tracker_event = this->tracker_events_.pop();
}
};
} // namespace esp32_ble_tracker
} // namespace esphome