#include "cpartwindow.h"
#include "ui_cpartwindow.h"

#include "cparteditdialog.h"

#include "common.h"

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

#include <QMenu>

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

#include <QMessageBox>
#include <QDebug>


cPartWindow::cPartWindow(QWidget *parent) :
	QWidget(parent),
	ui(new Ui::cPartWindow),
	m_lpPartGroupList(0),
	m_lpPartList(0),
	m_lpDistributorList(0),
	m_lpPartDistributorList(0)
{
	ui->setupUi(this);

	m_lpPartListModel	= new QStandardItemModel(0, 2);
	ui->m_lpPartList->setModel(m_lpPartListModel);
}

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

void cPartWindow::setList(cPartGroupList* lpPartGroupList, cPartList* lpPartList, cDistributorList *lpDistributorList, cPartDistributorList *lpPartDistributorList)
{
	m_lpPartGroupList		= lpPartGroupList;
	m_lpPartList			= lpPartList;
	m_lpDistributorList		= lpDistributorList;
	m_lpPartDistributorList	= lpPartDistributorList;

	showPartList();
}

void cPartWindow::showPartList(qint32 id)
{
	m_lpPartListModel->clear();

	QStringList	header;
	header << tr("Name") << tr("description") << tr("link");

	m_lpPartListModel->setHorizontalHeaderLabels(header);

	qint32			iOldPartGroupID	= -1;
	QStandardItem*	lpGroupItem		= 0;
	QStandardItem*	lpSelected		= 0;

	for(int x = 0;x < m_lpPartList->count();x++)
	{
		QList<QStandardItem*>	lpItems;
		cPart*					lpPart		= m_lpPartList->at(x);
		cPartGroup*				lpPartGroup	= lpPart->partGroup();

		if(lpPartGroup->id() != iOldPartGroupID)
		{
			iOldPartGroupID	= lpPartGroup->id();
			lpGroupItem	= new QStandardItem(lpPartGroup->name());
			m_lpPartListModel->appendRow(lpGroupItem);
		}

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

		lpItems.at(0)->setText(lpPart->name());
		lpItems.at(1)->setText(lpPart->description());
		lpItems.at(2)->setText(lpPart->link());

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

		lpGroupItem->appendRow(lpItems);

		if(id != -1 && lpPart->id() == id)
			lpSelected	= lpItems.at(0);
	}

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

	ui->m_lpPartList->expandAll();

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

	if(lpSelected)
	{
		ui->m_lpPartList->setCurrentIndex(lpSelected->index());
		ui->m_lpPartList->scrollTo(lpSelected->index());
	}
}


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

bool cPartWindow::groupSelected()
{
	QStandardItem*	lpItem	= m_lpPartListModel->itemFromIndex(ui->m_lpPartList->selectionModel()->currentIndex());
	if(!lpItem)
		return(false);

	if(lpItem->parent())
		return(false);
	return(true);
}

void cPartWindow::on_m_lpPartList_clicked(const QModelIndex &index)
{
	selectionChanged(index);
}

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

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

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

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

void cPartWindow::addPart()
{
	onAdd();
}

void cPartWindow::editPart()
{
	onEdit();
}

void cPartWindow::deletePart()
{
	onDelete();
}

void cPartWindow::onAdd()
{
	QString				szGroup;

	if(ui->m_lpPartList->selectionModel()->selectedIndexes().count())
	{
		QStandardItem*		lpItem			= m_lpPartListModel->itemFromIndex(ui->m_lpPartList->selectionModel()->selectedIndexes().at(0));

		if(lpItem)
		{
			QStandardItem*	lpRoot;

			lpRoot	= lpItem->parent();
			if(!lpRoot)
				szGroup	= lpItem->text();
			else
				szGroup	= lpRoot->text();
		}
	}

	cPartEditDialog*	lpDialog	= new cPartEditDialog(this);
	lpDialog->setValues(0, m_lpPartGroupList, m_lpDistributorList, m_lpPartDistributorList, szGroup);

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

	qint32	id	= lpDialog->id();

	delete lpDialog;

	emit partGroupChanged(0);
	emit partChanged(0);
	showPartList(id);
}

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

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

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

	cPartEditDialog*	lpDialog		= new cPartEditDialog(this);
	lpDialog->setValues(lpPart, m_lpPartGroupList, m_lpDistributorList, m_lpPartDistributorList);

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

	qint32	id	= lpPart->id();

	delete lpDialog;

	partChanged(0);
	showPartList(id);
}

void cPartWindow::onDelete()
{
	for(int x = 0;x < ui->m_lpPartList->selectionModel()->selectedIndexes().count();x++)
	{
		if(ui->m_lpPartList->selectionModel()->selectedIndexes().at(x).column())
			continue;

		QStandardItem*	lpItem			= m_lpPartListModel->itemFromIndex(ui->m_lpPartList->selectionModel()->selectedIndexes().at(x));
		if(!lpItem)
			continue;

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

		QSqlQuery	query;

		query.prepare("SELECT * FROM partlistitem WHERE partID = :partID;");
		query.bindValue(":partID", lpPart->id());
		if(!query.exec())
		{
			myDebug << query.lastError().text();
			continue;
		}

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

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

		query.prepare("DELETE FROM part_distributor WHERE partID = :partID;");
		query.bindValue(":partID", lpPart->id());
		if(!query.exec())
		{
			myDebug << query.lastError().text();
			continue;
		}

		query.prepare("DELETE FROM part WHERE id=:partID;");
		query.bindValue(":partID", lpPart->id());
		if(!query.exec())
		{
			myDebug << query.lastError().text();
			continue;
		}
	}

	partChanged(0);
	showPartList();
}

