#include "AgStateMachine.h"

#define LED_FAST_BLINK_DELAY 250  /** ms */
#define LED_SLOW_BLINK_DELAY 1000 /** ms */
#define LED_SHORT_BLINK_DELAY 500 /** ms */
#define LED_LONG_BLINK_DELAY 2000 /** ms */

void AgStateMachine::ledBarSingleLedAnimation(uint8_t r, uint8_t g, uint8_t b) {
  if (ledBarAnimationCount < 0) {
    ledBarAnimationCount = 0;
    ag.ledBar.setColor(r, g, b, ledBarAnimationCount);
  } else {
    ledBarAnimationCount++;
    if (ledBarAnimationCount >= ag.ledBar.getNumberOfLeds()) {
      ledBarAnimationCount = 0;
    }
    ag.ledBar.setColor(r, g, b, ledBarAnimationCount);
  }
}

void AgStateMachine::ledStatusBlinkDelay(uint32_t ms) {
  ag.statusLed.setOn();
  delay(ms);
  ag.statusLed.setOff();
  delay(ms);
}

void AgStateMachine::sensorLedHandle(void) {
  switch (config.getLedBarMode()) {
  case LedBarMode::LedBarModeCO2:
    co2LedHandle();
    break;
  case LedBarMode::LedBarModePm:
    pm25LedHandle();
    break;
  default:
    ag.ledBar.clear();
    break;
  }

  // if (config.getLedBarMode() == LedBarMode::LedBarModeCO2) {
  //   co2LedHandle();
  // }
}

void AgStateMachine::co2LedHandle(void) {
  int co2Value = value.CO2;
  if (co2Value <= 400) {
    /** G; 1 */
    ag.ledBar.setColor(0, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
  } else if (co2Value <= 700) {
    /** GG; 2 */
    ag.ledBar.setColor(0, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(0, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
  } else if (co2Value <= 1000) {
    /** YYY; 3 */
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 3);
  } else if (co2Value <= 1333) {
    /** YYYY; 4 */
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 4);
  } else if (co2Value <= 1666) {
    /** YYYYY; 5 */
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 5);
  } else if (co2Value <= 2000) {
    /** RRRRRR; 6 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
  } else if (co2Value <= 2666) {
    /** RRRRRRR; 7 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 7);
  } else if (co2Value <= 3333) {
    /** RRRRRRRR; 8 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 7);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 8);
  } else if (co2Value <= 4000) {
    /** RRRRRRRRR; 9 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 7);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 8);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 9);
  } else { /** > 4000 */
    /* PRPRPRPRP; 9 */
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 7);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 8);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 9);
  }
}

void AgStateMachine::pm25LedHandle(void) {
  int pm25Value = value.PM25;
  if (pm25Value <= 5) {
    /** G; 1 */
    ag.ledBar.setColor(0, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
  } else if (pm25Value <= 10) {
    /** GG; 2 */
    ag.ledBar.setColor(0, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(0, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
  } else if (pm25Value <= 20) {
    /** YYY; 3 */
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 3);
  } else if (pm25Value <= 35) {
    /** YYYY; 4 */
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 4);
  } else if (pm25Value <= 45) {
    /** YYYYY; 5 */
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 255, 0, ag.ledBar.getNumberOfLeds() - 5);
  } else if (pm25Value <= 55) {
    /** RRRRRR; 6 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
  } else if (pm25Value <= 65) {
    /** RRRRRRR; 7 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 7);
  } else if (pm25Value <= 150) {
    /** RRRRRRRR; 8 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 7);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 8);
  } else if (pm25Value <= 250) {
    /** RRRRRRRRR; 9 */
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 7);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 8);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 9);
  } else { /** > 250 */
    /* PRPRPRPRP; 9 */
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 1);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 2);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 3);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 4);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 5);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 6);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 7);
    ag.ledBar.setColor(255, 0, 0, ag.ledBar.getNumberOfLeds() - 8);
    ag.ledBar.setColor(153, 153, 0, ag.ledBar.getNumberOfLeds() - 9);
  }
}

AgStateMachine::AgStateMachine(AirGradient &ag, AgOledDisplay &disp,
                               Stream &log, AgValue &value, AgConfigure &config)
    : PrintLog(log, "AgStateMachine"), ag(ag), disp(disp), value(value),
      config(config) {}

