/***************************************************************************
 *   Copyright (C) 2005-2006 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"



KeepassEntryView::KeepassEntryView(QWidget* parent):QTreeWidget(parent){
	AutoResizeColumns=true;
	int sum=0;
	for(int i=0;i<NUM_COLUMNS;i++)
		sum+=config.ColumnSizes[i];
	for(int i=0;i<NUM_COLUMNS;i++)
		ColumnSizes << (float)config.ColumnSizes[i]/(float)sum;
	updateColumns();
	header()->setResizeMode(QHeaderView::Interactive);
	header()->setStretchLastSection(false);
	header()->setClickable(true);
	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)));
	Clipboard=QApplication::clipboard();
	ContextMenu=new QMenu(this);
	setAlternatingRowColors(config.AlternatingRowColors);
}

KeepassEntryView::~KeepassEntryView(){
	for(int i=0;i<ColumnSizes.size();i++){
		config.ColumnSizes[i]=(int)(ColumnSizes[i]*10000.0f);
	}
}

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);
	}
}

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);			
	}
	
	for(int i=0;i<Items.size();i++){
		Items[i]->EntryHandle->setVisualIndexDirectly(indexOfTopLevelItem(Items[i]));		
	}
}

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(config.Columns[0])item->setText(j++,entry->title());
	if(config.Columns[1]){
		if(config.ListView_HideUsernames)
			item->setText(j++,"******");
		else
			item->setText(j++,entry->username());}
		if(config.Columns[2]){item->setText(j++,entry->url());}
		if(config.Columns[3]){
			if(config.ListView_HidePasswords)
				item->setText(j++,"******");
			else{
				SecString password=entry->password();
				password.unlock();
				item->setText(j++,password.string());
			}
		}
		if(config.Columns[4]){
			item->setText(j++,entry->comment().section('\n',0,0));}
		if(config.Columns[5]){
			item->setText(j++,entry->expire().dateToString(Qt::LocalDate));}
		if(config.Columns[6]){
			item->setText(j++,entry->creation().dateToString(Qt::LocalDate));}
		if(config.Columns[7]){
			item->setText(j++,entry->lastMod().dateToString(Qt::LocalDate));}
		if(config.Columns[8]){
			item->setText(j++,entry->lastAccess().dateToString(Qt::LocalDate));}
		if(config.Columns[9]){
			item->setText(j++,entry->binaryDesc());}
		item->setIcon(0,db->icon(entry->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(){
	clear();
	Items.clear();
	createItems(SearchResults);		
}


void KeepassEntryView::showGroup(IGroupHandle* group){
	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(config.Columns[0])item->setText(j++,entries[i]->title());
		if(config.Columns[1]){
			if(config.ListView_HideUsernames)
				item->setText(j++,"******");
			else
				item->setText(j++,entries[i]->username());}
		if(config.Columns[2]){item->setText(j++,entries[i]->url());}
		if(config.Columns[3]){
			if(config.ListView_HidePasswords)
				item->setText(j++,"******");
			else{
				SecString password=entries[i]->password();
				password.unlock();
				item->setText(j++,password.string());
			}
		}
		if(config.Columns[4]){
			item->setText(j++,entries[i]->comment().section('\n',0,0));}
		if(config.Columns[5]){
			item->setText(j++,entries[i]->expire().dateToString(Qt::LocalDate));}
		if(config.Columns[6]){
			item->setText(j++,entries[i]->creation().dateToString(Qt::LocalDate));}
		if(config.Columns[7]){
			item->setText(j++,entries[i]->lastMod().dateToString(Qt::LocalDate));}
		if(config.Columns[8]){
			item->setText(j++,entries[i]->lastAccess().dateToString(Qt::LocalDate));}
		if(config.Columns[9]){
			item->setText(j++,entries[i]->binaryDesc());}
		Items.back()->setIcon(0,db->icon(entries[i]->image()));
		qDebug("%s  :  %i",entries[i]->title().toUtf8().data(),entries[i]->visualIndex());
	}
}

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::refreshItems(){
EntryViewItem *tmp=NULL;
for(int i=0;i<Items.size();i++){
  tmp=Items[i];
  IEntryHandle* entry=tmp->pEntry;

  int j=0;
  if(config.Columns[0]){
    tmp->setText(j++,entry->Title);}
  if(config.Columns[1]){
    if(config.ListView_HideUsernames)
      tmp->setText(j++,"******");
    else
      tmp->setText(j++,entry->UserName);}
  if(config.Columns[2]){
    tmp->setText(j++,entry->URL);}
  if(config.Columns[3]){
    if(config.ListView_HidePasswords)
      tmp->setText(j++,"******");
    else{
      entry->Password.unlock();
      tmp->setText(j++,entry->Password.string());
      entry->Password.lock();}}
  if(config.Columns[4]){
    tmp->setText(j++,entry->Additional.section('\n',0,0));}
  if(config.Columns[5]){
    tmp->setText(j++,entry->Expire.dateToString(Qt::LocalDate));}
  if(config.Columns[6]){
    tmp->setText(j++,entry->Creation.dateToString(Qt::LocalDate));}
  if(config.Columns[7]){
    tmp->setText(j++,entry->LastMod.dateToString(Qt::LocalDate));}
  if(config.Columns[8]){
    tmp->setText(j++,entry->LastAccess.dateToString(Qt::LocalDate));}
  if(config.Columns[9]){
   tmp->setText(j++,entry->BinaryDesc);}
  tmp->setIcon(0,db->icon(entry->ImageID));
}
}*/


