/*  MergeView.cpp
 *
 *  Copyright (C) 2016  Jim Evins <evins@snaught.com>
 *
 *  This file is part of gLabels-qt.
 *
 *  gLabels-qt 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 3 of the License, or
 *  (at your option) any later version.
 *
 *  gLabels-qt 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 gLabels-qt.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "MergeView.h"

#include "LabelModel.h"
#include "Merge/Factory.h"
#include <QFileDialog>
#include <QFileInfo>
#include <QtDebug>


///
/// Constructor
///
MergeView::MergeView( QWidget *parent )
	: QWidget(parent), mModel(0), mBlock(false)
{
	setupUi( this );

	mMergeFormatNames = merge::Factory::nameList();
	formatCombo->addItems( mMergeFormatNames );
}


///
/// Destructor
///
MergeView::~MergeView()
{
}


///
/// Set Model
///
void MergeView::setModel( LabelModel* model, UndoRedoModel* undoRedoModel )
{
	mModel = model;
	mUndoRedoModel = undoRedoModel;

	// Initialize CWD
	if ( model->fileName().isEmpty() )
	{
		mCwd = ".";
	}
	else
	{
		mCwd = QFileInfo( model->fileName() ).absolutePath();
	}

	onMergeChanged();
	connect( mModel, SIGNAL(mergeChanged()), this, SLOT(onMergeChanged()) );
}


///
/// Merge changed handler
///
void MergeView::onMergeChanged()
{
	int index = mMergeFormatNames.indexOf( merge::Factory::idToName( mModel->merge()->id() ) );
	mOldFormatComboIndex = index;
	formatCombo->setCurrentIndex( index );

	switch ( merge::Factory::idToType( mModel->merge()->id() ) )
	{
	case merge::Factory::NONE:
	case merge::Factory::FIXED:
		locationLabel->setEnabled( false );
		locationButton->setEnabled( false );
		locationButton->setText( "" );
		break;

	case merge::Factory::FILE:
		locationLabel->setEnabled( true );
		locationButton->setEnabled( true );
		if ( mModel->merge()->source().isEmpty() )
		{
			locationButton->setText( "Select file..." );
		}
		else
		{
			locationButton->setText( mModel->merge()->source() );
		}
		break;

	default:
		qWarning( "MergeView::onMergeChanged()::Should not be reached!" );
		break;
	}

	recordsTable->clear();
	recordsTable->setColumnCount( 0 );
	loadHeaders( mModel->merge() );
	loadTable( mModel->merge() );

	connect( mModel->merge(), SIGNAL(sourceChanged()), this, SLOT(onMergeSourceChanged()) );
	connect( mModel->merge(), SIGNAL(selectionChanged()), this, SLOT(onMergeSelectionChanged()) );

	connect( recordsTable, SIGNAL(cellChanged(int,int)), this, SLOT(onCellChanged(int,int)) );
}


///
/// Merge source changed handler
///
void MergeView::onMergeSourceChanged()
{
	locationButton->setText( mModel->merge()->source() );

	recordsTable->clear();
	recordsTable->setColumnCount( 0 );
	loadHeaders( mModel->merge() );
	loadTable( mModel->merge() );
}


///
/// Merge selection changed handler
///
void MergeView::onMergeSelectionChanged()
{
	mBlock = true;  // Don't recurse
	
	const QList<merge::Record*>& records = mModel->merge()->recordList();

	int iRow = 0;
	foreach ( merge::Record* record, records )
	{
		QTableWidgetItem* item = recordsTable->item( iRow, 0 );
		item->setCheckState( record->isSelected() ? Qt::Checked : Qt::Unchecked );

		iRow++;
	}

	mBlock = false;
}


///
/// Format combo changed handler
void MergeView::onFormatComboActivated()
{
	int index = formatCombo->currentIndex();
	if ( index != mOldFormatComboIndex )
	{
		mOldFormatComboIndex = index;

		mModel->setMerge( merge::Factory::createMerge( merge::Factory::indexToId(index) ) );
	}
}


///
/// Location button clicked handler
///
void MergeView::onLocationButtonClicked()
{
	QString fileName =
                QFileDialog::getOpenFileName( this,
                                              tr("Select merge file"),
                                              mCwd,
                                              tr("All files (*)") );
        if ( !fileName.isEmpty() )
        {
		mModel->merge()->setSource( fileName );
		mCwd = QFileInfo( fileName ).absolutePath(); // Update CWD
        }
}


///
/// Select all button clicked handler
///
void MergeView::onSelectAllButtonClicked()
{
	mModel->merge()->selectAll();
}


///
/// Unselect all button clicked handler
///
void MergeView::onUnselectAllButtonClicked()
{
	mModel->merge()->unselectAll();
}


///
/// Cell changed handler
///
void MergeView::onCellChanged( int iRow, int iCol )
{
	if ( !mBlock )
	{
		QTableWidgetItem* item = recordsTable->item( iRow, 0 );
		bool state = (item->checkState() == Qt::Unchecked) ? false : true;
		
		mModel->merge()->setSelected( iRow, state );
	}
}


///
/// Load headers
///
void MergeView::loadHeaders( merge::Merge* merge )
{
	mPrimaryKey = merge->primaryKey();
	mKeys = merge->keys();

	if ( mKeys.size() > 0 )
	{
		recordsTable->setColumnCount( mKeys.size() + 1 );  // Include extra column

		// First column = primay Key
		QTableWidgetItem* item = new QTableWidgetItem( mPrimaryKey );
		item->setFlags( Qt::ItemIsEnabled );
		recordsTable->setHorizontalHeaderItem( 0, item );

		// Starting on second column, one column per key, skip primary Key
		int iCol = 1;
		foreach ( QString key, mKeys )
		{
			if ( key != mPrimaryKey )
			{
				QTableWidgetItem* item = new QTableWidgetItem( key );
				item->setFlags( Qt::ItemIsEnabled );
				recordsTable->setHorizontalHeaderItem( iCol, item );

				iCol++;
			}

		}

		// Extra dummy column to fill any extra horizontal space
		QTableWidgetItem* fillItem = new QTableWidgetItem();
		fillItem->setFlags( Qt::NoItemFlags );
		recordsTable->setHorizontalHeaderItem( iCol, fillItem );
		recordsTable->horizontalHeader()->setStretchLastSection( true );
	}
}


///
/// Load table
///
void MergeView::loadTable( merge::Merge* merge )
{
	mBlock = true;

	const QList<merge::Record*>& records = merge->recordList();
	recordsTable->setRowCount( records.size() );

	int iRow = 0;
	foreach ( merge::Record* record, records )
	{
		// First column for primay field
		QTableWidgetItem* item = new QTableWidgetItem();
		if ( record->contains( mPrimaryKey ) )
		{
			item->setText( (*record)[mPrimaryKey] );
		}
		item->setFlags( Qt::ItemIsEnabled | Qt::ItemIsUserCheckable );
		item->setCheckState( record->isSelected() ? Qt::Checked : Qt::Unchecked );
		recordsTable->setItem( iRow, 0, item );
		recordsTable->resizeColumnToContents( 0 );

		// Starting on second column, one column per field (even if empty), skip primary field
		int iCol = 1;
		foreach ( QString key, mKeys )
		{
			if ( key != mPrimaryKey )
			{
				if ( record->contains( key ) )
				{
					QTableWidgetItem* item = new QTableWidgetItem( (*record)[key] );
					item->setFlags( Qt::ItemIsEnabled );
					recordsTable->setItem( iRow, iCol, item );
					recordsTable->resizeColumnToContents( iCol );
				}

				iCol++;
			}
		}

		// Extra dummy column to fill any extra horizontal space
		QTableWidgetItem* fillItem = new QTableWidgetItem();
		fillItem->setFlags( Qt::NoItemFlags );
		recordsTable->setItem( iRow, iCol, fillItem );

		iRow++;
	}

	mBlock = false;
}