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.
2 Commits
1 Branch
Tree: 066601c189
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '066601c189'
${ noResults }
esp-relais/src/main.cpp

main.cpp 16KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555 
  1. /*

  2.  * main.cpp

  3.  *

  4.  * ESP8266 / ESP32 Environmental Sensor

  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. #endif // ENABLE_INFLUXDB_LOGGING

  54. 

  55. #ifdef ENABLE_NTP

  56. #include <WiFiUdp.h>

  57. #include <NTPClient.h>

  58. 

  59. WiFiUDP ntpUDP;

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

  61. #endif // ENABLE_NTP

  62. 

  63. unsigned long last_server_handle_time = 0;

  64. unsigned long last_db_write_time = 0;

  65. unsigned long last_led_blink_time = 0;

  66. unsigned long last_timing_check_time = 0;

  67. 

  68. String relais_name[RELAIS_COUNT] = {

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

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

  71.     String(""),

  72.     String(" (Fan)"),

  73. };

  74. 

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

  76. String auto_set_state = String("");

  77. 

  78. #ifdef ENABLE_NTP

  79. void handleAutoTimingCheck(void) {

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

  81. 

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

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

  84.         relais_state[0] = true;

  85.         relais_state[1] = false;

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

  87.     } else {

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

  89.         relais_state[0] = false;

  90.         relais_state[1] = true;

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

  92.     }

  93. 

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

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

  96.     }

  97. }

  98. #endif // ENABLE_NTP

  99. 

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

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

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

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

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

  105. 

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

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

  108.         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"));

  109.         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"));

  110.     }

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

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

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

  114. 

  115.     if (mode >= 0) {

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

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

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

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

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

  121.     }

  122. 

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

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

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

  126.     }

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

  128. 

  129. #ifdef ENABLE_NTP

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

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

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

  133. 

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

  135.     message += auto_set_state;

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

  137. 

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

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

  140.     message += String(next_min);

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

  142. #endif // ENABLE_NTP

  143. 

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

  145. 

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

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

  148.     message += ESP_RELAIS_VERSION;

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

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

  151.     message += SENSOR_LOCATION;

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

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

  154.     message += WiFi.macAddress();

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

  156. 

  157. #if defined(ARDUINO_ARCH_ESP8266)

  158.     

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

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

  161.     message += ESP.getResetReason();

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

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

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

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

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

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

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

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

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

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

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

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

  174. 

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

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

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

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

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

  180.     }

  181.     

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

  183.     

  184. #elif defined(ARDUINO_ARCH_ESP32)

  185. 

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

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

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

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

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

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

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

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

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

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

  196.     

  197. #endif

  198. 

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

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

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

  202.     

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

  204. #ifdef ENABLE_INFLUXDB_LOGGING

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

  206.     message += INFLUXDB_DATABASE;

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

  208.     message += INFLUXDB_HOST;

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

  210.     message += String(INFLUXDB_PORT);

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

  212. #else

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

  214. #endif

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

  216. 

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

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

  219. }

  220. 

  221. void handleOn() {

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

  223.     int id = id_string.toInt();

  224. 

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

  226.         relais_set(id, 1);

  227.         relais_state[id] = true;

  228.     } else {

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

  230.             relais_set(i, 1);

  231.             relais_state[i] = true;

  232.         }

  233.     }

  234. 

  235.     // only reset to default after 15min

  236.     last_timing_check_time = millis();

  237. 

  238.     handlePage(1, id);

  239. }

  240. 

  241. void handleOff() {

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

  243.     int id = id_string.toInt();

  244. 

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

  246.         relais_set(id, 0);

  247.         relais_state[id] = false;

  248.     } else {

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

  250.             relais_set(i, 0);

  251.             relais_state[i] = false;

  252.         }

  253.     }

  254. 

  255.     // only reset to default after 15min

  256.     last_timing_check_time = millis();

  257. 

  258.     handlePage(0, id);

  259. }

  260. 

  261. void handleRoot() {

  262.     handlePage();

  263. }

  264. 

  265. #ifdef ENABLE_MQTT

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

  267.     int state = 0;

  268.     int id = 0;

  269. 

  270.     // TODO check topic matches

  271. 

  272.     if (((char)payload[0] == 't')

  273.             && ((char)payload[1] == 'u')

  274.             && ((char)payload[2] == 'r')

  275.             && ((char)payload[3] == 'n')

  276.             && ((char)payload[4] == ' ')

  277.             && ((char)payload[5] == 'o')) {

  278.         if (((char)payload[6] == 'n')

  279.             && ((char)payload[7] == ' ')) {

  280.             // turn on

  281.             state = 1;

  282.             id = payload[8];

  283.         } else if (((char)payload[6] == 'f')

  284.                 && ((char)payload[7] == 'f')

  285.                 && ((char)payload[8] == ' ')) {

  286.             // turn off

  287.             state = 0;

  288.             id = payload[9];

  289.         } else {

  290.             return;

  291.         }

  292.     } else {

  293.         return;

  294.     }

  295. 

  296.     relais_set(id - 1, state);

  297.     relais_state[id - 1] = state ? true : false;

  298. }

  299. 

  300. void mqttReconnect() {

  301.     // Loop until we're reconnected

  302.     //while (!mqtt.connected()) {

  303.         // Create a random client ID

  304.         String clientId = "ESP8266Client-";

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

  306. 

  307.         // Attempt to connect

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

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

  310.             //mqtt.publish("outTopic", "hello world");

  311.             // ... and resubscribe

  312.             mqtt.subscribe(SENSOR_LOCATION);

  313.         } else {

  314.             // Wait 5 seconds before retrying

  315.             delay(5000);

  316.         }

  317.     //}

  318. }

  319. #endif // ENABLE_MQTT

  320. 

  321. void setup() {

  322.     pinMode(BUILTIN_LED_PIN, OUTPUT);

  323.     

  324.     relais_init();

  325. 

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

  327.         relais_state[i] = false;

  328.     }

  329. 

  330.     Serial.println();

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

  332. 

  333.     // Blink LED for init

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

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

  336.         delay(LED_INIT_BLINK_INTERVAL);

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

  338.         delay(LED_INIT_BLINK_INTERVAL);

  339.     }

  340. 

  341.     // Build hostname string

  342.     String hostname = SENSOR_HOSTNAME_PREFIX;

  343.     hostname += SENSOR_LOCATION;

  344. 

  345. #if defined(ARDUINO_ARCH_ESP8266)

  346. 

  347.     // Connect to WiFi AP

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

  349.     WiFi.hostname(hostname);

  350.     WiFi.mode(WIFI_STA);

  351.     WiFi.begin(WIFI_SSID, WIFI_PASS);

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

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

  354.         Serial.print(" ");

  355. 

  356.         delay(LED_CONNECT_BLINK_INTERVAL);

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

  358.     }

  359.     Serial.println();

  360. 

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

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

  363.     

  364. #elif defined(ARDUINO_ARCH_ESP32)

  365. 

  366.     // Set hostname workaround

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

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

  369.     

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

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

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

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

  374.             esp_sleep_enable_timer_wakeup(10);

  375.             esp_deep_sleep_start();

  376.             delay(100);

  377.         }

  378.     }, WiFiEvent_t::SYSTEM_EVENT_STA_DISCONNECTED);

  379. 

  380.     // Connect to WiFi AP

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

  382.     WiFi.mode(WIFI_STA);

  383.     WiFi.begin(WIFI_SSID, WIFI_PASS);

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

  385.         Serial.print(".");

  386.         delay(LED_CONNECT_BLINK_INTERVAL);

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

  388.     }

  389.     Serial.println();

  390.     

  391.     // Set hostname workaround

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

  393. 

  394. #endif

  395. 

  396.     Serial.println("Seeding");

  397.     randomSeed(micros());

  398. 

  399. #ifdef ENABLE_MQTT

  400.     Serial.println("MQTT");

  401.     mqtt.setServer(MQTT_HOST, MQTT_PORT);

  402.     mqtt.setCallback(mqttCallback);

  403. #endif // ENABLE_MQTT

  404. 

  405. #ifdef ENABLE_INFLUXDB_LOGGING

  406.     // Setup InfluxDB Client

  407.     influx.setDb(INFLUXDB_DATABASE);

  408. #endif // ENABLE_INFLUXDB_LOGGING

  409. 

  410. #ifdef ENABLE_NTP

  411.     timeClient.begin();

  412. #endif // ENABLE_NTP

  413. 

  414.     // Setup HTTP Server

  415.     Serial.println("Server");

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

  417.     updater.setup(&server);

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

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

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

  421.     server.begin();

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

  423. 

  424.     Serial.println("Done");

  425. }

  426. 

  427. void handleServers() {

  428.     server.handleClient();

  429.     

  430. #if defined(ARDUINO_ARCH_ESP8266)

  431.     MDNS.update();

  432. #endif

  433. }

  434. 

  435. #ifdef ENABLE_INFLUXDB_LOGGING

  436. static boolean writeMeasurement(InfluxData &measurement) {

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

  438.     if (!success) {

  439.         error_count++;

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

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

  442.             delay(LED_ERROR_BLINK_INTERVAL);

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

  444.             delay(LED_ERROR_BLINK_INTERVAL);

  445.         }

  446.     }

  447.     return success;

  448. }

  449. 

  450. void writeDatabase() {

  451.     if (found_bme1) {

  452.         InfluxData measurement("environment");

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

  454.         measurement.addTag("placement", "1");

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

  456.         measurement.addTag("sensor", "bme280");

  457. 

  458.         measurement.addValue("temperature", bme1_temp());

  459.         measurement.addValue("pressure", bme1_pressure());

  460.         measurement.addValue("humidity", bme1_humid());

  461. 

  462.         writeMeasurement(measurement);

  463.     }

  464.     if (found_bme2) {

  465.         InfluxData measurement("environment");

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

  467.         measurement.addTag("placement", "2");

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

  469.         measurement.addTag("sensor", "bme280");

  470. 

  471.         measurement.addValue("temperature", bme2_temp());

  472.         measurement.addValue("pressure", bme2_pressure());

  473.         measurement.addValue("humidity", bme2_humid());

  474. 

  475.         writeMeasurement(measurement);

  476.     }

  477. 

  478.     if (found_sht) {

  479.         InfluxData measurement("environment");

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

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

  482.         measurement.addTag("sensor", "sht21");

  483. 

  484.         measurement.addValue("temperature", sht_temp());

  485.         measurement.addValue("humidity", sht_humid());

  486. 

  487.         writeMeasurement(measurement);

  488.     }

  489.     

  490.     for (int i = 0; i < moisture_count(); i++) {

  491.         int moisture = moisture_read(i);

  492.         if (moisture < moisture_max()) {

  493.             InfluxData measurement("moisture");

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

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

  496.             measurement.addTag("sensor", String(i + 1));

  497. 

  498.             measurement.addValue("value", moisture);

  499.             measurement.addValue("maximum", moisture_max());

  500. 

  501.             writeMeasurement(measurement);

  502.         }

  503.     }

  504. }

  505. #endif // ENABLE_INFLUXDB_LOGGING

  506. 

  507. void loop() {

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

  509.         last_server_handle_time = millis();

  510.         handleServers();

  511.     }

  512. 

  513. #ifdef ENABLE_NTP

  514.     timeClient.update();

  515. 

  516.     if (timeClient.isTimeSet()) {

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

  518.             last_timing_check_time = millis();

  519.             handleAutoTimingCheck();

  520.         }

  521.     }

  522. #endif // ENABLE_NTP

  523. 

  524. #ifdef ENABLE_MQTT

  525.     if (!mqtt.connected()) {

  526.         mqttReconnect();

  527.     }

  528.     mqtt.loop();

  529. #endif // ENABLE_MQTT

  530. 

  531. #ifdef ENABLE_INFLUXDB_LOGGING

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

  533.         last_db_write_time = millis();

  534.         writeDatabase();

  535.     }

  536.     

  537. #ifdef INFLUX_MAX_ERRORS_RESET

  538.     if (error_count >= INFLUX_MAX_ERRORS_RESET) {

  539.         ESP.restart();

  540.     }

  541. #endif // INFLUX_MAX_ERRORS_RESET

  542. #endif // ENABLE_INFLUXDB_LOGGING

  543. 

  544.     // blink heartbeat LED

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

  546.         last_led_blink_time = millis();

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

  548.     }

  549.     

  550.     // reset ESP every 48h to be safe

  551.     if (millis() >= (48 * 60 * 60 * 1000)) {

  552.         ESP.restart();

  553.     }

  554. }