#include <XInput.h>
#include <NintendoExtensionCtrl.h>


GuitarController guitar;


void setup() {
  guitar.begin();
  while (!guitar.connect()) {
    delay(1000);
  }
  XInput.begin();
  XInput.setRange(JOY_RIGHT, 0, 31);
}

void loop() {
  while (!guitar.connect()) {
    delay(1000);
  }

  boolean success = guitar.update();
  if (!success) {  // Ruh roh
    delay(1000);
  }

  // STRUM
  if (guitar.strumUp()) {
    XInput.press(DPAD_UP);
    XInput.release(DPAD_DOWN);
  }
  else if (guitar.strumDown()) {
    XInput.press(DPAD_DOWN);
    XInput.release(DPAD_UP);
  }
  else {
    XInput.release(DPAD_UP);
    XInput.release(DPAD_DOWN);
  }

  // FRETS
  boolean green = guitar.fretGreen();
  boolean red = guitar.fretRed();
  boolean yellow = guitar.fretYellow();
  boolean blue = guitar.fretBlue();
  boolean orange = guitar.fretOrange();

  if (green == true) {
    XInput.press(BUTTON_A);
  }
  else if (green == false) {
    XInput.release(BUTTON_A);
  }

  if (red == true) {
    XInput.press(BUTTON_B);
  }
  else if (red == false) {
    XInput.release(BUTTON_B);
  }

  if (yellow == true) {
    XInput.press(BUTTON_Y);
  }
  else if (yellow == false) {
    XInput.release(BUTTON_Y);
  }

  if (blue == true) {
    XInput.press(BUTTON_X);
  }
  else if (blue == false) {
    XInput.release(BUTTON_X);
  }

  if (orange == true) {
    XInput.press(BUTTON_LB);
  }
  else if (orange == false) {
    XInput.release(BUTTON_LB);
  }

  // WHAMMY
  uint8_t whammy = guitar.whammyBar();
  XInput.setJoystickX(JOY_RIGHT, whammy);

  // PLUS BUTTON

  boolean plusButton = guitar.buttonPlus();

  if (plusButton == true) {
    XInput.press(BUTTON_START);
  }
  else if (plusButton == false) {
    XInput.release(BUTTON_START);
  }

  // MINUS BUTTON

  boolean minusButton = guitar.buttonMinus();

  if (minusButton == true) {
    XInput.press(BUTTON_BACK);
  }
  else if (minusButton == false) {
    XInput.release(BUTTON_BACK);
  }
}