From 735e2d47b4ccd8fb8da8ae9fa13f24fa56a334c9 Mon Sep 17 00:00:00 2001
From: Nicolas MASSE <nicolas.masse@itix.fr>
Date: Sun, 27 Mar 2022 14:59:44 +0200
Subject: [PATCH] add fw upgrade & reboot ota commands

---
 main/Kconfig.projbuild |  6 ++++
 main/mqtt.c            | 65 ++++++++++++++++++++++++++++++++++++++++++
 sdkconfig.dev          |  1 +
 3 files changed, 72 insertions(+)

diff --git a/main/Kconfig.projbuild b/main/Kconfig.projbuild
index 49bdef0..2154ff2 100644
--- a/main/Kconfig.projbuild
+++ b/main/Kconfig.projbuild
@@ -53,6 +53,12 @@ menu "TIC-to-MQTT Configuration"
         help
             Topic where to send LWT (Last Will Testament) upon disconnect.
 
+    config MQTT_COMMAND_TOPIC
+        string "MQTT Last Will Testament Topic"
+        default "esp-tic/command"
+        help
+            Topic where to send system commands.
+
     config MQTT_TIC_VALUE_TOPIC
         string "Where to send TIC values"
         default "esp-tic/status/tic/%s"
diff --git a/main/mqtt.c b/main/mqtt.c
index 6150154..78be8a7 100644
--- a/main/mqtt.c
+++ b/main/mqtt.c
@@ -16,6 +16,7 @@
 #include <time.h>
 #include "cJSON.h"
 #include "common.h"
+#include "ota.h"
 
 static esp_mqtt_client_config_t mqtt_cfg;
 static esp_mqtt_client_handle_t client;
@@ -32,6 +33,9 @@ static const char *MQTT_LOGGER = "mqtt";
 #define JSON_BUFFER_SIZE 128
 #define MQTT_TOPIC_COMPONENT_SIZE 32
 
+#define SYSTEM_COMMAND_UPDATE "firmware-update"
+#define SYSTEM_COMMAND_REBOOT "reboot"
+
 void mqtt_publish_data(char* key, char* value) {
     char topic[sizeof(CONFIG_MQTT_TIC_VALUE_TOPIC) + MQTT_TOPIC_COMPONENT_SIZE];
     char payload[JSON_BUFFER_SIZE];
@@ -76,6 +80,55 @@ void mqtt_publish_data(char* key, char* value) {
     }
 }
 
+esp_err_t mqtt_process_system_command(char* data, int data_len) {
+    esp_err_t status = ESP_OK;
+    cJSON *json = cJSON_ParseWithLength(data, data_len);
+    if (json == NULL) {
+        ESP_LOGI(MQTT_LOGGER, "Error parsing MQTT system command as JSON");
+        status = ESP_FAIL;
+        goto end;
+    }
+
+    cJSON* command = cJSON_GetObjectItemCaseSensitive(json, "command");
+    if (!cJSON_IsString(command) || (command->valuestring == NULL)) {
+        ESP_LOGD(MQTT_LOGGER, "Expected a command name!");
+        status = ESP_FAIL;
+        goto end;
+    }
+
+    if (strcmp(command->valuestring, SYSTEM_COMMAND_REBOOT) == 0) {
+        ESP_LOGE(MQTT_LOGGER, "Received a reboot command. Rebooting now!");
+        esp_restart();
+        goto end;
+    }
+
+    if (strcmp(command->valuestring, SYSTEM_COMMAND_UPDATE) == 0) {
+        cJSON* args = cJSON_GetObjectItemCaseSensitive(json, "args");
+        if (!cJSON_IsObject(args)) {
+            ESP_LOGD(MQTT_LOGGER, "Expected a command argument!");
+            status = ESP_FAIL;
+            goto end;
+        }
+
+        cJSON* version = cJSON_GetObjectItemCaseSensitive(args, "version");
+        if (!cJSON_IsString(version) || (version->valuestring == NULL)) {
+            ESP_LOGD(MQTT_LOGGER, "Expected a version numer!");
+            status = ESP_FAIL;
+            goto end;
+        }
+
+        trigger_ota_update(version->valuestring);
+        goto end;
+    }
+
+    ESP_LOGW(MQTT_LOGGER, "Unknown system command %s!", command->valuestring);
+    status = ESP_FAIL;
+
+    end:
+    cJSON_Delete(json);
+    return status;
+}
+
 esp_err_t mqtt_event_handler_cb(esp_mqtt_event_handle_t event) {
     switch (event->event_id) {
         case MQTT_EVENT_CONNECTED:
@@ -84,6 +137,9 @@ esp_err_t mqtt_event_handler_cb(esp_mqtt_event_handle_t event) {
             if (esp_mqtt_client_publish(client, CONFIG_MQTT_LWT_TOPIC, "1", 0, MQTT_QOS_0, MQTT_RETAIN) == -1) {
                 ESP_LOGD(MQTT_LOGGER, "MQTT Message discarded!");
             }
+            if (esp_mqtt_client_subscribe(client, CONFIG_MQTT_COMMAND_TOPIC, MQTT_QOS_1) == -1) {
+                ESP_LOGD(MQTT_LOGGER, "Could not subscribe to " CONFIG_MQTT_COMMAND_TOPIC " MQTT topic");
+            }
             break;
         case MQTT_EVENT_DISCONNECTED:
             ESP_LOGI(MQTT_LOGGER, "MQTT_EVENT_DISCONNECTED");
@@ -105,6 +161,15 @@ esp_err_t mqtt_event_handler_cb(esp_mqtt_event_handle_t event) {
                 ESP_LOGW(MQTT_LOGGER, "Unknown error type: 0x%x", event->error_handle->error_type);
             }
             break;
+        case MQTT_EVENT_DATA:
+            if (strncmp(event->topic, CONFIG_MQTT_COMMAND_TOPIC, event->topic_len) == 0) {
+                ESP_LOGD(MQTT_LOGGER, "Received an MQTT system command!");
+                mqtt_process_system_command(event->data, event->data_len);
+            }
+            break;
+        case MQTT_EVENT_SUBSCRIBED:
+            // Expected event. Nothing to do.
+            break;
         default:
             ESP_LOGD(MQTT_LOGGER, "Other event id:%d", event->event_id);
             break;
diff --git a/sdkconfig.dev b/sdkconfig.dev
index 8390112..49e7622 100644
--- a/sdkconfig.dev
+++ b/sdkconfig.dev
@@ -1,6 +1,7 @@
 CONFIG_TIC_UART_RXD=16
 CONFIG_MQTT_LWT_TOPIC="test/esp-tic/connected"
 CONFIG_MQTT_TIC_VALUE_TOPIC="test/esp-tic/status/tic/%s"
+CONFIG_MQTT_COMMAND_TOPIC="test/esp-tic/command"
 CONFIG_MBEDTLS_CERTIFICATE_BUNDLE_DEFAULT_CMN=y
 CONFIG_ESPTOOLPY_FLASHSIZE_4MB=y
 CONFIG_PARTITION_TABLE_TWO_OTA=y