void cPartWindow::on_m_lpPartList_doubleClicked(const QModelIndex &/*index*/)
{
	onEdit();
}

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

void cPartWindow::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 cPartWindow::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				dGroup			= 0;
	qreal				dName			= 0;
	qreal				dDescription	= 0;
	qreal				dLink			= 0;
	qreal				dDistributor	= 0;
	qreal				dPartName		= 0;
	qreal				dPrice			= 0;

	qint32				iLine	= 1;
	QXlsx::Format		format;
	QXlsx::Format		formatMerged;
	QXlsx::Format		formatCurrency;
	QString				szNumberFormat("_-\"€\"\\ * #,##0.00_-;\\-\"€\"\\ * #,##0.00_-;_-\"€\"\\ * \"-\"??_-;_-@_-");

	format.setFontBold(true);

	formatMerged.setHorizontalAlignment(QXlsx::Format::AlignLeft);
	formatMerged.setVerticalAlignment(QXlsx::Format::AlignTop);

	formatCurrency.setNumberFormat(szNumberFormat);

	xlsx.write(iLine,  1, tr("Group"), format);
	xlsx.write(iLine,  2, tr("Name"), format);
	xlsx.write(iLine,  3, tr("Description"), format);
	xlsx.write(iLine,  4, tr("Link"), format);
	xlsx.write(iLine,  5, tr("Distributor"), format);
	xlsx.write(iLine,  6, tr("Part Name"), format);
	xlsx.write(iLine,  7, tr("Price"), format);

	metrics(fontBold, dGroup, iHeight, tr("Group"));
	metrics(fontBold, dName, iHeight, tr("Name"));
	metrics(fontBold, dDescription, iHeight, tr("Description"));
	metrics(fontBold, dLink, iHeight, tr("Link"));
	metrics(fontBold, dDistributor, iHeight, tr("Distributor"));
	metrics(fontBold, dPartName, iHeight, tr("Part Name"));
	metrics(fontBold, dPrice, iHeight, tr("Price"));
	iLine++;

	for(int x = 0;x < m_lpPartList->count();x++)
	{
		cPart*	lpPart	= m_lpPartList->at(x);

		xlsx.write(iLine,  1, lpPart->partGroup()->name());
		xlsx.write(iLine,  2, lpPart->name());
		xlsx.write(iLine,  3, lpPart->description());
		xlsx.write(iLine,  4, lpPart->link());

		metrics(font, dGroup, iHeight, lpPart->partGroup()->name());
		metrics(font, dName, iHeight, lpPart->name());
		metrics(font, dDescription, iHeight, lpPart->description());
		metrics(font, dLink, iHeight, lpPart->link());

		qint32	iLineTemp	= 0;
		for(int y = 0;y < m_lpPartDistributorList->count();y++)
		{
			cPartDistributor*	lpPartDistributor	= m_lpPartDistributorList->at(y);

			if(lpPartDistributor->part() == lpPart)
			{
				xlsx.write(iLine+iLineTemp, 5, lpPartDistributor->distributor()->name());
				xlsx.write(iLine+iLineTemp, 6, lpPartDistributor->name());
				xlsx.write(iLine+iLineTemp, 7, lpPartDistributor->price(), formatCurrency);

				metrics(font, dDistributor, iHeight, lpPartDistributor->distributor()->name());
				metrics(font, dPartName, iHeight, lpPartDistributor->name());
				metrics(font, dPrice, iHeight, QString("€ ") + QString::number(lpPartDistributor->price(), 'f', 2));

				iLineTemp++;
			}
		}
		if(iLineTemp)
		{
			xlsx.mergeCells(QString("A%1:A%2").arg(iLine).arg(iLine+iLineTemp-1), formatMerged);
			xlsx.mergeCells(QString("B%1:B%2").arg(iLine).arg(iLine+iLineTemp-1), formatMerged);
			xlsx.mergeCells(QString("C%1:C%2").arg(iLine).arg(iLine+iLineTemp-1), formatMerged);
			xlsx.mergeCells(QString("D%1:D%2").arg(iLine).arg(iLine+iLineTemp-1), formatMerged);
			iLine += iLineTemp;
		}
		else
			iLine++;
	}

	xlsx.setColumnWidth( 1, dGroup*1.2);
	xlsx.setColumnWidth( 2, dName*1.2);
	xlsx.setColumnWidth( 3, dDescription*1.2);
	xlsx.setColumnWidth( 4, dLink*1.2);
	xlsx.setColumnWidth( 5, dDistributor*1.2);
	xlsx.setColumnWidth( 6, dPartName*1.2);
	xlsx.setColumnWidth( 7, dPrice*1.2);

	xlsx.saveAs(szFileName);
}

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

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

	QTextStream out(&file);

	out << "\"" << tr("Group") << "\";\"" << tr("Name") << "\";\"" << tr("Description") << "\";\"" << tr("Link") << "\"\n";

	for(int x = 0;x < m_lpPartList->count();x++)
	{
		cPart*	lpPart	= m_lpPartList->at(x);
		out << "\"" << lpPart->partGroup()->name() << "\";\"" << lpPart->name() << "\";\"" << lpPart->description() << "\";\"" << lpPart->link() << "\";\"" << "\"\n";
	}

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

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

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