#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include <M5Cardputer.h>
#include <SD.h>
#include <SPI.h>
#include "src/dsp/miniacid_engine.h"
#include "cardputer_display.h"
#include <cstdarg>
#include <cstdio>
#include "src/ui/miniacid_display.h"
#include "miniacid_encoder8.h"
#include "scene_storage_cardputer.h"

static constexpr IGfxColor CP_BLACK = IGfxColor::Black();

CardputerDisplay g_display;
MiniAcidDisplay* g_miniDisplay = nullptr;
SceneStorageCardputer g_sceneStorage;

int16_t g_audioBuffer[AUDIO_BUFFER_SAMPLES];

TaskHandle_t g_audioTaskHandle = nullptr;

MiniAcid g_miniAcid(SAMPLE_RATE, &g_sceneStorage);
Encoder8Miniacid g_encoder8(g_miniAcid);

void audioTask(void *param) {
  while (true) {
    if (!g_miniAcid.isPlaying()) {
      vTaskDelay(10 / portTICK_PERIOD_MS);
      continue;
    }

    while (M5Cardputer.Speaker.isPlaying()) {
      vTaskDelay(1 / portTICK_PERIOD_MS);
    }

    g_miniAcid.generateAudioBuffer(g_audioBuffer, AUDIO_BUFFER_SAMPLES);

    M5Cardputer.Speaker.playRaw(g_audioBuffer, AUDIO_BUFFER_SAMPLES,
                                SAMPLE_RATE, false);
  }
}


void drawUI() {
  if (g_miniDisplay) g_miniDisplay->update();
}

void setup() {
  auto cfg = M5.config();
  M5Cardputer.begin(cfg);

  Serial.begin(115200);

  g_display.setRotation(1);
  g_display.begin();
  g_display.clear(CP_BLACK);

  M5Cardputer.Speaker.begin();
  M5Cardputer.Speaker.setVolume(200); // 0-255

  g_miniAcid.init();
  g_miniDisplay = new MiniAcidDisplay(g_display, g_miniAcid);

  xTaskCreatePinnedToCore(audioTask, "AudioTask",
                          4096, // stack
                          nullptr,
                          3, // priority
                          &g_audioTaskHandle,
                          1 // core
  );

  g_encoder8.initialize();

  drawUI();
}

