#include <tonc.h>
#include <string>
#include <cstring>
// #include <maxmod.h> //Music

#include "flash_mem.h"
#include "interrupt.h"
#include "gb_link.h"
#include "gameboy_colour.h"
#include "LinkGPIO.h"
#include "pokemon.h"
#include "random.h"
#include "text_engine.h"
#include "background_engine.h"
#include "pokemon_party.h"
#include "script_array.h"
#include "sprite_data.h"
#include "button_handler.h"
#include "main_menu.h"
#include "debug_mode.h"
#include "soundbank.h"
#include "soundbank_bin.h"
#include "dex_handler.h"
#include "pokedex.h"
#include "global_frame_counter.h"
#include "pkmn_font.h"
#include "save_data_manager.h"
#include "mystery_gift_injector.h"
#include "mystery_gift_builder.h"
#include "rom_data.h"
#include "save.h"

/*

TODO:
- Implement patch list (as described here: https://web.archive.org/web/20180508011842/https://vaguilar.js.org/posts/1/)
	But this isn't going to be relevant after the ACE is implemented, since we will be pulling directly from the box... right?
	is 0xFE being a terminator part of the SPI function or is it implemented specifically in the trade/battle code?

Post Beta TODO:
- Better custom sprites (Progress, Fennel, Title)
- Add a % or x/250 for the Dream Dex
- Determine if transfered Shiny Pokemon are square/star sparkles
- Music and sound effects
- Simplify the sprite initalization
- Smoother transitions
- Minigame
- Wii Channel
- Ditto gift (ignore Mew/Celebi?)
- MissingNo/Enigma Berry
- Text translations
- Add support for other languages
- Champion ribbons (u/NinjaKnight92)
- Doxygen generation
--------
*/

Pokemon_Party party = Pokemon_Party();
int delay_counter = 0;
bool skip = true;
rom_data curr_rom;

/*
int test_main(void) Music
{

	irq_init(NULL);
	// Initialize maxmod with default settings
	// pass soundbank address, and allocate 8 channels.

	irq_set(II_VBLANK, mmVBlank, 0);
	irq_enable(II_VBLANK);

	mmInitDefault((mm_addr)soundbank_bin, 8);

	mmStart(MOD_FLATOUTLIES, MM_PLAY_LOOP);
	// Song is playing now (well... almost)
	while (1)
	{
		// ..process game logic..

		// Update Maxmod
		mmFrame();

		// Wait for new frame (SWI 5)
		VBlankIntrWait();

		// ..update graphical data..
	}
}
*/

void initalization_script(void)
{
	// Initalizations
	linkGPIO->reset();

	REG_DISPCNT = DCNT_MODE0 | DCNT_BG0 | DCNT_BG1 | DCNT_BG2 | DCNT_BG3 | DCNT_OBJ | DCNT_OBJ_1D;

	irq_init(NULL);
	irq_enable(II_VBLANK);

	// Disable for save data read/write
	REG_IME = 0;
	REG_IE = 0;

	rand_set_seed(0x1216);
	add_script_party_var(party);

	// Sound bank init
	irq_init(NULL);
	// irq_set(II_VBLANK, mmVBlank, 0); //Music
	irq_enable(II_VBLANK);
	// mmInitDefault((mm_addr)soundbank_bin, 8); //Music
	// mmStart(MOD_FLATOUTLIES, MM_PLAY_LOOP); //Music

	// Graphics init
	oam_init(obj_buffer, 128);

	// Load opening background first so it hides everything else
	load_opening_background();
	load_background();
	load_textbox_background();
	load_testroid();
	load_professor();
	load_btn_t_l();
	load_btn_t_r();
	load_btn_p_l();
	load_btn_p_r();
	load_btn_c_l();
	load_btn_c_r();
	load_dex_l();
	load_dex_m();
	load_dex_r();
	load_btn_d_l();
	load_btn_d_r();
	load_btn_lang_eng();
	load_btn_lang_fre();
	load_btn_lang_ita();
	load_btn_lang_ger();
	load_btn_lang_spa();
	load_btn_lang_kor();
	load_lang_arrow();

	main_menu_btn_init(Button(btn_t_l, btn_t_r, 128, 160), BTN_TRANSFER);
	main_menu_btn_init(Button(btn_p_l, btn_p_r, 192, 224), BTN_POKEDEX);
	main_menu_btn_init(Button(btn_c_l, btn_c_r, 256, 288), BTN_CREDITS);
	main_menu_btn_init(Button(btn_d_l, btn_d_r, 416, 448), BTN_LANGUAGE);

	main_menu_btn_init(Button(btn_lang_eng, 480), BTN_ENG);
	main_menu_btn_init(Button(btn_lang_fre, 512), BTN_FRE);
	main_menu_btn_init(Button(btn_lang_ita, 544), BTN_ITA);
	main_menu_btn_init(Button(btn_lang_ger, 576), BTN_GER);
	main_menu_btn_init(Button(btn_lang_spa, 608), BTN_SPA);
	main_menu_btn_init(Button(btn_lang_kor, 640), BTN_KOR);
	main_menu_btn_init(Button(lang_arrow, 672), LANG_ARROW);

	// Prepare dialouge
	populate_dialogue();
	populate_script();
	init_text_engine();

	pokedex_init(); // Why does this cause the music to stop playing? Also the loop doesn't work

	// main_menu_init(transfer_btn, pokedex_btn, credits_btn, language_btn);

	text_disable();
};

