#ifndef zBLEConnect_h
#define zBLEConnect_h

#ifdef ESP32
#  include "NimBLEDevice.h"
#  include "TheengsCommon.h"
#  include "config_BT.h"

class zBLEConnect {
public:
  NimBLEClient* m_pClient;
  TaskHandle_t m_taskHandle = nullptr;
  zBLEConnect(NimBLEAddress& addr) {
    m_pClient = NimBLEDevice::createClient(addr);
    m_pClient->setConnectTimeout(5000);
  }
  virtual ~zBLEConnect() { NimBLEDevice::deleteClient(m_pClient); }
  virtual bool writeData(BLEAction* action);
  virtual bool readData(BLEAction* action);
  virtual bool processActions(std::vector<BLEAction>& actions);
  virtual void publishData() {}
  virtual NimBLERemoteCharacteristic* getCharacteristic(const NimBLEUUID& service, const NimBLEUUID& characteristic);
};

class LYWSD03MMC_connect : public zBLEConnect {
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  LYWSD03MMC_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  void publishData() override;
};

class DT24_connect : public zBLEConnect {
  std::vector<uint8_t> m_data;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  DT24_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  void publishData() override;
};

class BM2_connect : public zBLEConnect {
  //std::vector<uint8_t> m_data;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  BM2_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  void publishData() override;
};

class GENERIC_connect : public zBLEConnect {
  std::vector<uint8_t> m_data;

public:
  GENERIC_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
};

class HHCCJCY01HHCC_connect : public zBLEConnect {
  std::vector<uint8_t> m_data;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  HHCCJCY01HHCC_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  void publishData() override;
};

class XMWSDJ04MMC_connect : public zBLEConnect {
  std::vector<uint8_t> m_data;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  XMWSDJ04MMC_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  void publishData() override;
};

class SBS1_connect : public zBLEConnect {
  uint8_t m_notifyVal;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  SBS1_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  bool processActions(std::vector<BLEAction>& actions) override;
};

class SBBT_connect : public zBLEConnect {
  uint8_t m_notifyVal;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  SBBT_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  bool processActions(std::vector<BLEAction>& actions) override;
};

class SBCU_connect : public zBLEConnect {
  uint8_t m_notifyVal;
  void notifyCB(NimBLERemoteCharacteristic* pChar, uint8_t* pData, size_t length, bool isNotify);

public:
  SBCU_connect(NimBLEAddress& addr) : zBLEConnect(addr) {}
  bool processActions(std::vector<BLEAction>& actions) override;
};

#endif //ESP32
#endif //zBLEConnect_h
