#include "menu.hpp"

#include <string.h>
#include <gccore.h>

u8 m_categories[21];
u32 C_curPage;
bool gameSet;
u8 lastBtn;

void CMenu::_hideCategorySettings(bool instant)
{
	m_btnMgr.hide(m_categoryLblTitle, instant);
	m_btnMgr.hide(m_categoryBtnBack, instant);
	m_btnMgr.hide(m_categoryLblPage, instant);
	m_btnMgr.hide(m_categoryBtnPageM, instant);
	m_btnMgr.hide(m_categoryBtnPageP, instant);
	for(u32 i = 0; i < ARRAY_SIZE(m_categoryLblUser); ++i)
	{
		if(m_categoryLblUser[i] != -1u)
			m_btnMgr.hide(m_categoryLblUser[i], instant);
	}

	for(int i = 0; i < 21; ++i)
	{
		m_btnMgr.hide(m_categoryLblCat[i]);
		m_btnMgr.hide(m_categoryBtn[i]);
	}
}

void CMenu::_showCategorySettings(void)
{
	_setBg(m_categoryBg, m_categoryBg);
	for(u32 i = 0; i < ARRAY_SIZE(m_categoryLblUser); ++i)
	{
		if(m_categoryLblUser[i] != -1u)
			m_btnMgr.show(m_categoryLblUser[i]);
	}
	m_btnMgr.show(m_categoryLblTitle);
	m_btnMgr.show(m_categoryBtnBack);
	_updateCheckboxes();
}

void CMenu::_updateCheckboxes(void)
{
	if(m_max_categories > 10)
	{
		m_btnMgr.setText(m_categoryLblPage, wfmt(L"%i / 2", C_curPage));
		m_btnMgr.show(m_categoryLblPage);
		m_btnMgr.show(m_categoryBtnPageM);
		m_btnMgr.show(m_categoryBtnPageP);
	}
	for(int i = 0; i < 21; ++i)
	{
		m_btnMgr.hide(m_categoryBtn[i]);
		m_btnMgr.hide(m_categoryLblCat[i]);
	}
	string id;
	if(m_current_view != COVERFLOW_EMU)
		id = m_cf.getId();
	else
	{
		dir_discHdr *hdr = m_cf.getHdr();
		string tempname(hdr->path);
		tempname.erase(0, tempname.find_first_of('/')+1);
		string dirName = tempname.substr(0, tempname.find_first_of('/')+1);
		tempname.assign(&tempname[tempname.find_last_of('/') + 1]);
		if(tempname.find_last_of('.') != string::npos)
			tempname.erase(tempname.find_last_of('.'), tempname.size() - tempname.find_last_of('.'));
		id = dirName+tempname;
	}
	const char *catflags;
	if(gameSet) 
		catflags = m_cat.getString("CATEGORIES", id, "").c_str();
	else
		catflags = m_cat.getString(_domainFromView(), "categories", "100000000000000000000").c_str();
	memset(&m_categories, '0', sizeof(m_categories));
	if(strlen(catflags) == sizeof(m_categories))
		memcpy(&m_categories, catflags, sizeof(m_categories));

	if(C_curPage == 1)
	{
		int j = 11;
		if(m_max_categories < 11)
			j = m_max_categories;
		for(int i = 0; i < j; ++i)
		{
			if(i == 0 && gameSet)
				continue;
			m_btnMgr.show(m_categoryLblCat[i]);
			if(catflags[i] == '1')
				m_categoryBtn[i] = m_categoryBtnCats[i];
			else
				m_categoryBtn[i] = m_categoryBtnCat[i];
			m_btnMgr.show(m_categoryBtn[i]);
		}
	}
	else
	{
		for(int i = 11; i < m_max_categories; ++i)
		{
			m_btnMgr.show(m_categoryLblCat[i]);
			if(catflags[i] == '1')
				m_categoryBtn[i] = m_categoryBtnCats[i];
			else
				m_categoryBtn[i] = m_categoryBtnCat[i];
			m_btnMgr.show(m_categoryBtn[i]);
		}
	}
}

