#pragma once

#include <MeshCore.h>
#include <helpers/ui/DisplayDriver.h>
#include <stddef.h>

class UITask {
  DisplayDriver* _display;
  mesh::MainBoard* _board;
  unsigned long _next_refresh, _auto_off;
  bool _connected;
  uint32_t _pin_code;
  const char* _node_name;
  const char* _build_date;
  char _origin[62];
  char _msg[80];
  int _msgcount;

  void renderCurrScreen();
  void buttonHandler();
  void userLedHandler();

public:

  UITask(mesh::MainBoard* board) : _board(board), _display(NULL) {
      _next_refresh = 0; 
      _connected = false;
  }
  void begin(DisplayDriver* display, const char* node_name, const char* build_date, uint32_t pin_code);

  void setHasConnection(bool connected) { _connected = connected; }
  bool hasDisplay() const { return _display != NULL; }
  void clearMsgPreview();
  void msgRead(int msgcount);
  void newMsg(uint8_t path_len, const char* from_name, const char* text, int msgcount);
  void loop();
};