/***************************************************************************
 *   Copyright (C) 2005 by Jeff Ferr                                       *
 *   root@sat                                                              *
 *                                                                         *
 *   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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include "Stdafx.h"
#include "jfilechooserdialogbox.h"
#include "jfile.h"

namespace jgui {

struct ascending_sort {
	bool operator()(std::string a, std::string b)
	{
		if (a < b) {
			return true;
		}

		return false;
	}
};

FileChooserDialogBox::FileChooserDialogBox(std::string title, std::string directory, int x, int y, jfilechooser_type_t type):
	jgui::Frame(title, x, y, 1000, 600)
{
	jcommon::Object::SetClassName("jgui::FileChooserDialogBox");

	_label = NULL;
	_file = NULL;

	_base_dir = directory;
	_current_dir = directory;
	_has_parent = false;
	_type = type;
	_filter = JFF_FILE_AND_DIRECTORY;
	_extension_ignorecase = true;

	_list = new jgui::ListBox(_insets.left, _insets.top, _size.width-_insets.left-_insets.right, _size.height-_insets.top-_insets.bottom);

	Add(_list);

	if (_type == JFT_SAVE_FILE_DIALOG) {
		_label = new jgui::Label("File name", _insets.left, _list->GetY()+_list->GetHeight()+10, _size.width-_insets.left-_insets.right, 45);
		_file = new jgui::TextField(_insets.left, _label->GetY()+_label->GetHeight()+10, _size.width-_insets.left-_insets.right, 45);

		_label->SetBackgroundVisible(false);
		_label->SetBorder(JCB_EMPTY);

		Add(_label);
		Add(_file);
	}

	_list->RequestFocus();
	_list->RegisterSelectListener(this);

	ShowFiles(_current_dir);

	if (_type == JFT_OPEN_FILE_DIALOG) {
		AddSubtitle(_DATA_PREFIX"images/blue_icon.png", "Open");
	} else {
		AddSubtitle(_DATA_PREFIX"images/blue_icon.png", "Save");
	}

	Pack();
}

FileChooserDialogBox::~FileChooserDialogBox()
{
	jthread::AutoLock lock(&_mutex);

	_list->RemoveSelectListener(this);

	if (_list != NULL) {
		delete _list;
	}

	if (_label != NULL) {
		delete _label;
	}

	if (_file != NULL) {
		delete _file;
	}
}

std::string FileChooserDialogBox::GetFile()
{
	std::string path,
		selectedItem = _list->GetCurrentItem()->GetValue();

	if (_list->GetCurrentIndex() == 0) {
		selectedItem = "";
	}

	if (_type == JFT_OPEN_FILE_DIALOG) {
	} else {
		selectedItem = _file->GetText();
	}

#ifdef _WIN32
	if (_current_dir[_current_dir.size()-1] == '\\') {
#else
	if (_current_dir[_current_dir.size()-1] == '/') {
#endif
		path = _current_dir + selectedItem;
	} else {
		path = _current_dir + jio::File::GetDelimiter() + selectedItem;
	}

	return path;
}

std::string FileChooserDialogBox::GetName()
{
	return _list->GetCurrentItem()->GetValue();
}

std::string FileChooserDialogBox::GetCurrentDirectory()
{
	return _current_dir;
}

void FileChooserDialogBox::SetCurrentDirectory(std::string directory)
{
	_current_dir = directory;
}

void FileChooserDialogBox::AddExtension(std::string ext)
{
	_extensions.push_back(ext);
}

void FileChooserDialogBox::SetFileFilter(jfilechooser_filter_t filter)
{
	_filter = filter;
}

void FileChooserDialogBox::SetExtensionIgnoreCase(bool b)
{
	_extension_ignorecase = b;
}

bool FileChooserDialogBox::ShowFiles(std::string current_dir)
{
	std::vector<std::string> files = ListFiles(current_dir);

	if (files.size() == 0) {
		return false;
	}

	_list->RemoveItems();
	_list->AddImageItem("..", _DATA_PREFIX"/images/folder.png");

	std::sort(files.begin(), files.end(), ascending_sort());

	if (_filter == JFF_DIRECTORY_ONLY || _filter == JFF_FILE_AND_DIRECTORY) {
		for (unsigned int i=0; i<files.size(); i++) {
			if (files[i] == "." || files[i] == "..") {
				continue;
			}

			if (IsDirectory(current_dir +jio::File::GetDelimiter() + files[i])) {
				// adiciona um icone para o diretorio
				_list->AddImageItem(files[i], _DATA_PREFIX"/images/lockfolder.png"); 
			}
		}
	}

	if (_filter == JFF_FILE_ONLY || _filter == JFF_FILE_AND_DIRECTORY) {
		for (unsigned int i=0; i<files.size(); i++) {
			std::string ext,
				file = files[i];
			bool b = false;

			if (_extensions.size() == 0) {
				b = true;
			} else {
				for (std::vector<std::string>::iterator i=_extensions.begin(); i!=_extensions.end(); i++) {
					ext = (*i);

					if (file.size() > ext.size()) {
						if (_extension_ignorecase == true) {
#ifdef _WIN32
							if (_stricmp(file.c_str()+(file.size()-ext.size()), ext.c_str()) == 0) {
#else
							if (strcasecmp(file.c_str()+(file.size()-ext.size()), ext.c_str()) == 0) {
#endif
								b = true;
							}
						} else {
							if (strcmp(file.c_str()+(file.size()-ext.size()), ext.c_str()) == 0) {
								b = true;
							}
						}
					}
				}
			}

			if (b == true) {
				if (IsFile(current_dir + jio::File::GetDelimiter() + file)) {
					_list->AddImageItem(file, _DATA_PREFIX"/images/lockfile.png");
				}
			}
		}
	}

	_list->SetCurrentIndex(0);
	_list->Repaint();

	return true;
}

std::vector<std::string> FileChooserDialogBox::ListFiles(std::string dirPath)
{
	jio::File file(dirPath);

	return file.ListFiles();
}

bool FileChooserDialogBox::IsDirectory(std::string path)
{
	jio::File file(path);

	return file.IsDirectory();
}

bool FileChooserDialogBox::IsFile(std::string path)
{
	jio::File file(path);

	return file.IsFile();
}

void FileChooserDialogBox::ItemSelected(jgui::SelectEvent *event)
{
	jthread::AutoLock lock(&_mutex);

	std::string selectedItem = _list->GetCurrentItem()->GetValue();

#ifdef _WIN32
	if (selectedItem == "..\\") {
#else
	if (selectedItem == "../") {
#endif
		if (_has_parent) {
			std::string::size_type idx = _current_dir.substr(0, _current_dir.length()-1).rfind('/');
			std::string aux = _current_dir.substr(0, idx+1);

			if (aux.find(_base_dir) == 0) {
				if (aux == jio::File::GetDelimiter()) {
					_current_dir = aux;
					_has_parent = false;
				} else {
					_current_dir = aux;
					_has_parent = true;
				}

				ShowFiles(_current_dir);
			}
		}
	} else {
		std::string path;

#ifdef _WIN32
		if (_current_dir[_current_dir.size()-1] == '\\') {
#else
		if (_current_dir[_current_dir.size()-1] == '/') {
#endif
			path = _current_dir + selectedItem;
		} else {
			path = _current_dir + jio::File::GetDelimiter() + selectedItem;
		}

		if (IsDirectory(path)) {
			// INFO:: selecionar um diretorio
			if (ShowFiles(path) == true) {
				_current_dir = path;
				_has_parent = true;
			}
		} else { 
			// INFO:: Selecionar um arquivo
			if (_type == JFT_OPEN_FILE_DIALOG) {
				Release();

				_last_key_code = jgui::JKS_BLUE; 
			} else {
				_file->SetText(selectedItem);
			}
		}
	}
}

bool FileChooserDialogBox::ProcessEvent(jgui::KeyEvent *event)
{
	jthread::AutoLock lock(&_mutex);

	if (event->GetSymbol() == jgui::JKS_BLUE || event->GetSymbol() == jgui::JKS_F4) {
		if (_type == JFT_OPEN_FILE_DIALOG) {
			Release();
			_last_key_code = jgui::JKS_BLUE; 
		} else {
			if (_file->GetText() != "") {
				Release();

				_last_key_code = jgui::JKS_BLUE; 
			}
		}
	}

	return true;
}

}