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 19KB

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