#include "palettesscanpopup.h"

#include "toonzqt/gutil.h"

#include "tsystem.h"

#include <qpushbutton></qpushbutton>

#include <qlabel></qlabel>

//=============================================================================

/*! \class PalettesScanPopup

		\brief The PalettesScanPopup class provides a modal dialog to import

					 palette in StudioPalette folder.

		Inherits \b Dialog.

		This object is composed of three sub-object: a \b FileField to choose a

		folder, a label \b QLabel, generally empty, to show palette folder during

		them import, and a button bar to start import or cancel command.

		The class provides a method, \b setCurrentFolder(TFilePath path), to set

		folder wherein import palettes.

		Protected function allows to manage palette import.

*/

PalettesScanPopup::PalettesScanPopup()

	: Dialog(0, true, true, "PalettesScan"), m_folderPath(TFilePath()), m_timerId(0)

{

	setWindowTitle(tr("Search for Palettes"));

	setFixedWidth(250);

	m_field = new DVGui::FileField();

	addWidget(m_field);

	m_label = new QLabel();

	m_label->setFixedWidth(200);

	addWidget(m_label);

	QPushButton *okBtn = new QPushButton(tr("Ok"), this);

	okBtn->setDefault(true);

	QPushButton *cancelBtn = new QPushButton(tr("Cancel"), this);

	connect(okBtn, SIGNAL(clicked()), this, SLOT(onOkBtnClicked()));

	connect(cancelBtn, SIGNAL(clicked()), this, SLOT(reject()));

	addButtonBarWidget(okBtn, cancelBtn);

}

//-----------------------------------------------------------------------------

/*! Set current folder path to \b path.

*/

void PalettesScanPopup::setCurrentFolder(TFilePath path)

{

	m_folderPath = path;

}

//-----------------------------------------------------------------------------

/*! Return current folder path.

*/

TFilePath PalettesScanPopup::getCurrentFolder()

{

	return m_folderPath;

}

//-----------------------------------------------------------------------------

/*! Start palette import. Add \b FileField path to directories stack and start

		a timer.

*/

void PalettesScanPopup::onOkBtnClicked()

{

	m_timerId = startTimer(3);

	TFilePathSet fps;

	std::wstring s = m_field->getPath().toStdWString();

	int i = 0, len = s.length();

	while (i < len) {

		while (i < len && (s[i] == ' ' || s[i] == '\t'))

			i++;

		if (i >= len)

			break;

		int j = i;

		while (j < len && s[j] != ',')

			j++;

		int k = j;

		while (k > i && (s[k - 1] == ' ' || s[k - 1] == '\t'))

			k--;

		std::wstring token = s.substr(i, k - i);

		fps.push_back(TFilePath(token));

		i = j + 1;

	}

	push(fps);

}

//-----------------------------------------------------------------------------

/*! Set label text to path \b fp.

*/

void PalettesScanPopup::setLabel(const TFilePath &fp)

{

	QString elideStr = elideText(toQString(fp), m_label->font(), m_label->width());

	m_label->setText(elideStr);

}

//-----------------------------------------------------------------------------

/*! Proceed in palette import recalling \b step(); if step() return false

		stop timer, clear directories stack, and close popup sending accept event.

*/

void PalettesScanPopup::timerEvent(QTimerEvent *event)

{

	bool ret = step();

	if (!ret) {

		killTimer(m_timerId);

		clearStack();

		QDialog::accept();

	}

}

//-----------------------------------------------------------------------------

/*! Push TFilePath \b fp to the top of directories stack. Set label text to fp.

*/

void PalettesScanPopup::push(const TFilePath &fp)

{

	setLabel(fp);

	Directory *dir = new Directory();

	m_stack.push_back(dir);

	dir->m_path = fp;

	dir->m_files = TSystem::readDirectory(fp);

	dir->m_it = dir->m_files.begin();

}

//-----------------------------------------------------------------------------

/*! Push TFilePathSet \b fs to the top of directories stack. Set label text to

		"<files>".</files>

*/

void PalettesScanPopup::push(const TFilePathSet &fs)

{

	m_label->setText(tr("<files>"));</files>

	Directory *dir = new Directory();

	m_stack.push_back(dir);

	dir->m_path = TFilePath();

	dir->m_files = fs;

	dir->m_it = dir->m_files.begin();

}

//-----------------------------------------------------------------------------

/*! Removes the top item from the stack. If stack is empty return immediately.

*/

void PalettesScanPopup::pop()

{

	if (m_stack.empty())

		return;

	Directory *dir = m_stack.back();

	delete dir;

	m_stack.pop_back();

	if (!m_stack.empty()) {

		dir = m_stack.back();

		setLabel(dir->m_path);

	} else

		m_label->setText(tr(""));

}

//-----------------------------------------------------------------------------

/*! Return true if direcories stack is not empty; otherwise return false.

\n	If stack is not empty check current directory:

\li	if all files are checked recall \b pop();

\li	if its path is a directory recall \b push(const TFilePath &fp);

\li	if its path is a file check its extension and recall \b onPlt(const TFilePath &fp).

*/

bool PalettesScanPopup::step()

{

	if (m_stack.empty())

		return false;

	Directory *dir = m_stack.back();

	if (dir->m_it == dir->m_files.end())

		pop();

	else {

		TFilePath fp = *(dir->m_it)++;

		if (TFileStatus(fp).isDirectory())

			push(fp);

		else {

			setLabel(fp);

			std::string ext = fp.getType();

			if (ext == "plt" || ext == "tpl" || ext == "pli")

				onPlt(fp);

		}

	}

	return true;

}

//-----------------------------------------------------------------------------

/*! Resets the content of the directories stack and set label text empty.

*/

void PalettesScanPopup::clearStack()

{

	for (int i = 0; i < (int)m_stack.size(); i++)

		delete m_stack[i];

	m_stack.clear();

	m_label->setText(tr(""));

}

//-----------------------------------------------------------------------------

/*! Import palette, defined in path \b fp, in current \b StudioPalette folder.

*/

void PalettesScanPopup::onPlt(const TFilePath &fp)

{

	TFilePath root(m_field->getPath().toStdString());

	assert(root.isAncestorOf(fp));

	TFilePath q = fp.getParentDir() - root;

	StudioPalette::instance()->importPalette(m_folderPath + q, fp);

}