#include <Arduino.h>

#include <string>

#include "display.h"
#include "fingerprint.h"
#include "mqtt.h"
#include "wifi.h"

#define ADC_EN 14
#define ADC_PIN 34

// #include <Button2.h>
// #define BUTTON_1 35
// #define BUTTON_2 0

void setup() {
  Serial.begin(115200);
  Serial.println("Serial.begin");

  fingerprint::Setup();  // go early since the others display status on our LED
  display::Setup();
  display::Message("Hello world");
  wifi::Setup();
  mqtt::Setup();
}

namespace {
uint16_t LastComponentNumber(const std::string &s) {
  return atoi(s.substr(s.rfind("/") + 1).c_str());
}
}  // namespace

void loop() {
  Serial.println("--loop--");

  fingerprint::ExecuteAnyQueued();

  fingerprint::ScanLoop();

  if (mqtt::HasPendingMessage()) {
    std::pair<std::string, std::vector<byte>> msg = mqtt::PopPendingMessage();
    const std::string &topic = msg.first;
    const std::vector<byte> &payload = msg.second;
    const std::string payload_string(payload.begin(), payload.end());

    if (topic == "fingerprint/command") {
      if (payload_string == "enroll") {
        fingerprint::Enroll();
      } else if (payload_string == "show_success") {
        fingerprint::BlinkSuccess();
        while (!mqtt::HasPendingMessage()) yield();
        mqtt::PopPendingMessage();
        // hope it's "clear_success", but who cares
        fingerprint::BlinkClearSuccess();
      } else if (payload_string == "delete_all") {
        fingerprint::DeleteAll();
      } else if (payload_string.rfind("delete/model/", 0) == 0) {
        uint16_t fid = LastComponentNumber(payload_string);
        fingerprint::DeleteModel(fid);
      } else if (payload_string.rfind("download/model/", 0) == 0) {
        uint16_t fid = LastComponentNumber(payload_string);
        fingerprint::DownloadModel(fid);
      }
    } else if (topic.rfind("fingerprint/set/model/", 0) == 0) {
      uint16_t fid = LastComponentNumber(topic);

      fingerprint::SetModel(fid, payload);
    }
  }
}