PubSubClient.h

/*
 PubSubClient.h - A simple client for MQTT.
  Nicholas O'Leary
  http://knolleary.net
*/

#ifndef PubSubClient_h
#define PubSubClient_h

#ifdef ESP8266
#include <functional>
#include <pgmspace.h>
#endif

#include <Arduino.h>

#include "MQTT.h"

class PubSubClient {
public:
#ifdef _GLIBCXX_FUNCTIONAL
  typedef std::function<void(const MQTT::Publish&)> callback_t;
#else
  typedef void(*callback_t)(const MQTT::Publish&);
#endif

private:
   IPAddress server_ip;
   String server_hostname;
   uint16_t server_port;
   callback_t _callback;

   Client *_client;
   uint16_t nextMsgId, keepalive;
   uint8_t _max_retries;
   unsigned long lastOutActivity;
   unsigned long lastInActivity;
   bool pingOutstanding;


   MQTT::Message* _recv_message(void);


   bool _send_message(MQTT::Message& msg, bool need_reply = false);


   void _process_message(MQTT::Message* msg);


   bool _wait_for(MQTT::message_type wait_type, uint16_t wait_pid = 0);

   uint16_t _next_packet_id(void) {
     nextMsgId++;
     if (nextMsgId == 0) nextMsgId = 1;
     return nextMsgId;
   }

public:

   PubSubClient(Client& c);

   PubSubClient(Client& c, IPAddress &ip, uint16_t port = 1883);
   PubSubClient(Client& c, String hostname, uint16_t port = 1883);

   PubSubClient& set_server(IPAddress &ip, uint16_t port = 1883);
   PubSubClient& set_server(String hostname, uint16_t port = 1883);

   callback_t callback(void) const { return _callback; }
   PubSubClient& set_callback(callback_t cb) { _callback = cb; return *this; }
   PubSubClient& unset_callback(void) { _callback = NULL; return * this; }

   PubSubClient& set_max_retries(uint8_t mr) { _max_retries = mr; return *this; }


   bool connect(String id);


   bool connect(String id, String willTopic, uint8_t willQos, bool willRetain, String willMessage);

   void disconnect(void);


   bool publish(String topic, String payload);


   bool publish(String topic, const uint8_t *payload, uint32_t plength, bool retained = false);


   bool publish(String topic, MQTT::payload_callback_t pcb, uint32_t length, bool retained = false);


   bool publish_P(String topic, PGM_P payload, uint32_t plength, bool retained = false);


   bool subscribe(String topic, uint8_t qos = 0);

   bool unsubscribe(String topic);


   bool loop();

   bool connected();

   bool connect(MQTT::Connect &conn);
   bool publish(MQTT::Publish &pub);
   bool subscribe(MQTT::Subscribe &sub);
   bool unsubscribe(MQTT::Unsubscribe &unsub);
};


#endif