#include "QtListBox.h"

#include <QBoxLayout>
#include <QListWidget>
#include <QLabel>
#include <QListWidget>
#include <QScrollBar>

#include "QtIconButton.h"
#include "QtStringListBoxItem.h"
#include "utilityQt.h"
#include "QtTextEditDialog.h"
#include "ResourcePaths.h"
#include "utilityString.h"

QtListBox::QtListBox(QWidget *parent, const QString& listName)
	: QFrame(parent)
	, m_listName(listName)
{
	QBoxLayout* layout = new QVBoxLayout();
	layout->setSpacing(0);
	layout->setContentsMargins(0, 6, 0, 0);
	layout->setAlignment(Qt::AlignTop);

	m_list = new QListWidget(this);
	m_list->setObjectName("list");
	m_list->setAttribute(Qt::WA_MacShowFocusRect, 0);

	setStyleSheet(utility::getStyleSheet(ResourcePaths::getGuiPath().concatenate(L"window/listbox.css")).c_str());
	layout->addWidget(m_list, 5);

	QWidget* buttonContainer = new QWidget(this);
	buttonContainer->setObjectName("bar");

	QHBoxLayout* barLayout = new QHBoxLayout();
	barLayout->setContentsMargins(8, 4, 8, 2);
	barLayout->setSpacing(0);

	m_addButton = new QtIconButton(
		ResourcePaths::getGuiPath().concatenate(L"window/plus.png"),
		ResourcePaths::getGuiPath().concatenate(L"window/plus_hover.png")
	);
	m_addButton->setIconSize(QSize(16, 16));
	m_addButton->setObjectName("plusButton");
	m_addButton->setToolTip("add line");
	barLayout->addWidget(m_addButton);

	m_removeButton = new QtIconButton(
		ResourcePaths::getGuiPath().concatenate(L"window/minus.png"),
		ResourcePaths::getGuiPath().concatenate(L"window/minus_hover.png")
	);
	m_removeButton->setIconSize(QSize(16, 16));
	m_removeButton->setObjectName("minusButton");
	m_removeButton->setToolTip("remove line");
	barLayout->addWidget(m_removeButton);

	barLayout->addStretch();

	m_innerBarLayout = new QHBoxLayout();
	barLayout->addLayout(m_innerBarLayout);

	QPushButton* editButton = new QtIconButton(
		ResourcePaths::getGuiPath().concatenate(L"code_view/images/edit.png"),
		FilePath()
	);
	editButton->setIconSize(QSize(16, 16));
	editButton->setObjectName("editButton");
	editButton->setToolTip("edit plain text");
	barLayout->addWidget(editButton);

	buttonContainer->setLayout(barLayout);
	layout->addWidget(buttonContainer, 0);
	setLayout(layout);

	connect(m_addButton, &QPushButton::clicked, this, &QtListBox::addListBoxItem);
	connect(m_removeButton, &QPushButton::clicked, this, &QtListBox::removeListBoxItem);
	connect(editButton, &QPushButton::clicked, this, &QtListBox::showEditDialog);

	setAcceptDrops(true);
	setMaximumHeight(160);
}

void QtListBox::clear()
{
	m_list->clear();
}

void QtListBox::addWidgetToBar(QWidget* widget)
{
	if (m_innerBarLayout)
	{
		m_innerBarLayout->addWidget(widget);
	}
}

bool QtListBox::event(QEvent* event)
{
	// Prevent nested ScrollAreas from scrolling at the same time;
	if (event->type() == QEvent::Wheel)
	{
		QRect rect = m_list->viewport()->rect();
		QPoint pos = m_list->mapFromGlobal(dynamic_cast<QWheelEvent*>(event)->globalPos());
		QScrollBar* bar = m_list->verticalScrollBar();

		if (bar->minimum() != bar->maximum() && rect.contains(pos))
		{
			bool down = dynamic_cast<QWheelEvent*>(event)->angleDelta().y() < 0;

			if ((down && bar->value() != bar->maximum()) || (!down && bar->value() != bar->minimum()))
			{
				return true;
			}
		}
	}

	return QFrame::event(event);
}

QtListBoxItem* QtListBox::addListBoxItemWithText(const QString& text)
{
	QtListBoxItem* item = addListBoxItem();
	item->setText(text);
	return item;
}

void QtListBox::selectItem(QListWidgetItem* item)
{
	for (int i = 0; i < m_list->count(); i++)
	{
		m_list->item(i)->setSelected(false);
	}

	item->setSelected(true);
}

QtListBoxItem* QtListBox::addListBoxItem()
{
	QListWidgetItem* item = new QListWidgetItem(m_list);
	m_list->addItem(item);

	QtListBoxItem* widget = createListBoxItem(item);
	m_list->setItemWidget(item, widget);

	m_list->scrollToItem(item);
	widget->setFocus();

	return widget;
}

void QtListBox::removeListBoxItem()
{
	if (m_list->selectedItems().empty())
	{
		return;
	}

	int rowIndex = m_list->row(m_list->selectedItems().first());

	qDeleteAll(m_list->selectedItems());

	if (rowIndex == m_list->count())
	{
		rowIndex -= 1;
	}

	if (rowIndex >= 0)
	{
		m_list->setCurrentRow(rowIndex);
	}
}

void QtListBox::showEditDialog()
{
	if (!m_editDialog)
	{
		m_editDialog = std::make_shared<QtTextEditDialog>(m_listName, "Edit the list in plain text. Each line is one item.");
		m_editDialog->setup();

		std::vector<std::wstring> list;
		for (int i = 0; i < m_list->count(); ++i)
		{
			QtListBoxItem* item = dynamic_cast<QtListBoxItem*>(m_list->itemWidget(m_list->item(i)));
			if (!item->getReadOnly())
			{
				list.push_back(item->getText().toStdWString());
			}
		}

		m_editDialog->setText(utility::join(list, L"\n"));

		connect(m_editDialog.get(), &QtTextEditDialog::canceled, this, &QtListBox::canceledEditDialog);
		connect(m_editDialog.get(), &QtTextEditDialog::finished, this, &QtListBox::savedEditDialog);
	}

	m_editDialog->showWindow();
	m_editDialog->raise();
}

void QtListBox::canceledEditDialog()
{
	m_editDialog->hide();
	m_editDialog.reset();

	window()->raise();
}

void QtListBox::savedEditDialog()
{
	std::vector<std::wstring> readOnlyLines;
	for (int i = 0; i < m_list->count(); ++i)
	{
		QtListBoxItem* item = dynamic_cast<QtListBoxItem*>(m_list->itemWidget(m_list->item(i)));
		if (item->getReadOnly())
		{
			readOnlyLines.push_back(item->getText().toStdWString());
		}
	}

	std::vector<std::wstring> lines = utility::splitToVector(m_editDialog->getText(), L"\n");
	for (size_t i = 0; i < lines.size(); i++)
	{
		lines[i] = utility::trim(lines[i]);

		if (lines[i].empty())
		{
			lines.erase(lines.begin() + i);
			i--;
		}
	}

	clear();

	for (const std::wstring& line : readOnlyLines)
	{
		QtListBoxItem* itemWidget = addListBoxItemWithText(QString::fromStdWString(line));
		itemWidget->setReadOnly(true);
	}

	for (const std::wstring& line : lines)
	{
		QtListBoxItem* itemWidget = addListBoxItemWithText(QString::fromStdWString(line));
		itemWidget->setReadOnly(false);
	}

	canceledEditDialog();
}