#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 DVGui::INFORMATION:

		m_sim->appendRow(new QStandardItem(gGreenIcon, message));

		break;

	case DVGui::WARNING:

		m_sim->appendRow(new QStandardItem(gYellowIcon, message));

		break;

	case DVGui::CRITICAL:

		m_sim->appendRow(new QStandardItem(gRedIcon, message));

		break;

	}

	if (type == DVGui::CRITICAL || (type == DVGui::WARNING && !TMessageViewer::isTMsgVisible()))

		DVGui::MsgBoxInPopup(DVGui::MsgType(type), message);

}

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

std::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);

}