/*
 * Cantata
 *
 * Copyright (c) 2011-2022 Craig Drummond <craig.p.drummond@gmail.com>
 *
 * ----
 *
 * 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 2 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; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include "multipagewidget.h"
#include "gui/application.h"
#include "support/configuration.h"
#include "support/icon.h"
#include "support/proxystyle.h"
#include "support/squeezedtextlabel.h"
#include "support/utils.h"
#ifdef Q_OS_MAC
#include "support/osxstyle.h"
#endif
#include "listview.h"
#include "singlepagewidget.h"
#include "toolbutton.h"
#include <QGridLayout>
#include <QLabel>
#include <QScrollArea>
#include <QStylePainter>
#include <QVBoxLayout>
#include <algorithm>

class SelectorButton : public ToolButton {
public:
	SelectorButton(const QString& t, const QString& s, const QIcon& icnIn, QWidget* p)
		: ToolButton(p)
	{
		icn = icnIn;
		QGridLayout* layout = new QGridLayout(this);
		icon = new QLabel(this);
		mainText = new SqueezedTextLabel(this);
		subText = new SqueezedTextLabel(this);
		subText->setFont(Utils::smallFont(mainText->font()));
		layout->setSpacing(2);
		int textSize = mainText->sizeHint().height() + subText->sizeHint().height() + layout->spacing();
		textSize += 6;
		int size = textSize;
		if (size < 72) {
			size = Utils::scaleForDpi(32);
		}
		QColor col(mainText->palette().windowText().color());
		subText->setStyleSheet(QString("QLabel{color:rgba(%1, %2, %3, 0.5)}").arg(col.red()).arg(col.green()).arg(col.blue()));

		icon->setFixedSize(size, size);
		layout->addWidget(icon, 0, 0, 2, 1);
		layout->addItem(new QSpacerItem(Utils::layoutSpacing(this), 2, QSizePolicy::Fixed, QSizePolicy::Fixed), 0, 1);
		layout->addWidget(mainText, 0, 2, 1, 1);
		layout->addWidget(subText, 1, 2, 1, 1);
		mainText->setAlignment(Qt::AlignBottom);
		subText->setAlignment(Qt::AlignTop);
		icon->setAlignment(Qt::AlignCenter);
		double dpr = DEVICE_PIXEL_RATIO();
		QPixmap pix = Icon::getScaledPixmap(icn, icon->width() * dpr, icon->height() * dpr, 96 * dpr);
		pix.setDevicePixelRatio(dpr);
		icon->setPixmap(pix);
		setAutoRaise(true);
		setLayout(layout);
		setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::Preferred);
		mainText->setText(t);
		subText->setText(s);
		int left;
		int top;
		int right;
		int bottom;
		layout->getContentsMargins(&left, &top, &right, &bottom);
		layout->setContentsMargins(qMin(8, left), qMin(8, top), qMin(8, right), qMin(8, bottom));

		setMinimumHeight(qMax(textSize, size) + (layout->contentsMargins().top() * 2));
		updateToolTip();
		setFocusPolicy(Qt::TabFocus);

		connect(Icon::fa(), &fa::QtAwesome::defaultOptionsReset, this, &SelectorButton::iconsUpdated);
	}

	void changeEvent(QEvent* e) override
	{
		if (e->type() == QEvent::PaletteChange) {
			QColor col(qApp->palette().windowText().color());
			subText->setStyleSheet(QString("QLabel{color:rgba(%1, %2, %3, 0.5)}").arg(col.red()).arg(col.green()).arg(col.blue()));
		}
		ToolButton::changeEvent(e);
	}

	void setSubText(const QString& str)
	{
		subText->setText(str);
		updateToolTip();
	}

	void updateToolTip()
	{
		setToolTip("<b>" + mainText->fullText() + "</b><br/>" + subText->fullText());
	}

	void paintEvent(QPaintEvent* /*e*/) override
	{
		QStylePainter p(this);
		QStyleOptionToolButton opt;
		initStyleOption(&opt);

		if (opt.state & QStyle::State_Sunken || opt.state & QStyle::State_MouseOver) {
#ifdef Q_OS_MAC
			QColor col = OSXStyle::self()->viewPalette().highlight().color();
#else
			QColor col = qApp->palette().highlight().color();
#endif
			col.setAlphaF(opt.state & QStyle::State_Sunken ? 0.5 : 0.2);
			p.fillRect(opt.rect, col);
		}
		opt.state &= ~(QStyle::State_Sunken | QStyle::State_MouseOver | QStyle::State_Raised | QStyle::State_On);
		opt.state |= QStyle::State_AutoRaise;
		p.drawComplexControl(QStyle::CC_ToolButton, opt);
	}
private Q_SLOTS:
	void iconsUpdated()
	{
		double dpr = DEVICE_PIXEL_RATIO();
		QPixmap pix = Icon::getScaledPixmap(icn, icon->width() * dpr, icon->height() * dpr, 96 * dpr);
		pix.setDevicePixelRatio(dpr);
		icon->setPixmap(pix);
	}

private:
	SqueezedTextLabel* mainText;
	SqueezedTextLabel* subText;
	QLabel* icon;
	QIcon icn;
};

