#ifndef MQTT_H
#define MQTT_H

#if defined(__cplusplus)
extern "C" {
#endif

#include <stdlib.h>

typedef enum MqttConnectionStatus
{
    MqttConnectionAccepted = 0,
    MqttConnectionInvalidProtocolVersion,
    MqttConnectionIdentifierRejected,
    MqttConnectionServerUnavailable,
    MqttConnectionBadAuth,
    MqttConnectionNotAuthorized
} MqttConnectionStatus;

typedef enum MqttSubscriptionStatus
{
    MqttSubscriptionQos0 = 0,
    MqttSubscriptionQos1 = 1,
    MqttSubscriptionQos2 = 2,
    MqttSubscriptionFailure = 0x80
} MqttSubscriptionStatus;

typedef struct MqttClient MqttClient;

typedef void (*MqttClientOnConnectCallback)(MqttClient *client,
                                            MqttConnectionStatus status,
                                            int sessionPresent);

typedef void (*MqttClientOnSubscribeCallback)(MqttClient *client,
                                              int id,
                                              MqttSubscriptionStatus status);

typedef void (*MqttClientOnUnsubscribeCallback)(MqttClient *client, int id);

typedef void (*MqttClientOnMessageCallback)(MqttClient *client,
                                            const char *topic,
                                            const void *data, size_t size);

typedef void (*MqttClientOnPublishCallback)(MqttClient *client, int id);

MqttClient *MqttClientNew(const char *clientId, int cleanSession);

void MqttClientFree(MqttClient *client);

void MqttClientSetUserData(MqttClient *client, void *userData);

void *MqttClientGetUserData(MqttClient *client);

void MqttClientSetOnConnect(MqttClient *client, MqttClientOnConnectCallback cb);

void MqttClientSetOnSubscribe(MqttClient *client,
                              MqttClientOnSubscribeCallback cb);

void MqttClientSetOnUnsubscribe(MqttClient *client,
                                MqttClientOnUnsubscribeCallback cb);

void MqttClientSetOnMessage(MqttClient *client,
                            MqttClientOnMessageCallback cb);

void MqttClientSetOnPublish(MqttClient *client,
                            MqttClientOnPublishCallback cb);

int MqttClientConnect(MqttClient *client, const char *host, short port,
                      int keepAlive);

int MqttClientDisconnect(MqttClient *client);

int MqttClientRunOnce(MqttClient *client);

int MqttClientRun(MqttClient *client);

int MqttClientSubscribe(MqttClient *client, const char *topicFilter,
                        int qos);

int MqttClientUnsubscribe(MqttClient *client, const char *topicFilter);

int MqttClientPublish(MqttClient *client, int qos, int retain,
                      const char *topic, const void *data, size_t size);

int MqttClientPublishCString(MqttClient *client, int qos, int retain,
                             const char *topic, const char *msg);

void MqttClientSetPublishRetryTimeout(MqttClient *client, int timeout);

void MqttClientSetMaxMessagesInflight(MqttClient *client, int max);

void MqttClientSetMaxQueuedMessages(MqttClient *client, int max);

#if defined(__cplusplus)
}
#endif

#endif