/*-------------------------------------------------------------
 
regionchange.c -- Region Changing application
 
Copyright (C) 2008 tona
 
This software is provided 'as-is', without any express or implied
warranty.  In no event will the authors be held liable for any
damages arising from the use of this software.
 
Permission is granted to anyone to use this software for any
purpose, including commercial applications, and to alter it and
redistribute it freely, subject to the following restrictions:
 
1.The origin of this software must not be misrepresented; you
must not claim that you wrote the original software. If you use
this software in a product, an acknowledgment in the product
documentation would be appreciated but is not required.
 
2.Altered source versions must be plainly marked as such, and
must not be misrepresented as being the original software.
 
3.This notice may not be removed or altered from any source
distribution.
 
-------------------------------------------------------------*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <malloc.h>
#include <assert.h>
#include <string.h>
#include <ogcsys.h>
#include <gccore.h>
#include <ogc/system.h>
#include <wiiuse/wpad.h>

#include "wiibasics.h"
#include "runtimeiospatch.h"
#include "sysconf.h"
#include "id.h"
#include "detect_settings.h"
#include "gecko.h"

#define ITEMS 	11
#define SADR_LENGTH 0x1007 + 1
#define WARNING_SIGN "\x1b[30;1m\x1b[43;1m/!\\\x1b[37;1m\x1b[40m"
#define maxdata    256

u32 selected = 0;
char page_contents[ITEMS][64];

int lang, area, game, video, region, country, countrystr, eula;
u8 sadr[SADR_LENGTH];

char languages[][ITEMS] = {"Japanese ", "English  ", "German   ", "French   ", "Spanish  ", "Italian  ", "Dutch    "};
char areas[][ITEMS] = {"Japan   ", "USA     ", "Europe  ", "Australia", "Brazil  ", "Taiwan  ", "China   ", "Korea   ", "Hong Kong", "Asia    ", "Latin Am.", "S. Africa"};
char regions[][ITEMS] = {"Japan   ", "USA     ", "Europe  ", "Korea   "};
char vmodes[][ITEMS] = {"NTSC    ", "PAL     ", "MPAL    "};
char eulas[][ITEMS] = {"Unread  ", "Read    "};

void draw_credits() { 
	ClearScreen();
	PrintBanner();
     Console_SetPosition(3, 0); 
	printf("\t\t\t\tCREDITS:\n\n\n");
	printf("\tOriginal Program:\n");
	printf("\t- Tona\n");
	printf("\n\tOthers in Original Credits: \n");
	printf("\t- Bushing, svpe, Marcan, Waninkoko, Crediar, \n");
	printf("\t- ChipD, Sorak05, serlex, pcg, callmebob, and dieforit .\n");
	printf("\n\tARC ModMii Edition: \n");
	printf("\t- Scooby74029(FakeCoder) - Coding \n");
	printf("\t- Xflak - testing, overall help \n");
	printf("\t- TheShadowEevee - KR region support \n");
	printf("\t- To Many Coders To Thank For The Great Code I Used ..... \n");
	printf("\n");
	Console_SetPosition(26,0);
	ClearLine();
	Console_SetPosition(26, 20);
	printf("Press Any Button To Return ..... ");
}
void Draw_Disclaimer() {
	ClearScreen();
	PrintBanner();
	printf("\n\t  This software comes supplied with absolutely no warranty.\n");
	printf("\t\t\t\tUse this software at your own risk.\n");
	
	printf("\n\n\n\t\t\t\t" WARNING_SIGN " IMPORTANT BRICK INFORMATION " WARNING_SIGN "\n\n");
	printf("\n\tSemi Bricks are caused by the Console Area setting not matching\n");
	printf("\tyour System Menu region. A semi-brick in itself is not terribly\n");
	printf("\tharmful, but it can easily deteriorate into a full brick--there\n");
	printf("\tare multiple simple triggers to do so.\n");
	printf("\n\n\n\tIn order to practice proper safety when using this application, \n");
	printf("\tplease make sure that your Console Area and System Menu region \n");
	printf("\tare congruent before rebooting your system. A warning will be\n");
	printf("\tdisplayed if they are not in agreement.\n");
	
	sleep(5);
	printf("\n\n\n\n\t\t\t\t[A] Continue             [Home] Exit\n");
	wait_key(WPAD_BUTTON_A);
}
void handleError(const char* string, int errorval){
	printf("Unexpected Error: %s Value: %d\n", string, errorval);
	printf("Press any key to quit\n");
	wait_anyKey();
	exit(0);
}

void getSettings(void){
	int ret;
	lang = SYSCONF_GetLanguage();
	area = SYSCONF_GetArea();
	game = SYSCONF_GetRegion(); 
	video = SYSCONF_GetVideo();
	eula = SYSCONF_GetEULA();
	if (lang < 0 || area < 0 || game < 0 || video < 0 || (eula != SYSCONF_ENOENT && eula < 0)){
		printf("Error getting settings! %d, %d, %d, %d, %d\n", lang, area, game, video, eula);
		wait_anyKey();
		exit(1);
	}
	
	if (SYSCONF_GetLength("IPL.SADR") != SADR_LENGTH) handleError("IPL.SADR Length Incorrect", SYSCONF_GetLength("IPL.SADR"));
	ret = SYSCONF_Get("IPL.SADR", sadr, SADR_LENGTH);
	if (ret < 0 ) handleError("SYSCONF_Get IPL.SADR", ret);
	country = sadr[0];
	gprintf("\n\ncountry[%i] \n\n", country);
	
}

void saveSettings(void){
	Console_SetPosition(24, 0);
	int ret = 0;
	if (lang != SYSCONF_GetLanguage()) ret = SYSCONF_SetLanguage(lang);
	if (ret) handleError("SYSCONF_SetLanguage", ret);
	if (area != SYSCONF_GetArea()) ret = SYSCONF_SetArea(area);
	if (ret) handleError("SYSCONF_SetArea", ret);
	if(game != SYSCONF_GetRegion()) ret = SYSCONF_SetRegion(game);
	if (ret) handleError("SYSCONF_SetRegion", ret);
	if (video != SYSCONF_GetVideo()) ret  = SYSCONF_SetVideo(video);
	if (ret) handleError("SYSCONF_SetVideo", ret);
	if (eula != SYSCONF_GetEULA()) ret  = SYSCONF_SetEULA(eula);
	if (ret) handleError("SYSCONF_SetEULA", ret);
		
	if (country != sadr[0]){
		memset(sadr, 0, SADR_LENGTH);
		sadr[0] = country;
		ret = SYSCONF_Set("IPL.SADR", sadr, SADR_LENGTH);
		if (ret) handleError("SYSCONF_Set IPL.SADR", ret);
	}
	//wait_anyKey();
	printf("Saving...");
	ret = SYSCONF_SaveChanges();
	if (ret < 0) handleError("SYSCONF_SaveChanges", ret);
	else printf("OK!\n");
	printf("Press any key to continue .....");
	wait_anyKey();
}

void updateSelected(int delta){
	if (selected + delta >= ITEMS || selected + delta < 0) return;
	
	if (delta != 0){
		// Remove the cursor from the last selected item
		page_contents[selected][1] = ' ';
		page_contents[selected][45] = ' ';
		page_contents[selected][57] = ' ';
		// Set new cursor location
		selected += delta;
	}
	
	// Add the cursor to the now-selected item
	page_contents[selected][1] = '>';
	page_contents[selected][45] = '<';
	page_contents[selected][57] = '>';
}

void updatePage(void){
	
	sprintf(page_contents[0], "    %-40s   %10s  \n", "Language Setting:", languages[lang]);
	sprintf(page_contents[1], "    %-40s   %10s   \n", "Console Area Setting:", areas[area]);
	sprintf(page_contents[2], "    %-40s   %10s   \n", "Game Region Setting:", regions[game]);
	sprintf(page_contents[3], "    %-40s   %10s  \n", "Console Video Mode:", vmodes[video]);
	sprintf(page_contents[4], "    %-40s  %10d  \n", "Shop Country Code:", country);
	sprintf(page_contents[5], "    %-40s   %10s  \n", "Services EULA:", (eula == SYSCONF_ENOENT)?"Disabled":eulas[eula]);
	sprintf(page_contents[6], "    %-40s   %10s  \n", "Revert Settings", "Revert  ");
	sprintf(page_contents[7], "    %-40s   %10s  \n", "Save Settings", "Save    ");
	sprintf(page_contents[8], "    %-40s   %10s  \n", "Auto Fix Settings ", "Fix   ");
	sprintf(page_contents[9], "    %-40s   %10s  \n", "Exit to the Homebrew Channel", "Exit    ");
	sprintf(page_contents[10], "    %-40s   %10s  \n", "Reboot to System Menu", "Reboot  ");

	updateSelected(0);
}
char AREAtoSysMenuRegion(int area){
	// Data based on my own tests with AREA/Sysmenu
	switch (area){
		case 0:
		case 5:
		case 6:
			return 'J';
		case 1:
		case 4:
		case 7:
		case 8:
		case 9:
		case 10:
		case 11:
			return 'U';
		case 2:
		case 3:
			return 'E';
		default:
			return 0;
	}
}

//---------------------------------------------------------------------------------
int main(int argc, char **argv) {
//---------------------------------------------------------------------------------
	int ret = 0, i;
	u16 sysmenu_version;
	char sysmenu_region;
	bool needbreak = false;
	u32 buttons;
	int Current_Ios = 0;
	
	ret = IOS_ReloadIOS(236);
	if(ret != 0) {
		ret = IOS_ReloadIOS(249);
		if(ret != 0) {
			IosPatch_FULL(true, true, false, false, 58);
		}
	}
	basicInit();
	
	miscInit();
	if(InitGecko())
		USBGeckoOutput();
	Current_Ios = IOS_GetVersion();
	gprintf("\n\ncurrent_ios [%i] \n\n", Current_Ios);
	
	sysmenu_version = get_installed_title_version(TITLE_ID(1,2));
	sysmenu_region = get_sysmenu_region();
	gprintf("\n\nsysmenu_version[%u] sysmenu_region[%c] \n\n", sysmenu_version, sysmenu_region);
	gprintf("Init SYSCONF...");
	ret = SYSCONF_Init();
	if (ret < 0) handleError("SYSCONF_Init", ret);
	else gprintf("OK!\n");
	
	getSettings();
	region = game;
	Draw_Disclaimer();
	updatePage();
	
	while(1) {
	
		PrintBanner();
		printf("\n------------------------------------------------------------------------");
		printf("Edit Region Settings");
		if (sysmenu_region != 0 && sysmenu_region != AREAtoSysMenuRegion(area))
			printf("    " WARNING_SIGN " \x1b[41;1mWARNING: AREA/SysMenu MISMATCH!\x1b[40m " WARNING_SIGN);
		printf("\n------------------------------------------------------------------------");
		for (i = 0; i < 4; i++)
			printf(page_contents[i]);
		
		printf("\t    Country Codes: \t1[JPN] 49[USA] 110[UK] 136[KOR]\n");
		
		for (i = i; i < 8; i++)
			printf(page_contents[i]);
		printf("------------------------------------------------------------------------");
		printf("Auto Fix - SysMenu Region: %c (v%u)\n", sysmenu_region, sysmenu_version);
		printf("------------------------------------------------------------------------");
		for (i = i; i < 9; i++)
			printf(page_contents[i]);
		printf("------------------------------------------------------------------------");
		printf("Exiting Options\n");
		printf("------------------------------------------------------------------------");
		for (i = i; i < ITEMS; i++)
			printf(page_contents[i]);
		printf("------------------------------------------------------------------------");
		Console_SetPosition(26, 0);
		printf("Change Selection: [%s%s %s%s]\t\t\t\t\t\tSelect: [A]\tCredits: [1]", LEFT_ARROW, RIGHT_ARROW, UP_ARROW, DOWN_ARROW);
		buttons = wait_anyKey();
		gprintf("\n\nselected [%i] \n\n", selected);
		if (buttons & WPAD_BUTTON_DOWN)
			updateSelected(1);
		
		if (buttons & WPAD_BUTTON_UP)
			updateSelected(-1);
		
		if (buttons & WPAD_BUTTON_LEFT){
			switch(selected){
				case 0:
					if (--lang < 0) lang = 6;
				break;
				case 1:
					if (--area < 0) area = 11;
				break;
				case 2:
					if (--game < 0) game = 3;
				break;
				case 3:
					if (--video < 0) video = 2;
				break;
				case 4:
					if (country == 1) country = 136;
					else if (country == 49) country = 1;
					else if (country == 110) country = 49;
					else if (country == 136) country = 110;
					if (eula >= 0) eula = 0;
				break;
				case 5:
					if (eula >= 0) eula = !eula;
				break;
			}
		}
		
		if (buttons & WPAD_BUTTON_RIGHT){
			switch(selected){
				case 0:
					if (++lang == 7) lang = 0;
				break;
				case 1:
					if (++area == 12) area = 0;
				break;
				case 2:
					if (++game == 4) game = 0;
				break;
				case 3:
					if (++video == 3) video = 0;
				break;
				case 4:
					if (country == 1) country = 49;
					else if (country == 49) country = 110;
					else if (country == 110) country = 136;
					else if (country == 136) country = 1;
					if (eula >= 0) eula = 0;
				break;
				case 5:
					if (eula >= 0)	eula = !eula;
				break;
			}
		}
		
		if (buttons & WPAD_BUTTON_A){
			switch(selected){
				case 6:
					getSettings();
				break;
				case 7:
					saveSettings();
				break;
				case 8:
					if(sysmenu_region == 85) { // usa
						lang = 1;
						area = 1;
						game = 1;
						video = 0;
						country = 49;
					}
					else if(sysmenu_region == 74) { // jpn 
						lang = 0;
						area = 0;
						game = 0;
						video = 0;
						country = 1;
					}
					else if(sysmenu_region == 69) { // EUR/PAL
						lang = 1;
						area = 2;
						game = 2;
						video = 1;
						country = 110;
					}
					else if(sysmenu_region == 255) { // KOR
						lang = 1;
						area = 7;
						game = 3;
						video = 0;
						country = 136;
					}
					saveSettings();
				break;
				case 9:
					needbreak = true;
				break;
				case 10:
					STM_RebootSystem();
				break;
			}
		}
		
		if (buttons & WPAD_BUTTON_B) {
			selected = 9;
		}
		if (buttons & WPAD_BUTTON_1) {
			draw_credits();
			wait_anyKey();
		}
		updatePage();
		if(needbreak)
			break;
	}
	
	Console_SetPosition(26,30);
	Console_SetColors(BLACK, 0, GREEN, 0);
	printf("Exiting");
	miscDeInit();
	
	exit(0);
}