#include "toonzqt/tmessageviewer.h"
#include "toonzqt/dvdialog.h"
#include <QVBoxLayout>
#include <QPushButton>
#include <QListView>
#include <QStandardItemModel>
#include <QSortFilterProxyModel>
#include <QCheckBox>
#include <QLabel>

namespace
{

QIcon gRedIcon;
QIcon gGreenIcon;
QIcon gYellowIcon;

//-------------------------------------------------------------------------

class MyQListView : public QListView
{
public:
	MyQListView(QWidget *parent)
		: QListView(parent) {}

protected:
	void rowsInserted(const QModelIndex &parent, int start, int end)
	{
		QListView::rowsInserted(parent, start, end);
		scrollToBottom();
	}
};

//----------------------------------------------------------------------------------

} //namespace

class MySortFilterProxyModel : public QSortFilterProxyModel
{
public:
	MySortFilterProxyModel(QObject *parent)
		: QSortFilterProxyModel(parent) {}

	bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
	{
		TMessageViewer *v = dynamic_cast<TMessageViewer *>(parent());
		assert(v);

		QIcon ic = ((QStandardItemModel *)sourceModel())->item(sourceRow)->icon();
		if (ic.cacheKey() == gGreenIcon.cacheKey())
			return v->m_greenCheck->isChecked();
		else if (ic.cacheKey() == gYellowIcon.cacheKey())
			return v->m_yellowCheck->isChecked();
		else if (ic.cacheKey() == gRedIcon.cacheKey())
			return v->m_redCheck->isChecked();
		else
			return true;
	}
};

//----------------------------------------------------------------------------------

void TMessageViewer::onClicked(bool)
{
	TMessageRepository::instance()->clear();

	/*return;
	static int count=0;
	count++;
	DVGui::MsgBox((count%3)==1?INFORMATION:(count%3)==2?WARNING:CRITICAL, "messaggio "+QString::number(count));*/
}

//-----------------------------------------------------------

void TMessageViewer::refreshFilter(int)
{
	m_proxyModel->invalidate();
}

//---------------------------------------------------

void TMessageRepository::clear()
{
	m_sim->clear();
}

//---------------------------------------------------

TMessageRepository::TMessageRepository()
	: m_sim(new QStandardItemModel())
{
}

//---------------------------------------------------

TMessageRepository *TMessageRepository::instance()
{
	static TMessageRepository *theObject = 0;
	if (theObject == 0) {
		theObject = new TMessageRepository();
		bool ret = connect(TMsgCore::instance(), SIGNAL(sendMessage(int, const QString &)), theObject, SLOT(messageReceived(int, const QString &)));
		assert(ret);
		//TMsgCore::instance()->getConnectionName();
	}
	return theObject;
}

//---------------------------------------------------

void TMessageRepository::messageReceived(int type, const QString &message)
{
	if (gRedIcon.isNull()) {
		gRedIcon = QIcon(":Resources/tmsg_error.png");
		gYellowIcon = QIcon(":Resources/tmsg_warning.png");
		gGreenIcon = QIcon(":Resources/tmsg_info.png");
	}

	switch (type) {
	case INFORMATION:
		m_sim->appendRow(new QStandardItem(gGreenIcon, message));
		CASE WARNING : m_sim->appendRow(new QStandardItem(gYellowIcon, message));
		CASE CRITICAL : m_sim->appendRow(new QStandardItem(gRedIcon, message));
	DEFAULT:;
	}

	if (type == CRITICAL || (type == WARNING && !TMessageViewer::isTMsgVisible()))
		MsgBoxInPopup(MsgType(type), message);
}

//---------------------------------------------------------------------------------

vector<TMessageViewer *> TMessageViewer::m_tmsgViewers;

bool TMessageViewer::isTMsgVisible()
{
	for (int i = 0; i < (int)m_tmsgViewers.size(); i++)
		if (m_tmsgViewers[i]->isVisible())
			return true;
	return false;
}

//--------------------------------------------------

TMessageViewer::TMessageViewer(QWidget *parent)
	: QFrame(parent)
{
	m_tmsgViewers.push_back(this);

	bool ret = true;
	setFrameStyle(QFrame::StyledPanel);
	setObjectName("OnePixelMarginFrame");
	QVBoxLayout *vLayout = new QVBoxLayout();
	vLayout->setMargin(0);
	QFrame *fr = new QFrame();

	QHBoxLayout *hLayout = new QHBoxLayout();
	hLayout->setMargin(0);
	fr->setLayout(hLayout);
	fr->setFixedHeight(24);
	fr->setStyleSheet("background-color: rgb(210,210,210);");
	hLayout->addSpacing(4);

	hLayout->addWidget(new QLabel("Display:  "));

	m_redCheck = new QCheckBox(tr("Errors"));
	m_redCheck->setChecked(true);
	ret = ret && connect(m_redCheck, SIGNAL(stateChanged(int)), SLOT(refreshFilter(int)));
	hLayout->addWidget(m_redCheck);

	m_yellowCheck = new QCheckBox(tr("Warnings"));
	m_yellowCheck->setChecked(true);
	ret = ret && connect(m_yellowCheck, SIGNAL(stateChanged(int)), SLOT(refreshFilter(int)));
	hLayout->addWidget(m_yellowCheck);

	m_greenCheck = new QCheckBox(tr("Infos"));
	m_greenCheck->setChecked(true);
	ret = ret && connect(m_greenCheck, SIGNAL(stateChanged(int)), SLOT(refreshFilter(int)));
	hLayout->addWidget(m_greenCheck);

	hLayout->addStretch();

	QPushButton *pb = new QPushButton(tr(" Clear "));
	ret = ret && connect(pb, SIGNAL(clicked(bool)), SLOT(onClicked(bool)));
	hLayout->addWidget(pb);

	hLayout->addSpacing(4);

	vLayout->addWidget(fr);

	MyQListView *lv = new MyQListView(this);
	lv->setAlternatingRowColors(true);
	lv->setEditTriggers(QAbstractItemView::NoEditTriggers);
	lv->setAutoScroll(true);

	m_proxyModel = new MySortFilterProxyModel(this);
	m_proxyModel->setDynamicSortFilter(true);

	m_proxyModel->setSourceModel(TMessageRepository::instance()->getModel());
	lv->setModel(m_proxyModel);

	vLayout->addWidget(lv);
	setLayout(vLayout);

	assert(ret);
}