thomas
/
esp-relais
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
No Description
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
4 Commits
1 Branch
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
esp-relais/src/main.cpp

main.cpp 14KB
Permalink History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526 
  1. /*

  2.  * main.cpp

  3.  *

  4.  * ESP8266 / ESP32 Relais Actor

  5.  *

  6.  * ----------------------------------------------------------------------------

  7.  * "THE BEER-WARE LICENSE" (Revision 42):

  8.  * <xythobuz@xythobuz.de> wrote this file.  As long as you retain this notice

  9.  * you can do whatever you want with this stuff. If we meet some day, and you

  10.  * think this stuff is worth it, you can buy me a beer in return.   Thomas Buck

  11.  * ----------------------------------------------------------------------------

  12.  */

  13. 

  14. #include <Arduino.h>

  15. #include <Wire.h>

  16. #include <PubSubClient.h>

  17. 

  18. #if defined(ARDUINO_ARCH_ESP8266)

  19. 

  20. #include <ESP8266WiFi.h>

  21. #include <ESP8266WebServer.h>

  22. #include <ESP8266mDNS.h>

  23. 

  24. #elif defined(ARDUINO_ARCH_ESP32)

  25. 

  26. #include <WiFi.h>

  27. #include <WebServer.h>

  28. #include <ESPmDNS.h>

  29. 

  30. #endif

  31. 

  32. #include "config.h"

  33. #include "relais.h"

  34. #include "SimpleUpdater.h"

  35. 

  36. #define BUILTIN_LED_PIN 1

  37. 

  38. UPDATE_WEB_SERVER server(80);

  39. SimpleUpdater updater;

  40. 

  41. #ifdef ENABLE_MQTT

  42. WiFiClient mqttClient;

  43. PubSubClient mqtt(mqttClient);

  44. #endif // ENABLE_MQTT

  45. 

  46. #ifdef ENABLE_INFLUXDB_LOGGING

  47. #include <InfluxDb.h>

  48. 

  49. Influxdb influx(INFLUXDB_HOST, INFLUXDB_PORT);

  50. 

  51. #define INFLUX_MAX_ERRORS_RESET 10

  52. int error_count = 0;

  53. 

  54. void writeDatabase();

  55. #endif // ENABLE_INFLUXDB_LOGGING

  56. 

  57. #ifdef ENABLE_NTP

  58. #include <WiFiUdp.h>

  59. #include <NTPClient.h>

  60. 

  61. WiFiUDP ntpUDP;

  62. NTPClient timeClient(ntpUDP, "europe.pool.ntp.org", 2 * 60 * 60, 5 * 60 * 1000);

  63. #endif // ENABLE_NTP

  64. 

  65. unsigned long last_server_handle_time = 0;

  66. unsigned long last_db_write_time = 0;

  67. unsigned long last_led_blink_time = 0;

  68. unsigned long last_timing_check_time = 0;

  69. unsigned long last_mqtt_reconnect_time = 0;

  70. 

  71. String relais_name[RELAIS_COUNT] = {

  72.     String(" (Small Light)"),

  73.     String(" (Big Light)"),

  74.     String(""),

  75.     String(" (Fan)"),

  76. };

  77. 

  78. bool relais_state[RELAIS_COUNT] = { false, false, false, false };

  79. String auto_set_state = String("");

  80. 

  81. #ifdef ENABLE_NTP

  82. void handleAutoTimingCheck(void) {

  83.     relais_state[3] = true; // always keep fan on

  84. 

  85.     if ((timeClient.getHours() >= 20) || (timeClient.getHours() < 10)) {

  86.         // small light from 20:00 to 10:00

  87.         relais_state[0] = true;

  88.         relais_state[1] = false;

  89.         auto_set_state = String(F("Small Light at Night (")) + String(timeClient.getFormattedTime()) + String(F(")"));

  90.     } else {

  91.         // big light from 10:00 to 20:00

  92.         relais_state[0] = false;

  93.         relais_state[1] = true;

  94.         auto_set_state = String(F("Big Light while Daylight (")) + String(timeClient.getFormattedTime()) + String(F(")"));

  95.     }

  96. 

  97.     for (int i = 0; i < RELAIS_COUNT; i++) {

  98.         relais_set(i, relais_state[i] ? 1 : 0);

  99.     }

  100. }

  101. #endif // ENABLE_NTP

  102. 

  103. void handlePage(int mode = -1, int id = 0) {

  104.     String message = F("<html><head>\n");

  105.     message += F("<title>" ESP_PLATFORM_NAME " MQTT Relais</title>\n");

  106.     message += F("</head><body>\n");

  107.     message += F("<h1>" ESP_PLATFORM_NAME " MQTT Relais</h1>\n");

  108. 

  109.     message += F("\n<p>\n");

  110.     for (int i = 0; i < RELAIS_COUNT; i++) {

  111.         message += String(F("<a href=\"/on?id=")) + String(i) + String(F("\">Relais ")) + String(i) + String(F(" On")) + relais_name[i] + String(F("</a><br>\n"));

  112.         message += String(F("<a href=\"/off?id=")) + String(i) + String(F("\">Relais ")) + String(i) + String(F(" Off")) + relais_name[i] + String(F("</a><br>\n"));

  113.     }

  114.     message += String(F("<a href=\"/on?id=")) + String(RELAIS_COUNT) + String(F("\">All Relais On</a><br>\n"));

  115.     message += String(F("<a href=\"/off?id=")) + String(RELAIS_COUNT) + String(F("\">All Relais Off</a><br>\n"));

  116.     message += F("</p>\n");

  117. 

  118.     if (mode >= 0) {

  119.         message += F("<p>");

  120.         message += F("Turned Relais ");

  121.         message += (id < RELAIS_COUNT) ? String(id) : String(F("1-4"));

  122.         message += (mode ? String(F(" On")) : String(F(" Off")));

  123.         message += F("</p>\n");

  124.     }

  125. 

  126.     message += F("\n<p>\n");

  127.     for (int i = 0; i < RELAIS_COUNT; i++) {

  128.         message += String(F("Relais ")) + String(i) + relais_name[i] + String(F(" = ")) + (relais_state[i] ? String(F("On")) : String(F("Off"))) + String(F("<br>\n"));

  129.     }

  130.     message += F("</p>\n");

  131. 

  132. #ifdef ENABLE_NTP

  133.     message += F("<p>Time from NTP: ");

  134.     message += String(timeClient.getFormattedTime());

  135.     message += F("</p>\n");

  136. 

  137.     message += F("<p>Auto-State: ");

  138.     message += auto_set_state;

  139.     message += F("</p>\n");

  140. 

  141.     unsigned long next_min = (AUTO_TIMING_INTERVAL - (millis() - last_timing_check_time)) / (1000 * 60);

  142.     message += F("<p>Next Auto-State in ");

  143.     message += String(next_min);

  144.     message += F("min</p>\n");

  145. #endif // ENABLE_NTP

  146. 

  147.     message += F("<p><a href=\"/\">Refresh Page</a></p>\n");

  148. 

  149.     message += F("\n<p>\n");

  150.     message += F("Version: ");

  151.     message += ESP_RELAIS_VERSION;

  152.     message += F("\n<br>\n");

  153.     message += F("Location: ");

  154.     message += SENSOR_LOCATION;

  155.     message += F("\n<br>\n");

  156.     message += F("MAC: ");

  157.     message += WiFi.macAddress();

  158.     message += F("\n</p>\n");

  159. 

  160. #if defined(ARDUINO_ARCH_ESP8266)

  161.     

  162.     message += F("\n<p>\n");

  163.     message += F("Reset reason: ");

  164.     message += ESP.getResetReason();

  165.     message += F("\n<br>\n");

  166.     message += F("Free heap: ");

  167.     message += String(ESP.getFreeHeap());

  168.     message += F(" (");

  169.     message += String(ESP.getHeapFragmentation());

  170.     message += F("% fragmentation)");

  171.     message += F("\n<br>\n");

  172.     message += F("Free sketch space: ");

  173.     message += String(ESP.getFreeSketchSpace());

  174.     message += F("\n<br>\n");

  175.     message += F("Flash chip real size: ");

  176.     message += String(ESP.getFlashChipRealSize());

  177. 

  178.     if (ESP.getFlashChipSize() != ESP.getFlashChipRealSize()) {

  179.         message += F("\n<br>\n");

  180.         message += F("WARNING: sdk chip size (");

  181.         message += (ESP.getFlashChipSize());

  182.         message += F(") does not match!");

  183.     }

  184.     

  185.     message += F("\n</p>\n");

  186.     

  187. #elif defined(ARDUINO_ARCH_ESP32)

  188. 

  189.     message += F("\n<p>\n");

  190.     message += F("Free heap: ");

  191.     message += String(ESP.getFreeHeap() / 1024.0);

  192.     message += F("k\n<br>\n");

  193.     message += F("Free sketch space: ");

  194.     message += String(ESP.getFreeSketchSpace() / 1024.0);

  195.     message += F("k\n<br>\n");

  196.     message += F("Flash chip size: ");

  197.     message += String(ESP.getFlashChipSize() / 1024.0);

  198.     message += F("k\n</p>\n");

  199.     

  200. #endif

  201. 

  202.     message += F("<p>\n");

  203.     message += F("Try <a href=\"/update\">/update</a> for OTA firmware updates!\n");

  204.     message += F("</p>\n");

  205.     

  206.     message += F("<p>\n");

  207. #ifdef ENABLE_INFLUXDB_LOGGING

  208.     message += F("InfluxDB: ");

  209.     message += INFLUXDB_DATABASE;

  210.     message += F(" @ ");

  211.     message += INFLUXDB_HOST;

  212.     message += F(":");

  213.     message += String(INFLUXDB_PORT);

  214.     message += F("\n");

  215. #else

  216.     message += F("InfluxDB logging not enabled!\n");

  217. #endif

  218.     message += F("</p>\n");

  219. 

  220.     message += F("</body></html>\n");

  221.     server.send(200, "text/html", message);

  222. }

  223. 

  224. void handleOn() {

  225.     String id_string = server.arg("id");

  226.     int id = id_string.toInt();

  227. 

  228.     if ((id >= 0) && (id < RELAIS_COUNT)) {

  229.         relais_set(id, 1);

  230.         relais_state[id] = true;

  231.     } else {

  232.         for (int i = 0; i < RELAIS_COUNT; i++) {

  233.             relais_set(i, 1);

  234.             relais_state[i] = true;

  235.         }

  236.     }

  237. 

  238.     // only reset to default after 15min

  239.     last_timing_check_time = millis();

  240. 

  241. #ifdef ENABLE_INFLUXDB_LOGGING

  242.     writeDatabase();

  243. #endif // ENABLE_INFLUXDB_LOGGING

  244. 

  245.     handlePage(1, id);

  246. }

  247. 

  248. void handleOff() {

  249.     String id_string = server.arg("id");

  250.     int id = id_string.toInt();

  251. 

  252.     if ((id >= 0) && (id < RELAIS_COUNT)) {

  253.         relais_set(id, 0);

  254.         relais_state[id] = false;

  255.     } else {

  256.         for (int i = 0; i < RELAIS_COUNT; i++) {

  257.             relais_set(i, 0);

  258.             relais_state[i] = false;

  259.         }

  260.     }

  261. 

  262.     // only reset to default after 15min

  263.     last_timing_check_time = millis();

  264. 

  265. #ifdef ENABLE_INFLUXDB_LOGGING

  266.     writeDatabase();

  267. #endif // ENABLE_INFLUXDB_LOGGING

  268. 

  269.     handlePage(0, id);

  270. }

  271. 

  272. void handleRoot() {

  273.     handlePage();

  274. }

  275. 

  276. #ifdef ENABLE_MQTT

  277. void mqttCallback(char* topic, byte* payload, unsigned int length) {

  278.     int state = 0;

  279.     int id = 0;

  280. 

  281.     String ts(topic), ps((char *)payload);

  282. 

  283.     String our_topic(SENSOR_LOCATION);

  284.     our_topic += "/";

  285. 

  286.     if (!ts.startsWith(our_topic)) {

  287.         return;

  288.     }

  289. 

  290.     String ids = ts.substring(our_topic.length());

  291.     id = ids.toInt();

  292.     if ((id < 0) || (id >= RELAIS_COUNT)) {

  293.         return;

  294.     }

  295. 

  296.     if (ps.indexOf("on") != -1) {

  297.         state = 1;

  298.     } else if (ps.indexOf("off") != -1) {

  299.         state = 0;

  300.     } else {

  301.         return;

  302.     }

  303. 

  304.     if ((id >= 0) && (id < RELAIS_COUNT)) {

  305.         relais_set(id, state);

  306.         relais_state[id] = state ? true : false;

  307. 

  308. #ifdef ENABLE_INFLUXDB_LOGGING

  309.         writeDatabase();

  310. #endif // ENABLE_INFLUXDB_LOGGING

  311.     }

  312. }

  313. 

  314. void mqttReconnect() {

  315.     // Create a random client ID

  316.     String clientId = "ESP8266-Relais-";

  317.     clientId += String(random(0xffff), HEX);

  318. 

  319.     // Attempt to connect

  320. #if defined(MQTT_USER) && defined(MQTT_PASS)

  321.     if (mqtt.connect(clientId.c_str(), MQTT_USER, MQTT_PASS)) {

  322. #else

  323.     if (mqtt.connect(clientId.c_str())) {

  324. #endif

  325.         // Once connected, publish an announcement...

  326.         mqtt.publish(SENSOR_LOCATION, "hello world");

  327. 

  328.         // ... and resubscribe

  329.         mqtt.subscribe(SENSOR_LOCATION);

  330.         for (int i = 0; i < RELAIS_COUNT; i++) {

  331.             String topic(SENSOR_LOCATION);

  332.             topic += String("/") + String(i);

  333.             mqtt.subscribe(topic.c_str());

  334.         }

  335.     }

  336. }

  337. #endif // ENABLE_MQTT

  338. 

  339. void setup() {

  340.     pinMode(BUILTIN_LED_PIN, OUTPUT);

  341.     

  342.     relais_init();

  343. 

  344.     for (int i = 0; i < RELAIS_COUNT; i++) {

  345.         relais_state[i] = false;

  346.     }

  347. 

  348.     Serial.println();

  349.     Serial.println("Hello World From Relais Test");

  350. 

  351.     // Blink LED for init

  352.     for (int i = 0; i < 2; i++) {

  353.         digitalWrite(BUILTIN_LED_PIN, LOW); // LED on

  354.         delay(LED_INIT_BLINK_INTERVAL);

  355.         digitalWrite(BUILTIN_LED_PIN, HIGH); // LED off

  356.         delay(LED_INIT_BLINK_INTERVAL);

  357.     }

  358. 

  359.     // Build hostname string

  360.     String hostname = SENSOR_HOSTNAME_PREFIX;

  361.     hostname += SENSOR_LOCATION;

  362. 

  363. #if defined(ARDUINO_ARCH_ESP8266)

  364. 

  365.     // Connect to WiFi AP

  366.     Serial.print("Connecting to WiFi ");

  367.     WiFi.hostname(hostname);

  368.     WiFi.mode(WIFI_STA);

  369.     WiFi.begin(WIFI_SSID, WIFI_PASS);

  370.     while (WiFi.status() != WL_CONNECTED) {

  371.         Serial.print(WiFi.status());

  372.         Serial.print(" ");

  373. 

  374.         delay(LED_CONNECT_BLINK_INTERVAL);

  375.         digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));

  376.     }

  377.     Serial.println();

  378. 

  379.     Serial.print("Got local IP ");

  380.     Serial.println(WiFi.localIP().toString());

  381.     

  382. #elif defined(ARDUINO_ARCH_ESP32)

  383. 

  384.     // Set hostname workaround

  385.     WiFi.config(INADDR_NONE, INADDR_NONE, INADDR_NONE);

  386.     WiFi.setHostname(hostname.c_str());

  387.     

  388.     // Workaround for WiFi connecting only every 2nd reset

  389.     // https://github.com/espressif/arduino-esp32/issues/2501#issuecomment-513602522

  390.     WiFi.onEvent([](WiFiEvent_t event, WiFiEventInfo_t info) {

  391.         if (info.disconnected.reason == 202) {

  392.             esp_sleep_enable_timer_wakeup(10);

  393.             esp_deep_sleep_start();

  394.             delay(100);

  395.         }

  396.     }, WiFiEvent_t::SYSTEM_EVENT_STA_DISCONNECTED);

  397. 

  398.     // Connect to WiFi AP

  399.     Serial.print("Connecting to WiFi");

  400.     WiFi.mode(WIFI_STA);

  401.     WiFi.begin(WIFI_SSID, WIFI_PASS);

  402.     while (WiFi.status() != WL_CONNECTED) {

  403.         Serial.print(".");

  404.         delay(LED_CONNECT_BLINK_INTERVAL);

  405.         digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));

  406.     }

  407.     Serial.println();

  408.     

  409.     // Set hostname workaround

  410.     WiFi.setHostname(hostname.c_str());

  411. 

  412. #endif

  413. 

  414.     Serial.println("Seeding");

  415.     randomSeed(micros());

  416. 

  417. #ifdef ENABLE_MQTT

  418.     Serial.println("MQTT");

  419.     mqtt.setServer(MQTT_HOST, MQTT_PORT);

  420.     mqtt.setCallback(mqttCallback);

  421. #endif // ENABLE_MQTT

  422. 

  423. #ifdef ENABLE_INFLUXDB_LOGGING

  424.     // Setup InfluxDB Client

  425.     influx.setDb(INFLUXDB_DATABASE);

  426. #endif // ENABLE_INFLUXDB_LOGGING

  427. 

  428. #ifdef ENABLE_NTP

  429.     timeClient.begin();

  430. #endif // ENABLE_NTP

  431. 

  432.     // Setup HTTP Server

  433.     Serial.println("Server");

  434.     MDNS.begin(hostname.c_str());

  435.     updater.setup(&server);

  436.     server.on("/", handleRoot);

  437.     server.on("/on", handleOn);

  438.     server.on("/off", handleOff);

  439.     server.begin();

  440.     MDNS.addService("http", "tcp", 80);

  441. 

  442.     Serial.println("Done");

  443. }

  444. 

  445. void handleServers() {

  446.     server.handleClient();

  447.     

  448. #if defined(ARDUINO_ARCH_ESP8266)

  449.     MDNS.update();

  450. #endif

  451. }

  452. 

  453. #ifdef ENABLE_INFLUXDB_LOGGING

  454. static boolean writeMeasurement(InfluxData &measurement) {

  455.     boolean success = influx.write(measurement);

  456.     if (!success) {

  457.         error_count++;

  458.         for (int i = 0; i < 10; i++) {

  459.             digitalWrite(BUILTIN_LED_PIN, LOW); // LED on

  460.             delay(LED_ERROR_BLINK_INTERVAL);

  461.             digitalWrite(BUILTIN_LED_PIN, HIGH); // LED off

  462.             delay(LED_ERROR_BLINK_INTERVAL);

  463.         }

  464.     }

  465.     return success;

  466. }

  467. 

  468. void writeDatabase() {

  469.     for (int i = 0; i < RELAIS_COUNT; i++) {

  470.         InfluxData measurement("relais");

  471.         measurement.addTag("location", SENSOR_LOCATION);

  472.         measurement.addTag("device", WiFi.macAddress());

  473.         measurement.addTag("id", String(i));

  474.         measurement.addValue("state", relais_state[i] ? 1 : 0);

  475.         writeMeasurement(measurement);

  476.     }

  477. }

  478. #endif // ENABLE_INFLUXDB_LOGGING

  479. 

  480. void loop() {

  481.     if ((millis() - last_server_handle_time) >= SERVER_HANDLE_INTERVAL) {

  482.         last_server_handle_time = millis();

  483.         handleServers();

  484.     }

  485. 

  486. #ifdef ENABLE_NTP

  487.     timeClient.update();

  488. 

  489.     if (timeClient.isTimeSet()) {

  490.         if (((millis() - last_timing_check_time) >= AUTO_TIMING_INTERVAL) || (last_timing_check_time == 0)) {

  491.             last_timing_check_time = millis();

  492.             handleAutoTimingCheck();

  493.         }

  494.     }

  495. #endif // ENABLE_NTP

  496. 

  497. #ifdef ENABLE_MQTT

  498.     if (!mqtt.connected() && ((millis() - last_mqtt_reconnect_time) >= MQTT_RECONNECT_INTERVAL)) {

  499.         last_mqtt_reconnect_time = millis();

  500.         mqttReconnect();

  501.     }

  502. 

  503.     mqtt.loop();

  504. #endif // ENABLE_MQTT

  505. 

  506. #ifdef ENABLE_INFLUXDB_LOGGING

  507. #if DB_WRITE_INTERVAL > 0

  508.     if ((millis() - last_db_write_time) >= DB_WRITE_INTERVAL) {

  509.         last_db_write_time = millis();

  510.         writeDatabase();

  511.     }

  512. #endif // DB_WRITE_INTERVAL > 0

  513.     

  514. #ifdef INFLUX_MAX_ERRORS_RESET

  515.     if (error_count >= INFLUX_MAX_ERRORS_RESET) {

  516.         ESP.restart();

  517.     }

  518. #endif // INFLUX_MAX_ERRORS_RESET

  519. #endif // ENABLE_INFLUXDB_LOGGING

  520. 

  521.     // blink heartbeat LED

  522.     if ((millis() - last_led_blink_time) >= LED_BLINK_INTERVAL) {

  523.         last_led_blink_time = millis();

  524.         digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));

  525.     }

  526. }