// 3DS Controller Server

#define VERSION 0.01

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

#include "wireless.h"
#include "keys.h"
#include "general.h"
#include "joystick.h"
#include "settings.h"

int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPSTR lpCmd, int nShow) {
	printf("3DS Controller Server %.2f\n", VERSION);
	
	bool vJoy = true;
	UINT iInterface = 1;
	
	if(vJoy && !vJoyEnabled()) {
		printf("vJoy failed (1)! Buttons will still work, but joy stick won't work.\n");
		vJoy = false;
	}
	
	enum VjdStat status = GetVJDStatus(iInterface);
	if(vJoy && (status == VJD_STAT_OWN || (status == VJD_STAT_FREE && !AcquireVJD(iInterface)))) {
		printf("vJoy failed (2)! Buttons will still work, but joy stick won't work.\n");
		vJoy = false;
	}
	
	ContPovNumber = GetVJDContPovNumber(iInterface);
	//int DiscPovNumber = GetVJDDiscPovNumber(iInterface);
	
	if(vJoy && !updateJoystick(128 * 128, 128 * 128)) {
		printf("vJoy failed (3)! Buttons will still work, but joy stick won't work.\n");
		vJoy = false;
	}
	
	initNetwork();
	
	printf("Running on: %s\n\n", hostName);
	
	if(!readSettings()) {
		printf("Couldn't read settings file, using default key bindings.\n");
	}
	
	startListening();
	
	while(1) {
		//ZeroMemory(buffer, sizeof(struct packet));
		memset(&buffer, 0, sizeof(packet));
		
		while(receiveBuffer(sizeof(struct packet)) <= 0) {
			// Waiting
		}
		
		switch(buffer[0]) {
			case CONNECT:
				lastKeys = 0;
				currentKeys = 0;
				circlePad.x = 0;
				circlePad.y = 0;
				lastTouch.x = 0;
				lastTouch.y = 0;
				currentTouch.x = 0;
				currentTouch.y = 0;
				
				buffer[0] = CONNECT;
				printf("3DS Connected!\n");
				
				Sleep(50);
				sendBuffer(1);
				
				Sleep(50);
				sendBuffer(1);
				
				Sleep(50);
				sendBuffer(1);
				break;
			
			case KEYS:
				lastKeys = currentKeys;
				if(currentKeys & KEY_TOUCH) lastTouch = currentTouch;
				
				#pragma GCC diagnostic push
				#pragma GCC diagnostic ignored "-Wstrict-aliasing"
				memcpy(&currentKeys, &((struct packet *)buffer)->keys, 4);
				memcpy(&circlePad, &((struct packet *)buffer)->circlePad, 4);
				memcpy(&currentTouch, &((struct packet *)buffer)->touch, 4);
				#pragma GCC diagnostic pop
				
				handleKey(KEY_A, settings.A);
				handleKey(KEY_B, settings.B);
				handleKey(KEY_SELECT, settings.Select);
				handleKey(KEY_START, settings.Start);
				handleKey(KEY_DRIGHT, settings.Right);
				handleKey(KEY_DLEFT, settings.Left);
				handleKey(KEY_DUP, settings.Up);
				handleKey(KEY_DDOWN, settings.Down);
				handleKey(KEY_R, settings.R);
				handleKey(KEY_L, settings.L);
				handleKey(KEY_X, settings.X);
				handleKey(KEY_Y, settings.Y);
				handleKey(KEY_TOUCH, settings.Tap);
				//handleKey(KEY_LID, 'I');
				
				if(newpress(KEY_TOUCH)) {
					lastTouch.x = currentTouch.x;
					lastTouch.y = currentTouch.y;
				}
				
				if(currentKeys & KEY_TOUCH) {
					if(settings.touch == mouse) {
						POINT p;
						GetCursorPos(&p);
						SetCursorPos(p.x + (currentTouch.x - lastTouch.x), p.y + (currentTouch.y - lastTouch.y));
					}
					else if(settings.touch == joystick) {
						if(vJoy) updateJoystick((currentTouch.x) * 128, (240 - currentTouch.y) * 128);
					}
				}
				
				if(settings.circlePad == mouse) {
					POINT p;
					GetCursorPos(&p);
					SetCursorPos(p.x + circlePad.x / 32, p.y - circlePad.y / 32);
				}
				else if(settings.circlePad == joystick) {
					if(vJoy) updateJoystick((circlePad.x + 128) * 128, (128 - circlePad.y) * 128);
				}
				
				break;
		}
	}
	
	error("accept()");
	return 0;
}