#pragma once

#include <Arduino.h>
#include <TFT_eSPI.h>

class ScrollingTextDisplay {
private:
  TFT_eSprite* sprite;
  int yPos;
  int maxWidth;
  int xPos;
  int textWidth;
  unsigned long lastTick;
  unsigned long holdTick;
  bool isScrolling;
  std::function<void(TFT_eSprite*)> postDrawCallback;
  int usedH;
  bool hold;
  int xOffset;

  static const unsigned long SCROLL_INTERVAL = 5;
  static const unsigned long HOLD_DURATION = 2000;
  static const int SCROLL_GAP = 25;

public:
  ScrollingTextDisplay(TFT_eSprite* spr, int y, int maxW, int x = 35, int inuseH = -1 ) :
    sprite(spr), yPos(y), maxWidth(maxW), xPos(0), textWidth(0), lastTick(0), holdTick(0), isScrolling(false), postDrawCallback(nullptr), usedH(inuseH), hold(false), xOffset(x) {}

  void setPostDrawCallback(std::function<void(TFT_eSprite*)> callback) {
    postDrawCallback = callback;
  }

  void update(const String& text, bool status, uint16_t activeColor, uint16_t activeSmooth, uint16_t dropoutColor, uint16_t dropoutSmooth, uint16_t backgroundColor, uint8_t font = 255) {
    textWidth = sprite->textWidth(text);

    if(textWidth < maxWidth) {
      xPos = 0;
      isScrolling = false;
      drawText(text, status, activeColor, activeSmooth, dropoutColor, dropoutSmooth, backgroundColor, font);
    } else {
      if(!isScrolling) holdTick = millis();
      isScrolling = true;
      if(millis() - lastTick >= SCROLL_INTERVAL) {
        if(xPos <= -(textWidth + SCROLL_GAP)) {xPos += textWidth + SCROLL_GAP; hold = true;}

        if(hold) {
          if(millis() - holdTick >= HOLD_DURATION) {
            xPos--;
            holdTick = millis();
            hold = false;
          }
        } else {
          xPos--;
          holdTick = millis();
        }

        drawText(text, status, activeColor, activeSmooth, dropoutColor, dropoutSmooth, backgroundColor, font);
        lastTick = millis();
      }
    }
  }

  void reset() {
    xPos = 0;
    holdTick = millis();
    lastTick = millis();
  }

  void setYPosition(int y) {
    yPos = y;
  }

  bool getIsScrolling() const {
    return isScrolling;
  }

private:
  void drawText(const String& text, bool status, uint16_t activeColor, uint16_t activeSmooth, uint16_t dropoutColor, uint16_t dropoutSmooth, uint16_t backgroundColor, uint8_t font) {
    if(font > FONT_COUNT) font = sprite->textfont;

    if(usedH > 0) {
      sprite->fillSprite(TFT_TRANSPARENT);
      sprite->fillRect(0, 0, maxWidth, usedH, backgroundColor);

      sprite->setViewport(0, 0, maxWidth, usedH);

      if(status) sprite->setTextColor(activeColor, activeSmooth, false);
      else sprite->setTextColor(dropoutColor, dropoutSmooth, false);

      sprite->drawString(text, xPos, 0, font);
      if(isScrolling) sprite->drawString(text, xPos + textWidth + SCROLL_GAP, 0, font);

      sprite->resetViewport();
    } else {
      sprite->fillSprite(backgroundColor);

      if(status) sprite->setTextColor(activeColor, activeSmooth, false);
      else sprite->setTextColor(dropoutColor, dropoutSmooth, false);

      sprite->drawString(text, xPos, 0, font);
      if(isScrolling) sprite->drawString(text, xPos + textWidth + SCROLL_GAP, 0, font);
    }

    if(postDrawCallback) postDrawCallback(sprite);
    if(yPos != 0) sprite->pushSprite(xOffset, yPos, TFT_TRANSPARENT);
  }
};