/**************************************************************************
* Otter Browser: Web browser controlled by the user, not vice-versa.
* Copyright (C) 2017 - 2018 Michal Dutkiewicz aka Emdek <michal@emdek.pl>
*
* 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 3 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, see <http://www.gnu.org/licenses/>.
*
**************************************************************************/

#include "ItemModel.h"

#include <QtCore/QMimeData>

namespace Otter
{

ItemModel::Item::Item(ItemModel::ItemType type) : QStandardItem()
{
	setup(type);
}

ItemModel::Item::Item(const QString &title, ItemModel::ItemType type) : QStandardItem(title)
{
	setup(type);
}

ItemModel::Item::Item(const QIcon &icon, const QString &title, ItemModel::ItemType type) : QStandardItem(icon, title)
{
	setup(type);
}

void ItemModel::Item::setup(ItemModel::ItemType type)
{
	setData(type, TypeRole);

	switch (type)
	{
		case EntryType:
			setFlags(flags() | Qt::ItemNeverHasChildren);

			break;
		case SeparatorType:
			setData(QLatin1String("separator"), Qt::AccessibleDescriptionRole);
			setFlags(Qt::ItemIsEnabled | Qt::ItemNeverHasChildren);

			break;
		default:
			break;
	}
}

bool ItemModel::Item::isAncestorOf(QStandardItem *child) const
{
	if (child == nullptr || child == this)
	{
		return false;
	}

	QStandardItem *parent(child->parent());

	while (parent)
	{
		if (parent == this)
		{
			return true;
		}

		parent = parent->parent();
	}

	return false;
}

ItemModel::ItemModel(QObject *parent) : QStandardItemModel(parent),
	m_isExclusive(false),
	m_isIgnoringCheckStateReset(true)
{
}

void ItemModel::setupItem(QStandardItem *item, ItemModel::ItemType type)
{
	item->setData(type, TypeRole);

	if (type != FolderType)
	{
		item->setFlags((item->flags() & ~Qt::ItemIsDropEnabled) | Qt::ItemNeverHasChildren);
	}
}

void ItemModel::resetCheckState(const QModelIndex &parent)
{
	for (int i = 0; i < rowCount(parent); ++i)
	{
		const QModelIndex index(this->index(i, 0, parent));

		if (index.data(Qt::CheckStateRole).toInt() != Qt::Unchecked)
		{
			setData(index, Qt::Unchecked, Qt::CheckStateRole);
		}

		if (static_cast<ItemType>(index.data(TypeRole).toInt()) == FolderType)
		{
			resetCheckState(index);
		}
	}
}

void ItemModel::insertRow(QStandardItem *item, QStandardItem *parent, int row, ItemType type)
{
	if (!item)
	{
		item = new QStandardItem();
	}

	if (!parent)
	{
		parent = invisibleRootItem();
	}

	setupItem(item, type);

	if (row >= 0)
	{
		parent->insertRow(row, item);
	}
	else
	{
		parent->appendRow(item);
	}
}

void ItemModel::insertRow(const QList<QStandardItem*> &items, QStandardItem *parent, int row, ItemType type)
{
	if (!parent)
	{
		parent = invisibleRootItem();
	}

	for (int i = 0; i < items.count(); ++i)
	{
		setupItem(items.at(i), type);
	}

	if (row >= 0)
	{
		parent->insertRow(row, items);
	}
	else
	{
		parent->appendRow(items);
	}
}

void ItemModel::setExclusive(bool isExclusive)
{
	m_isExclusive = isExclusive;
}

QMimeData* ItemModel::mimeData(const QModelIndexList &indexes) const
{
	QMimeData *mimeData(QStandardItemModel::mimeData(indexes));

	if (indexes.count() == 1)
	{
		mimeData->setProperty("x-item-index", indexes.first());
	}

	return mimeData;
}

QVariant ItemModel::data(const QModelIndex &index, int role) const
{
	if (role == Qt::AccessibleDescriptionRole && static_cast<ItemType>(QStandardItemModel::data(index, TypeRole).toInt()) == SeparatorType)
	{
		return QLatin1String("separator");
	}

	return QStandardItemModel::data(index, role);
}

QVariantList ItemModel::getAllData(int role, int column, const QModelIndex &parent) const
{
	QVariantList data;
	const int rowAmount(rowCount(parent));
	const bool useAllColumns(column < 0);

	for (int i = 0; i < rowAmount; ++i)
	{
		const QModelIndex rowIndex(index(i, 0, parent));

		if (useAllColumns)
		{
			const int columnAmount(columnCount(rowIndex));

			for (int j = 0; j < columnAmount; ++j)
			{
				const QVariant value(index(i, j, parent).data(role));

				if (!value.isNull())
				{
					data.append(value);
				}
			}
		}
		else
		{
			const QVariant value(index(i, column, parent).data(role));

			if (!value.isNull())
			{
				data.append(value);
			}
		}

		if (static_cast<ItemType>(rowIndex.data(TypeRole).toInt()) == FolderType)
		{
			data.append(getAllData(role, column, rowIndex));
		}
	}

	return data;
}

bool ItemModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent)
{
	Q_UNUSED(column)

	if (static_cast<ItemType>(parent.data(TypeRole).toInt()) != FolderType)
	{
		return QStandardItemModel::dropMimeData(data, action, row, 0, parent);
	}

	QStandardItem *item(itemFromIndex(data->property("x-item-index").toModelIndex()));
	QStandardItem *targetItem(itemFromIndex(parent.sibling(parent.row(), 0)));

	if (!targetItem)
	{
		targetItem = invisibleRootItem();
	}

	if (!item || !targetItem)
	{
		return QStandardItemModel::dropMimeData(data, action, row, column, parent);
	}

	const QList<QStandardItem*> sourceItems(item->parent() ? item->parent()->takeRow(item->row()) : takeRow(item->row()));

	if (sourceItems.isEmpty())
	{
		return false;
	}

	if (row < 0)
	{
		targetItem->appendRow(sourceItems);
	}
	else
	{
		if (item->parent() == targetItem && item->row() < row)
		{
			--row;
		}

		targetItem->insertRow(row, sourceItems);
	}

	return true;
}

bool ItemModel::isExclusive() const
{
	return m_isExclusive;
}

bool ItemModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
	if (role == Qt::CheckStateRole && m_isExclusive)
	{
		if (m_isIgnoringCheckStateReset && !value.toBool())
		{
			return false;
		}

		if (value.toBool())
		{
			m_isIgnoringCheckStateReset = false;

			resetCheckState(QModelIndex());

			m_isIgnoringCheckStateReset = true;
		}
	}

	return QStandardItemModel::setData(index, value, role);
}

}