#include "cdistributorwindow.h"
#include "ui_cdistributorwindow.h"

#include "cdistributoreditdialog.h"

#include "common.h"

#include <QString>
#include <QSqlQuery>
#include <QSqlError>

#include <QMenu>

#include <QFileInfo>
#include <QFile>
#include <QTextStream>

#include <QFontMetrics>

#include <QMessageBox>
#include <QDebug>


cDistributorWindow::cDistributorWindow(QWidget *parent) :
	QWidget(parent),
	ui(new Ui::cDistributorWindow)
{
	ui->setupUi(this);

	m_lpDistributorListModel	= new QStandardItemModel(0, 3);
	ui->m_lpDistributorList->setModel(m_lpDistributorListModel);
}

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

void cDistributorWindow::setList(cDistributorList* lpDistributorList)
{
	m_lpDistributorList	= lpDistributorList;

	showDistributorList();
}

void cDistributorWindow::showDistributorList()
{
	m_lpDistributorListModel->clear();

	QStringList	header;
	header << tr("Name") << tr("Phone") << tr("Fax") << tr("Email") << tr("address") << tr("postal") << tr("city") << tr("country") << tr("link") << tr("description");
	m_lpDistributorListModel->setHorizontalHeaderLabels(header);

	for(int x = 0;x < m_lpDistributorList->count();x++)
	{
		QList<QStandardItem*>	lpItems;
		cDistributor*			lpDistributor	= m_lpDistributorList->at(x);

		for(int z = 0;z < header.count();z++)
			lpItems.append(new QStandardItem);

		lpItems.at(0)->setText(lpDistributor->name());
		lpItems.at(1)->setText(lpDistributor->phone());
		lpItems.at(2)->setText(lpDistributor->fax());
		lpItems.at(3)->setText(lpDistributor->eMail());
		lpItems.at(4)->setText(lpDistributor->address());
		if(lpDistributor->postalCode())
			lpItems.at(5)->setText(QString("%1").arg(lpDistributor->postalCode()));
		lpItems.at(6)->setText(lpDistributor->city());
		lpItems.at(7)->setText(lpDistributor->country());
		lpItems.at(8)->setText(lpDistributor->link());
		lpItems.at(9)->setText(lpDistributor->description());

		for(int z = 0;z < header.count();z++)
			lpItems.at(z)->setData(QVariant::fromValue(lpDistributor), Qt::UserRole);

		m_lpDistributorListModel->appendRow(lpItems);
	}

	for(int z = 0;z < header.count();z++)
		ui->m_lpDistributorList->resizeColumnToContents(z);
}

bool cDistributorWindow::somethingSelected()
{
	if(ui->m_lpDistributorList->selectionModel()->selectedRows().count())
		return(true);
	return(false);
}

void cDistributorWindow::on_m_lpDistributorList_clicked(const QModelIndex& index)
{
	selectionChanged(index);
}

void cDistributorWindow::on_m_lpDistributorList_customContextMenuRequested(const QPoint &pos)
{
	QMenu*	lpMenu	= new QMenu(this);

	lpMenu->addAction("add", this, SLOT(onAdd()));

	if(somethingSelected())
	{
		lpMenu->addAction("edit", this, SLOT(onEdit()));
		lpMenu->addAction("delete", this, SLOT(onDelete()));
	}

	lpMenu->popup(ui->m_lpDistributorList->viewport()->mapToGlobal(pos));
}

void cDistributorWindow::addDistributor()
{
	onAdd();
}

void cDistributorWindow::editDistributor()
{
	onEdit();
}

void cDistributorWindow::deleteDistributor()
{
	onDelete();
}

void cDistributorWindow::onAdd()
{
	cDistributorEditDialog*	lpDialog	= new cDistributorEditDialog(this);

	if(lpDialog->exec() == QDialog::Rejected)
	{
		delete lpDialog;
		return;
	}

	qint32	id	= lpDialog->id();

	delete lpDialog;

	distributorChanged(0);
	showDistributorList();

	for(int x = 0;x < m_lpDistributorListModel->rowCount();x++)
	{
		QStandardItem*	lpItem			= m_lpDistributorListModel->item(x, 0);
		if(!lpItem)
			continue;

		cDistributor*	lpDistributor	= qvariant_cast<cDistributor*>(lpItem->data(Qt::UserRole));
		if(!lpDistributor)
			continue;

		if(lpDistributor->id() == id)
		{
			ui->m_lpDistributorList->setCurrentIndex(lpItem->index());
			ui->m_lpDistributorList->scrollTo(lpItem->index());
			return;
		}
	}
}

