#include "DisplayManager.h"
#include "Config.h"

DisplayManager::DisplayManager()
  : _dash(_tft)
{
}

void DisplayManager::begin() {
  pinMode(PIN_LCD_BL, OUTPUT);
  setBacklight(true);

  _tft.init();
  _tft.setRotation(1);  // landscape: 480x320
  _tft.fillScreen(COL_BLACK);
}

void DisplayManager::setBacklight(bool on) {
  digitalWrite(PIN_LCD_BL, on ? HIGH : LOW);
}

TouchEvent DisplayManager::readTouch() {
  TouchEvent evt;
  uint16_t x, y;
  if (_tft.getTouch(&x, &y)) {
    evt.pressed = true;
    evt.x = x;
    evt.y = y;
  }
  return evt;
}

int DisplayManager::dashboardTouch(uint16_t x, uint16_t y) {
  if (_lastScreen != ScreenID::DASHBOARD) return -1;
  return _dash.handleTouch(x, y);
}

void DisplayManager::render(const ScreenState& state) {
  // Detect screen change → force full redraw
  if (state.screen != _lastScreen) {
    _needsFullRedraw = true;
    _lastScreen = state.screen;
  }

  switch (state.screen) {
    case ScreenID::BOOT_SPLASH:
      if (_needsFullRedraw) drawBootSplash(state);
      break;
    case ScreenID::WIFI_CONNECTING:
      if (_needsFullRedraw) drawWifiConnecting();
      break;
    case ScreenID::WIFI_CONNECTED:
      if (_needsFullRedraw) drawWifiConnected(state);
      break;
    case ScreenID::WIFI_FAILED:
      if (_needsFullRedraw) drawWifiFailed();
      break;
    case ScreenID::ALERT:
      if (_needsFullRedraw || state.blinkPhase != _lastBlink) {
        drawAlertScreen(state);
        _lastBlink = state.blinkPhase;
      }
      break;
    case ScreenID::STATUS:
      if (_needsFullRedraw) drawStatusScreen(state);
      break;
    case ScreenID::DASHBOARD:
      drawDashboard(state);
      break;
    case ScreenID::OFF:
      break;
  }

  _needsFullRedraw = false;
}

// ----- Dashboard -----

void DisplayManager::drawDashboard(const ScreenState& s) {
  if (_needsFullRedraw) {
    if (!_dashSpriteReady) {
      _dash.begin();               // create sprite (once)
      _dashSpriteReady = true;
    }
    _dash.drawAll();               // fill screen + draw all tiles
    _dash.refreshFromState(s);     // update with real data
    _lastDashRefresh = millis();
  } else if (millis() - _lastDashRefresh > 2000) {
    _lastDashRefresh = millis();
    _dash.refreshFromState(s);     // only redraws changed tiles
  }
}

// ----- Helpers -----

void DisplayManager::drawCentered(const char* txt, int y, int sz, uint16_t col) {
  _tft.setTextFont(1);    // ← ADDED: force GLCD font
  _tft.setTextSize(sz);
  _tft.setTextColor(col);
  int w = _tft.textWidth(txt);  // ← use TFT_eSPI's own width calc
  _tft.setCursor(max(0, (SCREEN_WIDTH - w) / 2), y);
  _tft.print(txt);
}

void DisplayManager::drawInfoLine(int x, int y, uint16_t col, const char* text) {
  _tft.setTextFont(1);    // ← ADDED
  _tft.setTextSize(1);
  _tft.setTextColor(col);
  _tft.setCursor(x, y);
  _tft.print(text);
}

void DisplayManager::drawHeaderBar(uint16_t col, const char* label, const char* timeStr) {
  _tft.setTextFont(1);    // ← ADDED
  _tft.setTextSize(2);
  _tft.setTextColor(col);
  _tft.setCursor(8, 8);
  _tft.print(label);

  int tw = _tft.textWidth(timeStr);  // ← proper width calc
  _tft.setCursor(SCREEN_WIDTH - tw - 8, 8);
  _tft.print(timeStr);
}
// ----- Screens -----

void DisplayManager::drawBootSplash(const ScreenState& s) {
  _tft.fillScreen(COL_BLACK);
  drawCentered("KLUBHAUS", 60, 4, COL_NEON_TEAL);
  drawCentered("ALERT", 110, 4, COL_HOT_FUCHSIA);
  drawCentered("v5.0 E32R35T", 180, 2, COL_DARK_GRAY);
  if (s.debugMode) {
    drawCentered("DEBUG MODE", 210, 2, COL_YELLOW);
  }
}

void DisplayManager::drawWifiConnecting() {
  _tft.fillScreen(COL_BLACK);
  drawCentered("Connecting", 130, 3, COL_NEON_TEAL);
  drawCentered("to WiFi...", 170, 3, COL_NEON_TEAL);
}

void DisplayManager::drawWifiConnected(const ScreenState& s) {
  _tft.fillScreen(COL_BLACK);
  drawCentered("Connected!", 100, 3, COL_GREEN);
  drawCentered(s.wifiSSID, 150, 2, COL_WHITE);
  drawCentered(s.wifiIP, 180, 2, COL_WHITE);
}

void DisplayManager::drawWifiFailed() {
  _tft.fillScreen(COL_BLACK);
  drawCentered("WiFi FAILED", 140, 3, COL_RED);
}

