#include "PinConfig.h"

#ifdef USB
#include "USBOutput.h"

char bottomRow[] = {'a', 'z', 's', 'x', 'd', 'c', 'f', 'v', 'g', 'b', 'h', 'n', 'j', 'm', 'k', ','};
char topRow[] = {'1', 'q', '2', 'w', '3', 'e', '4', 'r', '5', 't', '6', 'y', '7', 'u', '8', 'i'};

void USBOutput::sendKeyEvent(int key, bool pressed, bool doublePressed)
{
  if (pressed) {
    if (doublePressed) {
      writeKey(topRow[key]);
    }
    else {
      pressKey(bottomRow[key]);
    }
  }
  else {
    releaseKey(bottomRow[key]);
  }
}

void USBOutput::sendSensorEvent(float position)
{
  // Send hand up / hand down key
  if (position > lastPosition)
  {
    writeKey(KEY_PAGE_UP);
    if (wasAirHeld)
    {
      // Send Air Action
      writeKey(KEY_END);
    }
  }
  if (position < lastPosition)
  {
    writeKey(KEY_PAGE_DOWN);
    if (wasAirHeld)
    {
      // Send Air Action
      writeKey(KEY_END);
    }
  }

  // Send hand seen / unseen key
  if (position > 0.05f)
  {
    pressKey(KEY_HOME);
    wasAirHeld = true;
  }
  else {
    releaseKey(KEY_HOME);
    wasAirHeld = false;
  }

  // Save last position for next loop
  lastPosition = position;
}

USBOutput::USBOutput()
{
#ifndef TEENSY
  NKROKeyboard.begin();
#endif
  lastPosition = 0;
}

void USBOutput::writeKey(uint16_t key)
{
#ifndef TEENSY
  NKROKeyboard.write(key);
#else
  switch (key)
  {
    case KEY_PAGE_UP:
    case KEY_PAGE_DOWN:
    case KEY_HOME:
    case KEY_END:
      // Don't use ASCII method
      Nkro.set_key(key);
      Nkro.send_nkro_now();
      Nkro.reset_key(key);
      Nkro.send_nkro_now();
      break;
    default:
      // Use NKRO for ASCII
      Nkro.set_key_ascii(key);
      Nkro.send_nkro_now();
      Nkro.reset_key_ascii(key);
      Nkro.send_nkro_now();
      break;
  }
#endif
}

void USBOutput::pressKey(uint16_t key)
{
#ifndef TEENSY
  pressKey(key);
#else
  switch (key)
  {
    case KEY_PAGE_UP:
    case KEY_PAGE_DOWN:
    case KEY_HOME:
    case KEY_END:
      // Don't use ASCII method
      Nkro.set_key(key);
      Nkro.send_nkro_now();
      break;
    default:
      // Use NKRO for ASCII
      Nkro.set_key_ascii(key);
      Nkro.send_nkro_now();
      break;
  }
#endif
}

void USBOutput::releaseKey(uint16_t key)
{
#ifndef TEENSY
  NKROKeyboard.release(key);
#else
  switch (key)
  {
    case KEY_PAGE_UP:
    case KEY_PAGE_DOWN:
    case KEY_HOME:
    case KEY_END:
      // Don't use ASCII method
      Nkro.reset_key(key);
      Nkro.send_nkro_now();
      break;
    default:
      // Use NKRO for ASCII
      Nkro.reset_key_ascii(key);
      Nkro.send_nkro_now();
      break;
  }
#endif
}

#endif