Made some requested changes from reviews

This commit is contained in:
RobinVdB8
2023-11-26 13:33:47 +01:00
parent b50900e713
commit 1550ccd0ba
3 changed files with 91 additions and 86 deletions

View File

@@ -10,6 +10,6 @@
/** /**
* @brief Initialize MQTT application * @brief Initialize MQTT application
*/ */
void mqtt_application_init(void); uint8_t mqtt_application_init(void);
#endif /* INC_MQTTA_H_ */ #endif /* INC_MQTTA_H_ */

View File

@@ -28,7 +28,7 @@
#include "log.h" #include "log.h"
#include "llfs.h" #include "llfs.h"
#include "lcd_api.h" #include "lcd_api.h"
#include <mqtt_application.h> #include "mqtt_application.h"
#include "tftp.h" #include "tftp.h"
/* USER CODE END Includes */ /* USER CODE END Includes */

View File

@@ -4,7 +4,6 @@
* @author RobinVdB * @author RobinVdB
*/ */
#include <mqtt_application.h>
#include <string.h> #include <string.h>
#include "httpd.h" #include "httpd.h"
#include "lcd_api.h" #include "lcd_api.h"
@@ -12,6 +11,7 @@
#include "lwip/ip_addr.h" #include "lwip/ip_addr.h"
#include "mdns.h" #include "mdns.h"
#include "mqtt.h" #include "mqtt.h"
#include "mqtt_application.h"
#define LOGGER_LEVEL_INFO #define LOGGER_LEVEL_INFO
#include "log.h" #include "log.h"
@@ -64,9 +64,7 @@ static const char* TAG = "MQTT";
* @param[in] result Whether the publish was successful or not * @param[in] result Whether the publish was successful or not
*/ */
static void mqtt_pub_request_cb(void* arg, err_t result) { static void mqtt_pub_request_cb(void* arg, err_t result) {
if (result != ERR_OK) { LOG_DEBUG(TAG, "Publish result: %d", result);
LOG_DEBUG(TAG, "Publish result: %d", result);
}
} }
/** /**
@@ -79,15 +77,14 @@ static void mqtt_pub_request_cb(void* arg, err_t result) {
static void publish_data(mqtt_client_t* client, void* arg) { static void publish_data(mqtt_client_t* client, void* arg) {
char pub_payload[200] = {0}; char pub_payload[200] = {0};
err_t err; err_t err;
size_t max_files = MAX_FILES;
size_t num_files; size_t num_files;
llfs_file_t file_list[max_files]; llfs_file_t file_list[MAX_FILES];
u8_t qos = PUBLISH_QOS; u8_t qos = PUBLISH_QOS;
u8_t retain = PUBLISH_RETAIN; u8_t retain = PUBLISH_RETAIN;
LOG_DEBUG(TAG, "Entering publish"); LOG_DEBUG(TAG, "Entering publish");
num_files = llfs_file_list(file_list, max_files, "*.bmp"); num_files = llfs_file_list(file_list, MAX_FILES, "*.bmp");
if (num_files == 0) { if (num_files == 0) {
strncpy(pub_payload, "No images found", sizeof(pub_payload)); strncpy(pub_payload, "No images found", sizeof(pub_payload));
@@ -108,12 +105,12 @@ static void publish_data(mqtt_client_t* client, void* arg) {
pub_payload[0] = '\0'; pub_payload[0] = '\0';
num_files = llfs_file_list(file_list, max_files, "*.gif"); num_files = llfs_file_list(file_list, MAX_FILES, "*.gif");
if (num_files == 0) { if (num_files == 0) {
strcpy(pub_payload, "No gifs found"); strncpy(pub_payload, "No gifs found", sizeof(pub_payload));
} else { } else {
strcpy(pub_payload, "Available gifs: "); strncpy(pub_payload, "Available gifs: ", sizeof(pub_payload));
for (size_t i = 0; i < num_files; i++) { for (size_t i = 0; i < num_files; i++) {
// Concatenate file names into the payload string // Concatenate file names into the payload string
strncat(pub_payload, file_list[i].name, sizeof(pub_payload) - strlen(pub_payload) - 1); strncat(pub_payload, file_list[i].name, sizeof(pub_payload) - strlen(pub_payload) - 1);
@@ -140,16 +137,22 @@ static void mqtt_incoming_publish_cb(void* arg, const char* topic, uint32_t tot_
// Check for which topic a publish was received // Check for which topic a publish was received
if (strcmp(topic, "input/setText") == 0) { if (strcmp(topic, "input/setText") == 0) {
inpub_id = set_text; inpub_id = set_text;
} else if (strcmp(topic, "input/setImage") == 0) { return;
inpub_id = set_image;
} else if (strcmp(topic, "input/setTextColor") == 0) {
inpub_id = set_text_color;
} else if (strcmp(topic, "input/setColor") == 0) {
inpub_id = set_color;
} else {
// In case of wrong topic
inpub_id = other_topic;
} }
if (strcmp(topic, "input/setImage") == 0) {
inpub_id = set_image;
return;
}
if (strcmp(topic, "input/setTextColor") == 0) {
inpub_id = set_text_color;
return;
}
if (strcmp(topic, "input/setColor") == 0) {
inpub_id = set_color;
return;
}
// In case of wrong topic
inpub_id = other_topic;
} }
/** /**
@@ -165,46 +168,47 @@ static void mqtt_incoming_data_cb(void* arg, const uint8_t* data, uint16_t len,
char data_buffer[len + 1]; char data_buffer[len + 1];
lcd_gif_t* gif; lcd_gif_t* gif;
LOG_INFO(TAG, "Incoming publish payload with length %d, flags %u", len, (unsigned int)flags); LOG_INFO(TAG, "Incoming publish payload with length %d, flags %d", len, flags);
if (flags & MQTT_DATA_FLAG_LAST) { if (!(flags & MQTT_DATA_FLAG_LAST)) {
memcpy(data_buffer, data, len); LOG_WARN(TAG, "incoming data too big to fit in buffer.");
data_buffer[len] = '\0'; return;
switch (inpub_id) { }
case set_text: memcpy(data_buffer, data, len);
// Places text on the lcd data_buffer[len] = '\0';
LOG_INFO(TAG, "incoming data on input/setText: %s.", data_buffer); switch (inpub_id) {
lcd_clear_text(); case set_text:
lcd_display_text((const char*)data_buffer, xpos, ypos, color, bgcolor, font); // Places text on the lcd
break; LOG_INFO(TAG, "incoming data on input/setText: %s.", data_buffer);
case set_image: lcd_clear_text();
// Places an image on the lcd lcd_display_text((const char*)data_buffer, xpos, ypos, color, bgcolor, font);
LOG_INFO(TAG, "incoming data on input/setImage: %s.", data_buffer); break;
lcd_clear_images(); case set_image:
lcd_set_bg_color_layer0(bgcolor); // Places an image on the lcd
if (data_buffer[len - 3] == 'b') { LOG_INFO(TAG, "incoming data on input/setImage: %s.", data_buffer);
lcd_draw_img_from_fs((const char*)data_buffer, xpos, ypos); lcd_clear_images();
lcd_set_bg_color_layer0(bgcolor);
if (data_buffer[len - 3] == 'b') {
lcd_draw_img_from_fs((const char*)data_buffer, xpos, ypos);
}
if (data_buffer[len - 3] == 'g') {
gif = lcd_draw_gif_from_fs((const char*)data_buffer, xpos, ypos);
if (gif == NULL) {
LOG_INFO(TAG, "GIF could not be drawn");
} }
if (data_buffer[len - 3] == 'g') { }
gif = lcd_draw_gif_from_fs((const char*)data_buffer, xpos, ypos); break;
if (gif == NULL) { case set_text_color:
LOG_INFO(TAG, "GIF could not be drawn"); // Changes the text color for the next time text is written
} LOG_INFO(TAG, "incoming data on input/setTextColor: %s.", data_buffer);
} color = color_picker(data_buffer);
break; break;
case set_text_color: case set_color:
// Changes the text color for the next time text is written // Changes the background color for the next time text is written
LOG_INFO(TAG, "incoming data on input/setTextColor: %s.", data_buffer); LOG_INFO(TAG, "incoming data on input/setColor: %s.", data_buffer);
color = color_picker(data_buffer); bgcolor = color_picker(data_buffer);
break; break;
case set_color: default:
// Changes the background color for the next time text is written LOG_INFO(TAG, "Publish received on wrong topic, incoming data ignored.");
LOG_INFO(TAG, "incoming data on input/setColor: %s.", data_buffer);
bgcolor = color_picker(data_buffer);
break;
default:
LOG_INFO(TAG, "Publish received on wrong topic, incoming data ignored.");
}
} else {
} }
} }
@@ -230,22 +234,7 @@ static void mqtt_sub_request_cb(void* arg, err_t result) {
static void mqtt_connection_cb(mqtt_client_t* client, void* arg, mqtt_connection_status_t status) { static void mqtt_connection_cb(mqtt_client_t* client, void* arg, mqtt_connection_status_t status) {
err_t err; err_t err;
if (status == MQTT_CONNECT_ACCEPTED) { if (status != MQTT_CONNECT_ACCEPTED) {
LOG_INFO(TAG, "Successfully connected");
connection_attempt_counter = 0;
// Set up callback function for input
mqtt_set_inpub_callback(client, mqtt_incoming_publish_cb, mqtt_incoming_data_cb, arg);
// Subscribe to the topics setText, setImage, setColor and setTextcolor
err = mqtt_subscribe(client, "input/#", 1, mqtt_sub_request_cb, arg);
if (err != ERR_OK) {
LOG_DEBUG(TAG, "mqtt_subscribe return: %d", err);
}
// Publish list of images here
publish_data(client, NULL);
} else {
LOG_INFO(TAG, "mqtt_connection_cb: Disconnected, reason: %d", status); LOG_INFO(TAG, "mqtt_connection_cb: Disconnected, reason: %d", status);
while (connection_attempt_counter < ATTEMPT_RECONNECT_AMOUNT) { while (connection_attempt_counter < ATTEMPT_RECONNECT_AMOUNT) {
@@ -253,7 +242,22 @@ static void mqtt_connection_cb(mqtt_client_t* client, void* arg, mqtt_connection
// Try to reconnect // Try to reconnect
mosquitto_connect(client); mosquitto_connect(client);
} }
return;
} }
LOG_INFO(TAG, "Successfully connected");
connection_attempt_counter = 0;
// Set up callback function for input
mqtt_set_inpub_callback(client, mqtt_incoming_publish_cb, mqtt_incoming_data_cb, arg);
// Subscribe to the topics setText, setImage, setColor and setTextcolor
err = mqtt_subscribe(client, "input/#", 1, mqtt_sub_request_cb, arg);
if (err != ERR_OK) {
LOG_DEBUG(TAG, "mqtt_subscribe return: %d", err);
}
// Publish list of images here
publish_data(client, NULL);
} }
/** /**
@@ -277,17 +281,16 @@ static void mosquitto_connect(mqtt_client_t* client) {
err = mqtt_client_connect(client, &server_ip, server_port, mqtt_connection_cb, 0, &ci); err = mqtt_client_connect(client, &server_ip, server_port, mqtt_connection_cb, 0, &ci);
if (err != ERR_OK) { if (err != ERR_OK) {
LOG_DEBUG(TAG, "mqtt_connect return %d", err); LOG_DEBUG(TAG, "mqtt_connect return %d", err);
return;
} }
if (err == ERR_OK) { LOG_DEBUG(TAG, "Went into mqtt_client_connect; mqtt_connect return %d", err);
LOG_DEBUG(TAG, "Went into mqtt_client_connect; mqtt_connect return %d", err);
}
} }
/** /**
* @brief Init function for the mosquitto application of the assignment * @brief Init function for the mosquitto application of the assignment
* Gives the global variables a value and calls the mosquitto_connect function * Gives the global variables a value and calls the mosquitto_connect function
*/ */
void mqtt_application_init(void) { uint8_t mqtt_application_init(void) {
color = LCD_BLACK; color = LCD_BLACK;
bgcolor = LCD_WHITE; bgcolor = LCD_WHITE;
font = LCD_FONT16; font = LCD_FONT16;
@@ -296,10 +299,14 @@ void mqtt_application_init(void) {
connection_attempt_counter = 0; connection_attempt_counter = 0;
mqtt_client_t* client = mqtt_client_new(); mqtt_client_t* client = mqtt_client_new();
if (client != NULL) { if (client == NULL) {
LOG_DEBUG(TAG, "Starting connection test"); LOG_CRIT(TAG, "%s: client == NULL", __func__);
mosquitto_connect(client); return 1;
} }
LOG_DEBUG(TAG, "Starting connection test");
mosquitto_connect(client);
return 0;
} }
/** /**
@@ -309,8 +316,6 @@ void mqtt_application_init(void) {
* @return color Define to use with the LCD_API * @return color Define to use with the LCD_API
*/ */
uint32_t color_picker(char* color_in) { uint32_t color_picker(char* color_in) {
uint32_t output = LCD_BLACK;
if (strcmp((const char*)color_in, "blue") == 0) { if (strcmp((const char*)color_in, "blue") == 0) {
return LCD_BLUE; return LCD_BLUE;
} }
@@ -390,5 +395,5 @@ uint32_t color_picker(char* color_in) {
return LCD_TRANSPARENT; return LCD_TRANSPARENT;
} }
return output; return LCD_BLACK;
} }