SonOffS20.ino

#include <Homie.h>
#include <ArduinoOTA.h>
#include <OneButton.h>

#define PIN_LED     13      // PIN 12 / HSPI_MOSI; UART0_CTS MTCK
#define PIN_RELAY   12      // PIN 10 / HSPI_MISO MTDI
OneButton button(0, true);


#define FW_NAME "sonoff-s20"
#define FW_VERSION "2.0.1"

unsigned long WiFifix = 0;
unsigned long problemDetected = 0;
int problemCount = 0;
int deleteConfig = 0;
String problemCause;

HomieNode relayNode("outlet", "outlet", "outlet");

bool relayHandler(const HomieRange& range, const String& value) {
  String on = value;
  on.toLowerCase();

  if (on == "true") {
    digitalWrite(PIN_RELAY, HIGH);
  }
  else if (on == "false") {
    digitalWrite(PIN_RELAY, LOW);
  }
  else if (on == "reboot") {
    relayNode.setProperty("on/set").send(" ");
    delay(1000);
    Homie.reboot();
  }
  else if (on == "ota-restart") {
    relayNode.setProperty("on/set").send(" ");
    //ArduinoOTA.setHostname(Homie.getConfiguration().deviceId);
    //ArduinoOTA.begin();
  }
  relayNode.setProperty("on").send(value);
  return true;
}

void toggleRelay() {
  bool on = digitalRead(PIN_RELAY) == HIGH;
  digitalWrite(PIN_RELAY, on ? LOW : HIGH);
  relayNode.setProperty("on").send(on ? "false" : "true");
  relayNode.setProperty("on/set").send(on ? "false" : "true");
  Homie.getLogger() << "Switch is " << (on ? "off" : "on") << endl;
}

void click() {
  toggleRelay();
}

void doubleclick() {
  Homie.reboot();
}

void longclick() {
  Homie.reset();
}



void fixWiFi() {
  // Posts every 10 seconds the state of WiFi.status(), Homie.getMqttClient().connected() and Homie.isConfigured()
  // Within this interval the connectivity is checked and logged if a problem is detected
  // Then it disconnects Wifi, if Wifi or MQTT is not connected for 1 Minute (but only if Homie is configured)
  if ( WiFifix == 0 || ((millis() - WiFifix) > 10000)) {

    float rssi = WiFi.RSSI();
    //relayNode.setProperty("quality").send(String(rssi));
    Homie.getLogger() << "Wifi-state:" << WiFi.status() << " | Wi-Fi signal quality: " << rssi << " | MQTT-state:" << Homie.getMqttClient().connected() << " | HomieConfig-state:" << Homie.isConfigured() << endl;

    if (!Homie.getMqttClient().connected() || WiFi.status() != 3) {
      if (0 == problemDetected) {
        if (WiFi.status() != 3) {
          problemCause = "WiFi: Disconnected ";
        }
        if (!Homie.getMqttClient().connected()) {
          problemCause += "MQTT: Disconnected";
        }
        Homie.getLogger() << "Connectivity in problematic state --> " << problemCause << endl;
        problemDetected = millis();
      }
      else if ((millis() - problemDetected) > 120000 && (problemCount >= 5)) {
        Homie.getLogger() << "Connectivity in problematic state --> This remained for 10 minutes. Rebooting!" << endl;
        Homie.reboot();
      }
      else if ((millis() - problemDetected) > 120000 && problemCount < 5) {
        problemCount = (problemCount + 1);
        Homie.getLogger() << "Connectivity in problematic state --> " << problemCause << "/n This remained for 2 minutes. Disconnecting WiFi to start over." << endl;
        problemDetected = 0;
        problemCause = "";
        if (WiFi.status() != 0) {
          WiFi.disconnect();
        }
        if (WiFi.status() == 0) {
          WiFi.begin();
        }
      }
    }
    else if (problemCount != 0 && (Homie.getMqttClient().connected() || WiFi.status() == 3)) {
      problemCount = 0;
      ArduinoOTA.setHostname(Homie.getConfiguration().deviceId);
      ArduinoOTA.begin();
    }
  }
  WiFifix = millis();
}


void setup() {
  Homie_setBrand(FW_NAME);
  Homie_setFirmware(FW_NAME, FW_VERSION);
  Serial.begin(115200);
  Serial << endl << endl;
  pinMode(PIN_RELAY, OUTPUT);
  digitalWrite(PIN_RELAY, LOW);
  Homie.setLedPin(PIN_LED, LOW);
  button.attachClick(click);
  button.attachDoubleClick(doubleclick);
  button.attachLongPressStop(longclick);
  button.setClickTicks(500);
  button.setPressTicks(10000);
  relayNode.advertise("on").setName("on").setDatatype("boolean").settable(relayHandler);
  Homie.disableResetTrigger();
  Homie.setup();
  ArduinoOTA.setHostname(Homie.getConfiguration().deviceId);
  ArduinoOTA.begin();
}

void loop() {
  ArduinoOTA.handle();
  button.tick();
  Homie.loop();
  if (Homie.isConfigured() == 1) {
    fixWiFi();
  }
}