#include <Arduino.h>

#ifdef ROO_TESTING

#include "roo_testing/devices/display/st77xx/st77xx.h"
#include "roo_testing/microcontrollers/esp32/fake_esp32.h"
#include "roo_testing/transducers/ui/viewport/flex_viewport.h"
#include "roo_testing/transducers/ui/viewport/fltk/fltk_viewport.h"
#include "roo_testing/transducers/wifi/wifi.h"

using roo_testing_transducers::FlexViewport;
using roo_testing_transducers::FltkViewport;

struct Emulator {
  FltkViewport viewport;
  FlexViewport flexViewport;
  roo_testing_transducers::wifi::Environment wifi;

  FakeSt77xxSpi display;

  Emulator()
      : viewport(),
        flexViewport(viewport, 1, FlexViewport::kRotationLeft),
        display(flexViewport, 240, 320) {
    FakeEsp32().attachSpiDevice(display, 18, 19, 23);
    FakeEsp32().gpio.attachOutput(5, display.cs());
    FakeEsp32().gpio.attachOutput(2, display.dc());
    FakeEsp32().gpio.attachOutput(4, display.rst());

    auto ap = std::unique_ptr<roo_testing_transducers::wifi::AccessPoint>(
        new roo_testing_transducers::wifi::AccessPoint(
            roo_testing_transducers::wifi::MacAddress(1, 1, 1, 1, 1, 1),
            "*******"));
    ap->setAuthMode(roo_testing_transducers::wifi::AUTH_WEP);
    ap->setPasswd("**********");
    wifi.addAccessPoint(std::move(ap));

    FakeEsp32().setWifiEnvironment(wifi);
  }
} emulator;

#endif

#include <WiFi.h>
#include <time.h>

#include <string>

#include "roo_display.h"
#include "roo_display/font/font.h"
#include "roo_display/shape/basic.h"
#include "roo_display/ui/text_label.h"
#include "roo_fonts/NotoSans_Bold/60.h"
#include "roo_fonts/NotoSans_Regular/40.h"

using namespace roo_display;

//********* ENTER YOUR WIFI CREDENTIALS **********
const char* ssid = "*******";
const char* password = "**********";

//********* NTP TIME  **********
// Set TZ_INFO for your location:
// https://github.com/nayarsystems/posix_tz_db/blob/master/zones.csv
const char* TZ_INFO =
    "PST8PDT,M3.2.0,M11.1.0";  // e.g Los Angeles US Pacific and DST details
// Specify your nearest NTP server
const char* NTP_SERVER = "0.pool.ntp.org";

tm timeinfo;
time_t now;
bool us_datefmt = true;  // mon/day/year, set to false for day/mon/year format

//
// Change these two lines to use a different driver, transport, or pins.
#include "roo_display/driver/st7789.h"
St7789spi_172x320<5, 2, 4> device;
// St7789spi_240x240<5, 2, 4> device;

Display display(device);

// FWD Declarations
void initNTP(const std::string& s);
bool getNTPtime(int sec);
void stringifyTime(tm localTime, char* time, char* date);
void bootMsg(const std::string& s);
void initWiFi(std::string s);

void setup() {
  Serial.begin(9600);

  SPI.begin();  // Use default SPI pins, or specify your own here.

  display.init(color::AliceBlue);
  if (display.height() > display.width()) {
    display.setOrientation(Orientation().rotateRight());
  }
  display.setBackgroundColor(color::BlanchedAlmond);

  display.clear();

  DrawingContext dc(display);

  initWiFi("Init Wifi");
  initNTP("Init NTP");
  display.clear();

  // split screen background color vertically
  dc.draw(FilledRect(0, display.height() / 2 + 16, display.width() - 1,
                     display.height() - 1, color::BlueViolet));
  dc.draw(Line(0, display.height() / 2 + 15, display.width() - 1,
               display.height() / 2 + 15, color::Black));
}

