TinyCTSClientPico.h

Go to the documentation of this file.

 
#pragma once
#include "../TinyCTSClientBase.h"
#include "TinyCTSPicoBLE.h"
 
namespace tiny_cts {

class TinyCTSClient : public TinyCTSClientBase {
 public:
  explicit TinyCTSClient(const char* device = nullptr, uint32_t scanMs = 5000)
      : _scanMs(scanMs) {
    if (device != nullptr) _device = device;
  }

  bool begin() {
    if (!ctsPicoBegin("", false)) return false;
    return update();
  }

  bool update() {
    BLEClient* client = connect();
    if (client == nullptr) return false;
    bool ok = false;
    BLERemoteService* service = client->service(BLEUUID(CTS_SERVICE_UUID));
    if (service != nullptr) {
      uint8_t data[16];
      size_t len;
      if (read(service->characteristic(BLEUUID(CTS_LOCAL_TIME_INFO_UUID)), data, len)) {
        storeLocalInfo(data, len);
      }
      if (read(service->characteristic(BLEUUID(CTS_REFERENCE_TIME_INFO_UUID)), data, len)) {
        storeReferenceInfo(data, len);
      }
      BLERemoteCharacteristic* chr = service->characteristic(BLEUUID(CTS_CURRENT_TIME_UUID));
      uint32_t t0 = TINY_CTS_MILLIS();
      if (read(chr, data, len)) {
        ok = storeCurrentTime(data, len, t0, TINY_CTS_MILLIS());
      }
    }
    client->disconnect();
    return ok;
  }

  bool setTimeMs(uint64_t unixMs) override {
    BLEClient* client = connect();
    if (client == nullptr) return false;
    bool ok = false;
    BLERemoteService* service = client->service(BLEUUID(CTS_SERVICE_UUID));
    BLERemoteCharacteristic* chr =
        service != nullptr ? service->characteristic(BLEUUID(CTS_CURRENT_TIME_UUID)) : nullptr;
    if (chr != nullptr && chr->canWrite()) {
      uint8_t data[TinyCTSCurrentTime::Size];
      ok = chr->setValue(data, timeToWrite(unixMs, data));
    }
    client->disconnect();
    return ok;
  }
 
 protected:
  BLEClient* connect() {
    uint32_t seconds = (_scanMs + 999) / 1000;
    if (seconds == 0) seconds = 1;
    // the scan with the UUID keeps the devices that advertise the service
    BLEScanReport* report =
        _device.empty() ? BLE.scan(BLEUUID(CTS_SERVICE_UUID), seconds) : BLE.scan(seconds);
    BLEClient* client = BLE.client();
    bool connected = false;
    for (auto& item : *report) {
      std::string address = item.getAddress().toString().c_str();
      std::string name = item.getName();
      if (!_device.empty() && !isWanted(address, name)) continue;
      connected = client->connect(item, 10);
      if (connected) {
        _address = address;
        _name = name;
      }
      break;
    }
    BLE.clearScan();
    return connected ? client : nullptr;
  }

  static bool read(BLERemoteCharacteristic* chr, uint8_t* out, size_t& len) {
    if (chr == nullptr || !chr->canRead()) return false;
    chr->getUInt8();
    const void* value = chr->valueData();
    len = chr->valueLen();
    if (value == nullptr || len == 0) return false;
    if (len > 16) len = 16;
    memcpy(out, value, len);
    return true;
  }
 
  uint32_t _scanMs;
};
 
}  // namespace tiny_cts