web.cpp

   1 #include "Clock.h"
   2 #include <LittleFS.h>
   3 #include <ESPAsyncWebServer.h>
   4 #include <StreamString.h>
   5 #include <Ticker.h>
   6
   7 bool isApEnabled = false;
   8 bool isWebStarted = false;
   9
  10 bool pendingWiFi = false;
  11 bool pendingAuth = false;
  12
  13 AsyncWebServer server(80);
  14 AsyncEventSource events("/events");
  15
  16 Ticker tKeepalive;
  17
  18 char auth_user[32];
  19 char auth_pwd[32];
  20
  21 void reportChange(const __FlashStringHelper* name) {
  22   char buf[256];
  23   ConfigParameter* param = cfg.getParam(name);
  24   if (param) {
  25     switch (param->getType()) {
  26       case 'B':
  27         sprintf(buf,"{\"%s\":%s}", name, param->getBoolValue()?"true":"false");
  28         break;
  29       case 'I':
  30         sprintf(buf,"{\"%s\":%d}", name, param->getIntValue());
  31         break;
  32       case 'F':
  33         sprintf(buf,"{\"%s\":%f}", name, param->getFloatValue());
  34         break;
  35       case 'S':
  36         sprintf(buf,"{\"%s\":\"%s\"}", name, param->getCharValue());
  37         break;
  38     }
  39     events.send(buf,"update",millis());
  40   }
  41 }
  42
  43 void sendInitial(AsyncEventSourceClient *client) {
  44   String mac = WiFi.macAddress();
  45   char buf[256];
  46   sprintf(buf,"{\"_mac\":\"%s\",\"_weather\":\"%s\"}", mac.c_str(), weatherData);
  47   client->send(buf,"keepalive",millis());
  48   mac = String();
  49 }
  50
  51 void sendWeather() {
  52   char buf[256];
  53   sprintf(buf,"{\"_weather\":\"%s\"}",weatherData);
  54   events.send(buf,"keepalive",millis());
  55 }
  56
  57 void sendKeepalive() {
  58   static unsigned long lastMillis = 0;
  59   static unsigned long uptimeCorrection = 0;
  60   unsigned long currentMillis = millis();
  61   unsigned long uptime = millis()/1000;
  62   if (currentMillis < lastMillis) {
  63     uptimeCorrection += lastMillis/1000;
  64   }
  65   lastMillis = millis();
  66   uptime = uptime + uptimeCorrection;
  67   int days = uptime / 86400;
  68   uptime = uptime % 86400;
  69   int hrs = uptime / 3600;
  70   uptime = uptime % 3600;
  71   int mins = uptime / 60;
  72   uptime = uptime % 60;
  73   int heap = ESP.getFreeHeap();
  74   int rssi = WiFi.RSSI();
  75   struct tm* timeinfo = localtime(&last_sync);
  76   bool changed = cfg.getTimestamp() != 0;
  77   char sync[16] = "--:--:--";
  78   if (last_sync) {
  79     sprintf(sync, "%02d:%02d:%02d", timeinfo->tm_hour, timeinfo->tm_min, timeinfo->tm_sec);
  80   }
  81   char buf[256];
  82   if (days) {
  83     sprintf(buf,"{\"_uptime\":\"%d д %d ч %d % м %d с\", \"_date\":\"%02d.%2d.%04d\", \"_time\":\"%02d:%02d\",\"_heap\":\"%d б\", \"_rssi\":\"%d\",  \"_last_sync\":\"%s\", \"_changed\":%s}", days, hrs, mins, uptime, dd, mm, yy, hh, mi, heap, rssi, sync, changed?"true":"false");
  84   } else if (hrs) {
  85     sprintf(buf,"{\"_uptime\":\"%d ч %d % м %d с\", \"_date\":\"%02d.%2d.%04d\", \"_time\":\"%02d:%02d\",\"_heap\":\"%d б\", \"_rssi\":\"%d\",  \"_last_sync\":\"%s\", \"_changed\":%s}", hrs, mins, uptime, dd, mm, yy, hh, mi, heap, rssi, sync, changed?"true":"false");
  86   } else {
  87     sprintf(buf,"{\"_uptime\":\"%d м %d с\", \"_date\":\"%02d.%2d.%04d\", \"_time\":\"%02d:%02d\",\"_heap\":\"%d б\", \"_rssi\":\"%d\",  \"_last_sync\":\"%s\", \"_changed\":%s}", mins, uptime, dd, mm, yy, hh, mi, heap, rssi, sync, changed?"true":"false");
  88   }
  89   events.send(buf,"keepalive",millis());
  90 }
  91
  92 void apply(const char* name) {
  93   if (strcmp(name,"sta_ssid") == 0 || strcmp(name,"sta_psk") == 0) {
  94     if (!isApEnabled) {
  95       pendingWiFi = true;
  96     }
  97   } else if (strcmp(name,"ap_ssid") == 0 || strcmp(name,"ap_psk") == 0) {
  98     if (isApEnabled) {
  99       pendingWiFi = true;
 100     }
 101   } else if (strcmp(name,"auth_user") == 0 || strcmp(name,"auth_pwd") == 0) {
 102     pendingAuth = true;
 103   } else if (strcmp(name,"ntp_server") == 0 || strcmp(name,"tz") == 0) {
 104     setupTime();
 105   } else if (strcmp(name,"pin_sda") == 0 || strcmp(name,"pin_scl") == 0 ||
 106     strcmp(name,"i2c_speed") == 0 || strcmp(name,"enable_rtc") == 0 ||
 107     strcmp(name,"enable_button") == 0 || strcmp(name,"button_pin") == 0 || strcmp(name,"button_inversed") == 0 ||
 108     strcmp(name,"enable_buzzer") == 0 || strcmp(name,"buzzer_pin") == 0) {
 109     setupHardware();
 110   } else if (strcmp(name,"pin_din") == 0 || strcmp(name,"pin_clk") == 0 ||
 111     strcmp_P(name,"pin_cs") == 0 || strcmp(name,"led_modules") == 0){
 112     setupPanel();
 113   } else if (strcmp(name,"panel_brightness_day") == 0 || strcmp_P(name,"panel_brightness_night") == 0 ||
 114     strcmp_P(name,"day_from") == 0 || strcmp(name,"night_from") == 0){
 115     setPanelBrightness();
 116   }
 117 }
 118
 119 char* actionScheduled = nullptr;
 120 unsigned long millisScheduled = 0;
 121
 122 void setupWeb() {
 123   char buf[256];
 124
 125   if (isWebStarted) {
 126     tKeepalive.detach();
 127     server.end();
 128   }
 129
 130   isWebStarted = true;
 131
 132   Serial.println("Setting authentication...");
 133   strncpy(auth_user,cfg.getCharValue(F("auth_user")),31);
 134   strncpy(auth_pwd,cfg.getCharValue(F("auth_pwd")),31);
 135
 136   server.on("/action", HTTP_GET, [](AsyncWebServerRequest* request) {
 137     if (auth_user && auth_pwd && auth_user[0] && auth_pwd[0] && !request->authenticate(auth_user, auth_pwd)) {
 138       Serial.print("Failed auth as user: "); Serial.print(auth_user); Serial.print(':'); Serial.print(auth_pwd); Serial.println('!');
 139       return request-> requestAuthentication();
 140     }
 141     if(request->hasParam("name")) {
 142       const char* action = request->getParam("name")->value().c_str();
 143       if (strcmp(action,"restart") == 0) {
 144         if (pendingWiFi) {
 145           request->send(200,"application/json", "{\"result\":\"FAILED\",\"message\":\"Не применены настройки WiFi\", \"page\":\"wifi\"}");
 146         } else if (pendingAuth) {
 147           request->send(200,"application/json", "{\"result\":\"FAILED\",\"message\":\"Не применены настройки авторизации\", \"page\":\"system\"}");
 148         } else {
 149           request->send(200,"application/json", "{\"result\":\"OK\",\"message\":\"Перезагружаюсь\"}");
 150           millisScheduled = millis();
 151           actionScheduled = "restart";
 152         }
 153       } else if (strcmp(action,"wifi") == 0) {
 154         if (pendingAuth) {
 155           request->send(200,"application/json", "{\"result\":\"FAILED\",\"message\":\"Не применены настройки авторизации\", \"page\":\"system\"}");
 156         } else {
 157           request->send(200,"application/json", "{\"result\":\"OK\",\"message\":\"Применяю настройки\"}");
 158           millisScheduled = millis();
 159           actionScheduled = "wifi";
 160         }
 161       } else if (strcmp(action,"auth") == 0) {
 162         request->send(200,"application/json", "{\"result\":\"OK\",\"message\":\"Применяю настройки\"}");
 163         millisScheduled = millis();
 164         actionScheduled = "auth";
 165       } else if (strcmp(action,"save") == 0) {
 166         if (pendingWiFi) {
 167           request->send(200,"application/json", "{\"result\":\"FAILED\",\"message\":\"Не применены настройки WiFi\", \"page\":\"wifi\"}");
 168         } else if (pendingAuth) {
 169           request->send(200,"application/json", "{\"result\":\"FAILED\",\"message\":\"Не применены настройки авторизации\", \"page\":\"system\"}");
 170         } else {
 171           request->send(200,"application/json", "{\"result\":\"OK\",\"message\":\"Сохраняю настройки\"}");
 172           millisScheduled = millis();
 173           actionScheduled = "save";
 174         }
 175       }
 176     }
 177   });
 178
 179   server.on("/wifi/scan", HTTP_GET, [](AsyncWebServerRequest* request) {
 180     if (auth_user && auth_pwd && auth_user[0] && auth_pwd[0] && !request->authenticate(auth_user, auth_pwd)) {
 181       return request-> requestAuthentication();
 182     }
 183     String json = "[";
 184     int n = WiFi.scanComplete();
 185     if (n == -2) {
 186       WiFi.scanNetworks(true);
 187     } else if (n) {
 188       for (int i = 0; i < n; ++i) {
 189         if (i) json += ",";
 190         json += "{";
 191         json += "\"rssi\":" + String(WiFi.RSSI(i));
 192         json += ",\"ssid\":\"" + WiFi.SSID(i) + "\"";
 193         json += ",\"bssid\":\"" + WiFi.BSSIDstr(i) + "\"";
 194         json += ",\"channel\":" + String(WiFi.channel(i));
 195         json += ",\"secure\":" + String(WiFi.encryptionType(i));
 196         json += ",\"hidden\":" + String(WiFi.isHidden(i) ? "true" : "false");
 197         json += "}";
 198       }
 199       WiFi.scanDelete();
 200       if (WiFi.scanComplete() == -2) {
 201         WiFi.scanNetworks(true);
 202       }
 203     }
 204     json += "]";
 205     request->send(200, "application/json", json);
 206     json = String();
 207   });
 208
 209   server.on("/config/get", HTTP_GET, [](AsyncWebServerRequest* request) {
 210     if (auth_user && auth_pwd && auth_user[0] && auth_pwd[0] && !request->authenticate(auth_user, auth_pwd)) {
 211       return request-> requestAuthentication();
 212     }
 213     AsyncResponseStream* s = request->beginResponseStream("application/json");
 214     s->print("{");
 215     for (int i = 0; i < cfg.getParametersCount(); i++) {
 216       ConfigParameter* param = cfg.getParameter(i);
 217       if (i) s->print(",");
 218       s->print("\"");
 219       s->print(param->getID());
 220       s->print("\":");
 221       switch (param->getType()) {
 222         case 'B':
 223           s->print(param->getBoolValue() ? "true" : "false");
 224           break;
 225         case 'I':
 226           s->print(param->getIntValue());
 227           break;
 228         case 'F':
 229           s->print(param->getFloatValue());
 230           break;
 231         case 'S':
 232           s->print("\"");
 233           s->print(param->getCharValue());
 234           s->print("\"");
 235           break;
 236       }
 237     }
 238     s->print("}");
 239     request->send(s);
 240   });
 241
 242   server.on("/config/set", HTTP_GET, [](AsyncWebServerRequest* request) {
 243     if (auth_user && auth_pwd && auth_user[0] && auth_pwd[0] && !request->authenticate(auth_user, auth_pwd)) {
 244       return request-> requestAuthentication();
 245     }
 246     if(request->hasParam("name") && (request->hasParam("value"))) {
 247       const char* name = request->getParam("name")->value().c_str();
 248       const char* value = request->getParam("value")->value().c_str();
 249       Serial.print(name); Serial.print(" = "); Serial.println(value);
 250       char buf[256];
 251       ConfigParameter* param = cfg.getParam(name);
 252       if (param) {
 253         switch (param->getType()) {
 254           case 'B':
 255             cfg.setValue(name, strcmp(value, "true")==0);
 256             sprintf(buf,"{\"%s\":%s}", name, param->getBoolValue()?"true":"false");
 257             break;
 258           case 'I':
 259             cfg.setValue(name, atoi(value));
 260             sprintf(buf,"{\"%s\":%d}", name, param->getIntValue());
 261             break;
 262           case 'F':
 263             cfg.setValue(name, atof(value));
 264             sprintf(buf,"{\"%s\":%f}", name, param->getFloatValue());
 265             break;
 266           case 'S':
 267             cfg.setValue(name, value);
 268             sprintf(buf,"{\"%s\":\"%s\"}", name, param->getCharValue());
 269             break;
 270         }
 271         apply(name);
 272       } else {
 273           request->send(500, "text/plain", "Unknown parameter name");
 274           return;
 275       }
 276       request->send(200,"application/json",buf);
 277      } else {
 278       request->send(500, "text/plain", "Not all parameters set");
 279     }
 280   });
 281
 282   server.serveStatic("ui", LittleFS, "/ui.json");
 283
 284   server.serveStatic("/", LittleFS, "/web/").setDefaultFile("index.html");
 285
 286   server.onNotFound([](AsyncWebServerRequest *request){
 287     request->send(404,"text/plain","Not found");
 288   });
 289
 290   events.onConnect([](AsyncEventSourceClient *client){
 291     sendInitial(client);
 292   });
 293
 294   server.addHandler(&events);
 295
 296   server.begin();
 297
 298   tKeepalive.attach(2, sendKeepalive);
 299
 300 }
 301
 302 #define CFG_AUTOSAVE 15
 303
 304 void tickWeb() {
 305   static char storedSSID[64];
 306   static char storedPSK[64];
 307   static bool connectInProgress = false;
 308   static unsigned long connectMillis = 0;
 309   if (actionScheduled && millis()>millisScheduled+300) {
 310     Serial.print(F("Scheduled action ")); Serial.println(actionScheduled);
 311     //
 312     if (strcmp(actionScheduled,"restart") == 0) {
 313       server.end();
 314       reboot();
 315     } else if (strcmp(actionScheduled,"auth") == 0) {
 316       Serial.println("New authentication credentials");
 317       strncpy(auth_user,cfg.getCharValue(F("auth_user")),31);
 318       strncpy(auth_pwd,cfg.getCharValue(F("auth_pwd")),31);
 319       pendingAuth = false;
 320     } else if (strcmp(actionScheduled,"wifi") == 0) {
 321       Serial.println("New wifi credentials");
 322       strcpy(storedSSID,WiFi.SSID().c_str());
 323       strcpy(storedPSK,WiFi.psk().c_str());
 324       WiFi.mode(WIFI_STA);
 325       WiFi.begin(cfg.getCharValue("sta_ssid"),cfg.getCharValue("sta_psk"));
 326       connectInProgress = true;
 327       connectMillis = millis();
 328     } else if (strcmp(actionScheduled,"save") == 0 && !pendingWiFi && !pendingAuth) {
 329       saveConfig();
 330     }
 331     actionScheduled = nullptr;
 332   }
 333   if (connectInProgress && (millis() > connectMillis + 1000)) {
 334     if (WiFi.status() == WL_CONNECTED) {
 335       char buf[64];
 336       sprintf(buf,"Подключен к %s, IP=%s", WiFi.SSID(), WiFi.localIP().toString().c_str());
 337       Serial.println(buf);
 338       message(buf,1);
 339       pendingWiFi = false;
 340       connectInProgress = false;
 341     } else if (WiFi.status() == WL_CONNECT_FAILED || WiFi.status() == WL_NO_SSID_AVAIL || ((WiFi.status() == WL_CONNECTION_LOST || WiFi.status() == WL_DISCONNECTED) && (millis()>connectMillis+1000*cfg.getIntValue("sta_wait")))) {
 342       Serial.println(F("Подключение не удалось, возвращаю прежние настройки"));
 343       message(F("Подключение не удалось, возвращаю прежние настройки"),1);
 344       WiFi.begin(storedSSID, storedPSK);
 345       connectInProgress = false;
 346     }
 347   }
 348
 349   if (!pendingWiFi && !pendingAuth && cfg.getTimestamp() && cfg.getTimestamp() < now - CFG_AUTOSAVE) {
 350     saveConfig();
 351     Serial.println(F("Настройки сохранены"));
 352   }
 353 }