void cDistributorWindow::onEdit()
{
	if(!ui->m_lpDistributorList->selectionModel()->selectedRows().count())
		return;

	QStandardItem*	lpItem			= m_lpDistributorListModel->itemFromIndex(ui->m_lpDistributorList->selectionModel()->selectedIndexes().at(0));
	if(!lpItem)
		return;

	cDistributor*	lpDistributor	= qvariant_cast<cDistributor*>(lpItem->data(Qt::UserRole));
	if(!lpDistributor)
		return;

	cDistributorEditDialog*	lpDialog	= new cDistributorEditDialog(this);
	lpDialog->setValues(lpDistributor);

	if(lpDialog->exec() == QDialog::Rejected)
	{
		delete lpDialog;
		return;
	}

	qint32	id	= lpDialog->id();

	delete lpDialog;

	distributorChanged(0);
	showDistributorList();

	for(int x = 0;x < m_lpDistributorListModel->rowCount();x++)
	{
		QStandardItem*	lpItem			= m_lpDistributorListModel->item(x, 0);
		if(!lpItem)
			continue;

		cDistributor*	lpDistributor	= qvariant_cast<cDistributor*>(lpItem->data(Qt::UserRole));
		if(!lpDistributor)
			continue;

		if(lpDistributor->id() == id)
		{
			ui->m_lpDistributorList->setCurrentIndex(lpItem->index());
			ui->m_lpDistributorList->scrollTo(lpItem->index());
			return;
		}
	}
}

void cDistributorWindow::onDelete()
{
	for(int x = 0;x < ui->m_lpDistributorList->selectionModel()->selectedRows().count();x++)
	{
		QStandardItem*	lpItem			= m_lpDistributorListModel->item(ui->m_lpDistributorList->selectionModel()->selectedRows().at(x).row(), 0);
		if(!lpItem)
			continue;

		cDistributor*	lpDistributor	= qvariant_cast<cDistributor*>(lpItem->data(Qt::UserRole));
		if(!lpDistributor)
			continue;

		QSqlQuery	query;
		QString		szQuery;

		szQuery		= QString("SELECT * FROM partlistitem WHERE distributorID = %1;").arg(lpDistributor->id());
		if(!query.exec(szQuery))
		{
			myDebug << query.lastError().text();
			continue;
		}

		if(query.next())
		{
			QMessageBox::critical(this, "ERROR", QString("Distributor '%1' still in use.\nUnable to delete.").arg(lpDistributor->name()));
			continue;
		}

		if(QMessageBox::question(this, "DELETE", QString("Are you sure to delete distributor '%1'?").arg(lpDistributor->name())) == QMessageBox::No)
			continue;

		szQuery	= QString("DELETE FROM distributor WHERE id=%1;").arg(lpDistributor->id());
		if(!query.exec(szQuery))
		{
			myDebug << query.lastError().text();
			continue;
		}
	}

	distributorChanged(0);
	showDistributorList();
}

void cDistributorWindow::on_m_lpDistributorList_doubleClicked(const QModelIndex &/*index*/)
{
	onEdit();
}

bool cDistributorWindow::canClose()
{
	return(true);
}

void cDistributorWindow::exportList(const QString& szFileName)
{
	QFileInfo	fileInfo(szFileName);
	QString		szType	= fileInfo.suffix();

	if(!szType.compare("xlsx", Qt::CaseInsensitive))
		writeXLSX(szFileName);
	else if(!szType.compare("csv", Qt::CaseInsensitive))
		writeText(szFileName);
	else if(!szType.compare("txt", Qt::CaseInsensitive))
		writeText(szFileName);
	else if(!szType.compare("xml", Qt::CaseInsensitive))
		writeXML(szFileName);
	else if(!szType.compare("pdf", Qt::CaseInsensitive))
		writePDF(szFileName);
}

