ESPUI/src/ESPUI.h

#ifndef ESPUI_h
#define ESPUI_h

#define DEBUG_ESPUI true
#define WS_AUTHENTICATION false

#include "Arduino.h"
#include "ArduinoJson.h"
#include "FS.h"
#include "stdlib_noniso.h"

#if defined(ESP32)

#include "SPIFFS.h"
#include "WiFi.h"
#include <AsyncTCP.h>
#include <ESPAsyncWebServer.h>

#else

#include <ArduinoOTA.h>
#include <ESP8266WiFi.h>
#include <ESP8266mDNS.h>
#include <ESPAsyncTCP.h>
#include <ESPAsyncWebServer.h>
#include <FS.h>
#include <Hash.h>
#include <SPIFFSEditor.h>

#define FILE_WRITE "w"

#endif

// Message Types (and control types)

enum ControlType : uint8_t {
  // fixed controls
  Title = 0,

  // updatable controls
  Pad,
  PadWithCenter,
  Button,
  Label,
  Switcher,
  Slider,
  Number,
  Text,
  Graph,
  GraphPoint,
  Tab,
  Select,
  Option,
  Min,
  Max,
  Step,

  UpdateOffset = 100,
  UpdatePad = 101,
  UpdatePadWithCenter,
  ButtonButton,
  UpdateLabel,
  UpdateSwitcher,
  UpdateSlider,
  UpdateNumber,
  UpdateText,
  ClearGraph,
  UpdateTab,
  UpdateSelection,
  UpdateOption,
  UpdateMin,
  UpdateMax,
  UpdateStep,

  InitialGui = 200
};

#define UI_INITIAL_GUI ControlType::InitialGui

#define UI_TITLE ControlType::Title
#define UI_LABEL ControlType::Label
#define UI_BUTTON ControlType::Button
#define UI_SWITCHER ControlType::Switcher
#define UI_PAD ControlType::Pad
#define UI_CPAD ControlType::Cpad
#define UI_SLIDER ControlType::Slider
#define UI_NUMBER ControlType::Number
#define UI_TEXT_INPUT ControlType::Text
#define UI_GRAPH ControlType::Graph
#define UI_ADD_GRAPH_POINT ControlType::GraphPoint

#define UPDATE_LABEL ControlType::UpdateLabel
#define UPDATE_SWITCHER ControlType::UpdateSwitcher
#define UPDATE_SLIDER ControlType::UpdateSlider
#define UPDATE_NUMBER ControlType::UpdateNumber
#define UPDATE_TEXT_INPUT ControlType::UpdateText
#define CLEAR_GRAPH ControlType::ClearGraph

// Colors
enum ControlColor : uint8_t { Turquoise, Emerald, Peterriver, Wetasphalt, Sunflower, Carrot, Alizarin, Dark, None = 0xFF };
#define COLOR_TURQUOISE ControlColor::Turquoise
#define COLOR_EMERALD ControlColor::Emerald
#define COLOR_PETERRIVER ControlColor::Peterriver
#define COLOR_WETASPHALT ControlColor::Wetasphalt
#define COLOR_SUNFLOWER ControlColor::Sunflower
#define COLOR_CARROT ControlColor::Carrot
#define COLOR_ALIZARIN ControlColor::Alizarin
#define COLOR_DARK ControlColor::Dark
#define COLOR_NONE ControlColor::None

class Control {
public:
  ControlType type;
  uint16_t id; // just mirroring the id here for practical reasons
  const char *label;
  void (*callback)(Control *, int);
  String value;
  ControlColor color;
  uint16_t parentControl;
  Control *next;

  static constexpr uint16_t noParent = 0xffff;

  Control(ControlType type, const char *label, void (*callback)(Control *, int), String value, ControlColor color,
          uint16_t parentControl = Control::noParent)
      : type(type), label(label), callback(callback), value(value), color(color), parentControl(parentControl), next(nullptr) {
    id = idCounter++;
  }

