#define FASTLED_ESP8266_RAW_PIN_ORDER
#include "FastLED.h"

#include "math.h"

#define NUM_LEDS 150
#define DATA_PIN 0
#define led 13

CRGB leds[NUM_LEDS];

#include <ESP8266WiFi.h>
#include <ESP8266mDNS.h>
#include <ESP8266WebServer.h>
#include <WiFiUdp.h>
#include <EEPROM.h>
#include <ArduinoOTA.h>

const char* customssid = "led";
const int binterval = 10;
bool isBeating = false;
String beatingFun = "sin";
int beatingPeriod = 5000;
bool isOn = false;
CRGB lastColor = CRGB::White;

ESP8266WebServer httpServer(80);

int brightness = 255;

struct credential {
  String ssid;
  String pass;
};

int connected = 0;

//
//bool tryNetworks(){
//  WiFi.mode(WIFI_STA);
//  int n = WiFi.scanNetworks();
//  Serial.println("scan done");
//  if (n == 0)
//    Serial.println("no networks found");
//  else
//  {
//    Serial.print(n);
//    Serial.println("networks found");
//    for (int i = 0; i < n; ++i) {
//      Serial.print("Trying network:");
//      Serial.println(WiFi.SSID(i));
//      for (int j=0; j< ssids.length; j++) {
//      // Print SSID and RSSI for each network found
//        if (WiFi.SSID(i) == ssids[j]){
//          WiFi.mode(WIFI_STA);
//          WiFi.begin(ssid, password);
//          if (WiFi.waitForConnectResult() == WL_CONNECTED) {
//            Serial.println("Connection Established!...");
//            return true;
//          }
//        }
//      }
//     }
//  }
//  Serial.println("Could not connect to any network");
//  return false
//}

bool connect(struct credential cred) {
  WiFi.mode(WIFI_STA);
  Serial.println("Trying to connect to " + cred.ssid + " with " + cred.pass);
  if (cred.ssid.length() < 1) {
    return false;
  }
  Serial.println("Length " + cred.ssid.length());
  WiFi.begin(cred.ssid.c_str(), cred.pass.c_str());
  int c = 0;
  while ( c < 20 ) {
    if (WiFi.status() == WL_CONNECTED) {
      Serial.println("Ready");
      Serial.print("IP address: ");
      Serial.println(WiFi.localIP());
      return true;
    }
    delay(500);
    Serial.print(WiFi.status());
    c++;
  }
  Serial.println("Connect timed out");
  return false;
}

void setCredentials(struct credential cred) {
  String qsid = cred.ssid;
  Serial.println(qsid);
  Serial.println("");
  String qpass = cred.pass;
  Serial.println(qpass);
  Serial.println("");

  Serial.println("writing eeprom ssid:");
  for (int i = 0; i < qsid.length(); ++i)
  {
    EEPROM.write(i, qsid[i]);
    Serial.print("Wrote: ");
    Serial.println(qsid[i]);
  }
  for (int i = qsid.length(); i< 32; ++i){
    EEPROM.write(i, 0);

  }

  Serial.println("writing eeprom pass:");
  for (int i = 0; i < qpass.length(); ++i)
  {
    EEPROM.write(32 + i, qpass[i]);
    Serial.print("Wrote: ");
    Serial.println(qpass[i]);
  }
  for (int i = qpass.length(); i< 32; ++i){
    EEPROM.write(i+32, 0);

  }
  EEPROM.commit();
}

void handleSetCredentials() {
  struct credential cred;
  cred.ssid = "";
  cred.pass = "";

  cred.ssid = httpServer.arg("ssid");
  cred.pass = httpServer.arg("pass");

  if (cred.ssid != "" && cred.pass != "") {
    setCredentials(cred);
    String out = "Credentials set as: " + cred.ssid + " : " + cred.pass;
    httpServer.send(200, "text/plain", out);
    ESP.reset();
  } else {
    httpServer.send(400, "text/plain", "Invalid credentials");
  }
}