void loop() {
  M5Cardputer.update();

  g_encoder8.update();

  if (M5Cardputer.BtnA.wasClicked()) {
    if (g_miniAcid.isPlaying()) {
      g_miniAcid.stop();
    } else {
      g_miniAcid.start();
    }
    drawUI();
  }

  static constexpr unsigned long KEY_REPEAT_DELAY_MS = 350;
  static constexpr unsigned long KEY_REPEAT_INTERVAL_MS = 80;
  static Keyboard_Class::KeysState lastKeysState{};
  static bool hasLastKeys = false;
  static unsigned long nextRepeatAt = 0;

  auto handleWithFallback = [&](UIEvent evt) {
    evt.event_type = MINIACID_KEY_DOWN;
    bool handled = g_miniDisplay ? g_miniDisplay->handleEvent(evt) : false;
    if (handled) return;

    char c = evt.key;
    if (c == '\n' || c == '\r') {
      if (g_miniDisplay) g_miniDisplay->dismissSplash();
      drawUI();
    } else if (c == '[') {
      if (g_miniDisplay) g_miniDisplay->previousPage();
      drawUI();
    } else if (c == ']') {
      if (g_miniDisplay) g_miniDisplay->nextPage();
      drawUI();
    } else if (c == 'i' || c == 'I') {
      g_miniAcid.randomize303Pattern(0);
      drawUI();
    } else if (c == 'o' || c == 'O') {
      g_miniAcid.randomize303Pattern(1);
      drawUI();
    } else if (c == 'p' || c == 'P') {
      g_miniAcid.randomizeDrumPattern();
      drawUI();
    } else if (c == '1') {
      g_miniAcid.toggleMute303(0);
      drawUI();
    } else if (c == '2') {
      g_miniAcid.toggleMute303(1);
      drawUI();
    } else if (c == '3') {
      g_miniAcid.toggleMuteKick();
      drawUI();
    } else if (c == '4') {
      g_miniAcid.toggleMuteSnare();
      drawUI();
    } else if (c == '5') {
      g_miniAcid.toggleMuteHat();
      drawUI();
    } else if (c == '6') {
      g_miniAcid.toggleMuteOpenHat();
      drawUI();
    } else if (c == '7') {
      g_miniAcid.toggleMuteMidTom();
      drawUI();
    } else if (c == '8') {
      g_miniAcid.toggleMuteHighTom();
      drawUI();
    } else if (c == '9') {
      g_miniAcid.toggleMuteRim();
      drawUI();
    } else if (c == '0') {
      g_miniAcid.toggleMuteClap();
      drawUI();
    } else if (c == 'k' || c == 'K') {
      g_miniAcid.setBpm(g_miniAcid.bpm() - 5.0f);
      drawUI();
    } else if (c == 'l' || c == 'L') {
      g_miniAcid.setBpm(g_miniAcid.bpm() + 5.0f);
      drawUI();
    } else if (c == ' ') {
      if (g_miniAcid.isPlaying()) {
        g_miniAcid.stop();
      } else {
        g_miniAcid.start();
      }
      drawUI();
    }
  };

  auto processKeys = [&](const Keyboard_Class::KeysState& ks) {
    for (auto hid : ks.hid_keys) {
      UIEvent evt{};
      evt.alt = ks.alt;
      bool shouldSend = false;
      if (hid == 0x33) {
        evt.scancode = MINIACID_UP;
        shouldSend = true;
      } else if (hid == 0x37) {
        evt.scancode = MINIACID_DOWN;
        shouldSend = true;
      } else if (hid == 0x36) {
        evt.scancode = MINIACID_LEFT;
        shouldSend = true;
      } else if (hid == 0x38) {
        evt.scancode = MINIACID_RIGHT;
        shouldSend = true;
      } else if (hid == 0x28 || hid == 0x58) { // enter
        evt.key = '\n';
        shouldSend = true;
      } else if (hid == KEY_BACKSPACE) {
        evt.key = '\b';
        shouldSend = true;
      } else if (hid == KEY_TAB) {
        evt.key = '\t';
        shouldSend = true;
      }
      if (shouldSend) {
        handleWithFallback(evt);
      }
    }

    for (auto inputChar : ks.word) {
      UIEvent evt{};
      evt.alt = ks.alt;
      evt.key = inputChar;
      if (evt.key == '`' || evt.key == '~') { // escape
        evt.scancode = MINIACID_ESCAPE;
      }
      handleWithFallback(evt);
    }
  };

  bool keyChanged = M5Cardputer.Keyboard.isChange();
  bool keyPressed = M5Cardputer.Keyboard.isPressed();
  if (keyChanged && keyPressed) {
    if (g_miniDisplay) g_miniDisplay->dismissSplash();
    Keyboard_Class::KeysState ks = M5Cardputer.Keyboard.keysState();
    processKeys(ks);
    lastKeysState = ks;
    hasLastKeys = true;
    nextRepeatAt = millis() + KEY_REPEAT_DELAY_MS;
  } else if (!keyPressed) {
    hasLastKeys = false;
  } else if (hasLastKeys && millis() >= nextRepeatAt) {
    if (g_miniDisplay) g_miniDisplay->dismissSplash();
    processKeys(lastKeysState);
    nextRepeatAt = millis() + KEY_REPEAT_INTERVAL_MS;
  }

  static unsigned long lastUIUpdate = 0;
  if (millis() - lastUIUpdate > 80) {
    lastUIUpdate = millis();
    if (g_miniDisplay) g_miniDisplay->update();
  }

  delay(5);
}