thomas
/
giess-o-mat
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
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.
15 Commits
1 Branch
Tree: 9b3b5e21b4
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '9b3b5e21b4'
${ noResults }
giess-o-mat/src/WifiStuff.cpp

WifiStuff.cpp 19KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615 
  1. #include <Arduino.h>

  2. 

  3. #ifdef PLATFORM_ESP

  4. 

  5. #if defined(ARDUINO_ARCH_ESP8266)

  6. 

  7. #include <ESP8266WiFi.h>

  8. #include <ESP8266WebServer.h>

  9. #include <ESP8266mDNS.h>

  10. 

  11. #elif defined(ARDUINO_ARCH_ESP32)

  12. 

  13. #include <WiFi.h>

  14. #include <WebServer.h>

  15. #include <ESPmDNS.h>

  16. 

  17. #endif

  18. 

  19. #include <WebSocketsServer.h>

  20. 

  21. #include "wifi.h"

  22. #include "config.h"

  23. #include "config_pins.h"

  24. #include "Functionality.h"

  25. #include "SimpleUpdater.h"

  26. #include "DebugLog.h"

  27. #include "WifiStuff.h"

  28. 

  29. UPDATE_WEB_SERVER server(80);

  30. WebSocketsServer socket = WebSocketsServer(81);

  31. SimpleUpdater updater;

  32. unsigned long last_server_handle_time = 0;

  33. unsigned long last_websocket_update_time = 0;

  34. 

  35. String message_buffer_a;

  36. String message_buffer_b;

  37. String message_buffer_c;

  38. String message_buffer_d;

  39. 

  40. void wifi_set_message_buffer(String a, String b, String c, String d) {

  41.     message_buffer_a = a;

  42.     message_buffer_b = b;

  43.     message_buffer_c = c;

  44.     message_buffer_d = d;

  45. }

  46. 

  47. void wifi_schedule_websocket(void) {

  48.     last_websocket_update_time = 0;

  49. }

  50. 

  51. void wifi_send_status_broadcast(void) {

  52.     if (socket.connectedClients() <= 0) {

  53.         return;

  54.     }

  55.     

  56.     String a = message_buffer_a ;

  57.     String b = message_buffer_b;

  58.     String c = message_buffer_c;

  59.     String d = message_buffer_d;

  60.     

  61.     a.replace("\"", "'");

  62.     b.replace("\"", "'");

  63.     c.replace("\"", "'");

  64.     d.replace("\"", "'");

  65.     

  66.     String ws = "{\n";

  67.     

  68.     ws += "\"a\": \"" + a + "\",\n";

  69.     ws += "\"b\": \"" + b + "\",\n";

  70.     ws += "\"c\": \"" + c + "\",\n";

  71.     ws += "\"d\": \"" + d + "\",\n";

  72.     

  73.     ws += "\"state\": \"" + String(control_state_name()) + "\",\n";

  74.     

  75.     ws += F("\"valves\": [ ");

  76.     for (int i = 0; i < VALVE_COUNT; i++) {

  77.         ws += "\"";

  78.         ws += get_plants()->getValves()->getPin(i) ? "1" : "0";

  79.         ws += "\"";

  80.         

  81.         if (i < (VALVE_COUNT - 1)) {

  82.             ws += ", ";

  83.         }

  84.     }

  85.     ws += " ],\n";

  86.     

  87.     ws += F("\"pumps\": [ ");

  88.     for (int i = 0; i < PUMP_COUNT; i++) {

  89.         ws += "\"";

  90.         ws += get_plants()->getPumps()->getPin(i) ? "1" : "0";

  91.         ws += "\"";

  92.         

  93.         if (i < (PUMP_COUNT - 1)) {

  94.             ws += ", ";

  95.         }

  96.     }

  97.     ws += " ],\n";

  98.     

  99.     ws += F("\"switches\": [ ");

  100.     for (int i = 0; i < SWITCH_COUNT; i++) {

  101.         bool v = get_plants()->getSwitches()->getPin(i);

  102.         

  103. #ifdef INVERT_SENSOR_BOTTOM

  104.         if (i == 0) {

  105.             v = !v;

  106.         }

  107. #endif // INVERT_SENSOR_BOTTOM

  108.         

  109. #ifdef INVERT_SENSOR_TOP

  110.         if (i == 1) {

  111.             v = !v;

  112.         }

  113. #endif // INVERT_SENSOR_TOP

  114. 

  115.         ws += "\"";

  116.         ws += v ? "1" : "0";

  117.         ws += "\"";

  118.         

  119.         if (i < (SWITCH_COUNT - 1)) {

  120.             ws += ", ";

  121.         }

  122.     }

  123.     ws += " ],\n";

  124.     

  125.     ws += "\"switchstate\": \"";

  126.     Plants::Waterlevel wl = get_plants()->getWaterlevel();

  127.     if (wl == Plants::empty) {

  128.         ws += F("tank empty");

  129.     } else if (wl == Plants::inbetween) {

  130.         ws += F("tank half-filled");

  131.     } else if (wl == Plants::full) {

  132.         ws += F("tank full");

  133.     } else {

  134.         ws += F("invalid sensor state");

  135.     }

  136.     ws += "\"\n";

  137.     ws += "}";

  138.     

  139.     wifi_send_websocket(ws);

  140. }

  141. 

  142. void wifi_send_websocket(String s) {

  143.     socket.broadcastTXT(s);

  144. }

  145. 

  146. void handleRoot() {

  147.     String message = F("<!DOCTYPE html>\n");

  148.     message += F("<html><head>\n");

  149.     message += F("<meta charset='utf-8'/>\n");

  150.     message += F("<meta name='viewport' content='width=device-width, initial-scale=1'/>\n");

  151.     message += F("<title>Gieß-o-mat</title>\n");

  152.     

  153.     message += F("<style type='text/css'>\n");

  154.     message += F(".container {\n");

  155.     message += F("display: flex;\n");

  156.     message += F("}\n");

  157.     

  158.     message += F(".ui {\n");

  159.     message += F("width: max-content;\n");

  160.     message += F("height: max-content;\n");

  161.     message += F("margin-right: 1em;\n");

  162.     message += F("padding: 0 1.0em;\n");

  163.     message += F("border: 1px dashed black;\n");

  164.     message += F("}\n");

  165.     

  166.     message += F(".io {\n");

  167.     message += F("width: max-content;\n");

  168.     message += F("height: max-content;\n");

  169.     message += F("margin-right: 1em;\n");

  170.     message += F("padding: 0.8em 1.0em;\n");

  171.     message += F("border: 1px dashed black;\n");

  172.     message += F("font-family: monospace;\n");

  173.     message += F("}\n");

  174.     

  175.     message += F(".switch {\n");

  176.     message += F("width: max-content;\n");

  177.     message += F("border: 1px solid black;\n");

  178.     message += F("border-radius: 50%;\n");

  179.     message += F("padding: 2em;\n");

  180.     message += F("margin: 1em;\n");

  181.     message += F("}\n");

  182.     

  183.     message += F(".valve {\n");

  184.     message += F("width: max-content;\n");

  185.     message += F("border: 1px solid black;\n");

  186.     message += F("border-radius: 50%;\n");

  187.     message += F("padding: 2em;\n");

  188.     message += F("margin: 1em;\n");

  189.     message += F("}\n");

  190.     

  191.     message += F(".pump {\n");

  192.     message += F("width: max-content;\n");

  193.     message += F("border: 1px solid black;\n");

  194.     message += F("border-radius: 50%;\n");

  195.     message += F("padding: 2em;\n");

  196.     message += F("margin: 1em;\n");

  197.     message += F("}\n");

  198.     

  199.     message += F(".info {\n");

  200.     message += F("width: max-content;\n");

  201.     message += F("height: max-content;\n");

  202.     message += F("padding: 0 1.0em;\n");

  203.     message += F("border: 1px dashed black;\n");

  204.     message += F("font-family: monospace;\n");

  205.     message += F("}\n");

  206.     

  207.     message += F(".log {\n");

  208.     message += F("max-height: 300px;\n");

  209.     message += F("padding: 0 1.0em;\n");

  210.     message += F("margin: 1em 0;\n");

  211.     message += F("border: 1px dashed black;\n");

  212.     message += F("font-family: monospace;\n");

  213.     message += F("overflow-y: scroll;\n");

  214.     message += F("word-break: break-all;\n");

  215.     message += F("}\n");

  216.     

  217.     message += F("#logbuf {\n");

  218.     message += F("white-space: break-spaces;\n");

  219.     message += F("}\n");

  220.     

  221.     message += F(".pad {\n");

  222.     message += F("background: black;\n");

  223.     message += F("border: 3px solid black;\n");

  224.     message += F("border-radius: 20px;\n");

  225.     message += F("width: max-content;\n");

  226.     message += F("padding: 1.5em;\n");

  227.     message += F("margin-left: auto;\n");

  228.     message += F("margin-right: auto;\n");

  229.     message += F("}\n");

  230.     

  231.     message += F(".pad input {\n");

  232.     message += F("background: #fff0cf;\n");

  233.     message += F("border-radius: 6px;\n");

  234.     message += F("font-weight: bold;\n");

  235.     message += F("font-family: monospace;\n");

  236.     message += F("font-size: 1.2em;\n");

  237.     message += F("padding: 0.5em 1em;\n");

  238.     message += F("margin: 0.5em;\n");

  239.     message += F("}\n");

  240.     

  241.     // https://codepen.io/hawkz/pres/RpPaGK

  242.     message += F(".lcd {\n");

  243.     //message += F("background: #9ea18c;\n");

  244.     message += F("background: #9ed18c;\n");

  245.     message += F("border: 5px solid black;\n");

  246.     message += F("border-radius: 10px;\n");

  247.     message += F("width: max-content;\n");

  248.     message += F("padding: 0.65em 1em;\n");

  249.     message += F("box-shadow: inset 0 0 5px 5px rgba(0,0,0,.1);\n");

  250.     message += F("font-weight: bold;\n");

  251.     message += F("font-family: monospace;\n");

  252.     message += F("letter-spacing: 0.1em;\n");

  253.     message += F("font-size: 1.2em;\n");

  254.     message += F("line-height: 160%;\n");

  255.     message += F("color: #21230e;\n");

  256.     message += F("text-shadow: -1px 2px 1px rgba(0,0,0,.1);\n");

  257.     message += F("margin-left: auto;\n");

  258.     message += F("margin-right: auto;\n");

  259.     message += F("}\n");

  260.     

  261.     message += F("#state {\n");

  262.     message += F("text-align: center;\n");

  263.     message += F("}\n");

  264.     message += F("</style>\n");

  265.     

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

  267.     message += F("<h1>Gieß-o-mat</h1>\n");

  268.     

  269.     message += F("<div class='container'>\n");

  270.     message += F("<div class='ui'>\n");

  271.     message += F("<pre class='lcd'>\n");

  272.     message += message_buffer_a + '\n';

  273.     message += message_buffer_b + '\n';

  274.     message += message_buffer_c + '\n';

  275.     message += message_buffer_d + '\n';

  276.     message += F("</pre>\n");

  277.     

  278.     message += F("<form class='pad'>\n");

  279.     message += F("<input type='button' value='1'>");

  280.     message += F("<input type='button' value='2'>");

  281.     message += F("<input type='button' value='3'>");

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

  283.     message += F("<input type='button' value='4'>");

  284.     message += F("<input type='button' value='5'>");

  285.     message += F("<input type='button' value='6'>");

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

  287.     message += F("<input type='button' value='7'>");

  288.     message += F("<input type='button' value='8'>");

  289.     message += F("<input type='button' value='9'>");

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

  291.     message += F("<input type='button' value='*'>");

  292.     message += F("<input type='button' value='0'>");

  293.     message += F("<input type='button' value='#'>");

  294.     message += F("</form>\n");

  295.     

  296.     message += F("<p id='state'>\n");

  297.     message += F("State: ");

  298.     message += control_state_name();

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

  300.     

  301.     message += F("<div class='io'>\n");

  302.     message += F("Switches: <span id='switchstate'>");

  303.     

  304.     Plants::Waterlevel wl = get_plants()->getWaterlevel();

  305.     if (wl == Plants::empty) {

  306.         message += F("tank empty");

  307.     } else if (wl == Plants::inbetween) {

  308.         message += F("tank half-filled");

  309.     } else if (wl == Plants::full) {

  310.         message += F("tank full");

  311.     } else {

  312.         message += F("invalid sensor state");

  313.     }

  314.     message += F("</span>");

  315.     

  316.     message += F("<div class='container'>\n");

  317.     for (int i = 0; i < SWITCH_COUNT; i++) {

  318.         message += F("<div class='switch' style='background-color: ");

  319.         bool v = get_plants()->getSwitches()->getPin(i);

  320.         

  321. #ifdef INVERT_SENSOR_BOTTOM

  322.         if (i == 0) {

  323.             v = !v;

  324.         }

  325. #endif // INVERT_SENSOR_BOTTOM

  326.         

  327. #ifdef INVERT_SENSOR_TOP

  328.         if (i == 1) {

  329.             v = !v;

  330.         }

  331. #endif // INVERT_SENSOR_TOP

  332.         

  333.         if (v) {

  334.             message += F("red");

  335.         } else {

  336.             message += F("green");

  337.         }

  338.         message += F(";'>S");

  339.         message += String(i + 1);

  340.         message += F("</div>");

  341.     }

  342.     message += F("</div><hr>\n");

  343.     

  344.     message += F("Valves:\n");

  345.     message += F("<div class='container'>\n");

  346.     for (int i = 0; i < VALVE_COUNT; i++) {

  347.         message += F("<div class='valve' style='background-color: ");

  348.         if (get_plants()->getValves()->getPin(i)) {

  349.             message += F("red");

  350.         } else {

  351.             message += F("green");

  352.         }

  353.         message += F(";'>V");

  354.         message += String(i + 1);

  355.         message += F("</div>");

  356.     }

  357.     message += F("</div><hr>\n");

  358.     

  359.     message += F("Pumps:\n");

  360.     message += F("<div class='container'>\n");

  361.     for (int i = 0; i < PUMP_COUNT; i++) {

  362.         message += F("<div class='pump' style='background-color: ");

  363.         if (get_plants()->getPumps()->getPin(i)) {

  364.             message += F("red");

  365.         } else {

  366.             message += F("green");

  367.         }

  368.         message += F(";'>P");

  369.         message += String(i + 1);

  370.         message += F("</div>");

  371.     }

  372.     message += F("</div><hr>\n");

  373.     message += F("Green means valve is closed / pump is off / switch is not submersed.\n");

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

  375.     message += F("Red means valve is open / pump is running / switch is submersed.</div>\n");

  376.     

  377.     message += F("<div class='info'><p>\n");

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

  379.     message += FIRMWARE_VERSION;

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

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

  382.     message += WiFi.macAddress();

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

  384. 

  385. #if defined(ARDUINO_ARCH_ESP8266)

  386.     

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

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

  389.     message += ESP.getResetReason();

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

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

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

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

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

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

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

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

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

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

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

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

  402. 

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

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

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

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

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

  408.     }

  409.     

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

  411.     

  412. #elif defined(ARDUINO_ARCH_ESP32)

  413. 

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

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

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

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

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

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

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

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

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

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

  424.     

  425. #endif

  426. 

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

  428.     message += F("<p>Made by <a href='https://xythobuz.de'>xythobuz</a></p>\n");

  429.     message += F("<p><a href='https://git.xythobuz.de/thomas/giess-o-mat'>Project Repository</a></p>\n");

  430.     message += F("</div></div>\n");

  431.     

  432.     message += F("<div class='log'><pre id='logbuf'>\n");

  433.     message += debug.getBuffer();

  434.     message += F("</pre></div>\n");

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

  436.     

  437.     message += F("<script type='text/javascript'>\n");

  438.     message += F("var socket = new WebSocket('ws://' + window.location.hostname + ':81');\n");

  439.     message += F("socket.onmessage = function(e) {\n");

  440.     message += F(    "if (e.data.startsWith('log:')) {\n");

  441.     message += F(        "var log = document.getElementById('logbuf');\n");

  442.     message += F(        "var div = document.getElementsByClassName('log')[0];\n");

  443.     message += F(        "log.innerHTML += e.data.substring(4);\n");

  444.     message += F(        "if (log.innerHTML.length > (1024 * 1024)) {\n");

  445.     message += F(            "log.innerHTML = log.innerHTML.substring(1024 * 1024);\n");

  446.     message += F(        "}\n");

  447.     message += F(        "div.scrollTop = div.scrollHeight;\n");

  448.     message += F(        "return;\n");

  449.     message += F(    "}\n");

  450.     message += F(    "var msg = JSON.parse(e.data);\n");

  451.     message += F(    "var str = msg.a + '\\n' + msg.b + '\\n' + msg.c + '\\n' + msg.d;\n");

  452.     message += F(    "console.log(str);\n");

  453.     message += F(    "var lcd = document.getElementsByClassName('lcd');\n");

  454.     message += F(    "lcd[0].innerHTML = str;\n");

  455.     message += F(    "var state = document.getElementById('state');\n");

  456.     message += F(    "state.innerHTML = \"State: \" + msg.state;\n");

  457.     

  458.     message += F(    "for (let i = 0; i < ");

  459.     message += String(VALVE_COUNT);

  460.     message += F("; i++) {\n");

  461.     message += F(       "var valves = document.getElementsByClassName('valve');\n");

  462.     message += F(       "if (msg.valves[i] == '0') {\n");

  463.     message += F(           "valves[i].style = 'background-color: green;';\n");

  464.     message += F(       "} else {\n");

  465.     message += F(           "valves[i].style = 'background-color: red;';\n");

  466.     message += F(       "}\n");

  467.     message += F(    "}\n");

  468.     

  469.     message += F(    "for (let i = 0; i < ");

  470.     message += String(PUMP_COUNT);

  471.     message += F("; i++) {\n");

  472.     message += F(       "var pumps = document.getElementsByClassName('pump');\n");

  473.     message += F(       "if (msg.pumps[i] == '0') {\n");

  474.     message += F(           "pumps[i].style = 'background-color: green;';\n");

  475.     message += F(       "} else {\n");

  476.     message += F(           "pumps[i].style = 'background-color: red;';\n");

  477.     message += F(       "}\n");

  478.     message += F(    "}\n");

  479.     

  480.     message += F(    "for (let i = 0; i < ");

  481.     message += String(SWITCH_COUNT);

  482.     message += F("; i++) {\n");

  483.     message += F(       "var switches = document.getElementsByClassName('switch');\n");

  484.     message += F(       "if (msg.switches[i] == '0') {\n");

  485.     message += F(           "switches[i].style = 'background-color: green;';\n");

  486.     message += F(       "} else {\n");

  487.     message += F(           "switches[i].style = 'background-color: red;';\n");

  488.     message += F(       "}\n");

  489.     message += F(    "}\n");

  490.     

  491.     message += F(    "var switchstate = document.getElementById('switchstate');\n");

  492.     message += F(    "switchstate.innerHTML = msg.switchstate;\n");

  493.     message += F("};\n");

  494.     

  495.     message += F("var buttons = document.getElementsByTagName('input');\n");

  496.     message += F("for (let i = 0; i < buttons.length; i++) {\n");

  497.     message += F(    "buttons[i].addEventListener('click', updateButton);\n");

  498.     message += F("}\n");

  499.     message += F("function updateButton() {\n");

  500.     message += F(    "socket.send(this.value);\n");

  501.     message += F("}\n");

  502.     message += F("</script>\n");

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

  504. 

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

  506. }

  507. 

  508. void webSocketEvent(uint8_t num, WStype_t type, uint8_t *payload, size_t length) {

  509.     if ((type != WStype_TEXT) || (length != 1)) {

  510.         return;

  511.     }

  512.     

  513.     char c = payload[0];

  514.     if ((c >= '0') && (c <= '9')) {

  515.         control_act_input(c - '0');

  516.     } else if (c == '*') {

  517.         control_act_input(-1);

  518.     } else if (c == '#') {

  519.         control_act_input(-2);

  520.     }

  521. }

  522. 

  523. void wifi_setup() {

  524.     // Build hostname string

  525.     String hostname = "giess-o-mat";

  526. 

  527. #if defined(ARDUINO_ARCH_ESP8266)

  528. 

  529.     // Connect to WiFi AP

  530.     debug.println("WiFi: initializing");

  531.     WiFi.hostname(hostname);

  532.     WiFi.mode(WIFI_STA);

  533.     

  534.     debug.print("WiFi: connecting");

  535.     WiFi.begin(WIFI_SSID, WIFI_PW);

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

  537.         debug.print(".");

  538.         delay(LED_CONNECT_BLINK_INTERVAL);

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

  540.     }

  541.     debug.println();

  542.     

  543. #elif defined(ARDUINO_ARCH_ESP32)

  544. 

  545.     // Set hostname workaround

  546.     debug.println("WiFi: set hostname");

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

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

  549.     

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

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

  552.     debug.println("WiFi: connection work-around");

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

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

  555.             esp_sleep_enable_timer_wakeup(10);

  556.             esp_deep_sleep_start();

  557.             delay(100);

  558.         }

  559.     }, WiFiEvent_t::SYSTEM_EVENT_STA_DISCONNECTED);

  560. 

  561.     // Connect to WiFi AP

  562.     debug.println("WiFi: SSID=" WIFI_SSID);

  563.     debug.print("WiFi: connecting");

  564.     WiFi.mode(WIFI_STA);

  565.     WiFi.begin(WIFI_SSID, WIFI_PW);

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

  567.         debug.print(".");

  568.         delay(LED_CONNECT_BLINK_INTERVAL);

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

  570.     }

  571.     debug.println();

  572.     

  573.     // Set hostname workaround

  574.     debug.println("WiFi: set hostname work-around");

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

  576. 

  577. #endif

  578. 

  579.     // Setup HTTP Server

  580.     debug.println("WiFi: initializing HTTP server");

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

  582.     updater.setup(&server);

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

  584.     server.begin();

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

  586.     

  587.     socket.begin();

  588.     socket.onEvent(webSocketEvent);

  589.     

  590.     debug.println("WiFi: setup done");

  591. }

  592. 

  593. void wifi_run() {

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

  595.         last_server_handle_time = millis();

  596.         server.handleClient();

  597.         socket.loop();

  598.         

  599. #ifdef ARDUINO_ARCH_ESP8266

  600.         MDNS.update();

  601. #endif // ARDUINO_ARCH_ESP8266

  602.     }

  603.     

  604.     if ((millis() - last_websocket_update_time) >= WEBSOCKET_UPDATE_INTERVAL) {

  605.         last_websocket_update_time = millis();

  606.         wifi_send_status_broadcast();

  607.     }

  608.     

  609.     // reset ESP every 6h to be safe

  610.     if (millis() >= (6 * 60 * 60 * 1000)) {

  611.         ESP.restart();

  612.     }

  613. }

  614. 

  615. #endif // PLATFORM_ESP