Arduino/ESP8266-MQTT-MagicBox/ESP8266-MQTT-MagicBox.ino
2017-09-26 01:11:50 +02:00

253 lines
6.4 KiB
C++

/**********************************************************************
* The MIT License (MIT)
*
* Copyright (c) 2015 Nis Wechselberg
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
**********************************************************************/
/**********************************************************************
* ESP8266-MQTT-MagicBox
*
* Links the ESP8266 based control board to an mqtt broker.
* The program uses multiple topics for incoming and outgoing messages.
* Incoming topics:
* - <mqtt_device>/port
* Outgoing topics:
* - <mqtt_device>/portState
**********************************************************************/
#include <ESP8266WiFi.h>
#include <ESP8266mDNS.h>
#include <WiFiUdp.h>
#include <ArduinoOTA.h>
#include <PubSubClient.h>
#include "ESP8266-MQTT-MagicBox.h"
#define DEBUGTOSERIAL 1
// Timestamp for last publish
unsigned long lastMsg = 0;
long publishInterval = 5000;
// Message buffer
char msg[50];
// Last known power state
uint8_t portState = 0;
uint32_t codes[] = {0xFFAA55, 0xFFA857, 0xFFBA45, 0xFF38C7};
// Pin config
const uint8_t irSendPin = 12;
const uint8_t probe1 = 13;
const uint8_t probe2 = 14;
const uint8_t probe3 = 16;
const uint8_t probe4 = 5;
// Global mqtt client object
WiFiClient espClient;
PubSubClient client(espClient);
/*
* Initial setup for arduino
*/
void setup() {
// Configure pins
pinMode(irSendPin, OUTPUT);
digitalWrite(irSendPin, HIGH);
// Configure serial port
Serial.begin(115200);
delay(10);
// Prepare WiFi connection
setup_wifi();
// Connect to mqtt broker
client.setServer(mqtt_server, 1883);
client.setCallback(incoming_mqtt);
}
/*
* Prepares the wireless connection
*/
void setup_wifi() {
// Connect to the WiFi as a client
WiFi.mode(WIFI_STA);
// Do the connection
if (DEBUGTOSERIAL) {
Serial.println();
Serial.print("Connecting to ");
Serial.println(ssid);
}
WiFi.begin(ssid, password);
while (WiFi.status() != WL_CONNECTED) {
delay(500);
if (DEBUGTOSERIAL) {
Serial.print(".");
}
}
if (DEBUGTOSERIAL) {
Serial.println("");
Serial.println("WiFi connected");
// Print IP address to serial
Serial.print("My IP address: ");
Serial.println(WiFi.localIP());
}
ArduinoOTA.setHostname(ota_hostname);
ArduinoOTA.setPassword(ota_hostname);
ArduinoOTA.begin();
}
/*
* Callback method for incoming mqtt messages
*/
void incoming_mqtt(char* topic, byte* payload, unsigned int length) {
if (DEBUGTOSERIAL) {
Serial.print("Message arrived [");
Serial.print(topic);
Serial.print("] ");
for (unsigned int i = 0; i < length; i++) {
Serial.print((char)payload[i]);
}
Serial.println();
}
if (strcmp(topic, mqtt_topic_port) == 0) {
if (length == 1) {
if (DEBUGTOSERIAL) {
Serial.println("Setting new port active");
}
uint8_t desiredPortState = 0;
char request = (char) payload[0];
if (request == '1' || request == '2' || request == '3' || request == '4') {
desiredPortState = request - '0';
}
if (desiredPortState != 0 && desiredPortState != portState) {
changePort(desiredPortState);
}
}
}
}
void changePort(uint8_t newPort) {
if (newPort > 0 && newPort < 5) {
replicateNEC(irSendPin, codes[newPort - 1], 32);
}
}
void replicateNEC(int pin, uint32_t data, int dataLength) {
if (DEBUGTOSERIAL) {
Serial.print("Sending code ");
Serial.println(data, HEX);
}
// Prepare mask
uint32_t mask = 1L << (dataLength -1);
// Write initial 9ms+4.5ms pulse
digitalWrite(pin, LOW);
delayMicroseconds(9000);
digitalWrite(pin, HIGH);
delayMicroseconds(4500);
// Write pulses from data
while (mask > 0) {
digitalWrite(pin, LOW);
delayMicroseconds(562);
digitalWrite(pin, HIGH);
delayMicroseconds(562);
if (data & mask) {
delayMicroseconds(1124);
}
mask = mask >> 1;
}
// Write trailing pulse
digitalWrite(pin, LOW);
delayMicroseconds(562);
digitalWrite(pin, HIGH);
}
/*
* Blocking reconnect to the mqtt broker
*/
void reconnect() {
// Loop until we're reconnected
while (!client.connected()) {
if (DEBUGTOSERIAL) {
Serial.print("Attempting MQTT connection...");
}
// Attempt to connect
if (client.connect(mqtt_device)) {
if (DEBUGTOSERIAL) {
Serial.println("connected");
}
// subscribe to incoming topics
client.subscribe(mqtt_topic_port);
} else {
if (DEBUGTOSERIAL) {
Serial.print("failed, rc=");
Serial.print(client.state());
Serial.println(" try again in 5 seconds");
}
// Wait 5 seconds before retrying
delay(5000);
}
}
}
void updatePortState() {
// Read value from digital pin (Levels are inverted)
if (digitalRead(probe1)) portState = 1;
if (digitalRead(probe2)) portState = 2;
if (digitalRead(probe3)) portState = 3;
if (digitalRead(probe4)) portState = 4;
if (DEBUGTOSERIAL) {
Serial.print("Output port ");
Serial.print(portState);
Serial.println(" selected");
}
snprintf(msg, 49, "%d", portState);
client.publish(mqtt_topic_portState, msg);
}
void loop() {
// Check OTA update
ArduinoOTA.handle();
// Ensure MQTT connection
if (!client.connected()) {
reconnect();
}
// Check the inbox
client.loop();
// Maybe push the current status
long now = millis();
if (now - lastMsg > publishInterval) {
lastMsg = now;
updatePortState();
}
}