Move esp32_ble_server to its own component (#1898)

2025-09-11 07:42:26 +01:00 · 2021-06-12 08:31:15 +12:00
parent eb24da7c82
commit 5a2cfa2798
24 changed files with 192 additions and 146 deletions
--- a/esphome/components/esp32_ble_server/ble_server.cpp
+++ b/esphome/components/esp32_ble_server/ble_server.cpp
@@ -0,0 +1,168 @@
+#include "ble_server.h"
+
+#include "esphome/components/esp32_ble/ble.h"
+#include "esphome/core/log.h"
+#include "esphome/core/application.h"
+#include "esphome/core/version.h"
+
+#ifdef ARDUINO_ARCH_ESP32
+
+#include <nvs_flash.h>
+#include <freertos/FreeRTOSConfig.h>
+#include <esp_bt_main.h>
+#include <esp_bt.h>
+#include <freertos/task.h>
+#include <esp_gap_ble_api.h>
+
+namespace esphome {
+namespace esp32_ble_server {
+
+static const char *const TAG = "esp32_ble_server";
+
+static const uint16_t DEVICE_INFORMATION_SERVICE_UUID = 0x180A;
+static const uint16_t MODEL_UUID = 0x2A24;
+static const uint16_t VERSION_UUID = 0x2A26;
+static const uint16_t MANUFACTURER_UUID = 0x2A29;
+
+void BLEServer::setup() {
+  if (this->is_failed()) {
+    ESP_LOGE(TAG, "BLE Server was marked failed by ESP32BLE");
+    return;
+  }
+
+  ESP_LOGD(TAG, "Setting up BLE Server...");
+
+  global_ble_server = this;
+}
+
+void BLEServer::loop() {
+  switch (this->state_) {
+    case RUNNING:
+      return;
+
+    case INIT: {
+      esp_err_t err = esp_ble_gatts_app_register(0);
+      if (err != ESP_OK) {
+        ESP_LOGE(TAG, "esp_ble_gatts_app_register failed: %d", err);
+        this->mark_failed();
+        return;
+      }
+      this->state_ = REGISTERING;
+      break;
+    }
+    case REGISTERING: {
+      if (this->registered_) {
+        this->device_information_service_ = this->create_service(DEVICE_INFORMATION_SERVICE_UUID);
+
+        this->create_device_characteristics_();
+
+        this->state_ = STARTING_SERVICE;
+      }
+      break;
+    }
+    case STARTING_SERVICE: {
+      if (!this->device_information_service_->is_created()) {
+        break;
+      }
+      if (this->device_information_service_->is_running()) {
+        this->state_ = RUNNING;
+        this->can_proceed_ = true;
+        ESP_LOGD(TAG, "BLE server setup successfully");
+      } else if (!this->device_information_service_->is_starting()) {
+        this->device_information_service_->start();
+      }
+      break;
+    }
+  }
+}
+
+bool BLEServer::create_device_characteristics_() {
+  if (this->model_.has_value()) {
+    BLECharacteristic *model =
+        this->device_information_service_->create_characteristic(MODEL_UUID, BLECharacteristic::PROPERTY_READ);
+    model->set_value(this->model_.value());
+  } else {
+#ifdef ARDUINO_BOARD
+    BLECharacteristic *model =
+        this->device_information_service_->create_characteristic(MODEL_UUID, BLECharacteristic::PROPERTY_READ);
+    model->set_value(ARDUINO_BOARD);
+#endif
+  }
+
+  BLECharacteristic *version =
+      this->device_information_service_->create_characteristic(VERSION_UUID, BLECharacteristic::PROPERTY_READ);
+  version->set_value("ESPHome " ESPHOME_VERSION);
+
+  BLECharacteristic *manufacturer =
+      this->device_information_service_->create_characteristic(MANUFACTURER_UUID, BLECharacteristic::PROPERTY_READ);
+  manufacturer->set_value(this->manufacturer_);
+
+  return true;
+}
+
+BLEService *BLEServer::create_service(const uint8_t *uuid, bool advertise) {
+  return this->create_service(ESPBTUUID::from_raw(uuid), advertise);
+}
+BLEService *BLEServer::create_service(uint16_t uuid, bool advertise) {
+  return this->create_service(ESPBTUUID::from_uint16(uuid), advertise);
+}
+BLEService *BLEServer::create_service(const std::string &uuid, bool advertise) {
+  return this->create_service(ESPBTUUID::from_raw(uuid), advertise);
+}
+BLEService *BLEServer::create_service(ESPBTUUID uuid, bool advertise, uint16_t num_handles, uint8_t inst_id) {
+  ESP_LOGV(TAG, "Creating service - %s", uuid.to_string().c_str());
+  BLEService *service = new BLEService(uuid, num_handles, inst_id);
+  this->services_.push_back(service);
+  if (advertise) {
+    esp32_ble::global_ble->get_advertising()->add_service_uuid(uuid);
+  }
+  service->do_create(this);
+  return service;
+}
+
+void BLEServer::gatts_event_handler(esp_gatts_cb_event_t event, esp_gatt_if_t gatts_if,
+                                    esp_ble_gatts_cb_param_t *param) {
+  switch (event) {
+    case ESP_GATTS_CONNECT_EVT: {
+      ESP_LOGD(TAG, "BLE Client connected");
+      this->add_client_(param->connect.conn_id, (void *) this);
+      this->connected_clients_++;
+      for (auto *component : this->service_components_) {
+        component->on_client_connect();
+      }
+      break;
+    }
+    case ESP_GATTS_DISCONNECT_EVT: {
+      ESP_LOGD(TAG, "BLE Client disconnected");
+      if (this->remove_client_(param->disconnect.conn_id))
+        this->connected_clients_--;
+      esp32_ble::global_ble->get_advertising()->start();
+      for (auto *component : this->service_components_) {
+        component->on_client_disconnect();
+      }
+      break;
+    }
+    case ESP_GATTS_REG_EVT: {
+      this->gatts_if_ = gatts_if;
+      this->registered_ = true;
+      break;
+    }
+    default:
+      break;
+  }
+
+  for (auto *service : this->services_) {
+    service->gatts_event_handler(event, gatts_if, param);
+  }
+}
+
+float BLEServer::get_setup_priority() const { return setup_priority::BLUETOOTH - 10; }
+
+void BLEServer::dump_config() { ESP_LOGCONFIG(TAG, "ESP32 BLE Server:"); }
+
+BLEServer *global_ble_server = nullptr;
+
+}  // namespace esp32_ble_server
+}  // namespace esphome
+
+#endif