#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ogcsys.h>
#include <malloc.h>
#include <ctype.h>
#include <wiilight.h>
#include <wiidrc/wiidrc.h>

#include "sys.h"
#include "title.h"
#include "aes.h"
#include "gui.h"
#include "menu.h"
#include "restart.h"
#include "sys.h"
#include "video.h"
#include "wpad.h"
#include "wkb.h"
#include "fat.h"
#include "nand.h"
#include "globals.h"
#include "iospatch.h"
#include "fileops.h"

// Globals
CONFIG gConfig;

// Prototypes
extern void __exception_setreload(int t);
extern u32 WaitButtons (void);
void CheckPassword (void);
void SetDefaultConfig (void);
int ReadConfigFile (void);
int GetIntParam (char *inputStr);
int GetStartupPath (char *startupPath, char *inputStr);
int GetStringParam (char *outParam, char *inputStr, int maxChars);
int LaunchApp(const char* path, bool external);

// Default password Up-Down-Left-Right-Up-Down
//#define PASSWORD "UDLRUD"
void CheckPassword (void)
{
	char curPassword [11]; // Max 10 characters password, NULL terminated
	int count = 0;

	if (strlen (gConfig.password) == 0)
		return;

	// Ask user for a password. Press "B" to restart Wii
	printf("[+] [Enter Password to Continue]:\n\n");

	printf(">>  Press A to continue.\n");
	printf(">>  Press B button to cancel.\n");

	/* Wait for user answer */
	for (;;) 
	{
		u32 buttons = WaitButtons();

		if (buttons & WPAD_BUTTON_A)
		{
			// A button, validate the pw
			curPassword [count] = 0;
			//if (strcmp (curPassword, PASSWORD) == 0)
			if (strcmp (curPassword, gConfig.password) == 0)
			{
				printf(">>  Password Accepted...\n");
				break;
			}
			else
			{
				printf ("\n");
				printf(">>  Incorrect Password. Try again...\n");
				printf("[+] [Enter Password to Continue]:\n\n");
				printf(">>  Press A to continue.\n");
				printf(">>  Press B button to cancel.\n");
				count = 0;
			}
		}
		else if (buttons & WPAD_BUTTON_B)
			// B button, restart
			Restart();
		else
		{
			if (count < 10)
			{
				// Other buttons, build the password
				if (buttons & WPAD_BUTTON_LEFT)
				{
					curPassword [count++] = 'L';
					printf ("*");
				}
				else if (buttons & WPAD_BUTTON_RIGHT)
				{
					curPassword [count++] = 'R';
					printf ("*");
				}
				else if (buttons & WPAD_BUTTON_UP)
				{
					curPassword [count++] = 'U';
					printf ("*");
				}
				else if (buttons & WPAD_BUTTON_DOWN)
				{
					curPassword [count++] = 'D';
					printf ("*");
				}
				else if (buttons & WPAD_BUTTON_1)
				{
					curPassword [count++] = '1';
					printf ("*");
				}
				else if (buttons & WPAD_BUTTON_2)
				{
					curPassword [count++] = '2';
					printf ("*");
				}
			}
		}
	}
}

void Disclaimer(void)
{
	/* Print disclaimer */
	printf("[+] [DISCLAIMER]:\n\n");

	printf("    THIS APPLICATION COMES WITH NO WARRANTY AT ALL,\n");
	printf("    NEITHER EXPRESS NOR IMPLIED.\n");
	printf("    I DO NOT TAKE ANY RESPONSIBILITY FOR ANY DAMAGE IN YOUR\n");
	printf("    WII CONSOLE BECAUSE OF A IMPROPER USAGE OF THIS SOFTWARE.\n\n");

	printf(">>  If you agree, press A button to continue.\n");
	printf(">>  Otherwise, press B button to exit.\n");

	/* Wait for user answer */
	for (;;) {
		//u32 buttons = Wpad_WaitButtons();
		u32 buttons = WaitButtons();

		/* A button */
		if (buttons & WPAD_BUTTON_A)
			break;

		/* B button */
		if (buttons & WPAD_BUTTON_B)
			Restart();
	}
}

int main(int argc, char **argv)
{
	__exception_setreload(10);

	ES_GetBoot2Version(&boot2version);
	if (!AHBPROT_DISABLED)
/*
	We should just enable it tbh.
	Like, look at https://github.com/WiiLink24/wfc-patcher-wii/blob/main/launcher/source/IOS.cpp
	Awesome stuff.
 */
	{
		if (boot2version < 5)
		{
			if (!loadIOS(202)) if (!loadIOS(222)) if (!loadIOS(223)) if (!loadIOS(224)) if (!loadIOS(249)) loadIOS(36);
		} else {
			if (!loadIOS(249)) loadIOS(36);
		}
	}

	/* Initialize subsystems */
	Sys_Init();

	/* Set video mode */
	Video_SetMode();

	FatMount();

	/* Initialize console */
	Gui_InitConsole();

	/* Draw background */
	Gui_DrawBackground();

	/* Initialize Wiimote and GC Controller */
	Wpad_Init();
	PAD_Init();
	WiiDRC_Init();
	WKB_Initialize();
	WIILIGHT_Init();

	AES_Init();
	Title_SetupCommonKeys();

	/* Print disclaimer */
	//Disclaimer();
	
	// Set the defaults
	SetDefaultConfig ();

	// Read the config file
	ReadConfigFile();

	// Check password
	CheckPassword();

	/* Menu loop */
	Menu_Loop();

	FatUnmount();

	/* Restart Wii */
	Restart_Wait();

	return 0;
}

