/****************************************************************************
 * libwiigui
 *
 * Tantric 2009
 *
 * gui_filebrowser.cpp
 *
 * GUI class definitions
 ***************************************************************************/

#include "gui.h"
#include "prompts/filebrowser.h"
#include "settings/cfg.h"

/**
 * Constructor for the GuiFileBrowser class.
 */
GuiFileBrowser::GuiFileBrowser(int w, int h)
{
	width = w;
	height = h;
	selectedItem = 0;
	selectable = true;
	listChanged = true; // trigger an initial list update
	triggerdisabled = false; // trigger disable
	focus = 0; // allow focus

	trigA = new GuiTrigger;
	trigA->SetSimpleTrigger(-1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A);

	trigHeldA = new GuiTrigger;
	trigHeldA->SetHeldTrigger(-1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A);

	btnSoundOver = new GuiSound(button_over_pcm, button_over_pcm_size, Settings.sfxvolume);
	btnSoundClick = new GuiSound(button_click_pcm, button_click_pcm_size, Settings.sfxvolume);

	char imgPath[100];
	snprintf(imgPath, sizeof(imgPath), "%sbg_browser.png", CFG.theme_path);
	bgFileSelection = new GuiImageData(imgPath, bg_browser_png);
	bgFileSelectionImg = new GuiImage(bgFileSelection);
	bgFileSelectionImg->SetParent(this);
	bgFileSelectionImg->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);

	snprintf(imgPath, sizeof(imgPath), "%sbg_browser_selection.png", CFG.theme_path);
	bgFileSelectionEntry = new GuiImageData(imgPath, bg_browser_selection_png);
//	fileArchives = new GuiImageData(icon_archives_png);
//	fileDefault = new GuiImageData(icon_default_png);
	fileFolder = new GuiImageData(icon_folder_png);
//	fileGFX = new GuiImageData(icon_gfx_png);
//	filePLS = new GuiImageData(icon_pls_png);
//	fileSFX = new GuiImageData(icon_sfx_png);
//	fileTXT = new GuiImageData(icon_txt_png);
//	fileXML = new GuiImageData(icon_xml_png);
	
	snprintf(imgPath, sizeof(imgPath), "%sscrollbar.png", CFG.theme_path);
	scrollbar = new GuiImageData(imgPath, scrollbar_png);
	scrollbarImg = new GuiImage(scrollbar);
	scrollbarImg->SetParent(this);
	scrollbarImg->SetAlignment(ALIGN_RIGHT, ALIGN_TOP);
	scrollbarImg->SetPosition(0, 2);
	scrollbarImg->SetSkew(0,0,0,0,0,-30,0,-30);

	snprintf(imgPath, sizeof(imgPath), "%sscrollbar_arrowdown.png", CFG.theme_path);
	arrowDown = new GuiImageData(imgPath, scrollbar_arrowdown_png);
	arrowDownImg = new GuiImage(arrowDown);
	snprintf(imgPath, sizeof(imgPath), "%sscrollbar_arrowup.png", CFG.theme_path);
	arrowUp = new GuiImageData(imgPath, scrollbar_arrowup_png);
	arrowUpImg = new GuiImage(arrowUp);
	snprintf(imgPath, sizeof(imgPath), "%sscrollbar_box.png", CFG.theme_path);
	scrollbarBox = new GuiImageData(imgPath, scrollbar_box_png);
	scrollbarBoxImg = new GuiImage(scrollbarBox);

	arrowUpBtn = new GuiButton(arrowUpImg->GetWidth(), arrowUpImg->GetHeight());
	arrowUpBtn->SetParent(this);
	arrowUpBtn->SetImage(arrowUpImg);
	arrowUpBtn->SetAlignment(ALIGN_RIGHT, ALIGN_TOP);
	arrowUpBtn->SetPosition(12,-12);
	arrowUpBtn->SetSelectable(false);
	arrowUpBtn->SetClickable(false);
	arrowUpBtn->SetHoldable(true);
	arrowUpBtn->SetTrigger(trigHeldA);
	arrowUpBtn->SetSoundOver(btnSoundOver);
	arrowUpBtn->SetSoundClick(btnSoundClick);

	arrowDownBtn = new GuiButton(arrowDownImg->GetWidth(), arrowDownImg->GetHeight());
	arrowDownBtn->SetParent(this);
	arrowDownBtn->SetImage(arrowDownImg);
	arrowDownBtn->SetAlignment(ALIGN_RIGHT, ALIGN_BOTTOM);
	arrowDownBtn->SetPosition(12,12);
	arrowDownBtn->SetSelectable(false);
	arrowDownBtn->SetClickable(false);
	arrowDownBtn->SetHoldable(true);
	arrowDownBtn->SetTrigger(trigHeldA);
	arrowDownBtn->SetSoundOver(btnSoundOver);
	arrowDownBtn->SetSoundClick(btnSoundClick);

	scrollbarBoxBtn = new GuiButton(scrollbarBoxImg->GetWidth(), scrollbarBoxImg->GetHeight());
	scrollbarBoxBtn->SetParent(this);
	scrollbarBoxBtn->SetImage(scrollbarBoxImg);
	scrollbarBoxBtn->SetAlignment(ALIGN_RIGHT, ALIGN_TOP);
	scrollbarBoxBtn->SetPosition(-10, 0);
	scrollbarBoxBtn->SetMinY(-10);
	scrollbarBoxBtn->SetMaxY(156);
	scrollbarBoxBtn->SetSelectable(false);
	scrollbarBoxBtn->SetClickable(false);
	scrollbarBoxBtn->SetHoldable(true);
	scrollbarBoxBtn->SetTrigger(trigHeldA);

	for(int i=0; i<FILEBROWSERSIZE; i++)
	{
		fileListText[i] = new GuiText(NULL,20, (GXColor){0, 0, 0, 0xff});
		fileListText[i]->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);
		fileListText[i]->SetPosition(5,0);
		fileListText[i]->SetMaxWidth(bgFileSelectionImg->GetWidth() - (arrowDownImg->GetWidth()+20), GuiText::DOTTED);

		fileListTextOver[i] = new GuiText(NULL,20, (GXColor){0, 0, 0, 0xff});
		fileListTextOver[i]->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);
		fileListTextOver[i]->SetPosition(5,0);
		fileListTextOver[i]->SetMaxWidth(bgFileSelectionImg->GetWidth() - (arrowDownImg->GetWidth()+20), GuiText::SCROLL);

		fileListBg[i] = new GuiImage(bgFileSelectionEntry);
		//fileListArchives[i] = new GuiImage(fileArchives);
		//fileListDefault[i] = new GuiImage(fileDefault);
		fileListFolder[i] = new GuiImage(fileFolder);
		//fileListGFX[i] = new GuiImage(fileGFX);
		//fileListPLS[i] = new GuiImage(filePLS);
		//fileListSFX[i] = new GuiImage(fileSFX);
		//fileListTXT[i] = new GuiImage(fileTXT);
		//fileListXML[i] = new GuiImage(fileXML);
		fileList[i] = new GuiButton(350,30);
		fileList[i]->SetParent(this);
		fileList[i]->SetLabel(fileListText[i]);
		fileList[i]->SetLabelOver(fileListTextOver[i]);
		fileList[i]->SetImageOver(fileListBg[i]);
		fileList[i]->SetPosition(2,30*i+3);
		fileList[i]->SetTrigger(trigA);
		fileList[i]->SetRumble(false);
		fileList[i]->SetSoundClick(btnSoundClick);
	}
}

