#include "AgOledDisplay.h"
// #include "Libraries/U8g2/src/U8g2lib.h"
#include <U8g2lib.h>

#define DISP() ((U8G2_SH1106_128X64_NONAME_F_HW_I2C *)(this->u8g2))

void AgOledDisplay::showTempHum(void) {
  char buf[10];
  if (value.Temperature > -1001) {
    if (config.isTemperatureUnitInF()) {
      float tempF = (value.Temperature * 9) / 5 + 32;
      snprintf(buf, sizeof(buf), "%0.1f", tempF);
    } else {
      snprintf(buf, sizeof(buf), "-°F");
    }
  } else {
    if (value.Temperature > -1001) {
      snprintf(buf, sizeof(buf), "%.1f°C", value.Temperature);
    } else {
      snprintf(buf, sizeof(buf), "-°C");
    }
  }
  DISP()->drawUTF8(1, 10, buf);

  /** Show humidty */
  if (value.Humidity >= 0) {
    snprintf(buf, sizeof(buf), "%d%%", value.Humidity);
  } else {
    snprintf(buf, sizeof(buf), "%-%%");
  }

  if (value.Humidity > 99) {
    DISP()->drawStr(97, 10, buf);
  } else {
    DISP()->drawStr(105, 10, buf);
  }
}

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

AgOledDisplay::~AgOledDisplay() {}

/**
 * @brief Initialize display
 *
 * @return true Success
 * @return false Failure
 */
bool AgOledDisplay::begin(void) {
  if (isBegin) {
    logWarning("Already begin, call 'end' and try again");
    return true;
  }

  /** Create u8g2 instance */
  u8g2 = new U8G2_SH1106_128X64_NONAME_F_HW_I2C(U8G2_R0, U8X8_PIN_NONE);
  if (u8g2 == NULL) {
    logError("Create 'U8G2' failed");
    return false;
  }

  /** Init u8g2 */
  if (DISP()->begin() == false) {
    logError("U8G2 'begin' failed");
    return false;
  }

  isBegin = true;
  logInfo("begin");
  return true;
}

/**
 * @brief De-Initialize display
 *
 */
void AgOledDisplay::end(void) {
  if (!isBegin) {
    logWarning("Already end, call 'begin' and try again");
    return;
  }

  /** Free u8g2 */
  delete u8g2;
  u8g2 = NULL;

  isBegin = false;
  logInfo("end");
}

void AgOledDisplay::setStatus(String &status) { setStatus(status.c_str()); }

void AgOledDisplay::setStatus(const char *status) {}

void AgOledDisplay::setText(String &line1, String &line2, String &line3) {
  setText(line1.c_str(), line2.c_str(), line3.c_str());
}

void AgOledDisplay::setText(const char *line1, const char *line2,
                            const char *line3) {
  DISP()->firstPage();
  DISP()->setFont(u8g2_font_t0_16_tf);
  DISP()->drawStr(1, 10, line1);
  DISP()->drawStr(1, 30, line2);
  DISP()->drawStr(1, 50, line3);
}

void AgOledDisplay::setText(const char *text) {}

void AgOledDisplay::setText(String &text) {}

void AgOledDisplay::setText(String &line1, String &line2, String &line3,
                            String &line4) {
  setText(line1.c_str(), line2.c_str(), line3.c_str(), line4.c_str());
}

void AgOledDisplay::setText(const char *line1, const char *line2,
                            const char *line3, const char *line4) {
  DISP()->firstPage();
  DISP()->setFont(u8g2_font_t0_16_tf);
  DISP()->drawStr(1, 10, line1);
  DISP()->drawStr(1, 25, line2);
  DISP()->drawStr(1, 40, line3);
  DISP()->drawStr(1, 55, line4);
}

void AgOledDisplay::showDashboard(void) { showDashboard(NULL); }

void AgOledDisplay::showDashboard(const char *status) {
  char strBuf[10];

  DISP()->firstPage();

  DISP()->setFont(u8g2_font_t0_16_tf);
  if ((status == NULL) || (strlen(status) == 0)) {
    showTempHum();
  } else {
    String strStatus = "Show status: " + String(status);
    logInfo(strStatus);

    int strWidth = DISP()->getStrWidth(status);
    DISP()->drawStr((126 - strWidth) / 2, 10, status);

    /** Show WiFi NA*/
    if (strcmp(status, "WiFi N/A") == 0) {
      DISP()->setFont(u8g2_font_t0_12_tf);
      showTempHum();
    }
  }

  /** Draw horizonal line */
  DISP()->drawLine(1, 13, 128, 13);

  /** Show CO2 label */
  DISP()->setFont(u8g2_font_t0_12_tf);
  DISP()->drawUTF8(1, 27, "CO2");

  DISP()->setFont(u8g2_font_t0_22b_tf);
  if (value.CO2 > 0) {
    int val = 9999;
    if (value.CO2 < 10000) {
      val = value.CO2;
    }
    sprintf(strBuf, "%d", val);
  } else {
    sprintf(strBuf, "%s", "-");
  }
  DISP()->drawStr(1, 48, strBuf);

  /** Show CO2 value index */
  DISP()->setFont(u8g2_font_t0_12_tf);
  DISP()->drawStr(1, 61, "ppm");

  /** Draw vertical line */
  DISP()->drawLine(45, 14, 45, 64);
  DISP()->drawLine(82, 14, 82, 64);

  /** Draw PM2.5 label */
  DISP()->setFont(u8g2_font_t0_12_tf);
  DISP()->drawStr(48, 27, "PM2.5");

  /** Draw PM2.5 value */
  DISP()->setFont(u8g2_font_t0_22b_tf);
  if (config.isPmStandardInUSAQI()) {
    if (value.PM25 >= 0) {
      sprintf(strBuf, "%d", ag.pms5003.convertPm25ToUsAqi(value.PM25));
    } else {
      sprintf(strBuf, "%s", "-");
    }
    DISP()->drawStr(48, 48, strBuf);
    DISP()->setFont(u8g2_font_t0_12_tf);
    DISP()->drawUTF8(48, 61, "AQI");
  } else {
    if (value.PM25 >= 0) {
      sprintf(strBuf, "%d", value.PM25);
    } else {
      sprintf(strBuf, "%s", "-");
    }
    DISP()->drawStr(48, 48, strBuf);
    DISP()->setFont(u8g2_font_t0_12_tf);
    DISP()->drawUTF8(48, 61, "ug/m³");
  }

  /** Draw tvocIndexlabel */
  DISP()->setFont(u8g2_font_t0_12_tf);
  DISP()->drawStr(85, 27, "tvoc:");

  /** Draw tvocIndexvalue */
  if (value.TVOC >= 0) {
    sprintf(strBuf, "%d", value.TVOC);
  } else {
    sprintf(strBuf, "%s", "-");
  }
  DISP()->drawStr(85, 39, strBuf);

  /** Draw NOx label */
  DISP()->drawStr(85, 53, "NOx:");
  if (value.NOx >= 0) {
    sprintf(strBuf, "%d", value.NOx);
  } else {
    sprintf(strBuf, "%s", "-");
  }
  DISP()->drawStr(85, 63, strBuf);
}