void DisplayManager::drawAlertScreen(const ScreenState& s) {
  uint16_t bg = s.blinkPhase ? COL_NEON_TEAL : COL_HOT_FUCHSIA;
  uint16_t fg = s.blinkPhase ? COL_BLACK : COL_WHITE;

  _tft.fillScreen(bg);
  drawHeaderBar(fg, "ALERT", s.timeString);

  int sz = 5;
  int len = strlen(s.alertMessage);
  if (len > 10) sz = 4;
  if (len > 18) sz = 3;
  if (len > 30) sz = 2;

  if (len > 12) {
    String msg(s.alertMessage);
    int mid = len / 2;
    int sp = msg.lastIndexOf(' ', mid);
    if (sp < 0) sp = mid;
    String l1 = msg.substring(0, sp);
    String l2 = msg.substring(sp + 1);
    int lh = 8 * sz + 8;
    int y1 = (SCREEN_HEIGHT - lh * 2) / 2;
    drawCentered(l1.c_str(), y1, sz, fg);
    drawCentered(l2.c_str(), y1 + lh, sz, fg);
  } else {
    drawCentered(s.alertMessage, (SCREEN_HEIGHT - 8 * sz) / 2, sz, fg);
  }

  drawCentered("TAP TO SILENCE", SCREEN_HEIGHT - 25, 2, fg);
}

void DisplayManager::drawStatusScreen(const ScreenState& s) {
  _tft.fillScreen(COL_BLACK);
  drawHeaderBar(COL_MINT, "KLUBHAUS", s.timeString);

  drawCentered("MONITORING", 60, 3, COL_WHITE);

  char buf[80];
  int y = 110;
  int sp = 22;
  int x = 20;

  snprintf(buf, sizeof(buf), "WiFi: %s (%ddBm)",
    s.wifiConnected ? s.wifiSSID : "DOWN", s.wifiRSSI);
  drawInfoLine(x, y, COL_WHITE, buf); y += sp;

  snprintf(buf, sizeof(buf), "IP: %s",
    s.wifiConnected ? s.wifiIP : "---");
  drawInfoLine(x, y, COL_WHITE, buf); y += sp;

  snprintf(buf, sizeof(buf), "Up: %lu min  Heap: %lu KB",
    s.uptimeMinutes, s.freeHeapKB);
  drawInfoLine(x, y, COL_WHITE, buf); y += sp;

  snprintf(buf, sizeof(buf), "NTP: %s UTC",
    s.ntpSynced ? s.timeString : "not synced");
  drawInfoLine(x, y, COL_WHITE, buf); y += sp;

  const char* stName = s.deviceState == DeviceState::SILENT ? "SILENT" :
    s.deviceState == DeviceState::ALERTING ? "ALERTING" : "WAKE";
  snprintf(buf, sizeof(buf), "State: %s  Net: %s",
    stName, s.networkOK ? "OK" : "FAIL");
  uint16_t stCol = s.deviceState == DeviceState::ALERTING ? COL_RED :
    s.deviceState == DeviceState::SILENT ? COL_GREEN : COL_NEON_TEAL;
  drawInfoLine(x, y, stCol, buf); y += sp;

  if (s.alertHistoryCount > 0) {
    drawInfoLine(x, y, COL_MINT, "Recent Alerts:"); y += sp;
    for (int i = 0; i < s.alertHistoryCount; i++) {
      uint16_t col = (i == 0) ? COL_YELLOW : COL_DARK_GRAY;
      snprintf(buf, sizeof(buf), "%s  %.35s",
        s.alertHistory[i].timestamp,
        s.alertHistory[i].message);
      drawInfoLine(x, y, col, buf); y += sp;
    }
  } else {
    drawInfoLine(x, y, COL_DARK_GRAY, "No alerts yet"); y += sp;
  }

  if (s.debugMode) {
    drawInfoLine(x, y, COL_YELLOW, "DEBUG MODE - _test topics");
  }

  drawCentered("tap to dismiss", SCREEN_HEIGHT - 18, 1, COL_DARK_GRAY);
}

HoldState DisplayManager::updateHold(unsigned long requiredMs) {
  HoldState h;
  h.targetMs = requiredMs;

  uint16_t tx, ty;
  bool touching = _tft.getTouch(&tx, &ty);

  if (touching) {
    if (!_holdActive) {
      _holdActive  = true;
      _holdStartMs = millis();
      _holdX = tx;
      _holdY = ty;
    }
    h.active  = true;
    h.x       = _holdX;
    h.y       = _holdY;
    h.holdMs  = millis() - _holdStartMs;

    if (h.holdMs >= requiredMs) {
      h.completed = true;
      _holdActive = false;
    }
  } else {
    _holdActive = false;
  }

  return h;
}

void DisplayManager::drawSilenceProgress(float progress) {
  int barY = SCREEN_HEIGHT - 30;
  int barH = 20;
  int barW = (int)(SCREEN_WIDTH * progress);

  _tft.fillRect(0, barY, SCREEN_WIDTH, barH, COL_BLACK);
  _tft.fillRect(0, barY, barW, barH, COL_GREEN);
  _tft.drawRect(0, barY, SCREEN_WIDTH, barH, COL_WHITE);

  _tft.setTextFont(1);
  _tft.setTextSize(2);
  _tft.setTextColor(progress < 1.0f ? COL_WHITE : COL_BLACK);
  _tft.setTextDatum(MC_DATUM);
  _tft.drawString(
    progress < 1.0f ? "HOLD TO SILENCE" : "SILENCED",
    SCREEN_WIDTH / 2, barY + barH / 2);
}