  Control(const Control &control)
      : type(control.type), id(control.id), label(control.label), callback(control.callback), value(control.value), color(control.color),
        parentControl(control.parentControl), next(control.next) {}

private:
  static uint16_t idCounter;
};

// Values
#define B_DOWN -1
#define B_UP 1

#define P_LEFT_DOWN -2
#define P_LEFT_UP 2
#define P_RIGHT_DOWN -3
#define P_RIGHT_UP 3
#define P_FOR_DOWN -4
#define P_FOR_UP 4
#define P_BACK_DOWN -5
#define P_BACK_UP 5
#define P_CENTER_DOWN -6
#define P_CENTER_UP 6

#define S_ACTIVE -7
#define S_INACTIVE 7

#define SL_VALUE 8
#define N_VALUE 9
#define T_VALUE 10
#define S_VALUE 11

enum Verbosity : uint8_t { Quiet = 0, Verbose, VerboseJSON };

class ESPUIClass {
public:
  ESPUIClass() { verbosity = Verbosity::Quiet; }
  ´ void setVerbosity(Verbosity verbosity);
  void begin(const char *_title, const char *username = nullptr, const char *password = nullptr);       // Setup server and page in Memorymode
  void beginSPIFFS(const char *_title, const char *username = nullptr, const char *password = nullptr); // Setup server and page in SPIFFSmode

  void prepareFileSystem(); // Initially preps the filesystem and loads a lot of stuff into SPIFFS
  void list();              // Lists SPIFFS directory

  uint16_t addControl(ControlType type, const char *label, String value = String(""), ControlColor color = ControlColor::Turquoise,
                      uint16_t parentControl = Control::noParent, void (*callback)(Control *, int) = nullptr);

  int button(const char *label, void (*callback)(Control *, int), ControlColor color, String value = "");            // Create Event Button
  int switcher(const char *label, bool startState, void (*callback)(Control *, int), ControlColor color);            // Create Toggle Button
  int pad(const char *label, bool centerButton, void (*callback)(Control *, int), ControlColor color);               // Create Pad Control
  int slider(const char *label, void (*callback)(Control *, int), ControlColor color, String value);                 // Create Slider Control
  int number(const char *label, void (*callback)(Control *, int), ControlColor color, int number, int min, int max); // Create a Number Input Control
  int text(const char *label, void (*callback)(Control *, int), ControlColor color, String value = "");              // Create a Text Input Control

  // Output only
  int label(const char *label, ControlColor color, String value = ""); // Create Label
  int graph(const char *label, ControlColor color);                    // Create Graph display

  // Update Elements

  Control *getControl(uint16_t id);

  // Update Elements
  void updateControl(uint16_t id, String value, int clientId = -1);
  void updateControl(Control *control, String value, int clientId = -1);

  void updateControl(uint16_t id, int clientId = -1);
  void updateControl(Control *control, int clientId = -1);

  void print(uint16_t id, String value);
  void updateLabel(uint16_t id, String value);
  void updateSwitcher(uint16_t id, bool nValue, int clientId = -1);
  void updateSlider(uint16_t id, int nValue, int clientId = -1);
  void updateNumber(uint16_t id, int nValue, int clientId = -1);
  void updateText(uint16_t id, String nValue, int clientId = -1);
  void updateSelect(uint16_t id, String nValue, int clientId = -1);
  void clearGraph(uint16_t id, int clientId = -1);
  void addGraphPoint(uint16_t id, int nValue, int clientId = -1);

  // Variables
  const char *ui_title = "ESPUI"; // Store UI Title and Header Name
  Control *controls = nullptr;
  void jsonDom(AsyncWebSocketClient *client);

  Verbosity verbosity;

private:
  const char *basicAuthUsername = nullptr;
  const char *basicAuthPassword = nullptr;
  bool basicAuth = true;
  AsyncWebServer *server;
  AsyncWebSocket *ws;
};

extern ESPUIClass ESPUI;
#endif