void cDistributorWindow::writeXLSX(const QString& szFileName)
{
	QXlsx::Document		xlsx;
	QXlsx::Format		fmt;
	fmt.setFontBold(true);
	xlsx.write(1, 1, "empty");
	xlsx.write(2, 1, "empty", fmt);
	QFont				font		= xlsx.cellAt(1, 1)->format().font();
	QFont				fontBold	= xlsx.cellAt(2, 1)->format().font();
	fmt.setFontBold(false);
	xlsx.write(1, 1, "");
	xlsx.write(2, 1, "", fmt);

	qint16				iHeight			= 0;
	qreal				dName			= 0;
	qreal				dPhone			= 0;
	qreal				dFax			= 0;
	qreal				dEmail			= 0;
	qreal				dAddress		= 0;
	qreal				dPostal			= 0;
	qreal				dCity			= 0;
	qreal				dCountry		= 0;
	qreal				dLink			= 0;
	qreal				dDescription	= 0;

	writeXLSXLine(xlsx, 1, tr("Name"), tr("Phone"), tr("Fax"), tr("Email"), tr("address"), tr("postal"), tr("city"), tr("country"), tr("link"), tr("description"));

	metrics(fontBold, dName, iHeight, tr("Name"));
	metrics(fontBold, dPhone, iHeight, tr("Phone"));
	metrics(fontBold, dFax, iHeight, tr("Fax"));
	metrics(fontBold, dEmail, iHeight, tr("Email"));
	metrics(fontBold, dAddress, iHeight, tr("address"));
	metrics(fontBold, dPostal, iHeight, tr("postal"));
	metrics(fontBold, dCity, iHeight, tr("city"));
	metrics(fontBold, dCountry, iHeight, tr("country"));
	metrics(fontBold, dLink, iHeight, tr("link"));
	metrics(fontBold, dDescription, iHeight, tr("description"));
	xlsx.setRowHeight(1, iHeight);

	for(int x = 0;x < m_lpDistributorList->count();x++)
	{
		iHeight			= 0;

		cDistributor*	lpDistributor	= m_lpDistributorList->at(x);
		writeXLSXLine(xlsx, x+2, lpDistributor->name(), lpDistributor->phone(), lpDistributor->fax(), lpDistributor->eMail(), lpDistributor->address(), QString::number(lpDistributor->postalCode()), lpDistributor->city(), lpDistributor->country(), lpDistributor->link(), lpDistributor->description());

		metrics(font, dName, iHeight, lpDistributor->name());
		metrics(font, dPhone, iHeight, lpDistributor->phone());
		metrics(font, dFax, iHeight, lpDistributor->fax());
		metrics(font, dEmail, iHeight, lpDistributor->eMail());
		metrics(font, dAddress, iHeight, lpDistributor->address());
		metrics(font, dPostal, iHeight, QString::number(lpDistributor->postalCode()));
		metrics(font, dCity, iHeight, lpDistributor->city());
		metrics(font, dCountry, iHeight, lpDistributor->country());
		metrics(font, dLink, iHeight, lpDistributor->link());
		metrics(font, dDescription, iHeight, lpDistributor->description());

		xlsx.setRowHeight(x+2, iHeight);
	}

	xlsx.setColumnWidth( 1, dName*1.2);
	xlsx.setColumnWidth( 2, dPhone*1.2);
	xlsx.setColumnWidth( 3, dFax*1.2);
	xlsx.setColumnWidth( 4, dEmail*1.2);
	xlsx.setColumnWidth( 5, dAddress*1.2);
	xlsx.setColumnWidth( 6, dPostal*1.2);
	xlsx.setColumnWidth( 7, dCity*1.2);
	xlsx.setColumnWidth( 8, dCountry*1.2);
	xlsx.setColumnWidth( 9, dLink*1.2);
	xlsx.setColumnWidth(10, dDescription*1.2);

	xlsx.saveAs(szFileName);
}

void cDistributorWindow::writeXLSXLine(QXlsx::Document& xlsx, qint32 line, const QString& szName, const QString& szPhone, const QString& szFax, const QString& szEmail, const QString& szAddress, const QString& szPostal, const QString& szCity, const QString& szCountry, const QString& szLink, const QString& szDescription)
{
	QXlsx::Format	format;

	if(line == 1)
		format.setFontBold(true);

	xlsx.write(line,  1, szName, format);
	xlsx.write(line,  2, szPhone, format);
	xlsx.write(line,  3, szFax, format);
	xlsx.write(line,  4, szEmail, format);
	xlsx.write(line,  5, szAddress, format);
	xlsx.write(line,  6, szPostal, format);
	xlsx.write(line,  7, szCity, format);
	xlsx.write(line,  8, szCountry, format);
	xlsx.write(line,  9, szLink, format);
	xlsx.write(line, 10, szDescription, format);
}

void cDistributorWindow::writeText(const QString& szFileName)
{
	QFile	file(szFileName);

	if(!file.open(QFile::WriteOnly | QFile::Text))
		return;

	QTextStream out(&file);

	out << "\"" << tr("Name") << "\";\"" << tr("Phone") << "\";\"" << tr("Fax") << "\";\"" << tr("Email") << "\";\"" << tr("address") << "\";\"" << tr("postal") << "\";\"" << tr("city") << "\";\"" << tr("country") << "\";\"" << tr("link") << "\";\"" << tr("description") << "\"\n";

	for(int x = 0;x < m_lpDistributorList->count();x++)
	{
		cDistributor*	lpDistributor	= m_lpDistributorList->at(x);
		out << "\"" << lpDistributor->name() << "\";\"" << lpDistributor->phone() << "\";\"" << lpDistributor->fax() << "\";\"" << lpDistributor->eMail() << "\";\"" << lpDistributor->address() << "\";\"" << lpDistributor->postalCode() << "\";\"" << lpDistributor->city() << "\";\"" << lpDistributor->country() << "\";\"" << lpDistributor->link() << "\";\"" << lpDistributor->description() << "\"\n";
	}

	file.flush();
	file.close();
}

void cDistributorWindow::writeXML(const QString& szFileName)
{
}

void cDistributorWindow::writePDF(const QString& szFileName)
{
}