void game_load_error(void)
{
	REG_BG0CNT = (REG_BG0CNT & ~BG_PRIO_MASK) | BG_PRIO(3);
	REG_BG1CNT = (REG_BG1CNT & ~BG_PRIO_MASK) | BG_PRIO(2);
	REG_BG2CNT = (REG_BG2CNT & ~BG_PRIO_MASK) | BG_PRIO(1);
	REG_BG2VOFS = 0;
	tte_set_pos(40, 24);
	tte_set_margins(40, 24, 206, 104);
	tte_write("#{cx:0xF000}The Pok@mon save\nfile was not loaded successfully.\n\nPlease remove and\nreinsert the Game\nPak, and then press the A button.");
	// tte_write(std::to_string(get_gamecode() >> 8).c_str());
	key_poll();
	while (!key_hit(KEY_A))
	{
		key_poll();
		VBlankIntrWait();
	}
	tte_erase_screen();
	delay_counter = 0;

	while (delay_counter < 60)
	{
		delay_counter++;
		VBlankIntrWait();
	}
}

void first_load_message(void)
{
	tte_set_pos(8, 0);
	tte_set_ink(10);
	tte_write("#{cx:0xE000}Hello! Thank you for using\nPok@mon Mirror!\n\nJust as a word of caution- \nPok@mon Mirror WILL modify\nyour generation 3 save file.\nThe program is designed to\nnot corrupt anything, but if\nyou do not wish to modify\nyour save file, please turn\noff your Game Boy Advance.\n\nPlease note that Pok@mon\nMirror is still in beta, so\nsave file backups are HIGHLY\nrecommended before using.\nWith that all out of the\nway, please enjoy!\n\n      -The Gears of Progress");
	key_poll();
	while (!key_hit(KEY_A))
	{
		key_poll();
		VBlankIntrWait();
	}
	tte_erase_rect(0, 0, H_MAX, V_MAX);
}