void clearCredentials() {
  for (int i = 0; i < 96; ++i) {
    EEPROM.write(i, 0);
  }
  EEPROM.commit();
}


void handleClearCredentials() {
  clearCredentials();
  httpServer.send(200, "text/plain", "Cleared!");
}

void handleWhite() {
  lastColor = CRGB::White;
  handleOn();
}

void handleColor() {
  int r = httpServer.arg("r").toInt();
  int g = httpServer.arg("g").toInt();
  int b = httpServer.arg("b").toInt();
  char msg[400];
  lastColor = CRGB(r, g, b);
  fill_solid( leds, NUM_LEDS, lastColor);
  FastLED.show();
  snprintf ( msg, 400, "RGB=(%d, %d, %d)", r, g, b);
  httpServer.send(200, "text/plain", msg);
}

void handleOff() {
  isOn = false;
  fill_solid( leds, NUM_LEDS, CRGB::Black);
  FastLED.show();
  httpServer.send(200, "text/plain", "Off!");
}

void handleOn() {
  isOn = true;
  fill_solid( leds, NUM_LEDS, lastColor);
  FastLED.show();
  httpServer.send(200, "text/plain", "On!");
}

void handleToggle() {
  if ( isOn ){
    handleOff();
  } else {
    handleOn();
  }
}

void handleBrightnessUp() {
  // min/max don't work
  if (brightness > 255 - binterval) {
    brightness = 255;
   } else {
   brightness = brightness + binterval;
  }
  FastLED.setBrightness(brightness);
  httpServer.send(200, "text/plain", "Brightness set to " + String(brightness));
}

void handleBrightnessDown() {
    if (brightness < binterval) {
      brightness = 0;
    } else {

   brightness = brightness - binterval;
    }
  FastLED.setBrightness(brightness);
  httpServer.send(200, "text/plain", "Brightness set to " + String(brightness));
}


void handleBeat() {
  beatingPeriod = httpServer.arg("period").toInt();
  beatingFun = httpServer.arg("fun");
  // min/max don't work
  if (beatingPeriod < 0) {
    beatingPeriod = 5000;
   } 
  httpServer.send(200, "text/plain", "Beating period set to " + String(beatingPeriod) + " and function set to " + beatingFun);
}

void handleBeatOn() {
  isBeating = true;
  httpServer.send(200, "text/plain", "Now beating!");
}

void handleBeatOff() {
  isBeating = false;
  FastLED.setBrightness(brightness);
  httpServer.send(200, "text/plain", "Not beating now!");
}

int getBeatingBrightness(){
  
  double val = 255.0;
  if (beatingFun.equals("lin")){
    // statements
    val = (millis() % (beatingPeriod)) - (beatingPeriod/2.0);

    if (val < 0.0) {
      val = -val;
    }
    val = 255.0 * 2.0 * (val/beatingPeriod);
  }
  else {
   // statements
      val =  (cos(millis()*(2.0*PI)/((double) beatingPeriod))*127+127);
  }
  return (int) (val*(brightness/255.0));
}


void beat(){
  FastLED.setBrightness(getBeatingBrightness());
}

void handleBrightness() {
  int value = httpServer.arg("value").toInt();
  if (value > 0) {
    brightness = value;
 
    FastLED.setBrightness(brightness);
    httpServer.send(200, "text/plain", "Brightness set to " + String(brightness));
    FastLED.show();

  } else {
    httpServer.send(400, "text/plain", "Invalid value");
  }

}

void handleRoot() {
  digitalWrite ( led, 1 );
  char temp[400];
  int sec = millis() / 1000;
  int min = sec / 60;
  int hr = min / 60;

  snprintf ( temp, 400,

             "<html>\
    <head>\
      <meta http-equiv='refresh' content='5'/>\
      <title>ESP8266 Demo</title>\
      <style>\
        body { background-color: #cccccc; font-family: Arial, Helvetica, Sans-Serif; Color: #000088; }\
      </style>\
    </head>\
    <body>\
      <h1>Hello from ESP8266!</h1>\
      <p>Uptime: %02d:%02d:%02d</p>\
    </body>\
  </html>",

             hr, min % 60, sec % 60
           );
  httpServer.send ( 200, "text/html", temp );
  digitalWrite ( led, 0 );
}

