async_tcp_socket.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/core/defines.h"
 
#if !defined(USE_ESP32) && !defined(USE_ESP8266) && !defined(USE_RP2040) && !defined(USE_LIBRETINY) && \
    (defined(USE_SOCKET_IMPL_LWIP_SOCKETS) || defined(USE_SOCKET_IMPL_BSD_SOCKETS))
 
#include "esphome/components/socket/socket.h"
#include <functional>
#include <memory>
#include <string>
#include <utility>
 
namespace esphome::async_tcp {
 
class AsyncClient {
 public:
  using AcConnectHandler = std::function<void(void *, AsyncClient *)>;
  using AcDataHandler = std::function<void(void *, AsyncClient *, void *data, size_t len)>;
  using AcErrorHandler = std::function<void(void *, AsyncClient *, int8_t error)>;
 
  AsyncClient() = default;
  ~AsyncClient() = default;
 
  [[nodiscard]] bool connect(const char *host, uint16_t port);
  void close();
  [[nodiscard]] bool connected() const { return connected_; }
  size_t write(const char *data, size_t len);
 
  void onConnect(AcConnectHandler cb, void *arg = nullptr) {  // NOLINT(readability-identifier-naming)
    connect_cb_ = std::move(cb);
    connect_arg_ = arg;
  }
  void onDisconnect(AcConnectHandler cb, void *arg = nullptr) {  // NOLINT(readability-identifier-naming)
    disconnect_cb_ = std::move(cb);
    disconnect_arg_ = arg;
  }
  void onData(AcDataHandler cb, void *arg = nullptr) {  // NOLINT(readability-identifier-naming)
    data_cb_ = std::move(cb);
    data_arg_ = arg;
  }
  void onError(AcErrorHandler cb, void *arg = nullptr) {  // NOLINT(readability-identifier-naming)
    error_cb_ = std::move(cb);
    error_arg_ = arg;
  }
 
  // Must be called from loop()
  void loop();
 
 private:
  std::unique_ptr<esphome::socket::Socket> socket_;
 
  AcConnectHandler connect_cb_{nullptr};
  void *connect_arg_{nullptr};
  AcConnectHandler disconnect_cb_{nullptr};
  void *disconnect_arg_{nullptr};
  AcDataHandler data_cb_{nullptr};
  void *data_arg_{nullptr};
  AcErrorHandler error_cb_{nullptr};
  void *error_arg_{nullptr};
 
  bool connected_{false};
  bool connecting_{false};
};
 
}  // namespace esphome::async_tcp
 
// Expose AsyncClient in global namespace to match library behavior
using esphome::async_tcp::AsyncClient;  // NOLINT(google-global-names-in-headers)
#endif