void CMenu::_CategorySettings(bool fromGameSet)
{
	SetupInput();
	C_curPage = 1;
	gameSet = fromGameSet;
	_showCategorySettings();
	while(true)
	{
		_mainLoopCommon();
		if(!m_btnMgr.selected(lastBtn))
			m_btnMgr.noHover(false);
		if(BTN_HOME_PRESSED || BTN_B_PRESSED)
		{
			m_cat.save();
			break;
		}
		else if(BTN_UP_PRESSED)
			m_btnMgr.up();
		else if(BTN_DOWN_PRESSED)
			m_btnMgr.down();
		if(((BTN_MINUS_PRESSED || BTN_LEFT_PRESSED) && m_max_categories>11) || (BTN_A_PRESSED && m_btnMgr.selected(m_categoryBtnPageM)))
		{
			lastBtn = m_categoryBtnPageM;
			m_btnMgr.noHover(true);
			C_curPage = C_curPage == 1 ? 2 : 1;
			if(BTN_LEFT_PRESSED || BTN_MINUS_PRESSED)
				m_btnMgr.click(m_categoryBtnPageM);
			_updateCheckboxes();
		}
		else if(((BTN_PLUS_PRESSED || BTN_RIGHT_PRESSED) && m_max_categories>11) || (BTN_A_PRESSED && m_btnMgr.selected(m_categoryBtnPageP)))
		{
			lastBtn = m_categoryBtnPageP;
			m_btnMgr.noHover(true);
			C_curPage = C_curPage == 1 ? 2 : 1;
			if(BTN_RIGHT_PRESSED || BTN_PLUS_PRESSED)
				m_btnMgr.click(m_categoryBtnPageP);
			_updateCheckboxes();
		}
		if(BTN_A_PRESSED)
		{
			if(m_btnMgr.selected(m_categoryBtnBack))
			{
				m_cat.save();
				break;
			}
			for(int i = 0; i < 21; ++i)
			{
				if(m_btnMgr.selected(m_categoryBtn[i]))
				{
					lastBtn = m_categoryBtn[i];
					m_btnMgr.noHover(true);
					m_categories[i] = m_categories[i] == '1' ? '0' : '1';
					if(i == 0 && m_categories[i] == '1')
					{
						for(int j = 1; j < 21; ++j)
							m_categories[j] = '0';
					}
					else
						m_categories[0] = '0';
					char catflags[22];
					memset(&catflags, 0, sizeof(catflags));
					memcpy(&catflags, &m_categories, sizeof(m_categories));
					string id;
					if(m_current_view != COVERFLOW_EMU)
						id = m_cf.getId();
					else
					{
						dir_discHdr *hdr = m_cf.getHdr();
						string tempname(hdr->path);
						tempname.erase(0, tempname.find_first_of('/')+1);
						string dirName = tempname.substr(0, tempname.find_first_of('/')+1);
						tempname.assign(&tempname[tempname.find_last_of('/') + 1]);
						if(tempname.find_last_of('.') != string::npos)
							tempname.erase(tempname.find_last_of('.'), tempname.size() - tempname.find_last_of('.'));
						id = dirName+tempname;
					}
					if(string (catflags) == "000000000000000000000")
						catflags[0] = '1';
					if(gameSet)
						m_cat.setString("CATEGORIES", id, catflags);
					else
						m_cat.setString(_domainFromView(), "categories", catflags);
					_updateCheckboxes();
					break;
				}
			}
		}
	}
	_hideCategorySettings();
}

