esp8266_milight_hub/lib/WebServer/MiLightHttpServer.cpp

#include <FS.h>
#include <WiFiUdp.h>
#include <IntParsing.h>
#include <Settings.h>
#include <MiLightHttpServer.h>
#include <MiLightRadioConfig.h>
#include <GithubDownloader.h>

void MiLightHttpServer::begin() {
  applySettings(settings);
  
  server.on("/", HTTP_GET, handleServeFile(WEB_INDEX_FILENAME, "text/html"));
  server.on("/settings", HTTP_GET, handleServeFile(SETTINGS_FILE, "application/json"));
  server.on("/settings", HTTP_PUT, [this]() { handleUpdateSettings(); });
  server.on("/settings", HTTP_POST, [this]() { server.send(200, "text/plain", "success"); }, handleUpdateFile(SETTINGS_FILE));
  server.on("/radio_configs", HTTP_GET, [this]() { handleGetRadioConfigs(); });
  server.onPattern("/gateway_traffic/:type", HTTP_GET, [this](const UrlTokenBindings* b) { handleListenGateway(b); });
  server.onPattern("/gateways/:device_id/:type/:group_id", HTTP_PUT, [this](const UrlTokenBindings* b) { handleUpdateGroup(b); });
  server.onPattern("/gateways/:device_id/:type", HTTP_PUT, [this](const UrlTokenBindings* b) { handleUpdateGateway(b); });
  server.onPattern("/send_raw/:type", HTTP_PUT, [this](const UrlTokenBindings* b) { handleSendRaw(b); });
  server.onPattern("/download_update/:component", HTTP_GET, [this](const UrlTokenBindings* b) { handleDownloadUpdate(b); });
  server.on("/web", HTTP_POST, [this]() { server.send(200, "text/plain", "success"); }, handleUpdateFile(WEB_INDEX_FILENAME));
  server.on("/about", HTTP_GET, [this]() { handleAbout(); });
  server.on("/firmware", HTTP_POST, 
    [this](){
      server.sendHeader("Connection", "close");
      server.sendHeader("Access-Control-Allow-Origin", "*");
      server.send(200, "text/plain", (Update.hasError())?"FAIL":"OK");
      ESP.restart();
    },
    [this](){
      HTTPUpload& upload = server.upload();
      if(upload.status == UPLOAD_FILE_START){
        WiFiUDP::stopAll();
        uint32_t maxSketchSpace = (ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000;
        if(!Update.begin(maxSketchSpace)){//start with max available size
          Update.printError(Serial);
        }
      } else if(upload.status == UPLOAD_FILE_WRITE){
        if(Update.write(upload.buf, upload.currentSize) != upload.currentSize){
          Update.printError(Serial);
        }
      } else if(upload.status == UPLOAD_FILE_END){
        if(Update.end(true)){ //true to set the size to the current progress
        } else {
          Update.printError(Serial);
        }
      }
      yield();
    }
  );
  
  server.begin();
}

void MiLightHttpServer::handleClient() {
  server.handleClient();
}

void MiLightHttpServer::handleDownloadUpdate(const UrlTokenBindings* bindings) {
  GithubDownloader* downloader = new GithubDownloader();
  const String& component = bindings->get("component");
  
  if (component.equalsIgnoreCase("web")) {
    Serial.println("Attempting to update web UI...");
    
    const bool result = downloader->downloadFile(
      MILIGHT_GITHUB_USER,
      MILIGHT_GITHUB_REPO,
      MILIGHT_REPO_WEB_PATH,
      WEB_INDEX_FILENAME
    );
    
    Serial.println("Download complete!");
    
    if (result) {
      server.sendHeader("Location", "/");
      server.send(302);
    } else {
      server.send(500, "text/plain", "Failed to download update from Github. Check serial logs for more information.");
    }
  } else if (component.equalsIgnoreCase("firmware")) {
    String firmwareFile = String("dist/firmware-") + FIRMWARE_VARIANT + ".bin";
    Stream& stream = downloader->streamFile(
      MILIGHT_GITHUB_USER,
      MILIGHT_GITHUB_REPO,
      firmwareFile
    );
    
    if (stream.available()) {
      WiFiUDP::stopAll();
      uint32_t maxSketchSpace = (ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000;
      
      if(!Update.begin(maxSketchSpace)){//start with max available size
        Update.printError(Serial);
        return;
      }
      
      size_t read = 0;
      
      while (stream.available()) {
        size_t l = stream.readBytes(downloader->buffer, GITHUB_DOWNLOADER_BUFFER_SIZE);
        read += l;
    
        Serial.print("Read ");
        Serial.print(l);
        Serial.print(" bytes from Github (");
        Serial.print(read);
        Serial.println(" total)");
        
        if (Update.write(downloader->buffer, l) != l) {
          Update.printError(Serial);
          return;
        }
        
        Serial.println("Write to flash");
        
        yield();
      }
      
      Serial.println("Update complete");
      
      if (! Update.end(true)) {
        server.send(200, "text/plain", String(FIRMWARE_VARIANT));
      } else {
        Serial.println("Error completing update");
      }
    } else {
      server.send(500, "text/plain", "Couldn't open strem to firmware file on Github. Check serial logs.");
    }
  } else {
    String body = String("Unknown component: ") + component;
    server.send(400, "text/plain", body);
  }
  
  delete downloader;
}

void MiLightHttpServer::applySettings(Settings& settings) {
  if (server.authenticationRequired() && !settings.hasAuthSettings()) {
    server.disableAuthentication();
  } else {
    server.requireAuthentication(settings.adminUsername, settings.adminPassword);
  }
  
  milightClient->setResendCount(settings.packetRepeats);
}

void MiLightHttpServer::onSettingsSaved(SettingsSavedHandler handler) {
  this->settingsSavedHandler = handler;
}
  
void MiLightHttpServer::handleAbout() {
  DynamicJsonBuffer buffer;
  JsonObject& response = buffer.createObject();
  
  response["version"] = MILIGHT_HUB_VERSION;
  response["variant"] = FIRMWARE_VARIANT;
  
  String body;
  response.printTo(body);
  
  server.send(200, "application", body); 
}
  
void MiLightHttpServer::handleGetRadioConfigs() {
  DynamicJsonBuffer buffer;
  JsonArray& arr = buffer.createArray();
  
  for (size_t i = 0; i < MiLightRadioConfig::NUM_CONFIGS; i++) {
    const MiLightRadioConfig* config = MiLightRadioConfig::ALL_CONFIGS[i];
    arr.add(config->name);
  }
  
  String body;
  arr.printTo(body);
  
  server.send(200, "application/json", body);
}
  
ESP8266WebServer::THandlerFunction MiLightHttpServer::handleServeFile(
  const char* filename, 
  const char* contentType, 
  const char* defaultText) {
    
  return [this, filename, contentType, defaultText]() {
    if (!serveFile(filename)) {
      if (defaultText) {
        server.send(200, contentType, defaultText);
      } else {
        server.send(404);
      }
    }
  };
}

bool MiLightHttpServer::serveFile(const char* file, const char* contentType) {
  if (SPIFFS.exists(file)) {
    File f = SPIFFS.open(file, "r");
    server.send(200, contentType, f.readString());
    f.close();
    return true;
  }
  
  return false;
}

ESP8266WebServer::THandlerFunction MiLightHttpServer::handleUpdateFile(const char* filename) {
  return [this, filename]() {
    HTTPUpload& upload = server.upload();
    
    if (upload.status == UPLOAD_FILE_START) {
      updateFile = SPIFFS.open(filename, "w");
    } else if(upload.status == UPLOAD_FILE_WRITE){
      if (updateFile.write(upload.buf, upload.currentSize) != upload.currentSize) {
        Serial.println("Error updating web file");
      }
    } else if (upload.status == UPLOAD_FILE_END) {
      updateFile.close();
    }
  };
}

void MiLightHttpServer::handleUpdateSettings() {
  DynamicJsonBuffer buffer;
  const String& rawSettings = server.arg("plain");
  JsonObject& parsedSettings = buffer.parse(rawSettings);
  
  if (parsedSettings.success()) {
    settings.patch(parsedSettings);
    settings.save();
    
    this->applySettings(settings);
    this->settingsSavedHandler();
    
    server.send(200, "application/json", "true");
  } else {
    server.send(400, "application/json", "\"Invalid JSON\"");
  }
}

void MiLightHttpServer::handleListenGateway(const UrlTokenBindings* bindings) {
  bool available = false;
  MiLightRadioConfig* config = MiLightRadioConfig::fromString(bindings->get("type"));
  
  if (config == NULL) {
    String body = "Unknown device type: ";
    body += bindings->get("type");
    
    server.send(400, "text/plain", body);
    return;
  }
  
  milightClient->prepare(*config, 0, 0);
  
  while (!available) {
    if (!server.clientConnected()) {
      return;
    }
    
    if (milightClient->available()) {
      available = true;
    }
    
    yield();
  }
  
  uint8_t packet[config->getPacketLength()];
  milightClient->read(packet);
  
  String response = "Packet received (";
  response += String(sizeof(packet)) + " bytes)";
  response += ":\n";
  
  char ppBuffer[200];
  milightClient->formatPacket(packet, ppBuffer);
  response += String(ppBuffer);
  
  response += "\n\n";
  
  server.send(200, "text/plain", response);
}

void MiLightHttpServer::handleUpdateGroup(const UrlTokenBindings* urlBindings) {
  DynamicJsonBuffer buffer;
  JsonObject& request = buffer.parse(server.arg("plain"));
  
  if (!request.success()) {
    server.send(400, "text/plain", "Invalid JSON");
    return;
  }
  
  const uint16_t deviceId = parseInt<uint16_t>(urlBindings->get("device_id"));
  const uint8_t groupId = urlBindings->get("group_id").toInt();
  MiLightRadioConfig* config = MiLightRadioConfig::fromString(urlBindings->get("type"));
  
  if (config == NULL) {
    String body = "Unknown device type: ";
    body += urlBindings->get("type");
    
    server.send(400, "text/plain", body);
    return;
  }
  
  milightClient->setResendCount(
    settings.httpRepeatFactor * settings.packetRepeats
  );
  milightClient->prepare(*config, deviceId, groupId);
  
  if (request.containsKey("status")) {
    const String& statusStr = request.get<String>("status");
    MiLightStatus status = (statusStr == "on" || statusStr == "true") ? ON : OFF;
    milightClient->updateStatus(status);
  }
      
  if (request.containsKey("command")) {
    if (request["command"] == "unpair") {
      milightClient->unpair();
    }
    
    if (request["command"] == "pair") {
      milightClient->pair();
    }
    
    if (request["command"] == "set_white") {
      milightClient->updateColorWhite();
    }
    
    if (request["command"] == "level_up") {
      milightClient->increaseBrightness();
    }
    
    if (request["command"] == "level_down") {
      milightClient->decreaseBrightness();
    }
    
    if (request["command"] == "temperature_up") {
      milightClient->increaseTemperature();
    }
    
    if (request["command"] == "temperature_down") {
      milightClient->decreaseTemperature();
    }
    
    if (request["command"] == "next_mode") {
      milightClient->nextMode();
    }
    
    if (request["command"] == "previous_mode") {
      milightClient->previousMode();
    }
    
    if (request["command"] == "mode_speed_down") {
      milightClient->modeSpeedDown();
    }
    
    if (request["command"] == "mode_speed_up") {
      milightClient->modeSpeedUp();
    }
  }
  
  if (request.containsKey("hue")) {
    milightClient->updateHue(request["hue"]);
  }
    
  if (request.containsKey("level")) {
    milightClient->updateBrightness(request["level"]);
  }
    
  if (request.containsKey("temperature")) {
    milightClient->updateTemperature(request["temperature"]);
  }
  
  if (request.containsKey("saturation")) {
    milightClient->updateSaturation(request["saturation"]);
  }
  
  if (request.containsKey("mode")) {
    milightClient->updateMode(request["mode"]);
  }
  
  milightClient->setResendCount(settings.packetRepeats);
  
  server.send(200, "application/json", "true");
}

void MiLightHttpServer::handleUpdateGateway(const UrlTokenBindings* urlBindings) {
  DynamicJsonBuffer buffer;
  JsonObject& request = buffer.parse(server.arg("plain"));
  
  const uint16_t deviceId = parseInt<uint16_t>(urlBindings->get("device_id"));
  MiLightRadioConfig* config = MiLightRadioConfig::fromString(urlBindings->get("type"));
  
  if (config == NULL) {
    String body = "Unknown device type: ";
    body += urlBindings->get("type");
    
    server.send(400, "text/plain", body);
    return;
  }
  
  milightClient->prepare(*config, deviceId, 0);
  
  if (request.containsKey("status")) {
    if (request["status"] == "on") {
      milightClient->updateStatus(ON);
    } else if (request["status"] == "off") {
      milightClient->updateStatus(OFF);
    }
  }
  
  server.send(200, "application/json", "true");
}

void MiLightHttpServer::handleSendRaw(const UrlTokenBindings* bindings) {
  DynamicJsonBuffer buffer;
  JsonObject& request = buffer.parse(server.arg("plain"));
  MiLightRadioConfig* config = MiLightRadioConfig::fromString(bindings->get("type"));
  
  if (config == NULL) {
    String body = "Unknown device type: ";
    body += bindings->get("type");
    
    server.send(400, "text/plain", body);
    return;
  }
  
  uint8_t packet[config->getPacketLength()];
  const String& hexPacket = request["packet"];
  hexStrToBytes<uint8_t>(hexPacket.c_str(), hexPacket.length(), packet, config->getPacketLength());
  
  size_t numRepeats = MILIGHT_DEFAULT_RESEND_COUNT;
  if (request.containsKey("num_repeats")) {
    numRepeats = request["num_repeats"];
  }
  
  milightClient->prepare(*config, 0, 0);
  
  for (size_t i = 0; i < numRepeats; i++) {
    milightClient->write(packet);
  }
  
  server.send(200, "text/plain", "true");
}