/**
 * Destructor for the GuiFileBrowser class.
 */
GuiFileBrowser::~GuiFileBrowser()
{
	delete arrowUpBtn;
	delete arrowDownBtn;
	delete scrollbarBoxBtn;

	delete bgFileSelectionImg;
	delete scrollbarImg;
	delete arrowDownImg;
	delete arrowUpImg;
	delete scrollbarBoxImg;

	delete bgFileSelection;
	delete bgFileSelectionEntry;
	//delete fileArchives;
	//delete fileDefault;
	delete fileFolder;
	//delete fileGFX;
	//delete filePLS;
	//delete fileSFX;
	//delete fileTXT;
	//delete fileXML;
	delete scrollbar;
	delete arrowDown;
	delete arrowUp;
	delete scrollbarBox;

	delete btnSoundOver;
	delete btnSoundClick;
	delete trigHeldA;
	delete trigA;

	for(int i=0; i<FILEBROWSERSIZE; i++)
	{
		delete fileListText[i];
		delete fileListTextOver[i];
		delete fileList[i];
		delete fileListBg[i];
		//delete fileListArchives[i];
		//delete fileListDefault[i];
		delete fileListFolder[i];
		//delete fileListGFX[i];
		//delete fileListPLS[i];
		//delete fileListSFX[i];
		//delete fileListTXT[i];
		//delete fileListXML[i];
	}
}

void GuiFileBrowser::SetFocus(int f)
{
	LOCK(this);
	focus = f;

	for(int i=0; i<FILEBROWSERSIZE; i++)
		fileList[i]->ResetState();

	if(f == 1)
		fileList[selectedItem]->SetState(STATE_SELECTED);
}

void GuiFileBrowser::DisableTriggerUpdate(bool set)
{
	LOCK(this);
	triggerdisabled = set;
}

