groupview.cpp

KeePassX用于保护密码的安全

/***************************************************************************
 *   Copyright (C) 2005 by Tarek Saidi                                     *
 *   [e-mail]                                                  *
 *                                                                         *
 *   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 <QPaintEvent>
#include <QMouseEvent>
#include <QApplication>
#include <QFont>
#include <QFontMetrics>
#include <QSize>
#include <QPixmap>
#include <QPainter>
#include <QPen>
#include <QBrush>
#include <QMenu>
#include "main.h"
#include "EntryView.h"
#include "GroupView.h"
#define INSERT_AREA_WIDTH 4

KeepassGroupView::KeepassGroupView(QWidget* parent):QTreeWidget(parent){
InsertionMarker=QLine();
db=NULL;
LastHoverItem=NULL;
setHeaderLabels(QStringList()<<tr("Groups"));
ShowSearchGroup=false;
ContextMenu=new QMenu(this);
ContextMenuSearchGroup=new QMenu(this);
}

void KeepassGroupView::selectSearchGroup(){
Q_ASSERT(ShowSearchGroup);
setItemSelected(Items.back(),true);
}

void KeepassGroupView:: dragEnterEvent ( QDragEnterEvent * event ){
if(event->mimeData()->hasFormat("keepass/group")){
	DragType=GROUP;
	event->accept();
	return;
}
if(event->mimeData()->hasFormat("keepass/entry")){
	DragType=ENTRY;
	event->accept();
	return;
}
}

void KeepassGroupView::dragMoveEvent( QDragMoveEvent * event ){
GroupViewItem* item=(GroupViewItem*)itemAt(event->pos());
if(LastHoverItem){
	QFont f=LastHoverItem->font(0);
	f.setBold(false);
	LastHoverItem->setFont(0,f);
}

QLine LastMarker=InsertionMarker;
InsertionMarker=QLine();
if(isSearchResultGroup(item))
	event->setAccepted(false);
else 
if(DragType==GROUP){
	if(item){
	QRect ItemRect=visualItemRect(item);
		if(!db->isParentGroup(item->pGroup,DragItem->pGroup) && DragItem!=item){
			if((ItemRect.height()+ItemRect.y())-event->pos().y() > INSERT_AREA_WIDTH && event->pos().y() > INSERT_AREA_WIDTH){
				QFont f=item->font(0);
				f.setBold(true);
				item->setFont(0,f);
				LastHoverItem=item;
				event->setAccepted(true);
				///@FIXME does not work for top level groups
			}
			else{
				LastHoverItem=NULL;
				if(event->pos().y() > INSERT_AREA_WIDTH)
					InsertionMarker=QLine(ItemRect.x(),ItemRect.y()+ItemRect.height()
										,ItemRect.x()+ItemRect.width(),ItemRect.y()+ItemRect.height());
				else
					InsertionMarker=QLine(ItemRect.x(),0,ItemRect.x()+ItemRect.width(),0);
			}
		}
		else 
			event->setAccepted(false);
	}
else
  LastHoverItem=NULL;
}
else{
	if(item){
		QFont f=item->font(0);
		f.setBold(true);
		item->setFont(0,f);
		LastHoverItem=item;
		event->setAccepted(true);
	}
	else{
		event->setAccepted(false);
		LastHoverItem=NULL;
	}
}
if(!LastMarker.isNull()){
	///@FIXME
	//this is a very dirty work-around to force a redraw of items at the last marker position
	//should be replaced!!!
	GroupViewItem* i=(GroupViewItem*)itemAt(0,LastMarker.y1());
	if(i)i->setFont(0,i->font(0));
	i=(GroupViewItem*)itemAt(0,LastMarker.y1()-1);
	if(i)i->setFont(0,i->font(0));
}
update();
}

void KeepassGroupView:: dragLeaveEvent ( QDragLeaveEvent * event ){
InsertionMarker=QLine();
if(LastHoverItem){
  QFont f=LastHoverItem->font(0);
  f.setBold(false);
  LastHoverItem->setFont(0,f);
}
update();
}


void KeepassGroupView::dropEvent( QDropEvent * event ){
emit fileModified();
InsertionMarker=QLine();
if(LastHoverItem){
  QFont f=LastHoverItem->font(0);
  f.setBold(false);
  LastHoverItem->setFont(0,f);
  LastHoverItem=NULL;
}
GroupViewItem* item=(GroupViewItem*)itemAt(event->pos());
if(DragType==GROUP){
	if(item){
		QRect ItemRect=visualItemRect(item);
		if((ItemRect.height()+ItemRect.y())-event->pos().y() > INSERT_AREA_WIDTH && event->pos().y() > INSERT_AREA_WIDTH){
			db->moveGroup(DragItem->pGroup,item->pGroup);}
		else{
			if(event->pos().y() > INSERT_AREA_WIDTH){
				if(db->getNumberOfChilds(item->pGroup) > 0)
					db->moveGroup(DragItem->pGroup,item->pGroup,0);				
				else				
					db->moveGroupDirectly(DragItem->pGroup,item->pGroup);
			}
			else	db->moveGroupDirectly(DragItem->pGroup,NULL);
			}		
	}
	else db->moveGroup(DragItem->pGroup,NULL);
	updateItems();
}else{
	Q_ASSERT(item);
	QList<QTreeWidgetItem*>* pDragItems=(QList<QTreeWidgetItem*>*)*((QList<QTreeWidgetItem*>**)event->mimeData()->data("keepass/entry").data());
	for(int i=0;i<pDragItems->size();i++){
		db->moveEntry(((EntryViewItem*)(*pDragItems)[i])->pEntry,item->pGroup);
	}
	emit entryDropped();
}
}


void KeepassGroupView::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 KeepassGroupView::mouseMoveEvent(QMouseEvent *event){
 if (!(event->buttons() & Qt::LeftButton))
            return;
 if ((event->pos() - DragStartPos).manhattanLength() < QApplication::startDragDistance())
            return;
	DragItem=(GroupViewItem*)itemAt(DragStartPos);
 if (isSearchResultGroup(DragItem))
			return;
	if(!DragItem)return;
	QDrag *drag = new QDrag(this);
	QFontMetrics fontmet(DragItem->font(0));
	int DragPixmHeight=16;
	if(fontmet.height()>16)DragPixmHeight=fontmet.height();
	DragPixmap  = QPixmap(fontmet.width(DragItem->text(0))+19,DragPixmHeight);
	DragPixmap.fill(QColor(255,255,255));
	QPainter painter(&DragPixmap);
	painter.setPen(QColor(0,0,0));
	painter.setFont(DragItem->font(0));
	painter.drawPixmap(0,0,DragItem->icon(0).pixmap(QSize(16,16)));
	painter.drawText(19,DragPixmHeight-fontmet.strikeOutPos(),DragItem->text(0));	
        QMimeData *mimeData = new QMimeData;
	mimeData->setData("keepass/group",QByteArray((char*)&(DragItem->pGroup),sizeof(void*)));
        drag->setMimeData(mimeData);
        drag->setPixmap(DragPixmap);
	drag->start();
}

void KeepassGroupView::updateItems(){

clear();
Items.clear();
for(int i=0; i<db->numGroups();i++){
if(db->group(i).Level==0){
 if(Items.size()) Items.push_back(new GroupViewItem(this,getLastSameLevelItem(0)));
 else 		  Items.push_back(new GroupViewItem(this));
 Items.back()->setText(0,db->group(i).Name);
 Items.back()->pGroup=&db->group(i); 
}
 else{
 if(db->group(i).Level>db->group(i-1).Level){
 Items.push_back(new GroupViewItem(Items.back(),getLastSameLevelItem(db->group(i).Level)));
 Items.back()->setText(0,db->group(i).Name);
 Items.back()->pGroup=&db->group(i);
 }
 if(db->group(i).Level<=db->group(i-1).Level){
   GroupItemItr j;
   for(j=Items.end()-1;j!=Items.begin();j--){
    if((*j)->pGroup->Level<db->group(i).Level)break;}
     Items.push_back(new GroupViewItem((*j),getLastSameLevelItem(db->group(i).Level)));
     Items.back()->setText(0,db->group(i).Name);
     Items.back()->pGroup=&db->group(i);
     }
 }
Items.back()->setIcon(0,db->icon(db->group(i).ImageID));
}

for(int i=0;i<Items.size();i++){
 setItemExpanded(Items[i],Items[i]->pGroup->UI_ItemIsExpanded);
}
if(ShowSearchGroup){
 	 Items.push_back(new GroupViewItem(this));
     Items.back()->setText(0,tr("Search Results"));
     Items.back()->pGroup=NULL;
     QFont f=Items.back()->font(0);
     f.setItalic(true);
	 f.setBold(true);
     Items.back()->setFont(0,f);
}

}



GroupViewItem* KeepassGroupView::getLastSameLevelItem(int level){
for(int i=Items.size()-1;i>=0;i--){
 if(Items[i]->pGroup->Level==level){
return Items[i];}
}

return Items.back();

}

void KeepassGroupView::paintEvent(QPaintEvent* event){
QTreeWidget::paintEvent(event);
QPainter painter(viewport());
QPen pen(QColor(100,100,100));
pen.setWidth(2);
pen.setStyle(Qt::DotLine);
painter.setPen(pen);
//qDebug("UPDATE: (%i,%i) %ix%i",event->rect().x(),event->rect().y(),event->rect().width(),event->rect().height());
if(!InsertionMarker.isNull()){
	painter.drawLine(InsertionMarker);
}
}

bool KeepassGroupView::isSearchResultGroup(GroupViewItem* item){
if(ShowSearchGroup && (item == Items.back()))return true;
else return false;
}

void KeepassGroupView::contextMenuEvent(QContextMenuEvent* e){
if(!(GroupViewItem*)itemAt(e->pos()) && selectedItems().size()){
	setItemSelected(selectedItems()[0],false);
}
e->accept();
if(isSearchResultGroup((GroupViewItem*)itemAt(e->pos())))
	ContextMenuSearchGroup->popup(e->globalPos());
else
	ContextMenu->popup(e->globalPos());
}

GroupViewItem::GroupViewItem(QTreeWidget *parent):QTreeWidgetItem(parent){
}

GroupViewItem::GroupViewItem(QTreeWidget *parent, QTreeWidgetItem *preceding):QTreeWidgetItem(parent,preceding){
}

GroupViewItem::GroupViewItem(QTreeWidgetItem *parent):QTreeWidgetItem(parent){
}

GroupViewItem::GroupViewItem(QTreeWidgetItem *parent, QTreeWidgetItem *preceding):QTreeWidgetItem(parent,preceding){
}