MultiPageWidget::MultiPageWidget(QWidget* p)
	: StackedPageWidget(p)
{
	mainPage = new QWidget(this);
	QVBoxLayout* mainLayout = new QVBoxLayout(mainPage);
	infoLabel = new QLabel(mainPage);
	QWidget* sizer = new QWidget(mainPage);
	Application::fixSize(sizer);
	QScrollArea* scroll = new QScrollArea(this);
	view = new QWidget(scroll);
	QVBoxLayout* layout = new QVBoxLayout(view);
	scroll->setWidget(view);
	scroll->setWidgetResizable(true);
	view->setBackgroundRole(QPalette::Base);

	layout->addSpacerItem(new QSpacerItem(0, 0, QSizePolicy::Fixed, QSizePolicy::MinimumExpanding));
	view->setLayout(layout);
	scroll->setProperty(ProxyStyle::constModifyFrameProp, ProxyStyle::VF_Side | ProxyStyle::VF_Top);
	mainPage->setLayout(mainLayout);
	mainLayout->addWidget(scroll);
	mainLayout->setContentsMargins(0, 0, 0, 0);
	mainLayout->setSpacing(0);
	mainLayout->addWidget(infoLabel);
	infoLabel->setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::MinimumExpanding);
	mainLayout->addWidget(sizer);
	layout->setSpacing(0);
	layout->setSizeConstraint(QLayout::SetMinimumSize);
	int left;
	int top;
	int right;
	int bottom;
	layout->getContentsMargins(&left, &top, &right, &bottom);
	layout->setContentsMargins(qMin(4, left), qMin(4, top), qMin(4, right), qMin(4, bottom));
#ifdef Q_OS_MAC
	// TODO: This feels a bt of a hack...
	mainPage->setContentsMargins(-3, 0, -3, 0);
#endif
	addWidget(mainPage);
}

MultiPageWidget::~MultiPageWidget()
{
}

static const char* constCurrentPageKey = "currentPage";

void MultiPageWidget::load(Configuration& config)
{
	QString p = config.get(constCurrentPageKey, QString());

	if (!p.isEmpty()) {
		QMap<QString, Entry>::ConstIterator it = entries.find(p);
		if (it != entries.constEnd()) {
			setCurrentWidget(it.value().page);
		}
	}
}

void MultiPageWidget::save(Configuration& config) const
{
	QString p;
	QWidget* cw = currentWidget();

	QMap<QString, Entry>::ConstIterator it = entries.constBegin();
	QMap<QString, Entry>::ConstIterator end = entries.constEnd();

	for (; it != end; ++it) {
		if (it.value().page == cw) {
			p = it.key();
			break;
		}
	}
	config.set(constCurrentPageKey, p);
}
void MultiPageWidget::setInfoText(const QString& text)
{
	infoLabel->setText(text);
}

void MultiPageWidget::addPage(const QString& name, const QString& icon, const QString& text, const QString& subText, QWidget* widget)
{
	QIcon i;
	i.addFile(":" + icon);
	addPage(name, i, text, subText, widget);
}

void MultiPageWidget::addPage(const QString& name, const QIcon& icon, const QString& text, const QString& subText, QWidget* widget)
{
	if (entries.contains(name)) {
		return;
	}
	Entry e(new SelectorButton(text, subText, icon, view), widget);
	static_cast<QVBoxLayout*>(view->layout())->insertWidget(view->layout()->count() - 1, e.btn);
	addWidget(widget);
	entries.insert(name, e);
	connect(e.btn, SIGNAL(clicked()), SLOT(setPage()));
	infoLabel->setVisible(false);
	if (qobject_cast<SinglePageWidget*>(widget)) {
		connect(static_cast<SinglePageWidget*>(widget), SIGNAL(close()), this, SLOT(showMainView()));
	}
	else if (qobject_cast<StackedPageWidget*>(widget)) {
		connect(static_cast<StackedPageWidget*>(widget), SIGNAL(close()), this, SLOT(showMainView()));
	}
}

void MultiPageWidget::removePage(const QString& name)
{
	QMap<QString, Entry>::Iterator it = entries.find(name);
	if (it == entries.end()) {
		return;
	}
	if (it.value().page == currentWidget()) {
		setCurrentWidget(mainPage);
	}
	static_cast<QVBoxLayout*>(view->layout())->removeWidget(it.value().btn);
	it.value().btn->deleteLater();
	entries.erase(it);
}

void MultiPageWidget::updatePageSubText(const QString& name, const QString& text)
{
	QMap<QString, Entry>::Iterator it = entries.find(name);
	if (it == entries.end()) {
		return;
	}
	it.value().btn->setSubText(text);
}

void MultiPageWidget::showMainView()
{
	setCurrentWidget(mainPage);
}

void MultiPageWidget::setPage()
{
	QToolButton* btn = qobject_cast<QToolButton*>(sender());

	if (!btn) {
		return;
	}

	QMap<QString, Entry>::ConstIterator it = entries.constBegin();
	QMap<QString, Entry>::ConstIterator end = entries.constEnd();

	for (; it != end; ++it) {
		if (it.value().btn == btn) {
			setCurrentWidget(it.value().page);
			return;
		}
	}
}

void MultiPageWidget::sortItems()
{
	QList<QString> keys = entries.keys();
	std::sort(keys.begin(), keys.end());
	infoLabel->setVisible(0 == entries.count());
	QVBoxLayout* layout = static_cast<QVBoxLayout*>(view->layout());

	QMap<QString, Entry>::ConstIterator it = entries.constBegin();
	QMap<QString, Entry>::ConstIterator end = entries.constEnd();

	for (; it != end; ++it) {
		layout->removeWidget(it.value().btn);
	}

	for (const QString& key : keys) {
		layout->insertWidget(view->layout()->count() - 1, entries[key].btn);
	}
}

#include "moc_multipagewidget.cpp"