/*-----------------------------------------------------------------
 Copyright (C) 2005 - 2013
	Michael "Chishm" Chisholm
	Dave "WinterMute" Murphy
	Claudio "sverx"

 This program is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public License
 as published by the Free Software Foundation; either version 2
 of the License, or (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

------------------------------------------------------------------*/
#include <nds.h>
#include <stdio.h>
#include <fat.h>
#include <sys/stat.h>
#include <limits.h>

#include <string.h>
#include <unistd.h>

#include "args.h"
#include "file_browse.h"
#include "hbmenu_banner.h"
#include "font.h"
#include "consolebg.h"
#include "iconTitle.h"
#include "nrio_detect.h"
#include "nds_loader_arm9.h"

using namespace std;

void InitGUI (void) {
	iconTitleInit();
	videoSetModeSub(MODE_4_2D);
	vramSetBankC(VRAM_C_SUB_BG);
	int bgSub = bgInitSub(3, BgType_Bmp8, BgSize_B8_256x256, 1, 0);
	PrintConsole *console = consoleInit(0, 0, BgType_Text4bpp, BgSize_T_256x256, 4, 6, false, false);
	dmaCopy(consolebgBitmap, bgGetGfxPtr(bgSub), 256*256);
	ConsoleFont font;
	font.gfx = (u16*)fontTiles;
	font.pal = (u16*)fontPal;
	font.numChars = 95;
	font.numColors = (fontPalLen / 2);
	font.bpp = 4;
	font.asciiOffset = 32;
	font.convertSingleColor = true;
	consoleSetFont(console, &font);
	dmaCopy(consolebgPal, BG_PALETTE_SUB, 256*2);
	BG_PALETTE_SUB[255] = RGB15(31,31,31);
	keysSetRepeat(25,5);
}

int FileBrowser() {
	InitGUI();
	vector<string> extensionList = argsGetExtensionList();
	while(1) {
		string filename = browseForFile(extensionList);
		// Construct a command line
		vector<string> argarray;
		if (!argsFillArray(filename, argarray)) {
			iprintf("\n\nInvalid NDS or arg file selected\n");
		} else {
			iprintf("\n\nRunning %s with %d\nparameters\n", argarray[0].c_str(), argarray.size());
			// Make a copy of argarray using C strings, for the sake of runNdsFile
			vector<const char*> c_args;
			for (const auto& arg: argarray) { c_args.push_back(arg.c_str()); }
			// Try to run the NDS file with the given arguments
			int err = runNdsFile(c_args[0], c_args.size(), &c_args[0]);
			iprintf("\n\nStart failed. Error %i\n", err);
		}
		argarray.clear();
		while (1) {
			swiWaitForVBlank();
			scanKeys();
			if (!(keysHeld() & KEY_A)) break;
		}
	}
}


int RecoveryPrompt(void) {
	InitGUI();
	printf("<--------RECOVERY MODE--------->\n");
	printf("\n\nPress [A] to use built in uDisk.\n\n");
	printf("Hold [Start] + [Select] while\nbooting uDisk to enter format\nmenu.\n");
	while(1) {
		swiWaitForVBlank();
		scanKeys();
		if (keysDown() == 0)break;
	}
	while(1) {
		swiWaitForVBlank();
		scanKeys();
		if (keysDown() != 0)break;
	}
	return runSRLbinary(0); 
}

int main(void) {
	// Any error results in starting integrated uDisk 1.45 SRL as fall back
	scanKeys();
	u32 KeysHeld = keysHeld();
	if ((KeysHeld & KEY_L) && (KeysHeld & KEY_R) && (KeysHeld & KEY_A) && (KeysHeld & KEY_B) && (KeysHeld & KEY_UP)) {
		return runSRLbinary(1); // Recovery option if need to boot stage2. (only accessible if XuluMenu boots before stage2. Aka, N-Cards setup as bootleg games)
	} else if ((KeysHeld & KEY_L) && (KeysHeld & KEY_R)) {
		return RecoveryPrompt(); // Recovery option if FAT init hangs
	}
	nrio_usb_type_t usb = nrio_usb_detect();
	if (!(KeysHeld & KEY_SELECT) && !(KeysHeld & KEY_B) && (usb.board_type != 0)) {
		return runSRLbinary(2); // New open source version of uDisk with better performance/Linux compatiblity. Courtasy of asiekierka
	}	
	if (!fatInitDefault())return RecoveryPrompt();
	if (!(KeysHeld & KEY_B)) {
		if ((access("/boot.nds", F_OK) == 0) || (access("/udisk.nds", F_OK) == 0)) {
			if((access("/boot.nds", F_OK) == 0))return runNdsFile("/boot.nds", 0, NULL);
			if((access("/udisk.nds", F_OK) == 0))return runNdsFile("/udisk.nds", 0, NULL);
		} else {
			return runSRLbinary(0);
		}
	}
	FileBrowser();
	return RecoveryPrompt();
}