void setupAP() {
  WiFi.mode(WIFI_STA);
  WiFi.disconnect();
  delay(100);
  WiFi.softAP(customssid);
  Serial.println("softap");
  Serial.println("");
  Serial.println("Ready");
  Serial.print("IP address: ");
  Serial.println(WiFi.softAPIP());
}

void handleLoop() {
  for (int dot = 0; dot < NUM_LEDS; dot++) {
    leds[dot] = CRGB::Blue;
    FastLED.show();
    // clear this led for the next time around the loop
    leds[dot] = CRGB::Black;
    delay(30);
  }
  httpServer.send ( 404, "text/plain", "Done!" );
}


struct credential getCredentials() {
  Serial.println("Getting credentials from EEPROM");
  struct credential cred;
  cred.ssid = "";
  cred.pass = "";
  for (int i = 0; i < 32; ++i)
  {
    cred.ssid += char(EEPROM.read(i));
  }
  Serial.print("SSID: ");
  Serial.println(cred.ssid);
  Serial.println("Reading EEPROM pass");
  for (int i = 32; i < 96; ++i)
  {
    cred.pass += char(EEPROM.read(i));
  }
  Serial.print("PASS: ");
  Serial.println(cred.pass);
  return cred;
}

void setup() {
  Serial.begin(115200);
  Serial.println("Booting");

  EEPROM.begin(512);
  delay(10);
  WiFi.mode(WIFI_STA);

  struct credential cred = getCredentials();
  if (!connect(cred)) {
    setupAP();
  }

  // Port defaults to 8266
  // ArduinoOTA.setPort(8266);

  // Hostname defaults to esp8266-[ChipID]
  ArduinoOTA.setHostname("led");

  // No authentication by default
  ArduinoOTA.setPassword((const char *)"123");

  ArduinoOTA.onStart([]() {
    Serial.println("Start");
  });
  ArduinoOTA.onEnd([]() {
    Serial.println("\nEnd");
  });
  ArduinoOTA.onProgress([](unsigned int progress, unsigned int total) {
    Serial.printf("Progress: %u%%\r", (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();




  if ( MDNS.begin ( "esp8266" ) ) {
    Serial.println ( "MDNS responder started" );
  }
  MDNS.addService("http", "tcp", 80);
  httpServer.on ( "/loop", handleLoop );
  httpServer.on ( "/", handleRoot );
  httpServer.on ( "/clear", handleClearCredentials );
  httpServer.on ( "/credentials", handleSetCredentials );
  httpServer.on ( "/color", handleColor );
  httpServer.on ( "/off", handleOff );
  httpServer.on ( "/on", handleOn );
  httpServer.on ( "/toggle", handleToggle );
  httpServer.on ( "/white", handleWhite );
  httpServer.on ( "/beat/on", handleBeatOn );
  httpServer.on ( "/beat/off", handleBeatOff );
  httpServer.on ( "/beat/", handleBeat );
  httpServer.on ( "/brightness", handleBrightness );
  httpServer.on ( "/brightness/up", handleBrightnessUp );
  httpServer.on ( "/brightness/down", handleBrightnessDown );
  httpServer.begin();

  FastLED.addLeds<WS2812B, DATA_PIN, GRB>(leds, NUM_LEDS);
//  FastLED.setDither(0);
  FastLED.setBrightness(255);
  FastLED.setMaxPowerInVoltsAndMilliamps(5, 20000);
  if (isOn){
    fill_solid(leds, NUM_LEDS, lastColor);
    FastLED.show();
  }
}

void loop() {
  ArduinoOTA.handle();
  httpServer.handleClient();
  if (isBeating){
    beat();
  }
  FastLED.show();
}