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

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