#ifndef _MY_OTA_H
#define _MY_OTA_H

#include <ArduinoOTA.h>

void ota_set() {
  #ifdef USE_OTA
  //ArduinoOTA.setPort(8266);
  //ArduinoOTA.setHostname(clientID);
  //ArduinoOTA.setPassword(WiFi.macAddress().c_str());
  ArduinoOTA.onStart([]() {
    String type;
    displaySetStatus(1);
    if (ArduinoOTA.getCommand() == U_FLASH) {
      type = "sketch";
    } else { // U_FS
      type = "filesystem";
    }

    // NOTE: if updating FS this would be the place to unmount FS using FS.end()
    Serial.println("Start updating " + type);
  });
  ArduinoOTA.onEnd([]() {
    Serial.println("\nEnd");
    draw("DONE.", -1, -1, 100);
  });
  ArduinoOTA.onProgress([](unsigned int progress, unsigned int total) {
    Serial.printf("Progress: %u%%\r", (progress / (total / 100)));
    draw("Upgrading....", -1, -1, (progress / (total / 100)));
  });
  ArduinoOTA.onError([](ota_error_t error) {
    Serial.printf("Error[%u]: ", error);
    if (error == OTA_AUTH_ERROR) Serial.println("Auth Failed");
    else if (error == OTA_BEGIN_ERROR) Serial.println("Begin Failed");
    else if (error == OTA_CONNECT_ERROR) Serial.println("Connect Failed");
    else if (error == OTA_RECEIVE_ERROR) Serial.println("Receive Failed");
    else if (error == OTA_END_ERROR) Serial.println("End Failed");
  });
  ArduinoOTA.begin();
  #endif //USE_OTA
}

#endif //_MY_OTA_H