#include "qt/window/QtSelectPathsDialog.h"

#include <set>

#include <QLabel>
#include <QListWidget>
#include <QPushButton>

#include "utility/file/FilePath.h"

QtSelectPathsDialog::QtSelectPathsDialog(const QString& title, const QString& description, QWidget* parent)
	: QtTextEditDialog(title, description, parent)
{
}

std::vector<FilePath> QtSelectPathsDialog::getPathsList() const
{
	std::vector<FilePath> checkedPaths;

	for (int i = 0; i < m_list->count(); i++)
	{
		if (m_list->item(i)->checkState() ==  Qt::Checked)
		{
			checkedPaths.push_back(FilePath(m_list->item(i)->text().toStdWString()));
		}
	}

	return checkedPaths;
}

void QtSelectPathsDialog::setPathsList(const std::vector<FilePath>& paths, const std::vector<FilePath>& checkedPaths)
{
	std::set<FilePath> checked(checkedPaths.begin(), checkedPaths.end());

	for (const FilePath& s : paths)
	{
		QListWidgetItem* item = new QListWidgetItem(s.str().c_str(), m_list);
		item->setFlags(item->flags() | Qt::ItemIsUserCheckable); // set checkable flag

		if (checked.find(s) == checked.end())
		{
			item->setCheckState(Qt::Unchecked); // AND initialize check state
		}
		else
		{
			item->setCheckState(Qt::Checked);
		}

		if (!s.exists())
		{
			item->setTextColor(Qt::red);
			item->setToolTip("Path does not exist");
			item->setFlags( item->flags() & ~Qt::ItemIsEnabled );
		}
		else
		{
			item->setTextColor(Qt::black);
		}
	}
}

void QtSelectPathsDialog::checkSelected(bool checked)
{
	for (QListWidgetItem* item : m_list->selectedItems())
	{
		item->setCheckState( (checked ? Qt::Checked : Qt::Unchecked) );
	}
}

void QtSelectPathsDialog::populateWindow(QWidget* widget)
{
	QVBoxLayout* layout = new QVBoxLayout();
	layout->setContentsMargins(0, 0, 0, 0);

	QLabel* description = new QLabel(m_description);
	description->setObjectName("description");
	description->setWordWrap(true);
	layout->addWidget(description);

	m_list = new QListWidget();
	m_list->setObjectName("pathList");
	m_list->setEditTriggers(QAbstractItemView::NoEditTriggers);
	m_list->setSelectionMode(QAbstractItemView::ExtendedSelection);
	m_list->setAttribute(Qt::WA_MacShowFocusRect, 0);
	layout->addWidget(m_list);

	QHBoxLayout* buttonLayout = new QHBoxLayout();
	buttonLayout->setContentsMargins(0, 0, 0, 0);

	QPushButton* checkAllButton = new QPushButton("check all");
	checkAllButton->setObjectName("windowButton");
	connect(checkAllButton, &QPushButton::clicked,
		[=]()
		{
			m_list->selectAll();
			checkSelected(true);
			m_list->clearSelection();
		});
	buttonLayout->addWidget(checkAllButton);

	QPushButton* unCheckAllButton = new QPushButton("uncheck all");
	unCheckAllButton->setObjectName("windowButton");
	connect(unCheckAllButton, &QPushButton::clicked,
		[=]()
		{
			m_list->selectAll();
			checkSelected(false);
			m_list->clearSelection();
		});
	buttonLayout->addWidget(unCheckAllButton);

	QPushButton* checkSelectedButton = new QPushButton("check selected");
	checkSelectedButton->setObjectName("windowButton");
	connect(checkSelectedButton, &QPushButton::clicked,
		[=]()
		{
			checkSelected(true);
		});
	buttonLayout->addWidget(checkSelectedButton);

	QPushButton* unCheckSelectedButton = new QPushButton("uncheck selected");
	unCheckSelectedButton->setObjectName("windowButton");
	connect(unCheckSelectedButton, &QPushButton::clicked,
		[=]()
		{
			checkSelected(false);
		});
	buttonLayout->addWidget(unCheckSelectedButton);

	layout->addLayout(buttonLayout);

	widget->setLayout(layout);
}

void QtSelectPathsDialog::windowReady()
{
	updateNextButton("Save");
	updateCloseButton("Cancel");

	setPreviousVisible(false);

	updateTitle(m_title);
}