int ReadConfigFile()
{
	FILE* fptr;
	char tmpStr[MAX_FILE_PATH_LEN];
	char tmpOutStr[40], path[128];
	s32 i;
	bool found = false;

	for (i = 0; i < FatGetDeviceCount(); i++)
	{	
		snprintf(path, sizeof(path), "%s:%s", FatGetDevicePrefix(i), WM_CONFIG_FILE_PATH);
		if (FSOPFileExists(path))
		{
			found = true;
			break;
		}
	}
	
	if (!found)
		return -1;

	// Read the file
	// fptr = fopen(path, "rb"); // umm, why are we opening with mode rb and then using fgets?
	fptr = fopen(path, "r");
	if (!fptr) {
		// perror(path);
		return -1;
	}

	// Read the options
	while (true)
	{
		if (!fgets(tmpStr, MAX_FILE_PATH_LEN, fptr))
			break;

		else if (isalpha((int)tmpStr[0]))
		{
			// Get the password
			if (strncmp (tmpStr, "Password", 8) == 0)
			{
				// Get password
				// GetPassword (gConfig.password, tmpStr);
				GetStringParam (gConfig.password, tmpStr, MAX_PASSWORD_LENGTH);

				// If password is too long, ignore it
				if (strlen (gConfig.password) > 10)
				{
					gConfig.password [0] = 0;
					puts("Password longer than 10 characters; will be ignored. Press a button...");
					WaitButtons ();
				}
			}
				
			// Get startup path
			else if (strncmp (tmpStr, "StartupPath", 11) == 0)
			{
				// Get startup Path
				GetStartupPath (gConfig.startupPath, tmpStr);
			}

			// cIOS
			else if (strncmp (tmpStr, "cIOSVersion", 11) == 0)
			{
				// Get cIOSVersion
				gConfig.cIOSVersion = GetIntParam(tmpStr);
			}

			// FatDevice
			else if (strncmp (tmpStr, "FatDevice", 9) == 0)
			{
				// Get fatDevice
				GetStringParam (tmpOutStr, tmpStr, MAX_FAT_DEVICE_LENGTH);
				for (i = 0; i < 5; i++)
				{
					if (strncmp(FatGetDevicePrefix(i), tmpOutStr, 4) == 0)
					{
						gConfig.fatDeviceIndex = i;
					}
				}
			}
					
			// NandDevice
			else if (strncmp (tmpStr, "NANDDevice", 10) == 0)
			{
				// Get fatDevice
				GetStringParam (tmpOutStr, tmpStr, MAX_NAND_DEVICE_LENGTH);
				for (i = 0; i < 3; i++)
				{
					if (strncmp (ndevList[i].name, tmpOutStr, 2) == 0)
					{
						gConfig.nandDeviceIndex = i;
					}
				}
			}
		}
	} // EndWhile
			
	// Close the config file
	fclose (fptr);

	return 0;
} // ReadConfig


void SetDefaultConfig (void)
{
	// Default password is NULL or no password
	gConfig.password [0] = 0;
	
	// Default startup folder
	strcpy (gConfig.startupPath, WAD_ROOT_DIRECTORY);
	
	gConfig.cIOSVersion = CIOS_VERSION_INVALID;            // Means that user has to select later
	gConfig.fatDeviceIndex = FAT_DEVICE_INDEX_INVALID;     // Means that user has to select
	gConfig.nandDeviceIndex = NAND_DEVICE_INDEX_INVALID;   // Means that user has to select

} // SetDefaultConfig


int GetStartupPath (char *startupPath, char *inputStr)
{
	int i = 0;
	int len = strlen (inputStr);
	
	// Find the "="
	while ((inputStr [i] != '=') && (i < len))
	{
		i++;
	}
	i++;

	// Get to the "/"
	while ((inputStr [i] != '/') && (i < len))
	{
		i++;
	}

	// Get the startup Path
	int count = 0;
	while (isascii(inputStr [i]) && (i < len) && (inputStr [i] != '\n') && 
	         (inputStr [i] != '\r') && (inputStr [i] != ' '))
	{
		startupPath [count++] = inputStr [i++];
	}
	startupPath [count] = 0; // NULL terminate

	return (0);
} // GetStartupPath

int GetIntParam (char *inputStr)
{
	int retval = 0;
	int i = 0;
	int len = strlen (inputStr);
	char outParam [40];
	
	// Find the "="
	while ((inputStr [i] != '=') && (i < len))
	{
		i++;
	}
	i++;
	
	// Get to the first alpha numeric character
	while ((isdigit((int)inputStr[i]) == 0) && (i < len))
	{
		i++;
	}

	// Get the string param
	int outCount = 0;
	while ((isdigit((int)inputStr[i])) && (i < len) && (outCount < 40))
	{
		outParam [outCount++] = inputStr [i++];
	}
	outParam [outCount] = 0; // NULL terminate
	retval = atoi (outParam);
	
	return (retval);
} // GetIntParam


int GetStringParam (char *outParam, char *inputStr, int maxChars)
{
	int i = 0;
	int len = strlen (inputStr);
	
	// Find the "="
	while ((inputStr [i] != '=') && (i < len))
	{
		i++;
	}
	i++;
	
	// Get to the first alpha character
	while ((isalpha((int)inputStr[i]) == 0) && (i < len))
	{
		i++;
	}
	
	// Get the string param
	int outCount = 0;
	while ((isalnum((int)inputStr[i])) && (i < len) && (outCount < maxChars))
	{
		outParam [outCount++] = inputStr [i++];
	}
	outParam [outCount] = 0; // NULL terminate
	
	return (0);
} // GetStringParam