#include <malloc.h>
#include <setjmp.h>
#include <stdlib.h>
#include <string.h>

#include <3ds.h>
#include "3ds/services/gsplcd.h"
#include "3ds/services/hid.h"
#include "3ds/services/soc.h"

#include "drawing.h"
#include "input.h"
#include "keyboard.h"
#include "settings.h"
#include "wireless.h"

static jmp_buf exitJmp;

void hang(char *message) {
  while (aptMainLoop()) {
    hidScanInput();

    clearScreen();
    drawString(10, 10, "%s", message);
    drawString(10, 20, "Start and Select to exit");

    u32 kHeld = hidKeysHeld();
    if ((kHeld & KEY_START) && (kHeld & KEY_SELECT))
      longjmp(exitJmp, 1);

    gfxFlushBuffers();
    gspWaitForVBlank();
    gfxSwapBuffers();
  }
}

int main(void) {
  acInit();
  gfxInitDefault();

  gfxSetDoubleBuffering(GFX_TOP, true);
  gfxSetDoubleBuffering(GFX_BOTTOM, true);

  if (setjmp(exitJmp))
    goto exit;

  // preRenderKeyboard();

  // clearScreen();
  drawString(10, 10, "Initing FS...");
  gfxFlushBuffers();
  gfxSwapBuffers();

  fsInit();

  // clearScreen();
  drawString(10, 10, "Initing SOC...");
  gfxFlushBuffers();
  gfxSwapBuffers();

  socInit((u32 *)memalign(0x1000, 0x100000), 0x100000);

  u32 wifiStatus = 0;
  ACU_GetWifiStatus(&wifiStatus);
  if (!wifiStatus) {
    hang("No WiFi! Is your wireless slider on?");
  }

  clearScreen();
  drawString(10, 10, "Reading settings...");
  gfxFlushBuffers();
  gfxSwapBuffers();

  if (!readSettings()) {
    hang("Could not read 3DSController.ini!");
  }

  clearScreen();
  drawString(10, 10, "Connecting to %s on port %d...", settings.IPString,
             settings.port);
  gfxFlushBuffers();
  gfxSwapBuffers();

  openSocket(settings.port);
  sendConnectionRequest();

//   clearScreen();
  drawString(10, 10, "Sending data to %s on port %d...", settings.IPString,
             settings.port);
  gfxFlushBuffers();
  gfxSwapBuffers();

  GSPLCD_PowerOffAllBacklights();

  while (aptMainLoop()) {
    hidScanInput();
    irrstScanInput();

    u32 kHeld = hidKeysHeld();
    circlePosition circlePad;
    circlePosition cStick;
    angularRate gyro;
    accelVector accel;
    u8 vol8;
    u8 *volPointer = &vol8; // As a test for pointing at things.
    // u8 threeD;
    // u8 *threeDPointer = &threeD;
    hidCstickRead(&cStick);
    hidCircleRead(&circlePad);
    HIDUSER_EnableGyroscope();
    hidGyroRead(&gyro);
    HIDUSER_EnableAccelerometer();
    hidAccelRead(&accel);
    HIDUSER_GetSoundVolume(volPointer);
    // MCUHWC_GetSoundSliderLevel(threeDPointer);
    u32 volume = (u32)vol8; // Upscale to 32 for transmission
    // u32 threeD32 = (u32)threeD;
    touchPosition touch;
    touchRead(&touch);

    clearScreen();

    if ((kHeld & KEY_L) && (kHeld & KEY_R) && (kHeld & KEY_X)) {
      if (keyboardToggle) {
        keyboardActive = !keyboardActive;
        keyboardToggle = false;

        if (keyboardActive) {
          GSPLCD_PowerOnAllBacklights();
        } else {
          GSPLCD_PowerOffAllBacklights();
        }
      }
    } else
      keyboardToggle = true;

    if (keyboardActive) {
      drawKeyboard();

      if (touch.px >= 1 && touch.px <= 312 && touch.py >= 78 &&
          touch.py <= 208) {
        int x = (int)((float)touch.px * 12.0f / 320.0f);
        int y = (int)((float)(touch.py - 78) * 12.0f / 320.0f);
        int width = 24;
        int height = 24;

        if (keyboardChars[x + y * 12] == ' ') {
          while (keyboardChars[(x - 1) + y * 12] == ' ')
            x--;

          width = (int)(5.0f * 320.0f / 12.0f) - 1;
        }

        else if (keyboardChars[x + y * 12] == '\13') {
          while (keyboardChars[(x - 1) + y * 12] == '\13')
            x--;
          while (keyboardChars[x + (y - 1) * 12] == '\13')
            y--;

          width = (int)(2.0f * 320.0f / 12.0f) - 1;
          height = (int)(3.0f * 320.0f / 12.0f) - 1;
        }

        if (keyboardChars[x + y * 12])
          drawBox((int)((float)x * 320.0f / 12.0f) + 1,
                  (int)(78.0f + (float)y * 320.0f / 12.0f) + 1, width, height,
                  31, 31, 0);
      }
    }

    sendKeys(kHeld, circlePad, touch, cStick, volume, gyro, accel);
    // drawString(10, 10, "Volume: %x", volume);
    // receiveBuffer(sizeof(struct packet));

    if ((kHeld & KEY_START) && (kHeld & KEY_SELECT)) {
      sendKeys(0, circlePad, touch, cStick, volume, gyro, accel);
      longjmp(exitJmp, 1);
    }

    gfxFlushBuffers();
    gspWaitForVBlank();
    gfxSwapBuffers();
  }

exit:

  GSPLCD_PowerOnAllBacklights();

  socExit();

  svcCloseHandle(fileHandle);
  fsExit();

  gfxExit();
  acExit();

  return 0;
}