espleaf/src/mqtt.cpp

94 lines
3.3 KiB
C++

#include "mqtt.h"
#include "tools.h"
#include "prototypes.h"
void connect() {
LOGln("Connecting to MQTT broker...");
if (mqtt.connect(MQTT_CLIENT_ID, MQTT_USER, MQTT_PASS, MQTT_TOPIC_STATE, 0, true, "OFFLINE")) {
LOGln("Connected.");
mqtt.publish(MQTT_TOPIC_STATE, "ONLINE", true);
char buffer[40];
snprintf(buffer, 40, "ONLINE %s", wifi.localIP().toString().c_str());
mqtt.publish(MQTT_TOPIC_STATE_LONG, buffer, true);
mqtt.subscribe(MQTT_TOPIC_COMMANDS);
} else {
LOGln("Connection failed. Reason: %d", mqtt.state());
delay(1000);
}
}
void callback(char* topic, byte* pl, unsigned int length) {
pl[length] = 0;
String payload((char*)pl);
// Syntax: key=value&key2=value2...
uint16_t duration = 0;
AnimationMode new_mode = AM_NONE;
String current_part;
LOGln("Received command %s", payload.c_str());
while (payload.length() > 0) {
int offset = payload.indexOf("&");
if (offset != -1) {
current_part = payload.substring(0, offset);
payload = payload.substring(offset + 1);
} else {
current_part = payload;
payload = "";
}
offset = current_part.indexOf("=");
if (offset==-1) {
LOGln("Parameter without '=' detected: %s", current_part.c_str());
continue;
}
String key = current_part.substring(0, offset);
String value = current_part.substring(offset+1);
LOGln(" Processing key %s with value %s", key.c_str(), value.c_str());
if (key.equals("mode")) {
if (value.equals("nodes")) { new_mode = AM_NODES; }
else if (value.equals("first_node")) { new_mode = AM_FIRST_NODE; }
else if (value.equals("corners")) { new_mode = AM_CORNERS; }
else if (value.equals("first_corner")) { new_mode = AM_FIRST_CORNER; }
else if (value.equals("off")) { new_mode = AM_OFF; }
else if (value.equals("flash")) { new_mode = AM_FLASH; }
else { LOGln("Unknown mode '%s'.", value.c_str()); }
} else if (key.equals("duration")) {
duration = value.toInt();
} else if (key.equals("brightness")) {
if (temp_mode_until == 0) {
return_to_brightness = FastLED.getBrightness();
}
FastLED.setBrightness(value.toInt());
} else if (key.equals("color")) {
if (value.equals("red")) { color = CRGB::Red; }
else if (value.equals("green")) { color = CRGB::Green; }
else if (value.equals("blue")) { color = CRGB::Blue; }
else { LOGln("Unknown color name %s.", value.c_str());}
} else {
LOGln("Unknown key '%s'. (For reference: Value is '%s'.)", key.c_str(), value.c_str());
}
}
LOGln("Finished processing the command.");
if (new_mode != AM_NONE) {
if (duration > 0) {
temp_mode = new_mode;
temp_mode_until = millis() + duration*1000;
} else {
mode = new_mode;
}
}
}
void mqtt_setup() {
mqtt.setServer(MQTT_SERVER, MQTT_SERVER_PORT);
mqtt.setCallback(callback);
mqtt.setSocketTimeout(1);
connect();
}
void mqtt_loop() {
if (!mqtt.connected()) {
connect();
}
mqtt.loop();
}