void CMenu::_initCategorySettingsMenu(CMenu::SThemeData &theme)
{
	_addUserLabels(theme, m_categoryLblUser, ARRAY_SIZE(m_categoryLblUser), "CATEGORY");
	m_categoryBg = _texture(theme.texSet, "CATEGORY/BG", "texture", theme.bg);
	m_categoryLblTitle = _addTitle(theme, "CATEGORY/TITLE", theme.titleFont, L"", 20, 30, 600, 60, theme.titleFontColor, FTGX_JUSTIFY_CENTER | FTGX_ALIGN_MIDDLE);
	m_categoryBtnBack = _addButton(theme, "CATEGORY/BACK_BTN", theme.btnFont, L"", 420, 400, 200, 56, theme.btnFontColor);
	m_categoryLblPage = _addLabel(theme, "CATEGORY/PAGE_BTN", theme.btnFont, L"", 256, 400, 100, 56, theme.btnFontColor, FTGX_JUSTIFY_CENTER | FTGX_ALIGN_MIDDLE, theme.btnTexC);
	m_categoryBtnPageM = _addPicButton(theme, "CATEGORY/PAGE_MINUS", theme.btnTexMinus, theme.btnTexMinusS, 200, 400, 56, 56);
	m_categoryBtnPageP = _addPicButton(theme, "CATEGORY/PAGE_PLUS", theme.btnTexPlus, theme.btnTexPlusS, 356, 400, 56, 56);
	m_categoryLblCat[0] = _addLabel(theme, "CATEGORY/CAT_0", theme.lblFont, L"", 85, 390, 100, 48, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE);
	m_categoryBtnCat[0] = _addPicButton(theme, "CATEGORY/CAT_0_BTN", theme.checkboxoff, theme.checkboxoffs, 30, 390, 44, 48);
	m_categoryBtnCats[0] = _addPicButton(theme, "CATEGORY/CAT_0_BTNS", theme.checkboxon, theme.checkboxons, 30, 390, 44, 48);
	for(int i = 1; i < 6; ++i)
	{ 	// Page 1
		m_categoryLblCat[i] = _addLabel(theme, fmt("CATEGORY/CAT_%i", i), theme.lblFont, L"", 85, (42+i*58), 230, 48, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE);
		m_categoryBtnCat[i] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTN", i), theme.checkboxoff, theme.checkboxoffs, 30, (42+i*58), 44, 48);
		m_categoryBtnCats[i] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTNS", i), theme.checkboxon, theme.checkboxons, 30, (42+i*58), 44, 48);
		// right half
		m_categoryLblCat[i+5] = _addLabel(theme, fmt("CATEGORY/CAT_%i", i+5), theme.txtFont, L"", 380, (42+i*58), 230, 48, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE);
		m_categoryBtnCat[i+5] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTN", i+5), theme.checkboxoff, theme.checkboxoffs, 325, (42+i*58), 44, 48);
		m_categoryBtnCats[i+5] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTNS", i+5), theme.checkboxon, theme.checkboxons, 325, (42+i*58), 44, 48);
		// Page 2
		m_categoryLblCat[i+10] = _addLabel(theme, fmt("CATEGORY/CAT_%i", i+10), theme.lblFont, L"", 85, (42+i*58), 230, 48, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE);
		m_categoryBtnCat[i+10] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTN", i+10), theme.checkboxoff, theme.checkboxoffs, 30, (42+i*58), 44, 48);
		m_categoryBtnCats[i+10] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTNS", i+10), theme.checkboxon, theme.checkboxons, 30, (42+i*58), 44, 48);
		// right half
		m_categoryLblCat[i+15] = _addLabel(theme, fmt("CATEGORY/CAT_%i", i+15), theme.txtFont, L"", 380, (42+i*58), 230, 48, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE);
		m_categoryBtnCat[i+15] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTN", i+15), theme.checkboxoff, theme.checkboxoffs, 325, (42+i*58), 44, 48);
		m_categoryBtnCats[i+15] = _addPicButton(theme, fmt("CATEGORY/CAT_%i_BTNS", i+15), theme.checkboxon, theme.checkboxons, 325, (42+i*58), 44, 48);
	}
	_setHideAnim(m_categoryLblTitle, "CATEGORY/TITLE", 0, -200, 0.f, 1.f);
	_setHideAnim(m_categoryLblPage, "CATEGORY/PAGE_BTN", 0, 200, 1.f, 0.f);
	_setHideAnim(m_categoryBtnPageM, "CATEGORY/PAGE_MINUS", 0, 200, 1.f, 0.f);
	_setHideAnim(m_categoryBtnPageP, "CATEGORY/PAGE_PLUS", 0, 200, 1.f, 0.f);
	_setHideAnim(m_categoryBtnBack, "CATEGORY/BACK_BTN", 0, 200, 1.f, 0.f);
	for(int i = 0; i < 21; ++i)
	{
		_setHideAnim(m_categoryBtnCat[i], fmt("CATEGORY/CAT_%i_BTN", i), 0, 0, 1.f, 0.f);
		_setHideAnim(m_categoryBtnCats[i], fmt("CATEGORY/CAT_%i_BTNS", i), 0, 0, 1.f, 0.f);
		_setHideAnim(m_categoryLblCat[i], fmt("CATEGORY/CAT_%i", i), 0, 0, 1.f, 0.f);
		m_categoryBtn[i] = m_categoryBtnCat[i];
	}
	_hideCategorySettings(true);
	_textCategorySettings();
}

void CMenu::_textCategorySettings(void)
{
	m_btnMgr.setText(m_categoryLblTitle, _t("cfgg17", L"Select Categories"));
	m_btnMgr.setText(m_categoryBtnBack, _t("cd1", L"Back"));
	for(int i = 0; i < 21; ++i)
	{
		if(i == 0)
			m_btnMgr.setText(m_categoryLblCat[i], _t("dl3", L"All"));
		else
			m_btnMgr.setText(m_categoryLblCat[i], m_cat.getWString("GENERAL", fmt("cat%d",i), wfmt(L"Category %i",i).c_str()));
	}
}