void loop() {
  display.setBackgroundColor(color::BlanchedAlmond);

  char time[9] = "";
  char date[11] = "";
  const Font& timeFont = font_NotoSans_Bold_60();
  const Font& dateFont = font_NotoSans_Regular_40();

  getNTPtime(10);
  stringifyTime(timeinfo, time, date);

  {
    DrawingContext dc(
        display, Box(0, 0, display.width() - 1, display.height() / 2 + 14));
    dc.setFillMode(FILL_MODE_RECTANGLE);
    // Using tile to make sure that the entire 'anchor' extents are filled.
    // Otherwise, only the bounding box of the drawn text would be filled,
    // and those can differ for different digits, so e.g. '1' would leave
    // unfilled pixels from previous digits.
    // Anchor extents are always the same for all digits in the same font,
    // so this technique works well for clocks.
    auto label = StringViewLabel(time, timeFont, color::BlueViolet);
    dc.draw(MakeTileOf(label, label.anchorExtents()),
            kCenter | kMiddle);  // draw Time
  }
  {
    DrawingContext dc(display, Box(0, display.height() / 2 + 16,
                                   display.width() - 1, display.height() - 1));
    dc.setBackgroundColor(color::BlueViolet);
    dc.setFillMode(FILL_MODE_RECTANGLE);
    // See above.
    auto label = StringViewLabel(date, dateFont, color::BlanchedAlmond);
    dc.draw(MakeTileOf(label, label.anchorExtents()),
            kCenter | kMiddle);  // draw Date
  }
  delay(100);
}

// ntp methods (see
// https://github.com/SensorsIot/NTP-time-for-ESP8266-and-ESP32)
bool getNTPtime(int sec) {
  {
    uint32_t start = millis();
    do {
      time(&now);
      localtime_r(&now, &timeinfo);
      delay(10);
    } while (((millis() - start) <= (1000 * sec)) &&
             (timeinfo.tm_year < (2016 - 1900)));
    if (timeinfo.tm_year <= (2016 - 1900))
      return false;  // the NTP call was not successful
  }
  return true;
}

void initNTP(const std::string& s) {
  bootMsg(s);
  configTime(0, 0, NTP_SERVER);
  setenv("TZ", TZ_INFO, 1);

  if (getNTPtime(10)) {  // wait up to 10sec to sync
  } else {
    Serial.println("Time not set");
    ESP.restart();
  }
}

// Time/Date stringifier
void stringifyTime(tm localTime, char* time, char* date) {
  char int_st[32];
  sprintf(int_st, "%02d",
          (us_datefmt ? localTime.tm_mon + 1 : localTime.tm_mday));
  strcat(date, int_st);
  strcat(date, "/");
  sprintf(int_st, "%02d",
          (us_datefmt ? localTime.tm_mday : localTime.tm_mon + 1));
  strcat(date, int_st);
  strcat(date, "/");
  sprintf(int_st, "%04d", localTime.tm_year + 1900);
  strcat(date, int_st);

  if (localTime.tm_hour > 12) {  // PM
    localTime.tm_hour -= 12;
  }
  sprintf(int_st, "%02d", localTime.tm_hour);
  strcat(time, int_st);
  strcat(time, ":");
  sprintf(int_st, "%02d", localTime.tm_min);
  strcat(time, int_st);
  strcat(time, ":");
  sprintf(int_st, "%02d", localTime.tm_sec);
  strcat(time, int_st);
}

// Show short message during boot in vertical center of display
void bootMsg(const std::string& s) {
  const Font& font = font_NotoSans_Regular_40();
  DrawingContext dc(display);
  dc.draw(MakeTileOf(StringViewLabel(s, font, color::Black), display.extents(),
                     kLeft.shiftBy(10) | kMiddle));
}

// Connect to WiFi
void initWiFi(std::string s) {
  bootMsg(s);

  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    s.append(".");
    bootMsg(s);
    Serial.print(".");
  }
  WiFi.begin(ssid, password);
}