AgStateMachine::~AgStateMachine() {}

void AgStateMachine::displayHandle(AgStateMachineState state) {
  if (state > AgStateMachineNormal) {
    logError("displayHandle: State invalid");
    return;
  }

  dispState = state;

  switch (state) {
  case AgStateMachineWiFiManagerMode:
  case AgStateMachineWiFiManagerPortalActive: {
    if (wifiConnectCountDown >= 0) {
      String line1 = String(wifiConnectCountDown) + "s to connect";
      String line2 = "to WiFi hotspot:";
      String line3 = "\"airgradient-";
      String line4 = ag.deviceId() + "\"";
      disp.setText(line1, line2, line3, line4);
      // displayShowWifiText(String(wifiConnectCountDown) + "s to connect",
      //                     "to WiFi hotspot:", "\"airgradient-",
      //                     getDevId() + "\"");
      wifiConnectCountDown--;
    }
    break;
  }
  case AgStateMachineWiFiManagerStaConnecting: {
    // displayShowText("Trying to", "connect to WiFi", "...");
    disp.setText("Trying to", "connect to WiFi", "...");
    break;
  }
  case AgStateMachineWiFiManagerStaConnected: {
    // displayShowText("WiFi connection", "successful", "");
    disp.setText("WiFi connection", "successful", "");
    break;
  }
  case AgStateMachineWiFiOkServerConnecting: {
    // displayShowText("Connecting to", "Server", "...");
    disp.setText("Connecting to", "Server", "...");
    break;
  }
  case AgStateMachineWiFiOkServerConnected: {
    // displayShowText("Server", "connection", "successful");
    disp.setText("Server", "connection", "successful");
    break;
  }
  case AgStateMachineWiFiManagerConnectFailed: {
    // displayShowText("WiFi not", "connected", "");
    disp.setText("WiFi not", "connected", "");
    break;
  }
  case AgStateMachineWiFiOkServerConnectFailed: {
    // displayShowText("Server not", "reachable", "");
    break;
  }
  case AgStateMachineWiFiOkServerOkSensorConfigFailed: {
    // displayShowText("Monitor not", "setup on", "dashboard");
    disp.setText("Monitor not", "setup on", "dashboard");
    break;
  }
  case AgStateMachineWiFiLost: {
    // displayshowDashboard("WiFi N/A");
    disp.showDashboard("WiFi N/A");
    break;
  }
  case AgStateMachineServerLost: {
    // displayshowDashboard("Server N/A");
    disp.showDashboard("Server N/A");
    break;
  }
  case AgStateMachineSensorConfigFailed: {
    uint32_t ms = (uint32_t)(millis() - addToDashboardTime);
    if (ms >= 5000) {
      addToDashboardTime = millis();
      if (addToDashBoard) {
        // displayshowDashboard("Add to Dashboard");
        disp.showDashboard("Add to Dashboard");
      } else {
        // displayshowDashboard(getDevId());
        disp.showDashboard(ag.deviceId().c_str());
      }
      addToDashBoard = !addToDashBoard;
    }
    break;
  }
  case AgStateMachineNormal: {
    // displayshowDashboard("");
    disp.showDashboard();
  }
  default:
    break;
  }
}

void AgStateMachine::displayHandle(void) { displayHandle(dispState); }

void AgStateMachine::displaySetAddToDashBoard(void) {
  addToDashBoard = true;
  addToDashboardTime = millis();
}

void AgStateMachine::displayClearAddToDashBoard(void) {
  addToDashBoard = false;
}

void AgStateMachine::displayWiFiConnectCountDown(int count) {
  wifiConnectCountDown = count;
}

void AgStateMachine::ledAnimationInit(void) { ledBarAnimationCount = -1; }