void KeepassEntryView::updateColumns(){
	setColumnCount(0);
	QStringList cols;
	if(config.Columns[0]){
	cols << tr("Title");}
	if(config.Columns[1]){
	cols << tr("Username");}
	if(config.Columns[2]){
	cols << tr("URL");}
	if(config.Columns[3]){
	cols << tr("Password");}
	if(config.Columns[4]){
	cols << tr("Comments");}
	if(config.Columns[5]){
	cols << tr("Expires");}
	if(config.Columns[6]){
	cols << tr("Creation");}
	if(config.Columns[7]){
	cols << tr("Last Change");}
	if(config.Columns[8]){
	cols << tr("Last Access");}
	if(config.Columns[9]){
	cols << tr("Attachment");}
	setHeaderLabels(cols);
	resizeColumns();
}

void KeepassEntryView::resizeColumns(){
AutoResizeColumns=false;
if(!header()->count())return;

for(int i=0;i<NUM_COLUMNS;i++)
	if(!config.Columns[i])ColumnSizes[i]=0;

for(int i=0;i<NUM_COLUMNS;i++)
	if(config.Columns[i] && ColumnSizes[i]==0)ColumnSizes[i]=0.1f;

float sum=0;
for(int i=0;i<NUM_COLUMNS;i++)
	sum+=ColumnSizes[i];

for(int i=0;i<NUM_COLUMNS;i++)
	ColumnSizes[i]/=sum;

int w=viewport()->width();
int wx=0; int j=0;


for(int i=0;i<NUM_COLUMNS;i++){
	if(!config.Columns[i])continue;
	int NewWidth=(int)(ColumnSizes[i]*(float)w);
	wx+=NewWidth;
	header()->resizeSection(j++,NewWidth);
	//add rounding difference (w-wx) to the last column
	if(j==header()->count()){
		header()->resizeSection(j-1,header()->sectionSize(j-1)+(w-wx));
	}
}

AutoResizeColumns=true;

}


void KeepassEntryView::OnColumnResized(int index,int Old, int New){
if(!AutoResizeColumns)return;

int i=0; int c=-1;
for(i;i<ColumnSizes.size();i++){
	if(config.Columns[i])c++;
	if(c==index)break;
}

int j=0; c=-1; bool IsLastColumn=true;
for(j;j<ColumnSizes.size();j++){
	if(config.Columns[j])c++;
	if(c==(index+1)){IsLastColumn=false; break;}
}

if(IsLastColumn){
	j=0; c=-1;
	for(j;j<ColumnSizes.size();j++){
		if(config.Columns[j])c++;
		if(c==(index-1))break;
	}
}

int w=viewport()->width();
float div=(float)(New-Old)/(float)w;

if(((ColumnSizes[j]-div)*w > 2)){
	ColumnSizes[j]-=div;
	ColumnSizes[i]+=div;
}
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();
	//call base function
	QTreeWidget::mousePressEvent(event);
}

void KeepassEntryView::mouseMoveEvent(QMouseEvent *event){
	/*
if(IsSearchGroup)
	return;
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);
QFontMetrics fontmet(DragStartItem->font(0));
int DragPixmHeight=16;
if(fontmet.height()>16)DragPixmHeight=fontmet.height();
QString DragText;
if(DragItems.size()>1)DragText=QString(tr("%1 items")).arg(DragItems.size());
else DragText=((EntryViewItem*)DragItems[0])->pEntry->Title;
DragPixmap  = QPixmap(fontmet.width(DragText)+19,DragPixmHeight);
DragPixmap.fill(QColor(255,255,255));
QPainter painter(&DragPixmap);
painter.setPen(QColor(0,0,0));
painter.setFont(DragItems[0]->font(0));
painter.drawPixmap(0,0,DragItems[0]->icon(0).pixmap(QSize(16,16)));
painter.drawText(19,DragPixmHeight-fontmet.strikeOutPos(),DragText);	
QMimeData *mimeData = new QMimeData;
void* pDragItems=&DragItems;
mimeData->setData("keepass/entry",QByteArray((char*)&pDragItems,sizeof(void*)));
drag->setMimeData(mimeData);
drag->setPixmap(DragPixmap);
drag->start();
	*/
}

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){ //columns with string values (Title, Username, Password, URL, Comment)
		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;
}