DIY fertilizer mixer and plant watering machine https://www.xythobuz.de/giessomat.html
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.

WifiStuff.cpp 23KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737
  1. /*
  2. * Copyright (c) 2021 Thomas Buck <thomas@xythobuz.de>
  3. *
  4. * This file is part of Giess-o-mat.
  5. *
  6. * Giess-o-mat is free software: you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation, either version 3 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * Giess-o-mat is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License
  17. * along with Giess-o-mat. If not, see <https://www.gnu.org/licenses/>.
  18. */
  19. #include <Arduino.h>
  20. #ifdef PLATFORM_ESP
  21. #if defined(ARDUINO_ARCH_ESP8266)
  22. #include <ESP8266WiFi.h>
  23. #include <ESP8266WebServer.h>
  24. #include <ESP8266mDNS.h>
  25. #elif defined(ARDUINO_ARCH_ESP32)
  26. #include <WiFi.h>
  27. #include <WebServer.h>
  28. #include <ESPmDNS.h>
  29. #endif
  30. #include <WebSocketsServer.h>
  31. #include "wifi.h"
  32. #include "config.h"
  33. #include "config_pins.h"
  34. #include "Functionality.h"
  35. #include "SimpleUpdater.h"
  36. #include "DebugLog.h"
  37. #include "WifiStuff.h"
  38. UPDATE_WEB_SERVER server(80);
  39. WebSocketsServer socket = WebSocketsServer(81);
  40. SimpleUpdater updater;
  41. unsigned long last_server_handle_time = 0;
  42. unsigned long last_websocket_update_time = 0;
  43. String message_buffer_a;
  44. String message_buffer_b;
  45. String message_buffer_c;
  46. String message_buffer_d;
  47. #ifdef ENABLE_GPIO_TEST
  48. static bool runningGpioTest = false;
  49. static bool gpioTestState = false;
  50. unsigned long lastGpioTime = 0;
  51. #endif // ENABLE_GPIO_TEST
  52. #ifdef ENABLE_INFLUXDB_LOGGING
  53. #include <InfluxDb.h>
  54. Influxdb influx(INFLUXDB_HOST, INFLUXDB_PORT);
  55. #endif // ENABLE_INFLUXDB_LOGGING
  56. #ifdef ENABLE_GPIO_TEST
  57. void runGpioTest(bool state) {
  58. lastGpioTime = millis();
  59. for (int i = 0; i < VALVE_COUNT; i++) {
  60. get_plants()->getValves()->setPin(i, state);
  61. delay(GPIO_TEST_DELAY);
  62. }
  63. for (int i = 0; i < PUMP_COUNT; i++) {
  64. get_plants()->getPumps()->setPin(i, state);
  65. if (i < (PUMP_COUNT - 1)) {
  66. delay(GPIO_TEST_DELAY);
  67. }
  68. }
  69. }
  70. void handleGpioTest() {
  71. runningGpioTest = !runningGpioTest;
  72. gpioTestState = runningGpioTest;
  73. String message = F("GPIOs turned ");
  74. message += runningGpioTest ? "on" : "off";
  75. server.send(200, "text/html", message);
  76. runGpioTest(gpioTestState);
  77. }
  78. #endif // ENABLE_GPIO_TEST
  79. bool wifi_write_database(int duration, const char *type, int id) {
  80. bool success = false;
  81. #ifdef ENABLE_INFLUXDB_LOGGING
  82. InfluxData measurement(type);
  83. measurement.addTag("version", FIRMWARE_VERSION);
  84. measurement.addTag("device", WiFi.macAddress());
  85. measurement.addTag("id", String(id));
  86. measurement.addValue("duration", duration);
  87. success = influx.write(measurement);
  88. #endif // ENABLE_INFLUXDB_LOGGING
  89. return success;
  90. }
  91. void wifi_set_message_buffer(String a, String b, String c, String d) {
  92. message_buffer_a = a;
  93. message_buffer_b = b;
  94. message_buffer_c = c;
  95. message_buffer_d = d;
  96. }
  97. void wifi_schedule_websocket(void) {
  98. last_websocket_update_time = 0;
  99. }
  100. void wifi_send_status_broadcast(void) {
  101. if (socket.connectedClients() <= 0) {
  102. return;
  103. }
  104. String a = message_buffer_a ;
  105. String b = message_buffer_b;
  106. String c = message_buffer_c;
  107. String d = message_buffer_d;
  108. a.replace("\"", "'");
  109. b.replace("\"", "'");
  110. c.replace("\"", "'");
  111. d.replace("\"", "'");
  112. String ws = "{\n";
  113. ws += "\"a\": \"" + a + "\",\n";
  114. ws += "\"b\": \"" + b + "\",\n";
  115. ws += "\"c\": \"" + c + "\",\n";
  116. ws += "\"d\": \"" + d + "\",\n";
  117. ws += "\"state\": \"" + String(control_state_name()) + "\",\n";
  118. ws += F("\"valves\": [ ");
  119. for (int i = 0; i < VALVE_COUNT; i++) {
  120. ws += "\"";
  121. ws += get_plants()->getValves()->getPin(i) ? "1" : "0";
  122. ws += "\"";
  123. if (i < (VALVE_COUNT - 1)) {
  124. ws += ", ";
  125. }
  126. }
  127. ws += " ],\n";
  128. ws += F("\"pumps\": [ ");
  129. for (int i = 0; i < PUMP_COUNT; i++) {
  130. ws += "\"";
  131. ws += get_plants()->getPumps()->getPin(i) ? "1" : "0";
  132. ws += "\"";
  133. if (i < (PUMP_COUNT - 1)) {
  134. ws += ", ";
  135. }
  136. }
  137. ws += " ],\n";
  138. ws += F("\"switches\": [ ");
  139. for (int i = 0; i < SWITCH_COUNT; i++) {
  140. bool v = get_plants()->getSwitches()->getPin(i);
  141. #ifdef INVERT_SENSOR_BOTTOM
  142. if (i == 0) {
  143. v = !v;
  144. }
  145. #endif // INVERT_SENSOR_BOTTOM
  146. #ifdef INVERT_SENSOR_TOP
  147. if (i == 1) {
  148. v = !v;
  149. }
  150. #endif // INVERT_SENSOR_TOP
  151. ws += "\"";
  152. ws += v ? "1" : "0";
  153. ws += "\"";
  154. if (i < (SWITCH_COUNT - 1)) {
  155. ws += ", ";
  156. }
  157. }
  158. ws += " ],\n";
  159. ws += "\"switchstate\": \"";
  160. Plants::Waterlevel wl = get_plants()->getWaterlevel();
  161. if (wl == Plants::empty) {
  162. ws += F("tank empty");
  163. } else if (wl == Plants::inbetween) {
  164. ws += F("tank half-filled");
  165. } else if (wl == Plants::full) {
  166. ws += F("tank full");
  167. } else {
  168. ws += F("invalid sensor state");
  169. }
  170. ws += "\"\n";
  171. ws += "}";
  172. wifi_send_websocket(ws);
  173. }
  174. void wifi_send_websocket(String s) {
  175. socket.broadcastTXT(s);
  176. }
  177. void handleRoot() {
  178. String message = F("<!DOCTYPE html>\n");
  179. message += F("<html><head>\n");
  180. message += F("<meta charset='utf-8'/>\n");
  181. message += F("<meta name='viewport' content='width=device-width, initial-scale=1'/>\n");
  182. message += F("<title>Gieß-o-mat</title>\n");
  183. message += F("<style type='text/css'>\n");
  184. message += F(".container {\n");
  185. message += F("display: flex;\n");
  186. message += F("}\n");
  187. message += F(".ui {\n");
  188. message += F("width: max-content;\n");
  189. message += F("height: max-content;\n");
  190. message += F("margin-right: 1em;\n");
  191. message += F("padding: 0 1.0em;\n");
  192. message += F("border: 1px dashed black;\n");
  193. message += F("}\n");
  194. message += F(".io {\n");
  195. message += F("width: max-content;\n");
  196. message += F("height: max-content;\n");
  197. message += F("margin-right: 1em;\n");
  198. message += F("padding: 0.8em 1.0em;\n");
  199. message += F("border: 1px dashed black;\n");
  200. message += F("font-family: monospace;\n");
  201. message += F("}\n");
  202. message += F(".switch {\n");
  203. message += F("width: max-content;\n");
  204. message += F("border: 1px solid black;\n");
  205. message += F("border-radius: 50%;\n");
  206. message += F("padding: 2em;\n");
  207. message += F("margin: 1em;\n");
  208. message += F("}\n");
  209. message += F(".valve {\n");
  210. message += F("width: max-content;\n");
  211. message += F("border: 1px solid black;\n");
  212. message += F("border-radius: 50%;\n");
  213. message += F("padding: 2em;\n");
  214. message += F("margin: 1em;\n");
  215. message += F("}\n");
  216. message += F(".pump {\n");
  217. message += F("width: max-content;\n");
  218. message += F("border: 1px solid black;\n");
  219. message += F("border-radius: 50%;\n");
  220. message += F("padding: 2em;\n");
  221. message += F("margin: 1em;\n");
  222. message += F("}\n");
  223. message += F(".info {\n");
  224. message += F("width: max-content;\n");
  225. message += F("height: max-content;\n");
  226. message += F("padding: 0 1.0em;\n");
  227. message += F("border: 1px dashed black;\n");
  228. message += F("font-family: monospace;\n");
  229. message += F("}\n");
  230. message += F(".log {\n");
  231. message += F("max-height: 300px;\n");
  232. message += F("padding: 0 1.0em;\n");
  233. message += F("margin: 1em 0;\n");
  234. message += F("border: 1px dashed black;\n");
  235. message += F("font-family: monospace;\n");
  236. message += F("overflow-y: scroll;\n");
  237. message += F("word-break: break-all;\n");
  238. message += F("}\n");
  239. message += F("#logbuf {\n");
  240. message += F("white-space: break-spaces;\n");
  241. message += F("}\n");
  242. message += F(".pad {\n");
  243. message += F("background: black;\n");
  244. message += F("border: 3px solid black;\n");
  245. message += F("border-radius: 20px;\n");
  246. message += F("width: max-content;\n");
  247. message += F("padding: 1.5em;\n");
  248. message += F("margin-left: auto;\n");
  249. message += F("margin-right: auto;\n");
  250. message += F("}\n");
  251. message += F(".pad input {\n");
  252. message += F("background: #fff0cf;\n");
  253. message += F("border-radius: 6px;\n");
  254. message += F("font-weight: bold;\n");
  255. message += F("font-family: monospace;\n");
  256. message += F("font-size: 1.2em;\n");
  257. message += F("padding: 0.5em 1em;\n");
  258. message += F("margin: 0.5em;\n");
  259. message += F("}\n");
  260. // https://codepen.io/hawkz/pres/RpPaGK
  261. message += F(".lcd {\n");
  262. //message += F("background: #9ea18c;\n");
  263. message += F("background: #9ed18c;\n");
  264. message += F("border: 5px solid black;\n");
  265. message += F("border-radius: 10px;\n");
  266. message += F("width: max-content;\n");
  267. message += F("padding: 0.65em 1em;\n");
  268. message += F("box-shadow: inset 0 0 5px 5px rgba(0,0,0,.1);\n");
  269. message += F("font-weight: bold;\n");
  270. message += F("font-family: monospace;\n");
  271. message += F("letter-spacing: 0.1em;\n");
  272. message += F("font-size: 1.2em;\n");
  273. message += F("line-height: 160%;\n");
  274. message += F("color: #21230e;\n");
  275. message += F("text-shadow: -1px 2px 1px rgba(0,0,0,.1);\n");
  276. message += F("margin-left: auto;\n");
  277. message += F("margin-right: auto;\n");
  278. message += F("}\n");
  279. message += F("#state {\n");
  280. message += F("text-align: center;\n");
  281. message += F("}\n");
  282. message += F("</style>\n");
  283. message += F("</head><body>\n");
  284. message += F("<h1>Gieß-o-mat</h1>\n");
  285. message += F("<div class='container'>\n");
  286. message += F("<div class='ui'>\n");
  287. message += F("<pre class='lcd'>\n");
  288. message += message_buffer_a + '\n';
  289. message += message_buffer_b + '\n';
  290. message += message_buffer_c + '\n';
  291. message += message_buffer_d + '\n';
  292. message += F("</pre>\n");
  293. message += F("<form class='pad'>\n");
  294. message += F("<input type='button' value='1'>");
  295. message += F("<input type='button' value='2'>");
  296. message += F("<input type='button' value='3'>");
  297. message += F("<br>\n");
  298. message += F("<input type='button' value='4'>");
  299. message += F("<input type='button' value='5'>");
  300. message += F("<input type='button' value='6'>");
  301. message += F("<br>\n");
  302. message += F("<input type='button' value='7'>");
  303. message += F("<input type='button' value='8'>");
  304. message += F("<input type='button' value='9'>");
  305. message += F("<br>\n");
  306. message += F("<input type='button' value='*'>");
  307. message += F("<input type='button' value='0'>");
  308. message += F("<input type='button' value='#'>");
  309. message += F("</form>\n");
  310. message += F("<p id='state'>\n");
  311. message += F("State: ");
  312. message += control_state_name();
  313. message += F("</p></div>\n");
  314. message += F("<div class='io'>\n");
  315. message += F("Switches: <span id='switchstate'>");
  316. Plants::Waterlevel wl = get_plants()->getWaterlevel();
  317. if (wl == Plants::empty) {
  318. message += F("tank empty");
  319. } else if (wl == Plants::inbetween) {
  320. message += F("tank half-filled");
  321. } else if (wl == Plants::full) {
  322. message += F("tank full");
  323. } else {
  324. message += F("invalid sensor state");
  325. }
  326. message += F("</span>");
  327. message += F("<div class='container'>\n");
  328. for (int i = 0; i < SWITCH_COUNT; i++) {
  329. message += F("<div class='switch' style='background-color: ");
  330. bool v = get_plants()->getSwitches()->getPin(i);
  331. #ifdef INVERT_SENSOR_BOTTOM
  332. if (i == 0) {
  333. v = !v;
  334. }
  335. #endif // INVERT_SENSOR_BOTTOM
  336. #ifdef INVERT_SENSOR_TOP
  337. if (i == 1) {
  338. v = !v;
  339. }
  340. #endif // INVERT_SENSOR_TOP
  341. if (v) {
  342. message += F("red");
  343. } else {
  344. message += F("green");
  345. }
  346. message += F(";'>S");
  347. message += String(i + 1);
  348. message += F("</div>");
  349. }
  350. message += F("</div><hr>\n");
  351. message += F("Valves:\n");
  352. message += F("<div class='container'>\n");
  353. for (int i = 0; i < VALVE_COUNT; i++) {
  354. message += F("<div class='valve' style='background-color: ");
  355. if (get_plants()->getValves()->getPin(i)) {
  356. message += F("red");
  357. } else {
  358. message += F("green");
  359. }
  360. message += F(";'>V");
  361. message += String(i + 1);
  362. message += F("</div>");
  363. }
  364. message += F("</div><hr>\n");
  365. message += F("Pumps:\n");
  366. message += F("<div class='container'>\n");
  367. for (int i = 0; i < PUMP_COUNT; i++) {
  368. message += F("<div class='pump' style='background-color: ");
  369. if (get_plants()->getPumps()->getPin(i)) {
  370. message += F("red");
  371. } else {
  372. message += F("green");
  373. }
  374. message += F(";'>P");
  375. message += String(i + 1);
  376. message += F("</div>");
  377. }
  378. message += F("</div><hr>\n");
  379. message += F("Green means valve is closed / pump is off / switch is not submersed.\n");
  380. message += F("<br>\n");
  381. message += F("Red means valve is open / pump is running / switch is submersed.</div>\n");
  382. message += F("<div class='info'><p>\n");
  383. message += F("Version: ");
  384. message += FIRMWARE_VERSION;
  385. message += F("\n<br>\n");
  386. message += F("Build Date: ");
  387. message += __DATE__;
  388. message += F("\n<br>\n");
  389. message += F("Build Time: ");
  390. message += __TIME__;
  391. message += F("\n</p><p>\n");
  392. message += F("MAC: ");
  393. message += WiFi.macAddress();
  394. message += F("\n</p>\n");
  395. #if defined(ARDUINO_ARCH_ESP8266)
  396. message += F("\n<p>\n");
  397. message += F("Reset reason: ");
  398. message += ESP.getResetReason();
  399. message += F("\n<br>\n");
  400. message += F("Free heap: ");
  401. message += String(ESP.getFreeHeap());
  402. message += F(" (");
  403. message += String(ESP.getHeapFragmentation());
  404. message += F("% fragmentation)");
  405. message += F("\n<br>\n");
  406. message += F("Free sketch space: ");
  407. message += String(ESP.getFreeSketchSpace());
  408. message += F("\n<br>\n");
  409. message += F("Flash chip real size: ");
  410. message += String(ESP.getFlashChipRealSize());
  411. if (ESP.getFlashChipSize() != ESP.getFlashChipRealSize()) {
  412. message += F("\n<br>\n");
  413. message += F("WARNING: sdk chip size (");
  414. message += (ESP.getFlashChipSize());
  415. message += F(") does not match!");
  416. }
  417. message += F("\n</p>\n");
  418. #elif defined(ARDUINO_ARCH_ESP32)
  419. message += F("\n<p>\n");
  420. message += F("Free heap: ");
  421. message += String(ESP.getFreeHeap() / 1024.0);
  422. message += F("k\n<br>\n");
  423. message += F("Free sketch space: ");
  424. message += String(ESP.getFreeSketchSpace() / 1024.0);
  425. message += F("k\n<br>\n");
  426. message += F("Flash chip size: ");
  427. message += String(ESP.getFlashChipSize() / 1024.0);
  428. message += F("k\n</p>\n");
  429. #endif
  430. message += F("<p>\n");
  431. #ifdef ENABLE_INFLUXDB_LOGGING
  432. message += F("InfluxDB: ");
  433. message += INFLUXDB_DATABASE;
  434. message += F(" @ ");
  435. message += INFLUXDB_HOST;
  436. message += F(":");
  437. message += String(INFLUXDB_PORT);
  438. message += F("\n");
  439. #else
  440. message += F("InfluxDB logging not enabled!\n");
  441. #endif
  442. message += F("</p>\n");
  443. message += F("<p>Try <a href='/update'>/update</a> for OTA firmware updates!</p>\n");
  444. message += F("<p>Made by <a href='https://xythobuz.de'>xythobuz</a></p>\n");
  445. message += F("<p><a href='https://git.xythobuz.de/thomas/giess-o-mat'>Project Repository</a></p>\n");
  446. #ifdef ENABLE_GPIO_TEST
  447. message += F("<p><a href='/gpiotest'>GPIO Test</a></p>\n");
  448. #endif // ENABLE_GPIO_TEST
  449. message += F("</div></div>\n");
  450. message += F("<div class='log'><pre id='logbuf'>\n");
  451. message += debug.getBuffer();
  452. message += F("</pre></div>\n");
  453. message += F("</body>\n");
  454. message += F("<script type='text/javascript'>\n");
  455. message += F("var socket = new WebSocket('ws://' + window.location.hostname + ':81');\n");
  456. message += F("socket.onmessage = function(e) {\n");
  457. message += F( "if (e.data.startsWith('log:')) {\n");
  458. message += F( "var log = document.getElementById('logbuf');\n");
  459. message += F( "var div = document.getElementsByClassName('log')[0];\n");
  460. message += F( "log.innerHTML += e.data.substring(4);\n");
  461. message += F( "if (log.innerHTML.length > (1024 * 1024)) {\n");
  462. message += F( "log.innerHTML = log.innerHTML.substring(1024 * 1024);\n");
  463. message += F( "}\n");
  464. message += F( "div.scrollTop = div.scrollHeight;\n");
  465. message += F( "return;\n");
  466. message += F( "}\n");
  467. message += F( "var msg = JSON.parse(e.data);\n");
  468. message += F( "var str = msg.a + '\\n' + msg.b + '\\n' + msg.c + '\\n' + msg.d;\n");
  469. message += F( "console.log(str);\n");
  470. message += F( "var lcd = document.getElementsByClassName('lcd');\n");
  471. message += F( "lcd[0].innerHTML = str;\n");
  472. message += F( "var state = document.getElementById('state');\n");
  473. message += F( "state.innerHTML = \"State: \" + msg.state;\n");
  474. message += F( "for (let i = 0; i < ");
  475. message += String(VALVE_COUNT);
  476. message += F("; i++) {\n");
  477. message += F( "var valves = document.getElementsByClassName('valve');\n");
  478. message += F( "if (msg.valves[i] == '0') {\n");
  479. message += F( "valves[i].style = 'background-color: green;';\n");
  480. message += F( "} else {\n");
  481. message += F( "valves[i].style = 'background-color: red;';\n");
  482. message += F( "}\n");
  483. message += F( "}\n");
  484. message += F( "for (let i = 0; i < ");
  485. message += String(PUMP_COUNT);
  486. message += F("; i++) {\n");
  487. message += F( "var pumps = document.getElementsByClassName('pump');\n");
  488. message += F( "if (msg.pumps[i] == '0') {\n");
  489. message += F( "pumps[i].style = 'background-color: green;';\n");
  490. message += F( "} else {\n");
  491. message += F( "pumps[i].style = 'background-color: red;';\n");
  492. message += F( "}\n");
  493. message += F( "}\n");
  494. message += F( "for (let i = 0; i < ");
  495. message += String(SWITCH_COUNT);
  496. message += F("; i++) {\n");
  497. message += F( "var switches = document.getElementsByClassName('switch');\n");
  498. message += F( "if (msg.switches[i] == '0') {\n");
  499. message += F( "switches[i].style = 'background-color: green;';\n");
  500. message += F( "} else {\n");
  501. message += F( "switches[i].style = 'background-color: red;';\n");
  502. message += F( "}\n");
  503. message += F( "}\n");
  504. message += F( "var switchstate = document.getElementById('switchstate');\n");
  505. message += F( "switchstate.innerHTML = msg.switchstate;\n");
  506. message += F("};\n");
  507. message += F("var hist = document.getElementsByClassName('log')[0];\n");
  508. message += F("hist.scrollTop = hist.scrollHeight;\n");
  509. message += F("var buttons = document.getElementsByTagName('input');\n");
  510. message += F("for (let i = 0; i < buttons.length; i++) {\n");
  511. message += F( "buttons[i].addEventListener('click', updateButton);\n");
  512. message += F("}\n");
  513. message += F("function updateButton() {\n");
  514. message += F( "socket.send(this.value);\n");
  515. message += F("}\n");
  516. message += F("</script>\n");
  517. message += F("</html>\n");
  518. server.send(200, "text/html", message);
  519. }
  520. void webSocketEvent(uint8_t num, WStype_t type, uint8_t *payload, size_t length) {
  521. if ((type != WStype_TEXT) || (length != 1)) {
  522. return;
  523. }
  524. char c = payload[0];
  525. if ((c >= '0') && (c <= '9')) {
  526. control_act_input(c - '0');
  527. } else if (c == '*') {
  528. control_act_input(-1);
  529. } else if (c == '#') {
  530. control_act_input(-2);
  531. }
  532. }
  533. void wifi_setup() {
  534. // Build hostname string
  535. String hostname = "giess-o-mat";
  536. #if defined(ARDUINO_ARCH_ESP8266)
  537. // Connect to WiFi AP
  538. debug.println("WiFi: initializing");
  539. WiFi.hostname(hostname);
  540. WiFi.mode(WIFI_STA);
  541. debug.print("WiFi: connecting");
  542. WiFi.begin(WIFI_SSID, WIFI_PW);
  543. while (WiFi.status() != WL_CONNECTED) {
  544. debug.print(".");
  545. delay(LED_CONNECT_BLINK_INTERVAL);
  546. digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));
  547. }
  548. debug.println();
  549. #elif defined(ARDUINO_ARCH_ESP32)
  550. // Set hostname workaround
  551. debug.println("WiFi: set hostname");
  552. WiFi.config(INADDR_NONE, INADDR_NONE, INADDR_NONE);
  553. WiFi.setHostname(hostname.c_str());
  554. // Workaround for WiFi connecting only every 2nd reset
  555. // https://github.com/espressif/arduino-esp32/issues/2501#issuecomment-513602522
  556. debug.println("WiFi: connection work-around");
  557. WiFi.onEvent([](WiFiEvent_t event, WiFiEventInfo_t info) {
  558. if (info.disconnected.reason == 202) {
  559. esp_sleep_enable_timer_wakeup(10);
  560. esp_deep_sleep_start();
  561. delay(100);
  562. }
  563. }, WiFiEvent_t::SYSTEM_EVENT_STA_DISCONNECTED);
  564. // Connect to WiFi AP
  565. debug.println("WiFi: SSID=" WIFI_SSID);
  566. debug.print("WiFi: connecting");
  567. WiFi.mode(WIFI_STA);
  568. WiFi.begin(WIFI_SSID, WIFI_PW);
  569. while (WiFi.status() != WL_CONNECTED) {
  570. debug.print(".");
  571. delay(LED_CONNECT_BLINK_INTERVAL);
  572. digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));
  573. }
  574. debug.println();
  575. // Set hostname workaround
  576. debug.println("WiFi: set hostname work-around");
  577. WiFi.setHostname(hostname.c_str());
  578. #endif
  579. #ifdef ENABLE_INFLUXDB_LOGGING
  580. // Setup InfluxDB Client
  581. debug.println("WiFi: set InfluxDB database");
  582. influx.setDb(INFLUXDB_DATABASE);
  583. #endif // ENABLE_INFLUXDB_LOGGING
  584. // Setup HTTP Server
  585. debug.println("WiFi: initializing HTTP server");
  586. MDNS.begin(hostname.c_str());
  587. updater.setup(&server);
  588. server.on("/", handleRoot);
  589. #ifdef ENABLE_GPIO_TEST
  590. server.on("/gpiotest", handleGpioTest);
  591. #endif // ENABLE_GPIO_TEST
  592. server.begin();
  593. MDNS.addService("http", "tcp", 80);
  594. socket.begin();
  595. socket.onEvent(webSocketEvent);
  596. debug.println("WiFi: setup done");
  597. }
  598. void wifi_run() {
  599. if ((millis() - last_server_handle_time) >= SERVER_HANDLE_INTERVAL) {
  600. last_server_handle_time = millis();
  601. server.handleClient();
  602. socket.loop();
  603. #ifdef ARDUINO_ARCH_ESP8266
  604. MDNS.update();
  605. #endif // ARDUINO_ARCH_ESP8266
  606. }
  607. if ((millis() - last_websocket_update_time) >= WEBSOCKET_UPDATE_INTERVAL) {
  608. last_websocket_update_time = millis();
  609. wifi_send_status_broadcast();
  610. }
  611. // reset ESP every 6h to be safe
  612. if (millis() >= (6 * 60 * 60 * 1000)) {
  613. ESP.restart();
  614. }
  615. #ifdef ENABLE_GPIO_TEST
  616. if (runningGpioTest && ((millis() - lastGpioTime) >= GPIO_TEST_INTERVAL)) {
  617. gpioTestState = !gpioTestState;
  618. runGpioTest(gpioTestState);
  619. }
  620. #endif // ENABLE_GPIO_TEST
  621. }
  622. #endif // PLATFORM_ESP