void GuiFileBrowser::ResetState()
{
	LOCK(this);
	state = STATE_DEFAULT;
	stateChan = -1;
	selectedItem = 0;

	for(int i=0; i<FILEBROWSERSIZE; i++)
	{
		fileList[i]->ResetState();
	}
}

void GuiFileBrowser::TriggerUpdate()
{
	LOCK(this);
	listChanged = true;
}

/**
 * Draw the button on screen
 */
void GuiFileBrowser::Draw()
{
	LOCK(this);
	if(!this->IsVisible())
		return;

	bgFileSelectionImg->Draw();

	for(int i=0; i<FILEBROWSERSIZE; i++)
	{
		fileList[i]->Draw();
	}

	scrollbarImg->Draw();
	arrowUpBtn->Draw();
	arrowDownBtn->Draw();
	scrollbarBoxBtn->Draw();

	this->UpdateEffects();
}

void GuiFileBrowser::Update(GuiTrigger * t)
{
	LOCK(this);
	if(state == STATE_DISABLED || !t || triggerdisabled)
		return;

	int position = 0;
	int positionWiimote = 0;

	arrowUpBtn->Update(t);
	arrowDownBtn->Update(t);
	scrollbarBoxBtn->Update(t);

	// move the file listing to respond to wiimote cursor movement
	if(scrollbarBoxBtn->GetState() == STATE_HELD &&
		scrollbarBoxBtn->GetStateChan() == t->chan &&
		t->wpad.ir.valid &&
		browser->browserList.size() > FILEBROWSERSIZE
		)
	{
		scrollbarBoxBtn->SetPosition(20,-10);
		positionWiimote = t->wpad.ir.y - 60 - scrollbarBoxBtn->GetTop();

		if(positionWiimote < scrollbarBoxBtn->GetMinY())
			positionWiimote = scrollbarBoxBtn->GetMinY();
		else if(positionWiimote > scrollbarBoxBtn->GetMaxY())
			positionWiimote = scrollbarBoxBtn->GetMaxY();

		browser->pageIndex = (positionWiimote * browser->browserList.size())/136.0 - selectedItem;

		if(browser->pageIndex <= 0)
		{
			browser->pageIndex = 0;
		}
		else if(browser->pageIndex+FILEBROWSERSIZE >= (int)browser->browserList.size())
		{
			browser->pageIndex = browser->browserList.size()-FILEBROWSERSIZE;
		}
		listChanged = true;
		focus = false;
		

	}

	if(arrowDownBtn->GetState() == STATE_HELD && arrowDownBtn->GetStateChan() == t->chan)
	{
		t->wpad.btns_h |= WPAD_BUTTON_DOWN;
		if(!this->IsFocused())
			((GuiWindow *)this->GetParent())->ChangeFocus(this);
			
	}
	else if(arrowUpBtn->GetState() == STATE_HELD && arrowUpBtn->GetStateChan() == t->chan)
	{
		t->wpad.btns_h |= WPAD_BUTTON_UP;
		if(!this->IsFocused())
			((GuiWindow *)this->GetParent())->ChangeFocus(this);
			
	}

/*	// pad/joystick navigation
	if(!focus)
	{
		goto endNavigation; // skip navigation
		listChanged = false;
	}
*/
	if(t->Right())
	{
		if(browser->pageIndex < (int)browser->browserList.size() && browser->browserList.size() > FILEBROWSERSIZE)
		{
			browser->pageIndex += FILEBROWSERSIZE;
			if(browser->pageIndex+FILEBROWSERSIZE >= (int)browser->browserList.size())
				browser->pageIndex = browser->browserList.size()-FILEBROWSERSIZE;
			listChanged = true;
		}
	}
	else if(t->Left())
	{
		if(browser->pageIndex > 0)
		{
			browser->pageIndex -= FILEBROWSERSIZE;
			if(browser->pageIndex < 0)
				browser->pageIndex = 0;
			listChanged = true;
		}
	}
	else if(t->Down())
	{
		if(browser->pageIndex + selectedItem + 1 < (int)browser->browserList.size())
		{
			if(selectedItem == FILEBROWSERSIZE-1)
			{
				// move list down by 1
				browser->pageIndex++;
				listChanged = true;
			}
			else if(fileList[selectedItem+1]->IsVisible())
			{
				fileList[selectedItem]->ResetState();
				fileList[++selectedItem]->SetState(STATE_SELECTED, t->chan);
			}
		}
	}
	else if(t->Up())
	{
		if(selectedItem == 0 &&	browser->pageIndex + selectedItem > 0)
		{
			// move list up by 1
			browser->pageIndex--;
			listChanged = true;
		}
		else if(selectedItem > 0)
		{
			fileList[selectedItem]->ResetState();
			fileList[--selectedItem]->SetState(STATE_SELECTED, t->chan);
		}
	}

	//endNavigation:

	for(int i=0; i<FILEBROWSERSIZE; i++)
	{
		if(listChanged)
		{
			bool haveselected = false;
			if(browser->pageIndex+i < (int)browser->browserList.size())
			{
				if(fileList[i]->GetState() == STATE_DISABLED)
					fileList[i]->SetState(STATE_DEFAULT);

				if(fileList[i]->GetState() == STATE_SELECTED)
					haveselected = true;

				fileList[i]->SetVisible(true);

				fileListText[i]->SetText(browser->browserList[browser->pageIndex+i].displayname);
				fileListTextOver[i]->SetText(browser->browserList[browser->pageIndex+i].displayname);

				if(browser->browserList[browser->pageIndex+i].isdir) // directory
				{
					fileList[i]->SetIcon(fileListFolder[i]);
					fileListText[i]->SetPosition(30,0);
					fileListTextOver[i]->SetPosition(30,0);
				}
				else
				{
					/*
				    char *fileext = strrchr(browserList[browser.pageIndex+i].displayname, '.');
					fileListText[i]->SetPosition(32,0);
					fileListTextOver[i]->SetPosition(32,0);
				    if(fileext)
				    {
                        if(!strcasecmp(fileext, ".png") || !strcasecmp(fileext, ".jpg") || !strcasecmp(fileext, ".jpeg") ||
						   !strcasecmp(fileext, ".gif") || !strcasecmp(fileext, ".tga") || !strcasecmp(fileext, ".tpl") ||
						   !strcasecmp(fileext, ".bmp")) {
                            fileList[i]->SetIcon(fileListGFX[i]);
                        } else if(!strcasecmp(fileext, ".mp3") || !strcasecmp(fileext, ".ogg") || !strcasecmp(fileext, ".flac") ||
								  !strcasecmp(fileext, ".mpc") || !strcasecmp(fileext, ".m4a") || !strcasecmp(fileext, ".wav")) {
                            fileList[i]->SetIcon(fileListSFX[i]);
                        } else if(!strcasecmp(fileext, ".pls") || !strcasecmp(fileext, ".m3u")) {
                            fileList[i]->SetIcon(fileListPLS[i]);
                        } else if(!strcasecmp(fileext, ".txt")) {
                            fileList[i]->SetIcon(fileListTXT[i]);
                        } else if(!strcasecmp(fileext, ".xml")) {
                            fileList[i]->SetIcon(fileListXML[i]);
                        } else if(!strcasecmp(fileext, ".rar") || !strcasecmp(fileext, ".zip") ||
								  !strcasecmp(fileext, ".gz") || !strcasecmp(fileext, ".7z")) {
                            fileList[i]->SetIcon(fileListArchives[i]);
                        } else {
                            fileList[i]->SetIcon(fileListDefault[i]);
                        }
				    } else {
                      fileList[i]->SetIcon(fileListDefault[i]);
				    }
					*/				
					fileList[i]->SetIcon(NULL);
					fileListText[i]->SetPosition(10,0);
					fileListTextOver[i]->SetPosition(10,0);
				}
			}
			else
			{
				fileList[i]->SetVisible(false);
				fileList[i]->SetState(STATE_DISABLED);
			}
			if(!haveselected && browser->pageIndex < (int)browser->browserList.size())
				fileList[i]->SetState(STATE_SELECTED, t->chan);

		}

		if(i != selectedItem && fileList[i]->GetState() == STATE_SELECTED)
			fileList[i]->ResetState();
		else if(focus && i == selectedItem && fileList[i]->GetState() == STATE_DEFAULT)
			fileList[selectedItem]->SetState(STATE_SELECTED, t->chan);

		int currChan = t->chan;

		if(t->wpad.ir.valid && !fileList[i]->IsInside(t->wpad.ir.x, t->wpad.ir.y))
			t->chan = -1;

		fileList[i]->Update(t);
		t->chan = currChan;

		if(fileList[i]->GetState() == STATE_SELECTED)
		{
			selectedItem = i;
		}
	}

	// update the location of the scroll box based on the position in the file list
	if(positionWiimote > 0)
	{
		position = positionWiimote; // follow wiimote cursor
	}
	else
	{
		position = 136*(browser->pageIndex + FILEBROWSERSIZE/2.0) / (browser->browserList.size()*1.0);

		if(browser->pageIndex/(FILEBROWSERSIZE/2.0) < 1)
			position = -10;
		else if((browser->pageIndex+FILEBROWSERSIZE)/(FILEBROWSERSIZE*1.0) >= (browser->browserList.size())/(FILEBROWSERSIZE*1.0))
			position = 156;
	}

	scrollbarBoxBtn->SetPosition(12,position+26);

	listChanged = false;

	if(updateCB)
		updateCB(this);
}