diff options
author | Aqua-sama <aqua@iserlohn-fortress.net> | 2018-01-07 00:48:27 +0100 |
---|---|---|
committer | Aqua-sama <aqua@iserlohn-fortress.net> | 2018-01-07 00:48:27 +0100 |
commit | e0f64f9330fc30553dc9d8dcf308079f5d391526 (patch) | |
tree | 8c21a08f25208150ed4f0bcbf8496334c2e08bcc /lib/bookmarks/bookmarksmodel.cpp | |
parent | Updated ReadMe and pkgbuild (diff) | |
download | smolbote-e0f64f9330fc30553dc9d8dcf308079f5d391526.tar.xz |
Added BookmarksModel
TODO: Editing bookmarks
TODO: Saving bookmarks
BUG: Completer doesn't search through folders
Diffstat (limited to 'lib/bookmarks/bookmarksmodel.cpp')
-rw-r--r-- | lib/bookmarks/bookmarksmodel.cpp | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/lib/bookmarks/bookmarksmodel.cpp b/lib/bookmarks/bookmarksmodel.cpp new file mode 100644 index 0000000..ddd136e --- /dev/null +++ b/lib/bookmarks/bookmarksmodel.cpp @@ -0,0 +1,177 @@ +/* + * This file is part of smolbote. It's copyrighted by the contributors recorded + * in the version control history of the file, available from its original + * location: git://neueland.iserlohn-fortress.net/smolbote.git + * + * SPDX-License-Identifier: GPL-3.0 + */ + +#include "bookmarksmodel.h" +#include <QStyle> + +BookmarksModel::BookmarksModel(QStyle *style, QObject *parent) : QAbstractItemModel(parent) +{ + folderIcon.addPixmap(style->standardPixmap(QStyle::SP_DirOpenIcon), QIcon::Normal, QIcon::On); + folderIcon.addPixmap(style->standardPixmap(QStyle::SP_DirClosedIcon), QIcon::Normal, QIcon::Off); + bookmarkIcon.addPixmap(style->standardPixmap(QStyle::SP_FileIcon)); + + // create a root item + m_rootItem = new BookmarkItem(BookmarkItem::Root, nullptr); + m_rootItem->title = tr("Title"); + m_rootItem->href = tr("href"); +} + +BookmarksModel::~BookmarksModel() +{ + delete m_rootItem; +} + +void BookmarksModel::setRoot(BookmarkItem *root) +{ + Q_CHECK_PTR(root); + + delete m_rootItem; + m_rootItem = root; +} + +QVariant BookmarksModel::headerData(int section, Qt::Orientation orientation, int role) const +{ + if(orientation == Qt::Horizontal && role == Qt::DisplayRole) { + if(section == 0) { + return m_rootItem->title; + } else if(section == 1) { + return m_rootItem->href; + } else { + return QVariant(); + } + } + + return QVariant(); +} + +QModelIndex BookmarksModel::index(BookmarkItem *node) const +{ + BookmarkItem *parentItem = node->parentItem(); + if(!parentItem) { + return QModelIndex(); + } + + return createIndex(parentItem->childIndex(node), 0, node); +} + +QModelIndex BookmarksModel::index(int row, int column, const QModelIndex &parent) const +{ + // invalid item requested + if (!hasIndex(row, column, parent)) { + return QModelIndex(); + } + + // index exists + BookmarkItem *parentItem; + if(parent.isValid()) { + parentItem = static_cast<BookmarkItem*>(parent.internalPointer()); + } else { + parentItem = m_rootItem; + } + + BookmarkItem *childItem = parentItem->child(row); + if(childItem) { + return createIndex(row, column, childItem); + } + return QModelIndex(); +} + +QModelIndex BookmarksModel::parent(const QModelIndex &index) const +{ + if(!index.isValid()) { + return QModelIndex(); + } + + BookmarkItem *childItem = static_cast<BookmarkItem*>(index.internalPointer()); + BookmarkItem *parentItem = childItem->parentItem(); + + if(parentItem == m_rootItem) { + return QModelIndex(); + } + + return createIndex(parentItem->row(), 0, parentItem); +} + +int BookmarksModel::rowCount(const QModelIndex &parent) const +{ + if(parent.column() > 0) { + return 0; + } + + BookmarkItem *parentItem; + if(!parent.isValid()) { + parentItem = m_rootItem; + } else { + parentItem = static_cast<BookmarkItem*>(parent.internalPointer()); + } + + return parentItem->childCount(); +} + +int BookmarksModel::columnCount(const QModelIndex &parent) const +{ + if(parent.isValid()) { + return static_cast<BookmarkItem*>(parent.internalPointer())->columnCount(); + } else { + return m_rootItem->columnCount(); + } +} + +QVariant BookmarksModel::data(const QModelIndex &index, int role) const +{ + // get data of invalid index? + if (!index.isValid()) { + return QVariant(); + } + + const BookmarkItem *item = static_cast<BookmarkItem *>(index.internalPointer()); + switch (role) { + case Qt::DisplayRole: + case Qt::EditRole: + if(index.column() == 0) { + return item->title; + } else if(index.column() == 1) { + return item->href; + } else { + return QVariant(); + } + + case Qt::DecorationRole: + if(index.column() == 0) { + if(item->type() == BookmarkItem::Folder) { + return folderIcon; + } else if(item->type() == BookmarkItem::Bookmark) { + return bookmarkIcon; + } else { + return QVariant(); + } + } else { + return QVariant(); + } + + case OpenUrlRole: + if(item->type() == BookmarkItem::Bookmark) { + return item->href; + } else { + return QVariant(); + } + + default: + return QVariant(); + } +} + +bool BookmarksModel::hasChildren(const QModelIndex &parent) const +{ + if(!parent.isValid()) { + return true; + } + + const BookmarkItem *parentItem = static_cast<BookmarkItem *>(parent.internalPointer()); + return (parentItem->type() == BookmarkItem::Folder); +} |