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

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