/***************************************************************************
 *   Copyright (C) 2005-2007 by Tarek Saidi                                *
 *   tarek.saidi@arcor.de                                                  *
 *                                                                         *
 *   This program 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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program 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 this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/


#include <QDragEnterEvent>
#include <QDragMoveEvent>
#include <QDragLeaveEvent>
#include <QDropEvent>
#include <QMouseEvent>
#include <QHeaderView>
#include <QTime>
#include <QApplication>
#include <QPainter>
#include <QPair>
#include "main.h"
#include "PwmConfig.h"
#include "EntryView.h"
#include "dialogs/EditEntryDlg.h"
#include "lib/AutoType.h"


// just for the lessThan funtion
QList<EntryViewItem*>* pItems;
KeepassEntryView* pEntryView;

KeepassEntryView::KeepassEntryView(QWidget* parent):QTreeWidget(parent){
	AutoResizeColumns=true;	
	header()->setResizeMode(QHeaderView::Interactive);
	header()->setStretchLastSection(false);
	header()->setClickable(true);
	header()->setCascadingSectionResizes(true);
	ColumnSizes.resize(NUM_COLUMNS);
	QStringList ColumnSizeConfig=settings->value("Ui/ColumnSizes",QStringList()<<"1"<<"1"<<"1"<<"1"<<"1"<<"1"<<"1"<<"1"<<"1"<<"1"<<"1").toStringList();
	for(int i=0;i<NUM_COLUMNS;i++)
		ColumnSizes[i]=(float)ColumnSizeConfig[i].toInt();	
	QStringList DefaultColumnConfig=QStringList()<<"1"<<"1"<<"1"<<"1"<<"1"<<"0"<<"0"<<"0"<<"0"<<"0"<<"1";
	QStringList ColumnConfig=settings->value("Ui/ShowColumns",DefaultColumnConfig).toStringList();
	Columns.resize(NUM_COLUMNS);
	if(ColumnConfig.size()<NUM_COLUMNS)ColumnConfig=DefaultColumnConfig;
	for(int i=0;i<ColumnConfig.size();i++){
		Columns[i]=(bool)ColumnConfig[i].toInt();	
	}
	ColumnOrder.resize(NUM_COLUMNS);
	QStringList ColumnOrderConfig=settings->value("Ui/ColumnOrder",QStringList()<<"100"<<"100"<<"100"<<"100"<<"100"<<"100"<<"100"<<"100"<<"100"<<"100"<<"100").toStringList();
	for(int i=0;i<NUM_COLUMNS;i++){
		if(i<ColumnOrderConfig.size()){
			ColumnOrder[i]=ColumnOrderConfig[i].toInt();			
		}
		else
			ColumnOrder[i]=100;		
	}	
	
	updateColumns();
	
	connect(header(),SIGNAL(sectionResized(int,int,int)),this,SLOT(OnColumnResized(int,int,int)));
	connect(this,SIGNAL(itemSelectionChanged()),this,SLOT(OnItemsChanged()));
	connect(&ClipboardTimer, SIGNAL(timeout()), this, SLOT(OnClipboardTimeOut()));
	connect(header(),SIGNAL(sectionClicked(int)),this,SLOT(OnHeaderSectionClicked(int)));
	connect(header(),SIGNAL(sectionMoved(int,int,int)),this,SLOT(OnColumnMoved(int,int,int)));
	Clipboard=QApplication::clipboard();
	ContextMenu=new QMenu(this);
	setAlternatingRowColors(config.AlternatingRowColors);
	
	pItems=&Items;
	pEntryView=this;
}

KeepassEntryView::~KeepassEntryView(){
	QStringList ColumnSizesConfig;
	for(int i=0;i<ColumnSizes.size();i++){
		ColumnSizesConfig<<QString::number((int)(ColumnSizes[i]));
	}
	settings->setValue("Ui/ColumnSizes",ColumnSizesConfig);
	QStringList ColumnConfig;
	for(int i=0;i<Columns.size();i++){
		if(Columns[i])
			ColumnConfig<<"1";
		else
			ColumnConfig<<"0";		
	}
	settings->setValue("Ui/ShowColumns",ColumnConfig);
	QStringList ColumnOrderConfig;
	for(int i=0;i<NUM_COLUMNS;i++){
		ColumnOrderConfig << QString::number(ColumnOrder[i]);
	}
	settings->setValue("Ui/ColumnOrder",ColumnOrderConfig);
}

void KeepassEntryView::OnGroupChanged(IGroupHandle* group){
	CurrentGroup=group;
	showGroup(group);
}

void KeepassEntryView::OnShowSearchResults(){
	CurrentGroup=NULL;
	showSearchResults();	
}


void KeepassEntryView::OnItemsChanged(){
	switch(selectedItems().size()){
		case 0:	emit selectionChanged(NONE);
				break;
		case 1:	emit selectionChanged(SINGLE);
				break;	
		default:emit selectionChanged(MULTIPLE);
	}
}

bool sortSearchResultsLessThan(const IEntryHandle* a, const IEntryHandle* b){
	int indexA=0;
	int indexB=0;
	for(indexA;indexA<pItems->size();indexA++){
		if((*pItems)[indexA]->EntryHandle==a)break;		
	}
	for(indexB;indexB<pItems->size();indexB++){
		if((*pItems)[indexB]->EntryHandle==b)break;		
	}
	return pEntryView->indexOfTopLevelItem((*pItems)[indexA])<pEntryView->indexOfTopLevelItem((*pItems)[indexB]);
}


void KeepassEntryView::OnHeaderSectionClicked(int index){
	if(header()->isSortIndicatorShown() && header()->sortIndicatorSection()==index){
		header()->setSortIndicator(index,header()->sortIndicatorOrder() ? Qt::DescendingOrder : Qt::AscendingOrder);
		sortItems(index,header()->sortIndicatorOrder());
	}
	else{
		header()->setSortIndicator(index,Qt::AscendingOrder);
		header()->setSortIndicatorShown(true);
		sortItems(index,Qt::AscendingOrder);			
	}
		
	if(ViewMode==Normal){
		for(int i=0;i<Items.size();i++){
			Items[i]->EntryHandle->setVisualIndexDirectly(indexOfTopLevelItem(Items[i]));		
		}
	}
	else if(ViewMode==ShowSearchResults){
		if(header()->sortIndicatorOrder()==Qt::AscendingOrder)
			qSort(SearchResults.begin(),SearchResults.end(),sortSearchResultsLessThan);
		else
			qSort(SearchResults.end(),SearchResults.begin(),sortSearchResultsLessThan);
		
	}
}

void KeepassEntryView::OnSaveAttachment(){
	Q_ASSERT(selectedItems().size()==1);
	CEditEntryDlg::saveAttachment(((EntryViewItem*)selectedItems()[0])->EntryHandle,this);
}

void KeepassEntryView::OnCloneEntry(){	
	QList<QTreeWidgetItem*> entries=selectedItems();
	for(int i=0; i<entries.size();i++){
		Items.append(new EntryViewItem(this));
		Items.back()->EntryHandle=
			db->cloneEntry(((EntryViewItem*)entries[i])->EntryHandle);
		updateEntry(Items.back());
	}
	emit fileModified();
}

void KeepassEntryView::OnDeleteEntry(){
	QList<QTreeWidgetItem*> entries=selectedItems();
	for(int i=0; i<entries.size();i++){
		db->deleteEntry(((EntryViewItem*)entries[i])->EntryHandle);
		Items.removeAt(Items.indexOf((EntryViewItem*)entries[i]));
		delete entries[i];
	}
	emit fileModified();
}



void KeepassEntryView::updateEntry(EntryViewItem* item){
	IEntryHandle* entry = item->EntryHandle;
	int j=0;
	if(Columns[0]){
		item->setText(j++,entry->title());
		item->setIcon(0,db->icon(entry->image()));
	}
	if(Columns[1]){
		if(config.ListView_HideUsernames)
			item->setText(j++,"******");
		else
			item->setText(j++,entry->username());}
		if(Columns[2]){item->setText(j++,entry->url());}
		if(Columns[3]){
			if(config.ListView_HidePasswords)
				item->setText(j++,"******");
			else{
				SecString password=entry->password();
				password.unlock();
				item->setText(j++,password.string());
			}
		}
	if(Columns[4]){
	item->setText(j++,entry->comment().section('\n',0,0));}
	if(Columns[5]){
		item->setText(j++,entry->expire().dateToString(Qt::LocalDate));}
	if(Columns[6]){
		item->setText(j++,entry->creation().dateToString(Qt::LocalDate));}
	if(Columns[7]){
		item->setText(j++,entry->lastMod().dateToString(Qt::LocalDate));}
	if(Columns[8]){
		item->setText(j++,entry->lastAccess().dateToString(Qt::LocalDate));}
	if(Columns[9]){
		item->setText(j++,entry->binaryDesc());}
	if(Columns[10]){
		item->setText(j,entry->group()->title());
		item->setIcon(j++,db->icon(entry->group()->image()));}	
}

void KeepassEntryView::editEntry(EntryViewItem* item){
	CEditEntryDlg dlg(db,item->EntryHandle,this,true);
	switch(dlg.exec()){
		case 0: //canceled or no changes
			break;
		case 1: //modifications but same group
			updateEntry(item);
			emit fileModified();
			break;
		case 2: //entry moved to another group	
			delete item;
			Items.removeAt(Items.indexOf(item));		
			emit fileModified();
			break;
	}
	
}


void KeepassEntryView::OnNewEntry(){
	IEntryHandle* NewEntry=db->newEntry(CurrentGroup);
	CEditEntryDlg dlg(db,NewEntry,this,true);
	if(!dlg.exec()){
		db->deleteLastEntry();
	}
	else{
		Items.append(new EntryViewItem(this));
		Items.back()->EntryHandle=NewEntry;
		updateEntry(Items.back());
		emit fileModified();
	}
	
}

void KeepassEntryView::OnEntryActivated(QTreeWidgetItem* item,int Column){
	int i=0;
	int c=-1;
	for(i;i<NUM_COLUMNS;i++){
		if(config.Columns[i])c++;
		if(c==Column)break;
	}
	if(c==-1)return;
	switch(i){
		case 0:	editEntry((EntryViewItem*)item);
		break;
		case 1: OnUsernameToClipboard();
		break;
		case 2: //OnEditOpenUrl();
		break;
		case 3:	OnPasswordToClipboard();
		break;
	}
	
}

void KeepassEntryView::OnEditEntry(){
	Q_ASSERT(selectedItems().size()==1);
	editEntry((EntryViewItem*)selectedItems()[0]);	
}

void KeepassEntryView::OnUsernameToClipboard(){
	Clipboard->setText(((EntryViewItem*)selectedItems()[0])->EntryHandle->username(),  QClipboard::Clipboard);
	ClipboardTimer.setSingleShot(true);
	ClipboardTimer.start(config.ClipboardTimeOut*1000);
}

void KeepassEntryView::OnPasswordToClipboard(){
	SecString password;
	password=((EntryViewItem*)selectedItems()[0])->EntryHandle->password();
	password.unlock();
	Clipboard->setText(password.string(),QClipboard::Clipboard);
	ClipboardTimer.setSingleShot(true);
	ClipboardTimer.start(config.ClipboardTimeOut*1000);
}

void KeepassEntryView::OnClipboardTimeOut(){
	Clipboard->clear(QClipboard::Clipboard);
}


void KeepassEntryView::contextMenuEvent(QContextMenuEvent* e){
	if(itemAt(e->pos())){
		EntryViewItem* item=(EntryViewItem*)itemAt(e->pos());
		if(selectedItems().size()==0){
			setItemSelected(item,true);
		}
		else{
				if(!isItemSelected(item)){
					while(selectedItems().size()){
						setItemSelected(selectedItems()[0],false);
					}
					setItemSelected(item,true);
				}
			}
	}
	else
	{while(selectedItems().size()){
		setItemSelected(selectedItems()[0],false);}
	}
	e->accept();
	ContextMenu->popup(e->globalPos());
}

void KeepassEntryView::resizeEvent(QResizeEvent* e){
	resizeColumns();
	e->accept();
}


void KeepassEntryView::showSearchResults(){
	ViewMode=ShowSearchResults;
	clear();
	Items.clear();
	createItems(SearchResults);
}


void KeepassEntryView::showGroup(IGroupHandle* group){
	ViewMode=Normal;
	clear();
	Items.clear();
	if(group==NULL)return;
	QList<IEntryHandle*>entries=db->entries(group);
	createItems(entries);	
}

void KeepassEntryView::createItems(QList<IEntryHandle*>& entries){
	header()->setSortIndicatorShown(false);
	for(int i=0;i<entries.size();i++){
		if(!entries[i]->isValid())continue;
		EntryViewItem* item=new EntryViewItem(this);
		Items.push_back(item);
		Items.back()->EntryHandle=entries[i];
		int j=0;
		if(Columns[0]){
			item->setText(j++,entries[i]->title());
			item->setIcon(0,db->icon(entries[i]->image()));}
		if(Columns[1]){
			if(config.ListView_HideUsernames)
				item->setText(j++,"******");
			else
				item->setText(j++,entries[i]->username());}
		if(Columns[2]){item->setText(j++,entries[i]->url());}
		if(Columns[3]){
			if(config.ListView_HidePasswords)
				item->setText(j++,"******");
			else{
				SecString password=entries[i]->password();
				password.unlock();
				item->setText(j++,password.string());
			}
		}
		if(Columns[4]){
			item->setText(j++,entries[i]->comment().section('\n',0,0));}
		if(Columns[5]){
			item->setText(j++,entries[i]->expire().dateToString(Qt::LocalDate));}
		if(Columns[6]){
			item->setText(j++,entries[i]->creation().dateToString(Qt::LocalDate));}
		if(Columns[7]){
			item->setText(j++,entries[i]->lastMod().dateToString(Qt::LocalDate));}
		if(Columns[8]){
			item->setText(j++,entries[i]->lastAccess().dateToString(Qt::LocalDate));}
		if(Columns[9]){
			item->setText(j++,entries[i]->binaryDesc());}
		if(Columns[10]){
			item->setText(j,entries[i]->group()->title());
			item->setIcon(j++,db->icon(entries[i]->group()->image()));}
	}
}

void KeepassEntryView::updateIcons(){
	for(int i=0;i<Items.size();i++){
		Items[i]->setIcon(0,db->icon(Items[i]->EntryHandle->image()));		
	}	
}


void KeepassEntryView::setEntry(IEntryHandle* entry){

}

void KeepassEntryView::updateColumns(){
	setColumnCount(0);
	QStringList cols;
	if(Columns[0]){
	cols << tr("Title");}
	if(Columns[1]){
	cols << tr("Username");}
	if(Columns[2]){
	cols << tr("URL");}
	if(Columns[3]){
	cols << tr("Password");}
	if(Columns[4]){
	cols << tr("Comments");}
	if(Columns[5]){
	cols << tr("Expires");}
	if(Columns[6]){
	cols << tr("Creation");}
	if(Columns[7]){
	cols << tr("Last Change");}
	if(Columns[8]){
	cols << tr("Last Access");}
	if(Columns[9]){
	cols << tr("Attachment");}
	if(Columns[10]){
	cols << tr("Group");}	
	setHeaderLabels(cols);
	
	QMap<int,int> Order;
	for(int i=NUM_COLUMNS-1;i>=0;i--)
		Order.insertMulti(ColumnOrder[i],i);
	
	QMapIterator<int, int> i(Order);
 	while (i.hasNext()) {
		i.next();
		int index=i.value();
		if(!Columns[index])continue;
		header()->moveSection(header()->visualIndex(logicalColIndex(index)),header()->count()-1);		
	}

	resizeColumns();
}

void KeepassEntryView::OnColumnMoved(int LogIndex,int OldVisIndex,int NewVisIndex){
	for(int i=0;i<header()->count();i++){
		ColumnOrder[columnListIndex(header()->logicalIndex(i))]=i;
	}	
}

int KeepassEntryView::logicalColIndex(int LstIndex){
	qDebug("%i",LstIndex);
	int c=-1;
	for(int i=0;i<NUM_COLUMNS;i++){
		if(Columns[i])c++;
		if(i==LstIndex)return c;		
	}
	Q_ASSERT(false);	
}

void KeepassEntryView::resizeColumns(){	
	AutoResizeColumns=false;
	int w=viewport()->width();
	int sum=0;
	for(int i=0;i<header()->count();i++){
		sum+=ColumnSizes[columnListIndex(i)];		
	}
	float stretch=((float)w)/((float)sum);
	sum=0;
	for(int i=0;i<header()->count();i++){
		int lstIndex=columnListIndex(header()->logicalIndex(i));
		int NewSize=qRound(stretch*(float)ColumnSizes[lstIndex]);
		sum+=NewSize;
		if(i==header()->count()-1){
			NewSize+=(w-sum);	// add rounding difference to the last column
		}
		//qDebug("i=%i lstIndex=%i NewSize=%f",i,lstIndex,NewSize);
		header()->resizeSection(header()->logicalIndex(i),NewSize);
		ColumnSizes[lstIndex]=NewSize;
	}	
	AutoResizeColumns=true;
}

int KeepassEntryView::columnListIndex(int LogicalIndex){
	int c=-1; int i=0;
	for(i;i<NUM_COLUMNS;i++){
		if(Columns[i])c++;
		if(c==LogicalIndex)break;
	}
	return i;
}


void KeepassEntryView::OnColumnResized(int lindex,int Old, int New){	
	if(!AutoResizeColumns)return;
	for(int i=0;i<header()->count();i++){
		ColumnSizes[columnListIndex(i)]=header()->sectionSize(i);
	}
	resizeColumns();
}

void KeepassEntryView::mousePressEvent(QMouseEvent *event){
	//save event position - maybe this is the start of a drag
	if (event->button() == Qt::LeftButton)
				DragStartPos = event->pos();
	QTreeWidget::mousePressEvent(event);
}

void KeepassEntryView::mouseMoveEvent(QMouseEvent *event){
	if (!(event->buttons() & Qt::LeftButton))
		return;
	if ((event->pos() - DragStartPos).manhattanLength() < QApplication::startDragDistance())
		return;
	
	DragItems.clear();
	EntryViewItem* DragStartItem=(EntryViewItem*)itemAt(DragStartPos);
	if(!DragStartItem){
		while(selectedItems().size()){
			setItemSelected(selectedItems()[0],false);}
		return;
	}
	if(selectedItems().size()==0){
			setItemSelected(DragStartItem,true);}
	else{
		bool AlreadySelected=false;
		for(int i=0;i<selectedItems().size();i++){
			if(selectedItems()[i]==DragStartItem){AlreadySelected=true; break;}
		}
		if(!AlreadySelected){
			while(selectedItems().size()){
				setItemSelected(selectedItems()[0],false);
			}
			setItemSelected(DragStartItem,true);
		}
	}
	
	DragItems=selectedItems();
	QDrag *drag = new QDrag(this);
	QMimeData *mimeData = new QMimeData;
	void* pDragItems=&DragItems;
	mimeData->setData("text/plain;charset=UTF-8",DragItems[0]->text(0).toUtf8());
	mimeData->setData("application/x-keepassx-entry",QByteArray((char*)&pDragItems,sizeof(void*)));
	drag->setMimeData(mimeData);
	drag->setPixmap(DragPixmap);
	drag->start();

}

void KeepassEntryView::removeDragItems(){
	for(int i=0;i<DragItems.size();i++){
		for(int j=0;j<Items.size();j++){
			if(Items[j]==DragItems[i]){
				Items.removeAt(j);
				j--;
				delete DragItems[i];				
			}			
		}	
	}	
}

void KeepassEntryView::OnAutoType(){
	Q_ASSERT(selectedItems().size()==1);
	QString error;
	AutoType::perform(((EntryViewItem*)selectedItems()[0])->EntryHandle,error);
}

void KeepassEntryView::paintEvent(QPaintEvent * event){
QTreeWidget::paintEvent(event);
}


EntryViewItem::EntryViewItem(QTreeWidget *parent):QTreeWidgetItem(parent){

}

EntryViewItem::EntryViewItem(QTreeWidget *parent, QTreeWidgetItem *preceding):QTreeWidgetItem(parent,preceding){

}

EntryViewItem::EntryViewItem(QTreeWidgetItem *parent):QTreeWidgetItem(parent){

}

EntryViewItem::EntryViewItem(QTreeWidgetItem *parent, QTreeWidgetItem *preceding):QTreeWidgetItem(parent,preceding){

}


bool EntryViewItem::operator<(const QTreeWidgetItem& other)const{
	int SortCol=treeWidget()->sortColumn();
	if(SortCol < 5 || SortCol==9 || SortCol==10){ //columns with string values (Title, Username, Password, URL, Comment, Group)
		if(QString::localeAwareCompare(text(SortCol),other.text(SortCol)) < 0)
			return true;
		else 
			return false;
	}
	KpxDateTime DateThis;
	KpxDateTime DateOther;

	
	switch(SortCol){
		case 5: DateThis=EntryHandle->expire();
				DateOther=((EntryViewItem&)other).EntryHandle->expire();
				break;
		case 6: DateThis=EntryHandle->creation();
				DateOther=((EntryViewItem&)other).EntryHandle->creation();
				break;
		case 7: DateThis=EntryHandle->lastMod();
				DateOther=((EntryViewItem&)other).EntryHandle->lastMod();
				break;
		case 8: DateThis=EntryHandle->lastAccess();
				DateOther=((EntryViewItem&)other).EntryHandle->lastAccess();
				break;
		default:Q_ASSERT(false);
	}
	return DateThis < DateOther;
}