565 lines
19 KiB
C++
565 lines
19 KiB
C++
#include "./entrymodel.h"
|
|
|
|
#ifdef PASSWORD_MANAGER_GUI_QTWIDGETS
|
|
# include "./gui/undocommands.h"
|
|
#endif
|
|
|
|
#include <passwordfile/io/entry.h>
|
|
|
|
#include <c++utilities/io/catchiofailure.h>
|
|
|
|
#include <QIcon>
|
|
#include <QBuffer>
|
|
#include <QMimeData>
|
|
#include <QDebug>
|
|
|
|
#include <sstream>
|
|
|
|
using namespace std;
|
|
using namespace Io;
|
|
|
|
namespace QtGui {
|
|
|
|
/*!
|
|
* \class EntryModel
|
|
* \brief The EntryModel class provides a model interface for a hierarchy of Entry instances.
|
|
*
|
|
* When building the Qt Widgets GUI, the model also supports Qt Widgets' undo framework.
|
|
* \sa http://qt-project.org/doc/qt-5/qabstractitemmodel.html
|
|
* \sa http://qt-project.org/doc/qt-5/qundo.html
|
|
*/
|
|
|
|
/*!
|
|
* \brief Constructs a new entry model.
|
|
*/
|
|
EntryModel::EntryModel(QObject *parent) :
|
|
QAbstractItemModel(parent),
|
|
m_rootEntry(nullptr),
|
|
m_insertType(EntryType::Node)
|
|
{}
|
|
|
|
#ifdef PASSWORD_MANAGER_GUI_QTWIDGETS
|
|
/*!
|
|
* \brief Constructs a new entry model with the specified \a undoStack.
|
|
*
|
|
* This constructor is only available when PASSWORD_MANAGER_GUI_QTWIDGETS is defined.
|
|
*/
|
|
EntryModel::EntryModel(QUndoStack *undoStack, QObject *parent) :
|
|
QAbstractItemModel(parent),
|
|
StackSupport(undoStack),
|
|
m_rootEntry(nullptr),
|
|
m_insertType(EntryType::Node)
|
|
{}
|
|
#endif
|
|
|
|
/*!
|
|
* \brief Returns the Entry for the specified \a index of nullptr if the \a index is invalid.
|
|
*
|
|
* Modifications should be done using the common methods defined by the QAbstractItemModel class.
|
|
* This method is intended to be used only internally to implement the QAbstractItemModel interface
|
|
* or when implementing a QUndoCommand to support Qt's undo framework.
|
|
*/
|
|
Entry *EntryModel::entry(const QModelIndex &index)
|
|
{
|
|
if(index.isValid()) {
|
|
return static_cast<Entry *>(index.internalPointer());
|
|
} else {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
/*!
|
|
* \brief Removes \a count number of entries at the specified \a row within the specified \a parent.
|
|
* \returns Returns the removed entries (rather then deleting them).
|
|
*
|
|
* Modifications should be done using the common methods defined by the QAbstractItemModel class.
|
|
* This method is intended to be used only internally to implement the QAbstractItemModel interface
|
|
* or when implementing a QUndoCommand to support Qt's undo framework.
|
|
*/
|
|
QList<Entry *> EntryModel::takeEntries(int row, int count, const QModelIndex &parent)
|
|
{
|
|
QList<Entry *> res;
|
|
if(Entry *parentEntry = entry(parent)) {
|
|
if(parentEntry->type() == EntryType::Node) {
|
|
NodeEntry *parentNodeEntry = static_cast<NodeEntry *>(parentEntry);
|
|
int lastIndex = row + count - 1;
|
|
const vector<Entry *> &children = parentNodeEntry->children();
|
|
if(lastIndex < 0 || static_cast<size_t>(lastIndex) >= children.size()) {
|
|
lastIndex = children.size() - 1;
|
|
}
|
|
beginRemoveRows(parent, row, lastIndex);
|
|
for(int index = lastIndex; index >= row; --index) {
|
|
Entry *child = children.at(index);
|
|
child->setParent(nullptr);
|
|
res << child;
|
|
}
|
|
endRemoveRows();
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
/*!
|
|
* \brief Inserts the specified \a entries before the specified \a row within the specified \a parent.
|
|
*
|
|
* Modifications should be done using the common methods defined by the QAbstractItemModel class.
|
|
* This method is intended to be used only internally to implement the QAbstractItemModel interface
|
|
* or when implementing a QUndoCommand to support Qt's undo framework.
|
|
*/
|
|
bool EntryModel::insertEntries(int row, const QModelIndex &parent, const QList<Entry *> &entries)
|
|
{
|
|
if(entries.isEmpty()) {
|
|
return true;
|
|
}
|
|
if(Entry *parentEntry = entry(parent)) {
|
|
if(parentEntry->type() == EntryType::Node) {
|
|
NodeEntry *parentNodeEntry = static_cast<NodeEntry *>(parentEntry);
|
|
const vector<Entry *> &children = parentNodeEntry->children();
|
|
if(row < 0 || static_cast<size_t>(row) > children.size()) {
|
|
row = children.size();
|
|
}
|
|
beginInsertRows(parent, row, row + entries.size() - 1);
|
|
foreach(Entry *entry, entries) {
|
|
entry->setParent(parentNodeEntry, row);
|
|
++row;
|
|
}
|
|
endInsertRows();
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
QModelIndex EntryModel::index(int row, int column, const QModelIndex &parent) const
|
|
{
|
|
if(parent.isValid()) {
|
|
if(Entry *parentEntry = static_cast<Entry *>(parent.internalPointer())) {
|
|
switch(parentEntry->type()) {
|
|
case EntryType::Node: {
|
|
const std::vector<Entry *> &children = static_cast<NodeEntry *>(parentEntry)->children();
|
|
if(row >= 0 && static_cast<size_t>(row) < children.size()) {
|
|
return createIndex(row, column, children.at(row));
|
|
}
|
|
break;
|
|
} case EntryType::Account:
|
|
;
|
|
}
|
|
}
|
|
} else if(m_rootEntry && row == 0) {
|
|
return createIndex(row, column, m_rootEntry);
|
|
}
|
|
return QModelIndex();
|
|
}
|
|
|
|
/*!
|
|
* \brief Returns the index of the specified \a entry.
|
|
* \remarks It is up to the caller to ensure that \a entry is a child of the root element.
|
|
*/
|
|
QModelIndex EntryModel::index(Entry *entry) const
|
|
{
|
|
if(entry->parent()) {
|
|
return createIndex(entry->index(), 0, entry);
|
|
} else {
|
|
return createIndex(0, 0, m_rootEntry);
|
|
}
|
|
}
|
|
|
|
QModelIndex EntryModel::parent(const QModelIndex &child) const
|
|
{
|
|
if(child.isValid()) {
|
|
if(Entry *entry = static_cast<Entry *>(child.internalPointer())) {
|
|
NodeEntry *parent = entry->parent();
|
|
if(parent && (child.row() >= 0 && static_cast<size_t>(child.row()) < parent->children().size())) {
|
|
return createIndex(parent->index() > 0 ? parent->index() : 0, 0, parent);
|
|
}
|
|
}
|
|
}
|
|
return QModelIndex();
|
|
}
|
|
|
|
bool EntryModel::hasChildren(const QModelIndex &parent) const
|
|
{
|
|
if(parent.isValid()) {
|
|
if(Entry *entry = static_cast<Entry *>(parent.internalPointer())) {
|
|
if(entry->type() == EntryType::Node) {
|
|
return static_cast<NodeEntry *>(entry)->children().size();
|
|
}
|
|
}
|
|
} else {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/*!
|
|
* \brief Returns an indication whether the specified \a parent might have children.
|
|
* \remarks Only node entries might have childs.
|
|
*/
|
|
bool EntryModel::isNode(const QModelIndex &parent) const
|
|
{
|
|
if(parent.isValid()) {
|
|
if(Entry *entry = static_cast<Entry *>(parent.internalPointer())) {
|
|
return entry->type() == EntryType::Node;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
QVariant EntryModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if(index.isValid()) {
|
|
if(Entry *entry = static_cast<Entry *>(index.internalPointer())) {
|
|
switch(role) {
|
|
case Qt::DisplayRole:
|
|
case Qt::EditRole:
|
|
switch(index.column()) {
|
|
case 0:
|
|
return QString::fromStdString(entry->label());
|
|
default:
|
|
;
|
|
}
|
|
break;
|
|
case Qt::DecorationRole:
|
|
if(index.column() == 0 && entry->type() == EntryType::Node) {
|
|
static const QVariant folderIcon = QIcon::fromTheme(QStringLiteral("folder"));
|
|
return folderIcon;
|
|
}
|
|
break;
|
|
case SerializedRole: {
|
|
stringstream ss(stringstream::in | stringstream::out | stringstream::binary);
|
|
ss.exceptions(std::stringstream::failbit | std::stringstream::badbit);
|
|
try {
|
|
entry->make(ss);
|
|
string string = ss.str();
|
|
return QByteArray(string.data(), string.size());
|
|
} catch(...) {
|
|
IoUtilities::catchIoFailure();
|
|
return false;
|
|
}
|
|
}
|
|
break;
|
|
case DefaultExpandedRole:
|
|
return entry->type() == EntryType::Node && static_cast<NodeEntry *>(entry)->isExpandedByDefault();
|
|
default:
|
|
;
|
|
}
|
|
}
|
|
}
|
|
return QVariant();
|
|
}
|
|
|
|
QMap<int, QVariant> EntryModel::itemData(const QModelIndex &index) const
|
|
{
|
|
QMap<int, QVariant> roles;
|
|
roles.insert(Qt::DisplayRole, data(index, Qt::DisplayRole));
|
|
roles.insert(SerializedRole, data(index, SerializedRole));
|
|
return roles;
|
|
}
|
|
|
|
bool EntryModel::setData(const QModelIndex &index, const QVariant &value, int role)
|
|
{
|
|
#ifdef PASSWORD_MANAGER_GUI_QTWIDGETS
|
|
if(undoStack()) {
|
|
return push(new EntryModelSetValueCommand(this, index, value, role));
|
|
}
|
|
#endif
|
|
if(index.isValid()) {
|
|
if(Entry *entry = static_cast<Entry *>(index.internalPointer())) {
|
|
switch(role) {
|
|
case Qt::DisplayRole:
|
|
case Qt::EditRole:
|
|
switch(index.column()) {
|
|
case 0:
|
|
entry->setLabel(value.toString().toStdString());
|
|
emit dataChanged(index, index, QVector<int>() << role);
|
|
return true;
|
|
default:
|
|
;
|
|
}
|
|
break;
|
|
case SerializedRole: {
|
|
NodeEntry *parent = entry->parent();
|
|
QModelIndex parentIndex = index.parent();
|
|
if(parent && parentIndex.isValid()) {
|
|
stringstream ss(stringstream::in | stringstream::out | stringstream::binary);
|
|
ss.exceptions(std::stringstream::failbit | std::stringstream::badbit);
|
|
QByteArray array = value.toByteArray();
|
|
if(array.size()) {
|
|
try {
|
|
ss.write(array.data(), array.size());
|
|
Entry *newEntry = Entry::parse(ss);
|
|
int row = entry->index();
|
|
beginRemoveRows(parentIndex, row, row);
|
|
delete entry;
|
|
endRemoveRows();
|
|
beginInsertRows(parentIndex, row, row);
|
|
newEntry->setParent(parent, row);
|
|
endInsertRows();
|
|
return true;
|
|
} catch(...) {
|
|
IoUtilities::catchIoFailure();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case DefaultExpandedRole:
|
|
switch(entry->type()) {
|
|
case EntryType::Account:
|
|
return false;
|
|
case EntryType::Node:
|
|
static_cast<NodeEntry *>(entry)->setExpandedByDefault(value.toBool());
|
|
emit dataChanged(index, index, QVector<int>() << role);
|
|
return true;
|
|
}
|
|
break;
|
|
default:
|
|
;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool EntryModel::setItemData(const QModelIndex &index, const QMap<int, QVariant> &roles)
|
|
{
|
|
for(QMap<int, QVariant>::ConstIterator it = roles.constBegin(); it != roles.constEnd(); ++it) {
|
|
setData(index, it.value(), it.key());
|
|
}
|
|
return true;
|
|
}
|
|
|
|
Qt::ItemFlags EntryModel::flags(const QModelIndex &index) const
|
|
{
|
|
return isNode(index)
|
|
? QAbstractItemModel::flags(index) | Qt::ItemIsEditable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled
|
|
: QAbstractItemModel::flags(index) | Qt::ItemIsEditable | Qt::ItemIsDragEnabled;
|
|
}
|
|
|
|
QVariant EntryModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
switch(orientation) {
|
|
case Qt::Horizontal:
|
|
switch(role) {
|
|
case Qt::DisplayRole:
|
|
switch(section) {
|
|
case 0:
|
|
return tr("Name");
|
|
default:
|
|
;
|
|
}
|
|
break;
|
|
default:
|
|
;
|
|
}
|
|
break;
|
|
default:
|
|
;
|
|
}
|
|
return QVariant();
|
|
}
|
|
|
|
int EntryModel::rowCount(const QModelIndex &parent) const
|
|
{
|
|
if(parent.isValid()) {
|
|
if(Entry *parentEntry = static_cast<Entry *>(parent.internalPointer())) {
|
|
switch(parentEntry->type()) {
|
|
case EntryType::Node:
|
|
return static_cast<NodeEntry *>(parentEntry)->children().size();
|
|
case EntryType::Account:
|
|
;
|
|
}
|
|
}
|
|
} else if(m_rootEntry) {
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int EntryModel::columnCount(const QModelIndex &) const
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
bool EntryModel::insertRows(int row, int count, const QModelIndex &parent)
|
|
{
|
|
#ifdef PASSWORD_MANAGER_GUI_QTWIDGETS
|
|
if(undoStack()) {
|
|
return push(new EntryModelInsertRowsCommand(this, row, count, parent));
|
|
}
|
|
#endif
|
|
if(parent.isValid()) {
|
|
if(Entry *parentEntry = static_cast<Entry *>(parent.internalPointer())) {
|
|
if(parentEntry->type() == EntryType::Node) {
|
|
beginInsertRows(parent, row, row + count - 1);
|
|
for(int end = row + count; row < end; ++row) {
|
|
Entry *newEntry;
|
|
switch(m_insertType) {
|
|
case EntryType::Node:
|
|
newEntry = new NodeEntry;
|
|
break;
|
|
case EntryType::Account:
|
|
newEntry = new AccountEntry;
|
|
break;
|
|
default:
|
|
return false; // should never be reached, just to suppress compiler warning
|
|
}
|
|
newEntry->setParent(static_cast<NodeEntry *>(parentEntry), row);
|
|
}
|
|
endInsertRows();
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool EntryModel::removeRows(int row, int count, const QModelIndex &parent)
|
|
{
|
|
#ifdef PASSWORD_MANAGER_GUI_QTWIDGETS
|
|
if(undoStack()) {
|
|
return push(new EntryModelRemoveRowsCommand(this, row, count, parent));
|
|
}
|
|
#endif
|
|
if(parent.isValid() && count > 0) {
|
|
if(Entry *parentEntry = static_cast<Entry *>(parent.internalPointer())) {
|
|
if(parentEntry->type() == EntryType::Node) {
|
|
beginRemoveRows(parent, row, row + count - 1);
|
|
static_cast<NodeEntry *>(parentEntry)->deleteChildren(row, row + count);
|
|
endRemoveRows();
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool EntryModel::moveRows(const QModelIndex &sourceParent, int sourceRow, int count, const QModelIndex &destinationParent, int destinationChild)
|
|
{
|
|
#ifdef PASSWORD_MANAGER_GUI_QTWIDGETS
|
|
if(undoStack()) {
|
|
return push(new EntryModelMoveRowsCommand(this, sourceParent, sourceRow, count, destinationParent, destinationChild));
|
|
}
|
|
#endif
|
|
// check validation of specified arguments
|
|
if(sourceParent.isValid() && destinationParent.isValid()
|
|
&& sourceRow >= 0 && count > 0
|
|
&& entry(sourceParent)->type() == EntryType::Node // source and destination parent entries
|
|
&& entry(destinationParent)->type() == EntryType::Node) { // need to be node entries
|
|
// determine the source parent entry and dest parent entry as node entries
|
|
NodeEntry *srcParentEntry = static_cast<NodeEntry *>(sourceParent.internalPointer());
|
|
NodeEntry *destParentEntry = static_cast<NodeEntry *>(destinationParent.internalPointer());
|
|
// source rows must be within the valid range
|
|
if(static_cast<size_t>(sourceRow + count) <= srcParentEntry->children().size()
|
|
// if source and destination parent are the same the destination child mustn't be in the source range
|
|
&& (srcParentEntry != destParentEntry || (destinationChild < sourceRow || (sourceRow + count) < destinationChild))) {
|
|
// do not move a row to one of its own children! -> check before
|
|
for(int index = 0; index < count; ++index) {
|
|
Entry *toMove = srcParentEntry->children().at(sourceRow + index);
|
|
if(toMove->type() == EntryType::Node) {
|
|
if(destParentEntry->isIndirectChildOf(static_cast<NodeEntry *>(toMove))) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
// actually perform the move operation
|
|
beginMoveRows(sourceParent, sourceRow, sourceRow + count - 1, destinationParent, destinationChild);
|
|
for(int index = 0; index < count; ++index) {
|
|
Entry *toMove = srcParentEntry->children().at(sourceRow + index);
|
|
if(srcParentEntry == destParentEntry && sourceRow < destinationChild) {
|
|
toMove->setParent(destParentEntry, destinationChild + index - 1);
|
|
} else {
|
|
toMove->setParent(destParentEntry, destinationChild + index);
|
|
}
|
|
}
|
|
endMoveRows();
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
QStringList EntryModel::mimeTypes() const
|
|
{
|
|
return QStringList() << QStringLiteral("application/x-entrymodelpathlistmove") << QStringLiteral("text/plain");
|
|
}
|
|
|
|
QMimeData *EntryModel::mimeData(const QModelIndexList &indexes) const
|
|
{
|
|
if(indexes.count() <= 0) {
|
|
return nullptr;
|
|
}
|
|
QStringList types = mimeTypes();
|
|
if(types.isEmpty()) {
|
|
return nullptr;
|
|
}
|
|
QMimeData *data = new QMimeData();
|
|
QStringList plainTextParts;
|
|
QByteArray encoded;
|
|
QDataStream dataStream(&encoded, QIODevice::WriteOnly);
|
|
foreach(const QModelIndex &index, indexes) {
|
|
if(index.isValid()) {
|
|
Entry *entry = static_cast<Entry *>(index.internalPointer());
|
|
list<string> path = entry->path();
|
|
dataStream << static_cast<quint32>(path.size());
|
|
for(const string &part : path) {
|
|
dataStream << QString::fromStdString(part);
|
|
}
|
|
plainTextParts << QString::fromStdString(entry->label());
|
|
}
|
|
}
|
|
data->setData(types.at(0), encoded);
|
|
data->setText(plainTextParts.join(QStringLiteral(", ")));
|
|
return data;
|
|
}
|
|
|
|
bool EntryModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent)
|
|
{
|
|
if(!m_rootEntry || !data || action != Qt::MoveAction) {
|
|
return false;
|
|
}
|
|
QStringList types = mimeTypes();
|
|
if(types.isEmpty()) {
|
|
return false;
|
|
}
|
|
QString format = types.at(0);
|
|
if(!data->hasFormat(format)) {
|
|
return false;
|
|
}
|
|
if(row > rowCount(parent) || row < 0) {
|
|
row = rowCount(parent);
|
|
}
|
|
if(column > columnCount(parent) || column < 0) {
|
|
column = 0;
|
|
}
|
|
// decode and insert
|
|
QByteArray encoded = data->data(format);
|
|
QDataStream stream(&encoded, QIODevice::ReadOnly);
|
|
int moved = 0;
|
|
while(!stream.atEnd()) {
|
|
quint32 size;
|
|
stream >> size;
|
|
list<string> path;
|
|
for(quint32 i = 0; i < size; ++i) {
|
|
QString part;
|
|
stream >> part;
|
|
path.push_back(part.toStdString());
|
|
}
|
|
if(Entry *entry = m_rootEntry->entryByPath(path, true)) {
|
|
if(NodeEntry *srcParentEntry = entry->parent()) {
|
|
if(moveRows(index(srcParentEntry), entry->index(), 1, parent, row)) {
|
|
++moved;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Qt::DropActions EntryModel::supportedDropActions() const
|
|
{
|
|
return Qt::MoveAction;
|
|
}
|
|
|
|
}
|