/* Copyright 2015-2017 MultiMC Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "PackagesPage.h"
#include "ui_PackagesPage.h"

#include <QDateTime>
#include <QSortFilterProxyModel>
#include <QRegularExpression>

#include "dialogs/ProgressDialog.h"
#include "VersionProxyModel.h"

#include "meta/Index.h"
#include "meta/VersionList.h"
#include "meta/Version.h"
#include "Env.h"
#include "MultiMC.h"

using namespace Meta;

static QString formatRequires(const VersionPtr &version)
{
	QStringList lines;
	auto & reqs = version->requires();
	auto iter = reqs.begin();
	while (iter != reqs.end())
	{
		auto &uid = iter.key();
		auto &version = iter.value();
		const QString readable = ENV.metadataIndex()->hasUid(uid) ? ENV.metadataIndex()->get(uid)->humanReadable() : uid;
		lines.append(QString("%1 (%2)").arg(readable, version));
		iter++;
	}
	return lines.join('\n');
}

PackagesPage::PackagesPage(QWidget *parent) :
	QWidget(parent),
	ui(new Ui::PackagesPage)
{
	ui->setupUi(this);
	ui->tabWidget->tabBar()->hide();

	m_fileProxy = new QSortFilterProxyModel(this);
	m_fileProxy->setSortRole(Qt::DisplayRole);
	m_fileProxy->setSortCaseSensitivity(Qt::CaseInsensitive);
	m_fileProxy->setFilterCaseSensitivity(Qt::CaseInsensitive);
	m_fileProxy->setFilterRole(Qt::DisplayRole);
	m_fileProxy->setFilterKeyColumn(0);
	m_fileProxy->sort(0);
	m_fileProxy->setSourceModel(ENV.metadataIndex().get());
	ui->indexView->setModel(m_fileProxy);

	m_filterProxy = new QSortFilterProxyModel(this);
	m_filterProxy->setSortRole(VersionList::SortRole);
	m_filterProxy->setFilterCaseSensitivity(Qt::CaseInsensitive);
	m_filterProxy->setFilterRole(Qt::DisplayRole);
	m_filterProxy->setFilterKeyColumn(0);
	m_filterProxy->sort(0, Qt::DescendingOrder);
	ui->versionsView->setModel(m_filterProxy);

	m_versionProxy = new VersionProxyModel(this);
	m_filterProxy->setSourceModel(m_versionProxy);

	connect(ui->indexView->selectionModel(), &QItemSelectionModel::currentChanged, this, &PackagesPage::updateCurrentVersionList);
	connect(ui->versionsView->selectionModel(), &QItemSelectionModel::currentChanged, this, &PackagesPage::updateVersion);
	connect(m_filterProxy, &QSortFilterProxyModel::dataChanged, this, &PackagesPage::versionListDataChanged);

	updateCurrentVersionList(QModelIndex());
	updateVersion();
}

PackagesPage::~PackagesPage()
{
	delete ui;
}

QIcon PackagesPage::icon() const
{
	return MMC->getThemedIcon("packages");
}

void PackagesPage::on_refreshIndexBtn_clicked()
{
	ENV.metadataIndex()->load();
}
void PackagesPage::on_refreshFileBtn_clicked()
{
	VersionListPtr list = ui->indexView->currentIndex().data(Index::ListPtrRole).value<VersionListPtr>();
	if (!list)
	{
		return;
	}
	list->load();
}
void PackagesPage::on_refreshVersionBtn_clicked()
{
	VersionPtr version = ui->versionsView->currentIndex().data(VersionList::VersionPtrRole).value<VersionPtr>();
	if (!version)
	{
		return;
	}
	version->load();
}

void PackagesPage::on_fileSearchEdit_textChanged(const QString &search)
{
	if (search.isEmpty())
	{
		m_fileProxy->setFilterFixedString(QString());
	}
	else
	{
		QStringList parts = search.split(' ');
		std::transform(parts.begin(), parts.end(), parts.begin(), &QRegularExpression::escape);
		m_fileProxy->setFilterRegExp(".*" + parts.join(".*") + ".*");
	}
}
void PackagesPage::on_versionSearchEdit_textChanged(const QString &search)
{
	if (search.isEmpty())
	{
		m_filterProxy->setFilterFixedString(QString());
	}
	else
	{
		QStringList parts = search.split(' ');
		std::transform(parts.begin(), parts.end(), parts.begin(), &QRegularExpression::escape);
		m_filterProxy->setFilterRegExp(".*" + parts.join(".*") + ".*");
	}
}

void PackagesPage::updateCurrentVersionList(const QModelIndex &index)
{
	if (index.isValid())
	{
		VersionListPtr list = index.data(Index::ListPtrRole).value<VersionListPtr>();
		ui->versionsBox->setEnabled(true);
		ui->refreshFileBtn->setEnabled(true);
		ui->fileUidLabel->setEnabled(true);
		ui->fileUid->setText(list->uid());
		ui->fileNameLabel->setEnabled(true);
		ui->fileName->setText(list->name());
		m_versionProxy->setSourceModel(list.get());
		ui->refreshFileBtn->setText(tr("Refresh %1").arg(list->humanReadable()));
		list->load();
	}
	else
	{
		ui->versionsBox->setEnabled(false);
		ui->refreshFileBtn->setEnabled(false);
		ui->fileUidLabel->setEnabled(false);
		ui->fileUid->clear();
		ui->fileNameLabel->setEnabled(false);
		ui->fileName->clear();
		m_versionProxy->setSourceModel(nullptr);
		ui->refreshFileBtn->setText(tr("Refresh"));
	}
}

void PackagesPage::versionListDataChanged(const QModelIndex &tl, const QModelIndex &br)
{
	if (QItemSelection(tl, br).contains(ui->versionsView->currentIndex()))
	{
		updateVersion();
	}
}

void PackagesPage::updateVersion()
{
	VersionPtr version = std::dynamic_pointer_cast<Version>(
				ui->versionsView->currentIndex().data(VersionList::VersionPointerRole).value<BaseVersionPtr>());
	if (version)
	{
		ui->refreshVersionBtn->setEnabled(true);
		ui->versionVersionLabel->setEnabled(true);
		ui->versionVersion->setText(version->version());
		ui->versionTimeLabel->setEnabled(true);
		ui->versionTime->setText(version->time().toString("yyyy-MM-dd HH:mm"));
		ui->versionTypeLabel->setEnabled(true);
		ui->versionType->setText(version->type());
		ui->versionRequiresLabel->setEnabled(true);
		ui->versionRequires->setText(formatRequires(version));
		ui->refreshVersionBtn->setText(tr("Refresh %1").arg(version->version()));
	}
	else
	{
		ui->refreshVersionBtn->setEnabled(false);
		ui->versionVersionLabel->setEnabled(false);
		ui->versionVersion->clear();
		ui->versionTimeLabel->setEnabled(false);
		ui->versionTime->clear();
		ui->versionTypeLabel->setEnabled(false);
		ui->versionType->clear();
		ui->versionRequiresLabel->setEnabled(false);
		ui->versionRequires->clear();
		ui->refreshVersionBtn->setText(tr("Refresh"));
	}
}

void PackagesPage::opened()
{
	ENV.metadataIndex()->load();
}