Copied from sourcehut.

1 files changed, 69 insertions, 0 deletions

diff --git a/main/mqtt.c b/main/mqtt.c
new file mode 100644
index 0000000..6a06aea
--- /dev/null
+++ b/main/mqtt.c
@@ -0,0 +1,69 @@
+#include <freertos/FreeRTOS.h>
+#include <freertos/task.h>
+#include "freertos/event_groups.h"
+
+#include <esp_event.h>
+#include <esp_netif.h>
+#include <esp_log.h>
+
+#include "net.h"
+
+#define MQTT_CONNECTED_BIT BIT0
+
+static const char *tag = "mqtt";
+
+static EventGroupHandle_t mqtt_evt_group;
+
+static void mqtt_evt_handler(void *args, esp_event_base_t eb, int32_t evt_id, void *data)
+{
+	esp_mqtt_event_handle_t evt = (esp_mqtt_event_handle_t) data;
+	switch (evt_id) {
+		case MQTT_EVENT_CONNECTED:
+			ESP_LOGI(tag, "MQTT_EVENT_CONNECTED");
+			xEventGroupSetBits(mqtt_evt_group, MQTT_CONNECTED_BIT);
+			break;
+		case MQTT_EVENT_PUBLISHED:
+			ESP_LOGI(tag, "MQTT_EVENT_PUBLISHED, msg_id=%d", evt->msg_id);	
+			break;
+		default:
+			ESP_LOGI(tag, "MQTT event id:%d", evt->event_id);
+			break;
+	}
+}
+
+esp_mqtt_client_handle_t mqtt_connect(void)
+{
+	esp_mqtt_client_handle_t client = NULL;
+
+	const esp_mqtt_client_config_t cfg = {
+		.host = CONFIG_BROKER_HOST,
+		.username = CONFIG_BROKER_USERNAME,
+		.password = CONFIG_BROKER_PASSWORD,
+		.client_id = CONFIG_BROKER_CLIENT_ID,
+		.transport = MQTT_TRANSPORT_OVER_SSL,
+		.protocol_ver = MQTT_PROTOCOL_V_3_1_1
+	};
+
+	if (!(client = esp_mqtt_client_init(&cfg))) {
+		ESP_LOGI(tag, "esp_mqtt_client_init() failed");
+		return NULL;
+	}
+	
+	ESP_LOGI(tag, "MQTT broker: %s", cfg.host);
+
+	esp_mqtt_client_register_event(client,
+		ESP_EVENT_ANY_ID,
+		mqtt_evt_handler,
+		client);
+
+	mqtt_evt_group = xEventGroupCreate();
+	ESP_ERROR_CHECK(esp_mqtt_client_start(client));
+
+	xEventGroupWaitBits(mqtt_evt_group,
+		MQTT_CONNECTED_BIT,
+		pdFALSE,
+		pdFALSE,
+		portMAX_DELAY);
+
+	return client;
+}