int main(void)
{
	initalization_script();

	// Check if the game has been loaded correctly.
	while (!curr_rom.load_rom())
	{
		game_load_error();
		initalization_script();
	}

	// Initalize memory and save data after loading the game
	init_bank();
	initalize_memory_locations();
	load_custom_save_data();

	if (get_tutorial_flag() == false)
	{
		first_load_message();
		// initalize_save_data();
	}

	// Legal mumbo jumbo
	tte_set_pos(8, 0);
	tte_write("#{cx:0xE000}\n\nPokemon Mirror was created\nout of love and appreciation\nfor the Pokemon franchise\nwith no profit in mind.\nIt will ALWAYS be free.\n\nPlease support the original developers-\nNintendo and GAME FREAK.\n\nAll Pokemon names, sprites, and music are owned by \nNintendo, Creatures Inc, and\nGAME FREAK Inc.");
	tte_write("#{cx:0xF000}"); // Set the color to grey
	while (delay_counter < (15 * 60))
	{
		delay_counter++;
		rand_next_frame();
		key_poll();
		if (key_hit(KEY_A))
		{
			delay_counter = (15 * 60);
		}
		VBlankIntrWait();
	}
	key_poll();

	// Gears of Progress
	tte_erase_rect(0, 0, 240, 160);
	REG_BG1VOFS = 0;
	delay_counter = 0;
	while (delay_counter < (15 * 60))
	{
		delay_counter++;
		rand_next_frame();
		key_poll();
		if (key_hit(KEY_A))
		{
			delay_counter = (15 * 60);
		}
		VBlankIntrWait();
	}
	key_poll();
	REG_BG1CNT = REG_BG1CNT | BG_PRIO(3);

	// Set up blend to fade to white/black

	int curr_lang_btn_num = get_def_lang_num();
	int old_lang_btn_num = -1;
	set_arrow_point(curr_lang_btn_num);

#define CREDITS_ARRAY_SIZE 15
	int curr_credits_num = 0;
	std::string credits_array[CREDITS_ARRAY_SIZE] = {
		"Developed by:\n\n\nThe Gears\nof Progress",
		"Built using:\n\n\n-DevkitPro\n-LibTonc\n-LibGBA",
		"Inspired by the\nworks of:\n\n-Goppier\n-Lorenzooone\n-im a blisy\n-RETIRE",
		"Programs used:\n\n\n-HexManiacAdvance\n-PKHeX\n-WC3Tool\n-Usenti\n",
		"Open Source Code and\nLibraries:\n\n-libtonc-examples\n-PokemonGen3toGenX\n-gba-link-connection\n-awesome-gbadev\n-arduino-poke-gen2",
		"Research resources:\n\n-arm-docs\n-PokemonGen3toGenX\n\nFull links can be\nfound on this\nprogram's GitHub",
		"ROM data obtained\nfrom decompilations created by the PRET team",
		"Pok@mon data\nobtained from:\n\n-Bulbapedia\n-Serebii\n-PokeAPI.com",
		"Discord community\nassistance:\n\n-Hex Maniac Advance\n Development\n-gbadev\n-pret",
		"Sprite work:\n\n\n-lite_thespark",
		"Writing assistance:\n\n\n-Mad",
		"An immense thanks to\nLorenzooone for\ntheir assistance in\nreading/writing save\ndata. Without them,\nthis project would\nnot have been\npossible.",
		"Special thanks:\n\n\nEnn, Roku, Luca,\nArctic, and everyone\nwho has listened to me talk about this\nfor months",
		"All Pok@mon names,\nsprites, and names\nof related resources\nare copyright\nNintendo,\nCreatures Inc.,\nand GAME FREAK Inc.",
		"This project is not endorsed or\nsupported by\nGameFreak/Nintendo.\n\nPlease support the\noriginal developers.",
	};

	curr_rom.load_rom();

	// MAIN LOOP
	while (1)
	{
		if (DEBUG_MODE)
		{
			print_mem_section();
			curr_rom.print_rom_info();
		}
		switch (main_menu_loop())
		{
		case (BTN_TRANSFER):
			text_enable();
			break;
		case (BTN_POKEDEX):
			pokedex_show();
			pokedex_loop();
			if (key_hit(KEY_B))
			{
				pokedex_hide();
				main_menu_exit();
			}
			break;
		case (BTN_LANGUAGE):
			REG_BG2CNT = (REG_BG2CNT & ~BG_PRIO_MASK) | BG_PRIO(1); // Enable text box
			show_lang_btns();
			tte_set_pos(LEFT, TOP);
			tte_write("Choose the default language\nfor transfering from a non\nJapanese game. Setting will\nbe saved after next transfer.");
			if (key_hit(KEY_LEFT) && ((curr_lang_btn_num % 3) != 0))
			{
				curr_lang_btn_num--;
			}
			if (key_hit(KEY_RIGHT) && ((curr_lang_btn_num % 3) != 2))
			{
				curr_lang_btn_num++;
			}
			if (key_hit(KEY_UP) && ((curr_lang_btn_num > 2)))
			{
				curr_lang_btn_num -= 3;
			}
			if (key_hit(KEY_DOWN) && ((curr_lang_btn_num <= 2)))
			{
				curr_lang_btn_num += 3;
			}
			if (curr_lang_btn_num != old_lang_btn_num)
			{
				highlight_lang_btn(old_lang_btn_num, false);
				highlight_lang_btn(curr_lang_btn_num, true);
				old_lang_btn_num = curr_lang_btn_num;
			}
			if (key_hit(KEY_A))
			{
				set_arrow_point(curr_lang_btn_num);
				set_def_lang(curr_lang_btn_num);
			}
			if (key_hit(KEY_B))
			{
				hide_lang_btns();
				main_menu_exit();
				tte_erase_screen();
				hide_text_box();
			}
			break;
		case (BTN_CREDITS):
			tte_set_pos(40, 24);
			tte_set_margins(40, 24, 206, 104);
			tte_erase_screen();
			tte_write(credits_array[curr_credits_num].c_str());
			REG_BG0CNT = (REG_BG0CNT & ~BG_PRIO_MASK) | BG_PRIO(3);
			REG_BG2CNT = (REG_BG2CNT & ~BG_PRIO_MASK) | BG_PRIO(2);
			REG_BG2VOFS = 0;
			if (key_hit(KEY_B))
			{
				tte_erase_rect(0, 0, H_MAX, V_MAX);
				REG_BG0CNT = (REG_BG0CNT & ~BG_PRIO_MASK) | BG_PRIO(2);
				REG_BG2CNT = (REG_BG2CNT & ~BG_PRIO_MASK) | BG_PRIO(3);
				REG_BG2VOFS = 96;
				main_menu_exit();
			}
			if (key_hit(KEY_LEFT) && curr_credits_num > 0)
			{
				curr_credits_num--;
			}
			if (key_hit(KEY_RIGHT) && curr_credits_num < (CREDITS_ARRAY_SIZE - 1))
			{
				curr_credits_num++;
			}
			break;
		}

		key_poll();
		rand_next_frame();
		tte_set_pos(0, 0);
		// tte_write(std::to_string(get_rand_u32()).c_str());
		background_frame();
		text_next_frame();
		oam_copy(oam_mem, obj_buffer, 35);

		VBlankIntrWait();
		// mmFrame(); //Music
		global_next_frame();
	}
}