wip

components/esp32_presense/BleFingerprint.cpp

@@ -19,7 +19,7 @@ class ClientCallbacks : public BLEClientCallbacks {
 static ClientCallbacks clientCB;
 
 BleFingerprint::BleFingerprint(BLEAdvertisedDevice *advertisedDevice, float fcmin, float beta, float dcutoff) : filteredDistance{FilteredDistance(fcmin, beta, dcutoff)} {
-    firstSeenMillis = millis();
+    firstSeenMillis = esphome::millis();
     address = NimBLEAddress(advertisedDevice->getAddress());
     addressType = advertisedDevice->getAddressType();
     rssi = advertisedDevice->getRSSI();
@@ -70,10 +70,10 @@ bool BleFingerprint::setId(const std::string &newId, short newIdType, const std:
                 qryAttempts = 0;
                 if (rssi < -80) {
                     qryDelayMillis = 30000;
-                    lastQryMillis = millis();
+                    lastQryMillis = esphome::millis();
                 } else if (rssi < -70) {
                     qryDelayMillis = 5000;
-                    lastQryMillis = millis();
+                    lastQryMillis = esphome::millis();
                 }
             }
         }
@@ -411,7 +411,7 @@ void BleFingerprint::fingerprintManufactureData(NimBLEAdvertisedDevice *advertis
 }
 
 bool BleFingerprint::seen(BLEAdvertisedDevice *advertisedDevice) {
-    lastSeenMillis = millis();
+    lastSeenMillis = esphome::millis();
     reported = false;
 
     seenCount++;
@@ -443,17 +443,17 @@ bool BleFingerprint::fill(JsonObject *doc) {
     (*doc)["rssi@1m"] = get1mRssi();
     (*doc)["rssi"] = rssi;
 
-    if (isnormal(raw)) (*doc)["raw"] = serialized(std::string(raw, 2));
+    if (isnormal(raw)) (*doc)["raw"] = serialized(std::to_string(raw));
-    if (isnormal(dist)) (*doc)["distance"] = serialized(std::string(dist, 2));
+    if (isnormal(dist)) (*doc)["distance"] = serialized(std::to_string(dist));
-    if (isnormal(vari)) (*doc)["var"] = serialized(std::string(vari, 2));
+    if (isnormal(vari)) (*doc)["var"] = serialized(std::to_string(vari));
     if (close) (*doc)["close"] = true;
 
     (*doc)["int"] = (millis() - firstSeenMillis) / seenCount;
 
     if (mv) (*doc)["mV"] = mv;
     if (battery != 0xFF) (*doc)["batt"] = battery;
-    if (temp) (*doc)["temp"] = serialized(std::string(temp, 1));
+    if (temp) (*doc)["temp"] = serialized(std::to_string(temp));
-    if (humidity) (*doc)["rh"] = serialized(std::string(humidity, 1));
+    if (humidity) (*doc)["rh"] = serialized(std::to_string(humidity));
     return true;
 }
 
@@ -465,7 +465,7 @@ bool BleFingerprint::report(JsonObject *doc) {
     if (maxDistance > 0 && dist > maxDistance)
         return false;
 
-    auto now = millis();
+    auto now = esphome::millis();
     if ((abs(dist - lastReported) < BleFingerprintCollection::skipDistance) && (lastReportedMillis > 0) && (now - lastReportedMillis < BleFingerprintCollection::skipMs))
         return false;
 
@@ -483,7 +483,7 @@ bool BleFingerprint::query() {
     if (!allowQuery || isQuerying) return false;
     if (rssi < -90) return false; // Too far away
 
-    auto now = millis();
+    auto now = esphome::millis();
     if (now - lastSeenMillis > 5) return false; // Haven't seen lately
     if (now - lastQryMillis < qryDelayMillis) return false; // Too soon
 

components/esp32_presense/BleFingerprint.h

@@ -20,8 +20,6 @@
 #include "esphome/core/time.h"
 
 #define TAG "esp32_presense"
-#define millis esphome::millis
-// #define delay esphome::delay
 
 #define NO_RSSI int8_t(-128)
 

components/esp32_presense/MiFloraHandler.cpp

@@ -1,4 +1,5 @@
 #include "MiFloraHandler.h"
+#include "NimBLEAttValue.h"
 
 namespace MiFloraHandler {
 

components/esp32_presense/__init__.py

@@ -30,23 +30,19 @@ CONFIG_SCHEMA = cv.Schema({
 
 async def to_code(config):
   var = cg.new_Pvariable(config[CONF_ID])
-  # await cg.register_component(var, config)
+  await cg.register_component(var, config)
 
   cg.add(var.set_room(config[CONF_ROOM_KEY]))
   # cg.add(var.set_base_topic(config[CONF_BASE_TOPIC_KEY]))
   # cg.add(var.set_addresses(config[CONF_MAC_KEY]))
-  # cg.add(var.set_max_distance(config[CONF_MAX_DISTANCE]))
+  cg.add(var.set_max_distance(config[CONF_MAX_DISTANCE]))
 
   add_idf_sdkconfig_option("CONFIG_BT_ENABLED", True)
   add_idf_sdkconfig_option("CONFIG_BT_BLUEDROID_ENABLED", False)
   add_idf_sdkconfig_option("CONFIG_BT_NIMBLE_ENABLED", True)
   add_idf_sdkconfig_option("CONFIG_MBEDTLS_HARDWARE_AES", False)
 
-  cg.add_library(
+  cg.add_library("esp-nimble-cpp", repository="https://github.com/h2zero/esp-nimble-cpp#v1.4.1", version="v1.4.1")
-    name="esp-nimble-cpp",
-    repository="https://github.com/h2zero/esp-nimble-cpp",
-    version="1.4.1"
-  )
 
   # await nimble_tracker.device_listener_to_code(var, config)
   # await nimble_tracker.register_ble_device(var, config)

components/esp32_presense/defaults.h

@@ -0,0 +1,5 @@
+#define BLE_SCAN_INTERVAL 0x80
+#define BLE_SCAN_WINDOW 0x80
+#define SCAN_TASK_STACK_SIZE 2562
+
+#define CHANNEL "espresense"

components/esp32_presense/esp32_presense.cpp

@@ -9,16 +9,101 @@ namespace esphome
         unsigned int totalFpQueried = 0;
         unsigned int totalFpReported = 0;
         
+        TimerHandle_t reconnectTimer;
+        TaskHandle_t scanTaskHandle;
 
+        unsigned long updateStartedMillis = 0;
+        unsigned long lastTeleMillis = 0;
+        int reconnectTries = 0;
+        int teleFails = 0;
+        int reportFailed = 0;
+        bool online = false;         // Have we successfully sent status=online
+        bool sentDiscovery = false;  // Have we successfully sent discovery
+        UBaseType_t bleStack = 0;
+
+        DynamicJsonDocument doc(1024);
+        std::string _id, roomsTopic;
+
+        bool discovery, publishTele, publishRooms, publishDevices;
+
+        class MyAdvertisedDeviceCallbacks : public BLEAdvertisedDeviceCallbacks {
+            void onResult(BLEAdvertisedDevice *advertisedDevice) {
+                bleStack = uxTaskGetStackHighWaterMark(nullptr);
+                BleFingerprintCollection::Seen(advertisedDevice);
+            }
+        };
+
+        void scanTask(void *parameter) {
+            NimBLEDevice::init("ESPresense");
+            // Enrollment::Setup();
+            NimBLEDevice::setMTU(23);
+    
+            auto pBLEScan = NimBLEDevice::getScan();
+            pBLEScan->setInterval(BLE_SCAN_INTERVAL);
+            pBLEScan->setWindow(BLE_SCAN_WINDOW);
+            pBLEScan->setAdvertisedDeviceCallbacks(new MyAdvertisedDeviceCallbacks(), true);
+            pBLEScan->setActiveScan(false);
+            pBLEScan->setDuplicateFilter(false);
+            pBLEScan->setMaxResults(0);
+            if (!pBLEScan->start(0, nullptr, false))
+                ESP_LOGE(TAG, "Error starting continuous ble scan");
+
+            while (true) {
+                for (auto &f : BleFingerprintCollection::fingerprints)
+                    if (f->query())
+                        totalFpQueried++;
+
+                // Enrollment::Loop();
+
+                if (!pBLEScan->isScanning()) {
+                    if (!pBLEScan->start(0, nullptr, true))
+                        ESP_LOGE(TAG, "Error re-starting continuous ble scan");
+                    delay(3000);  // If we stopped scanning, don't query for 3 seconds in order for us to catch any missed broadcasts
+                } else {
+                    delay(100);
+                }
+            }
+        }
+
+        void ESP32Presense::set_room(std::string room) { 
+            _id = slugify(room);
+            roomsTopic = std::string(CHANNEL) + std::string("/rooms/") + _id;
+        }
+        
+        void ESP32Presense::set_max_distance(float maxDistance) {
+            BleFingerprintCollection::maxDistance = maxDistance;
+        }
+
+        void ESP32Presense::setup()
+        {
+            BleFingerprintCollection::Setup();
+            xTaskCreatePinnedToCore(scanTask, "scanTask", SCAN_TASK_STACK_SIZE, nullptr, 1, &scanTaskHandle, CONFIG_BT_NIMBLE_PINNED_TO_CORE);
+            publishDevices = true;
+        }
+
+        void ESP32Presense::loop()
+        {
+            reportLoop();
+        }
+
+        bool ESP32Presense::reportBuffer(BleFingerprint *f) {
+            auto report = f->getReport();
+            std::string topic = Sprintf(CHANNEL "/devices/%s/%s/%s", f->getId().c_str(), _id.c_str(), report.getId().c_str());
+            return this->publish(topic, report.getPayload());
+        }
 
         void ESP32Presense::reportLoop() 
         {
             auto copy = BleFingerprintCollection::GetCopy();
             unsigned int count = 0;
+
             for (auto &i : copy)
                 if (i->shouldCount())
                     count++;
+
+            yield();
             sendTelemetry(totalSeen, totalFpSeen, totalFpQueried, totalFpReported, count);
+            yield();
 
             auto reported = 0;
             for (auto &f : copy) {
@@ -28,20 +113,47 @@ namespace esphome
                     totalFpSeen++;
                 }
 
-                /*
                 if (f->hasReport()) {
-                    if (reportBuffer(f))
+                    if (reportBuffer(f)) {
                         f->clearReport();
                     }
+                }
                 
-                if (reportDevice(f)) {
+
+                if (this->reportDevice(f)) {
                     totalFpReported++;
                     reported++;
                 }
-                */
             }
         }
 
+        bool ESP32Presense::reportDevice(BleFingerprint *f) {
+            doc.clear();
+            JsonObject obj = doc.to<JsonObject>();
+            if (!f->report(&obj))
+                return false;
+
+            std::string buffer;
+            serializeJson(doc, buffer);
+            std::string devicesTopic = Sprintf(CHANNEL "/devices/%s/%s", f->getId().c_str(), _id.c_str());
+
+            bool p1 = false, p2 = false;
+            for (int i = 0; i < 10; i++) {
+                if (!p1 && (!publishRooms || this->publish(roomsTopic.c_str(), buffer.c_str())))
+                p1 = true;
+
+                if (!p2 && (!publishDevices || this->publish(devicesTopic.c_str(), buffer.c_str())))
+                    p2 = true;
+
+                if (p1 && p2)
+                    return true;
+                delay(20);
+            }
+
+            reportFailed++;
+            return false;
+        }
+        
         bool ESP32Presense::sendTelemetry(
             unsigned int totalSeen, 
             unsigned int totalFpSeen, 
@@ -49,17 +161,17 @@ namespace esphome
             unsigned int totalFpReported, 
             unsigned int count
         ) {
-            this->publish(this->room_ + "/status", "online");
+            this->publish(roomsTopic + "/status", "online");
-            this->publish(this->room_ + "/max_distance", BleFingerprintCollection::maxDistance);
+            this->publish(roomsTopic + "/max_distance", BleFingerprintCollection::maxDistance);
-            this->publish(this->room_ + "/absorption", BleFingerprintCollection::absorption);
+            this->publish(roomsTopic + "/absorption", BleFingerprintCollection::absorption);
-            this->publish(this->room_ + "/tx_ref_rssi", BleFingerprintCollection::txRefRssi);
+            this->publish(roomsTopic + "/tx_ref_rssi", BleFingerprintCollection::txRefRssi);
-            this->publish(this->room_ + "/rx_adj_rssi", BleFingerprintCollection::rxAdjRssi);
+            this->publish(roomsTopic + "/rx_adj_rssi", BleFingerprintCollection::rxAdjRssi);
-            this->publish(this->room_ + "/query", BleFingerprintCollection::query);
+            this->publish(roomsTopic + "/query", BleFingerprintCollection::query);
-            this->publish(this->room_ + "/include", BleFingerprintCollection::include);
+            this->publish(roomsTopic + "/include", BleFingerprintCollection::include);
-            this->publish(this->room_ + "/exclude", BleFingerprintCollection::exclude);
+            this->publish(roomsTopic + "/exclude", BleFingerprintCollection::exclude);
-            this->publish(this->room_ + "/known_macs", BleFingerprintCollection::knownMacs);
+            this->publish(roomsTopic + "/known_macs", BleFingerprintCollection::knownMacs);
-            this->publish(this->room_ + "/known_irks", BleFingerprintCollection::knownIrks);
+            this->publish(roomsTopic + "/known_irks", BleFingerprintCollection::knownIrks);
-            this->publish(this->room_ + "/count_ids", BleFingerprintCollection::countIds);
+            this->publish(roomsTopic + "/count_ids", BleFingerprintCollection::countIds);
 
             return true;
         }

components/esp32_presense/esp32_presense.h

@@ -1,8 +1,10 @@
 #pragma once
 
+#include <ArduinoJson.h>
 #include "esphome/core/component.h"
 #include "esphome/components/mqtt/custom_mqtt_device.h"
 #include "BleFingerprintCollection.h"
+#include "defaults.h"
 // #include "esphome/components/nimble_distance_custom/nimble_distance_custom.h"
 
 namespace esphome
@@ -10,12 +12,11 @@ namespace esphome
     namespace esp32_presense
     {
         class ESP32Presense :
+            public Component,
             public mqtt::CustomMQTTDevice 
             // public nimble_distance_custom::NimbleDistanceCustomComponent
         {
             protected:
-                std::string room_;
-
                 BleFingerprint *ble_fingerprint = NULL;
             
                 bool sendTelemetry(
@@ -26,11 +27,14 @@ namespace esphome
                     unsigned int count
                 );
             public:
-                void setup() {};
+                void setup() override;
-                void loop() {};
+                void loop() override;
                 void reportLoop();
                 // void on_result(nimble_distance_custom::NimbleDistanceCustomResult&) override;
-                void set_room(std::string room) { room_ = room; }
+                void set_room(std::string room);
+                bool reportBuffer(BleFingerprint *f);
+                bool reportDevice(BleFingerprint *f);
+                void set_max_distance(float maxDistance);
                 //void set_base_topic(std::string base_topic) { base_topic_ = base_topic; }
         };
     } // namespace esp32_nimble_tracker