Initial Commit

README.md

@@ -0,0 +1,6 @@
+Warning: messy code!
+
+This code is for an ESP8266 based IoT smartswitch, which connects to an MQTT server (via WiFi) and listens for instructions. 
+The code is based on the example Arduino code for MQTT communication from the Adafruit MQTT library.
+
+For this particular implementation, I used an ESP-01 module, and since it only gives 2 outputs (GPIO 0 and GPIO 2), I also used the serial TX/RX pins to give 2 more outputs. The serial code is still there, but commented. 

mqtts_esp8266_sw_quad.ino

@@ -0,0 +1,338 @@
+/***************************************************
+  Adafruit MQTT Library ESP8266 Example
+
+  Must use ESP8266 Arduino from:
+    https://github.com/esp8266/Arduino
+
+  Works great with Adafruit's Huzzah ESP board & Feather
+  ----> https://www.adafruit.com/product/2471
+  ----> https://www.adafruit.com/products/2821
+
+  Adafruit invests time and resources providing this open source code,
+  please support Adafruit and open-source hardware by purchasing
+  products from Adafruit!
+
+  Written by Tony DiCola for Adafruit Industries.
+  MIT license, all text above must be included in any redistribution
+ ****************************************************/
+#include <ESP8266WiFi.h>
+#include "Adafruit_MQTT.h"
+#include "Adafruit_MQTT_Client.h"
+
+/************************* WiFi Access Point *********************************/
+
+#define WLAN_SSID       "HOME1234"
+#define WLAN_PASS       "supersecurewifipassword"
+
+//#define led D4 //LoLin
+//#define led 2  //LoLin
+#define out2 2
+#define out1 0
+
+#define out3 3   //RX = 3
+#define out4 1   //TX = 1
+/************************* Adafruit.io Setup *********************************/
+
+#define AIO_SERVER      "mqtt.server.net"
+#define AIO_SERVERPORT  8883                   // use 8883 for SSL
+#define AIO_USERNAME    "username"
+#define AIO_KEY         "password"
+
+/************ Global State (you don't need to change this!) ******************/
+
+// Create an ESP8266 WiFiClient class to connect to the MQTT server.
+//WiFiClient client;
+// or... use WiFiClientSecure for SSL
+WiFiClientSecure client;
+
+// Setup the MQTT client class by passing in the WiFi client and MQTT server and login details.
+Adafruit_MQTT_Client mqtt(&client, AIO_SERVER, AIO_SERVERPORT, AIO_USERNAME, AIO_KEY);
+
+/****************************** Feeds ***************************************/
+
+// Setup feeds for publishing.
+// Notice MQTT paths for AIO follow the form: <username>/feeds/<feedname>
+Adafruit_MQTT_Publish statusfeed = Adafruit_MQTT_Publish(&mqtt, AIO_USERNAME "/feeds/status");
+Adafruit_MQTT_Publish out1status = Adafruit_MQTT_Publish(&mqtt, AIO_USERNAME "/feeds/out1");
+Adafruit_MQTT_Publish out2status = Adafruit_MQTT_Publish(&mqtt, AIO_USERNAME "/feeds/out2");
+Adafruit_MQTT_Publish out3status = Adafruit_MQTT_Publish(&mqtt, AIO_USERNAME "/feeds/out3");
+Adafruit_MQTT_Publish out4status = Adafruit_MQTT_Publish(&mqtt, AIO_USERNAME "/feeds/out4");
+
+Adafruit_MQTT_Publish mainstatusfeed = Adafruit_MQTT_Publish(&mqtt, "status"); //shared feed shows devices connecting
+
+// Setup a feed called 'onoff' for subscribing to changes.
+Adafruit_MQTT_Subscribe onoff1 = Adafruit_MQTT_Subscribe(&mqtt, AIO_USERNAME "/feeds/onoff1");
+Adafruit_MQTT_Subscribe onoff2 = Adafruit_MQTT_Subscribe(&mqtt, AIO_USERNAME "/feeds/onoff2");
+Adafruit_MQTT_Subscribe onoff3 = Adafruit_MQTT_Subscribe(&mqtt, AIO_USERNAME "/feeds/onoff3");
+Adafruit_MQTT_Subscribe onoff4 = Adafruit_MQTT_Subscribe(&mqtt, AIO_USERNAME "/feeds/onoff4");
+
+
+/*************************** Sketch Code ************************************/
+
+// Bug workaround for Arduino 1.6.6, it seems to need a function declaration
+// for some reason (only affects ESP8266, likely an arduino-builder bug).
+void MQTT_connect();
+
+char statusMessage[32] = "ESP-01 Switcher Status:      ";
+char out1stat, out2stat, out3stat, out4stat;
+uint8_t out1val, out2val, out3val, out4val;
+
+void setup() {
+
+  //********** CHANGE PIN FUNCTION  TO GPIO **********
+  //GPIO 1 (TX) swap the pin to a GPIO.
+ // pinMode(1, FUNCTION_3);
+  //GPIO 3 (RX) swap the pin to a GPIO.
+  //pinMode(3, FUNCTION_3);
+  //**************************************************
+pinMode(1, OUTPUT);
+pinMode(3, OUTPUT);
+
+  //pinMode(led, OUTPUT);
+  pinMode(out1, OUTPUT);
+  pinMode(out2, OUTPUT);
+  digitalWrite(out1, LOW); //switch to HIGH as needed, depending on what sort of device you have connected
+  digitalWrite(out2, LOW);
+  digitalWrite(out3, LOW);
+  digitalWrite(out4, LOW);
+
+  /*
+    digitalWrite(led, LOW);
+    delay(500);
+    digitalWrite(led, HIGH);
+    delay(500);
+    digitalWrite(led, LOW);
+    delay(500);
+    digitalWrite(led, HIGH);
+  */
+  //Serial.begin(115200);
+  delay(10);
+
+  //Serial.println(F("RuhNet MQTTS switcher"));
+
+  // Connect to WiFi access point.
+  //Serial.println(); Serial.println();
+  //Serial.print("Connecting to ");
+  //Serial.println(WLAN_SSID);
+
+  WiFi.begin(WLAN_SSID, WLAN_PASS);
+  while (WiFi.status() != WL_CONNECTED) {
+    delay(500);
+    //Serial.print(".");
+  }
+  //Serial.println();
+
+  //Serial.println("WiFi connected");
+  //Serial.println("IP address: "); Serial.println(WiFi.localIP());
+
+  // Setup MQTT subscription for onoff feed.
+  mqtt.subscribe(&onoff1);
+  mqtt.subscribe(&onoff2);
+  mqtt.subscribe(&onoff3);
+  mqtt.subscribe(&onoff4);
+
+}
+
+void loop() {
+  // Ensure the connection to the MQTT server is alive (this will make the first
+  // connection and automatically reconnect when disconnected).  See the MQTT_connect
+  // function definition further below.
+  MQTT_connect();
+
+  // this is our 'wait for incoming subscription packets' busy subloop
+  // try to spend your time here
+
+  Adafruit_MQTT_Subscribe *subscription;
+  while ((subscription = mqtt.readSubscription(5000))) {
+    if (subscription == &onoff1) {
+      //Serial.print(F("Got: "));
+      //Serial.println((char *)onoff1.lastread);
+      if        (strcmp((char *)onoff1.lastread, "on1") == 0) {
+        digitalWrite(out1, HIGH);
+      } else if (strcmp((char *)onoff1.lastread, "off1") == 0) {
+        digitalWrite(out1, LOW);
+      } else if (strcmp((char *)onoff1.lastread, "1") == 0) {
+        digitalWrite(out1, HIGH);
+      } else if (strcmp((char *)onoff1.lastread, "0") == 0) {
+        digitalWrite(out1, LOW);
+      } else if (strcmp((char *)onoff1.lastread, "on2") == 0) {
+        digitalWrite(out2, HIGH);
+      } else if (strcmp((char *)onoff1.lastread, "off2") == 0) {
+        digitalWrite(out2, LOW);
+      } else if (strcmp((char *)onoff1.lastread, "onall") == 0) {
+        digitalWrite(out1, HIGH);
+        digitalWrite(out2, HIGH);
+      } else if (strcmp((char *)onoff1.lastread, "offall") == 0) {
+        digitalWrite(out1, LOW);
+        digitalWrite(out2, LOW);
+     digitalWrite(out3, LOW);
+        digitalWrite(out4, LOW);
+      } else if (strcmp((char *)onoff1.lastread, "toggle1") == 0) {
+        digitalWrite(out1, !digitalRead(out1));
+      } else if (strcmp((char *)onoff1.lastread, "toggle2") == 0) {
+        digitalWrite(out2, !digitalRead(out2));
+      }
+      //Serial.println((char *)onoff1.lastread);
+    } else if (subscription == &onoff2) {
+      //Serial.print(F("Got: "));
+      //Serial.println((char *)onoff2.lastread);
+      if        (strcmp((char *)onoff2.lastread, "on1") == 0) {
+        digitalWrite(out1, HIGH);
+      } else if (strcmp((char *)onoff2.lastread, "off1") == 0) {
+        digitalWrite(out1, LOW);
+      } else if (strcmp((char *)onoff2.lastread, "on2") == 0) {
+        digitalWrite(out2, HIGH);
+      } else if (strcmp((char *)onoff2.lastread, "off2") == 0) {
+        digitalWrite(out2, LOW);
+      } else if (strcmp((char *)onoff2.lastread, "1") == 0) {
+        digitalWrite(out2, HIGH);
+      } else if (strcmp((char *)onoff2.lastread, "0") == 0) {
+        digitalWrite(out2, LOW);
+      } else if (strcmp((char *)onoff2.lastread, "onall") == 0) {
+        digitalWrite(out1, HIGH);
+        digitalWrite(out2, HIGH);
+      } else if (strcmp((char *)onoff2.lastread, "offall") == 0) {
+        digitalWrite(out1, LOW);
+        digitalWrite(out2, LOW);          
+        digitalWrite(out3, LOW);
+      digitalWrite(out4, LOW);
+
+      } else if (strcmp((char *)onoff2.lastread, "toggle1") == 0) {
+        digitalWrite(out1, !digitalRead(out1));
+      } else if (strcmp((char *)onoff2.lastread, "toggle2") == 0) {
+        digitalWrite(out2, !digitalRead(out2));
+      }
+      //Serial.println((char *)onoff2.lastread);
+
+
+    } else if (subscription == &onoff3) {
+      //Serial.print(F("Got: "));
+      //Serial.println((char *)onoff2.lastread);
+      if        (strcmp((char *)onoff3.lastread, "on") == 0) {
+        digitalWrite(out3, HIGH);
+      } else if (strcmp((char *)onoff3.lastread, "off") == 0) {
+        digitalWrite(out3, LOW);
+      } else if (strcmp((char *)onoff3.lastread, "on3") == 0) {
+        digitalWrite(out3, HIGH);
+      } else if (strcmp((char *)onoff3.lastread, "off3") == 0) {
+        digitalWrite(out3, LOW);
+      } else if (strcmp((char *)onoff3.lastread, "1") == 0) {
+        digitalWrite(out3, HIGH);
+      } else if (strcmp((char *)onoff3.lastread, "0") == 0) {
+        digitalWrite(out3, LOW);
+      } else if (strcmp((char *)onoff3.lastread, "toggle") == 0) {
+        digitalWrite(out3, !digitalRead(out3));
+      }
+      //Serial.println((char *)onoff3.lastread);
+
+
+    } else if (subscription == &onoff4) {
+      //Serial.print(F("Got: "));
+      //Serial.println((char *)onoff4.lastread);
+      if        (strcmp((char *)onoff4.lastread, "on") == 0) {
+        digitalWrite(out4, HIGH);
+      } else if (strcmp((char *)onoff4.lastread, "off") == 0) {
+        digitalWrite(out4, LOW);
+      } else if (strcmp((char *)onoff4.lastread, "on4") == 0) {
+        digitalWrite(out4, HIGH);
+      } else if (strcmp((char *)onoff4.lastread, "off4") == 0) {
+        digitalWrite(out4, LOW);
+      } else if (strcmp((char *)onoff4.lastread, "1") == 0) {
+        digitalWrite(out4, HIGH);
+      } else if (strcmp((char *)onoff4.lastread, "0") == 0) {
+        digitalWrite(out4, LOW);
+      } else if (strcmp((char *)onoff4.lastread, "toggle") == 0) {
+        digitalWrite(out4, !digitalRead(out4));
+      }
+      //Serial.println((char *)onoff4.lastread);
+
+    }
+  }
+
+  // Now we can publish stuff!
+  //Serial.print(F("\nPublishing status to statusfeed, out1status, out2status: "));
+  //Serial.print(statusMessage);
+  //Serial.print("\n");
+  if (digitalRead(out1) == 0) {
+    out1stat = '0';
+    out1val = 0;
+  } else {
+    out1stat = '1';
+    out1val = 1;
+  }
+  if (digitalRead(out2) == 0) {
+    out2stat = '0';
+    out2val = 0;
+  } else {
+    out2stat = '1';
+    out2val = 1;
+  }
+  if (digitalRead(out3) == 0) {
+    out3stat = '0';
+    out3val = 0;
+  } else {
+    out3stat = '1';
+    out3val = 1;
+  }
+  if (digitalRead(out4) == 0) {
+    out4stat = '0';
+    out4val = 0;
+  } else {
+    out4stat = '1';
+    out4val = 1;
+  }
+
+  out1status.publish(out1val);
+  out2status.publish(out2val);
+  out3status.publish(out3val);
+  out4status.publish(out4val);
+
+  statusMessage[26] = out1stat;
+  statusMessage[27] = ' ';
+  statusMessage[28] = out2stat;
+statusMessage[29] = out3stat;
+  statusMessage[30] = ' ';
+  statusMessage[31] = out4stat;
+
+  if (! statusfeed.publish(statusMessage)) {
+    //Serial.println(F("Failed"));
+  } else {
+    //Serial.println(F("OK!"));
+  }
+
+  // ping the server to keep the mqtt connection alive
+  // NOT required if you are publishing once every KEEPALIVE seconds
+  /*
+    if(! mqtt.ping()) {
+    mqtt.disconnect();
+    }
+  */
+}
+
+// Function to connect and reconnect as necessary to the MQTT server.
+// Should be called in the loop function and it will take care if connecting.
+void MQTT_connect() {
+  int8_t ret;
+
+  // Stop if already connected.
+  if (mqtt.connected()) {
+    return;
+  }
+
+  //Serial.print("Connecting to MQTT... ");
+
+  uint8_t retries = 3;
+  while ((ret = mqtt.connect()) != 0) { // connect will return 0 for connected
+    //Serial.println(mqtt.connectErrorString(ret));
+    //Serial.println("Retrying MQTT connection in 5 seconds...");
+    mqtt.disconnect();
+    delay(5000);  // wait 5 seconds
+    retries--;
+    if (retries == 0) {
+      // basically die and wait for WDT to reset me
+      while (1);
+    }
+  }
+  //Serial.println("MQTT Connected!");
+  mainstatusfeed.publish("Connected!");
+}