forked from earlephilhower/psychoplug
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mqtt.cpp
123 lines (104 loc) · 3.43 KB
/
mqtt.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
PsychoPlug
ESP8266 based remote outlet with standalone timer and MQTT integration
Copyright (C) 2017 Earle F. Philhower, III
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <Arduino.h>
#include <ESP8266WiFi.h>
#include <MQTTClient.h>
#include "psychoplug.h"
#include "log.h"
#include "mqtt.h"
#include "settings.h"
#include "relay.h"
// MQTT interface
static WiFiClient *wifiMQTT = NULL;
static MQTTClient mqttClient;
// Callback for the MQTT library
void messageReceived(String& topic, String& payload)
{
char t[64], p[32];
topic.toCharArray(t, sizeof(t));
payload.toCharArray(p, sizeof(p));
LogPrintf("MQTT: '%s'='%s'\n", t, p);
char topicStr[64];
snprintf_P(topicStr, sizeof(topicStr), PSTR("%s/remotepower"), settings.mqttTopic);
if (!strcmp(topicStr, t)) {
if (!strcasecmp_P(p, PSTR("on")) || !strcmp_P(p, PSTR("1")))
SetRelay(true);
else if (!strcasecmp_P(p, PSTR("off")) || !strcmp_P(p, PSTR("0")))
SetRelay(false);
else if (!strcasecmp_P(p, PSTR("toggle")) )
SetRelay(!GetRelay());
}
}
void StartMQTT()
{
LogPrintf("Starting MQTT\n");
LogPrintf("Free heap = %d\n", ESP.getFreeHeap());
LogPrintf("Connecting MQTT...\n");
if (settings.mqttEnable) {
if (settings.mqttSSL) wifiMQTT = new WiFiClientSecure();
else wifiMQTT = new WiFiClient();
mqttClient.begin(settings.mqttHost, settings.mqttPort, *wifiMQTT);
mqttClient.onMessage(messageReceived);
mqttClient.connect(settings.mqttClientID, settings.mqttUser, settings.mqttPass);
if (mqttClient.connected() ) {
char topic[64];
snprintf_P(topic, sizeof(topic), PSTR("%s/remotepower"), settings.mqttTopic);
mqttClient.subscribe(topic);
LogPrintf("Subscribed to:", topic);
}
}
LogPrintf("Free heap = %d after connection\n", ESP.getFreeHeap());
}
void ManageMQTT()
{
if (!settings.mqttEnable) return;
// Only have MQTT loop if we're connected and configured
if (mqttClient.connected()) {
mqttClient.loop();
delay(10);
} else {
LogPrintf("MQTT disconnected, reconnecting\n");
mqttClient.connect(settings.mqttClientID, settings.mqttUser, settings.mqttPass);
if (mqttClient.connected() ) {
char topic[64];
snprintf_P(topic, sizeof(topic), PSTR("%s/remotepower"), settings.mqttTopic);
mqttClient.subscribe(topic);
}
delay(10);
}
}
void StopMQTT()
{
if (settings.mqttEnable) {
wifiMQTT->flush();
wifiMQTT->stop();
mqttClient.disconnect();
}
}
void MQTTPublish(const char *key, const char *value)
{
if (isSetup && settings.mqttEnable && mqttClient.connected()) {
char topic[64];
snprintf_P(topic, sizeof(topic), PSTR("%s/%s"), settings.mqttTopic, key);
mqttClient.publish(topic, value);
}
}
void MQTTPublishInt(const char *key, const int value)
{
char str[16];
snprintf_P(str, sizeof(str), PSTR("%d"), value);
MQTTPublish(key, str);
}