void AgStateMachine::ledHandle(AgStateMachineState state) {
  if (state > AgStateMachineNormal) {
    logError("ledHandle: state invalid");
    return;
  }

  ledState = state;
  if (ag.isOneIndoor()) {
    ag.ledBar.clear(); // Set all LED OFF
  }
  switch (state) {
  case AgStateMachineWiFiManagerMode: {
    /** In WiFi Manager Mode */
    /** Turn LED OFF */
    /** Turn midle LED Color */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(0, 0, 255, ag.ledBar.getNumberOfLeds() / 2);
    } else {
      ag.statusLed.setToggle();
    }
    break;
  }
  case AgStateMachineWiFiManagerPortalActive: {
    /** WiFi Manager has connected to mobile phone */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(0, 0, 255);
    } else {
      ag.statusLed.setOn();
    }
    break;
  }
  case AgStateMachineWiFiManagerStaConnecting: {
    /** after SSID and PW entered and OK clicked, connection to WiFI network is
     * attempted */
    if (ag.isOneIndoor()) {
      ledBarSingleLedAnimation(255, 255, 255);
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiManagerStaConnected: {
    /** Connecting to WiFi worked */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(255, 255, 255);
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiOkServerConnecting: {
    /** once connected to WiFi an attempt to reach the server is performed */
    if (ag.isOneIndoor()) {
      ledBarSingleLedAnimation(0, 255, 0);
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiOkServerConnected: {
    /** Server is reachable, all fine */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(0, 255, 0);
    } else {
      ag.statusLed.setOff();

      /** two time slow blink, then off */
      for (int i = 0; i < 2; i++) {
        ledStatusBlinkDelay(LED_SLOW_BLINK_DELAY);
      }

      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiManagerConnectFailed: {
    /** Cannot connect to WiFi (e.g. wrong password, WPA Enterprise etc.) */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(255, 0, 0);
    } else {
      ag.statusLed.setOff();

      for (int j = 0; j < 3; j++) {
        for (int i = 0; i < 3; i++) {
          ledStatusBlinkDelay(LED_FAST_BLINK_DELAY);
        }
        delay(2000);
      }
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiOkServerConnectFailed: {
    /** Connected to WiFi but server not reachable, e.g. firewall block/
     * whitelisting needed etc. */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(233, 183, 54); /** orange */
    } else {
      ag.statusLed.setOff();
      for (int j = 0; j < 3; j++) {
        for (int i = 0; i < 4; i++) {
          ledStatusBlinkDelay(LED_FAST_BLINK_DELAY);
        }
        delay(2000);
      }
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiOkServerOkSensorConfigFailed: {
    /** Server reachable but sensor not configured correctly */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(139, 24, 248); /** violet */
    } else {
      ag.statusLed.setOff();
      for (int j = 0; j < 3; j++) {
        for (int i = 0; i < 5; i++) {
          ledStatusBlinkDelay(LED_FAST_BLINK_DELAY);
        }
        delay(2000);
      }
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineWiFiLost: {
    /** Connection to WiFi network failed credentials incorrect encryption not
     * supported etc. */
    if (ag.isOneIndoor()) {
      /** WIFI failed status LED color */
      ag.ledBar.setColor(255, 0, 0, 0);
      /** Show CO2 or PM color status */
      // sensorLedColorHandler();
      sensorLedHandle();
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineServerLost: {
    /** Connected to WiFi network but the server cannot be reached through the
     * internet, e.g. blocked by firewall */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(233, 183, 54, 0);

      /** Show CO2 or PM color status */
      sensorLedHandle();
      // sensorLedColorHandler();
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineSensorConfigFailed: {
    /** Server is reachable but there is some configuration issue to be fixed on
     * the server side */
    if (ag.isOneIndoor()) {
      ag.ledBar.setColor(139, 24, 248, 0);

      /** Show CO2 or PM color status */
      // sensorLedColorHandler();
      sensorLedHandle();
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  case AgStateMachineNormal: {
    if (ag.isOneIndoor()) {
      // sensorLedColorHandler();
      sensorLedHandle();
    } else {
      ag.statusLed.setOff();
    }
    break;
  }
  default:
    break;
  }

  if (ag.isOneIndoor()) {
    ag.ledBar.show();
  }
}

void AgStateMachine::ledHandle(void) { ledHandle(ledState); }

// void AgStateMachine::setState(AgStateMachineState state) {
//   this->state = state;
// }

// AgStateMachineState AgStateMachine::getState(void) { return state; }

void AgStateMachine::setDisplayState(AgStateMachineState state) {
  dispState = state;
}

AgStateMachineState AgStateMachine::getDisplayState(void) { return dispState; }