mirror of
https://github.com/YACReader/yacreader
synced 2025-05-27 19:00:29 -04:00
149 lines
4.6 KiB
C++
149 lines
4.6 KiB
C++
#ifndef TREEMODEL_H
|
|
#define TREEMODEL_H
|
|
|
|
#include <QAbstractItemModel>
|
|
#include <QSortFilterProxyModel>
|
|
#include <QModelIndex>
|
|
#include <QVariant>
|
|
#include <QSqlQuery>
|
|
#include <QSqlDatabase>
|
|
#include <QUrl>
|
|
#include <QIcon>
|
|
|
|
#include "folder.h"
|
|
#include "folder_query_result_processor.h"
|
|
#include "yacreader_global.h"
|
|
|
|
class FolderItem;
|
|
|
|
class FolderModelProxy : public QSortFilterProxyModel
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit FolderModelProxy(QObject *parent = nullptr);
|
|
~FolderModelProxy() override;
|
|
|
|
void setFilterData(QMap<unsigned long long, FolderItem *> *filteredItems, FolderItem *root);
|
|
void clear();
|
|
|
|
bool filterAcceptsRow(int source_row, const QModelIndex &source_parent) const override;
|
|
|
|
protected:
|
|
FolderItem *rootItem;
|
|
QMap<unsigned long long int, FolderItem *> filteredItems; // relación entre folders
|
|
|
|
bool filterEnabled;
|
|
};
|
|
|
|
class FolderModel : public QAbstractItemModel
|
|
{
|
|
Q_OBJECT
|
|
|
|
friend class FolderModelProxy;
|
|
friend class YACReader::FolderQueryResultProcessor;
|
|
|
|
public:
|
|
explicit FolderModel(QObject *parent = nullptr);
|
|
~FolderModel() override;
|
|
|
|
// QAbstractItemModel methods
|
|
QVariant data(const QModelIndex &index, int role) const override;
|
|
Qt::ItemFlags flags(const QModelIndex &index) const override;
|
|
QVariant headerData(int section, Qt::Orientation orientation,
|
|
int role = Qt::DisplayRole) const override;
|
|
QModelIndex index(int row, int column,
|
|
const QModelIndex &parent = QModelIndex()) const override;
|
|
QModelIndex index(qulonglong folderId) const;
|
|
QModelIndex parent(const QModelIndex &index) const override;
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
int columnCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
QHash<int, QByteArray> roleNames() const override;
|
|
|
|
// Convenience methods
|
|
void reload();
|
|
void reload(const QModelIndex &index);
|
|
void setupModelData(QString path);
|
|
QString getDatabase();
|
|
QString getFolderPath(const QModelIndex &folder);
|
|
|
|
void updateFolderCompletedStatus(const QModelIndexList &list, bool status);
|
|
void updateFolderFinishedStatus(const QModelIndexList &list, bool status);
|
|
void updateFolderType(const QModelIndexList &list, YACReader::FileType type);
|
|
void updateTreeType(YACReader::FileType type);
|
|
void setCustomFolderCover(const QModelIndex &index, const QString &path);
|
|
void resetFolderCover(const QModelIndex &index);
|
|
|
|
QStringList getSubfoldersNames(const QModelIndex &mi);
|
|
FolderModel *getSubfoldersModel(const QModelIndex &mi); // it creates a model that contains just the direct subfolders
|
|
|
|
Folder getFolder(const QModelIndex &mi);
|
|
QModelIndex getIndexFromFolderId(qulonglong folderId, const QModelIndex &parent = QModelIndex());
|
|
QModelIndex getIndexFromFolder(const Folder &folder, const QModelIndex &parent = QModelIndex());
|
|
|
|
QModelIndex addFolderAtParent(const QString &folderName, const QModelIndex &parent);
|
|
|
|
Q_INVOKABLE QUrl getCoverUrlPathForComicHash(const QString &hash) const;
|
|
Q_INVOKABLE QUrl getCoverUrlPathForFolderId(qulonglong folderId) const;
|
|
|
|
void setShowRecent(bool showRecent);
|
|
void setRecentRange(int days);
|
|
|
|
enum Columns {
|
|
Name = 0,
|
|
Path,
|
|
Finished,
|
|
Completed,
|
|
NumChildren,
|
|
FirstChildHash,
|
|
CustomImage,
|
|
Type, // FileType
|
|
Added,
|
|
Updated,
|
|
};
|
|
|
|
enum Roles {
|
|
FinishedRole = Qt::UserRole + 1,
|
|
CompletedRole,
|
|
IdRole,
|
|
CoverPathRole,
|
|
FolderNameRole,
|
|
NumChildrenRole,
|
|
TypeRole,
|
|
AddedRole,
|
|
UpdatedRole,
|
|
ShowRecentRole,
|
|
RecentRangeRole,
|
|
};
|
|
|
|
bool isSubfolder;
|
|
public slots:
|
|
void deleteFolder(const QModelIndex &mi);
|
|
void updateFolderChildrenInfo(qulonglong folderId);
|
|
|
|
private:
|
|
struct ModelData {
|
|
FolderItem *rootItem; // items tree
|
|
QMap<unsigned long long int, FolderItem *> items; // items lookup
|
|
};
|
|
|
|
void setModelData(const ModelData &modelData);
|
|
ModelData createModelData(const QString &path) const;
|
|
ModelData createModelData(QSqlQuery &sqlquery, FolderItem *parent) const;
|
|
|
|
// parent contains the current data in the model (parentModelIndex is its index), updated contains fresh info loaded from the DB,
|
|
void takeUpdatedChildrenInfo(FolderItem *parent, const QModelIndex &parentModelIndex, FolderItem *updated);
|
|
|
|
FolderItem *rootItem; // items tree
|
|
QMap<unsigned long long int, FolderItem *> items; // items lookup
|
|
|
|
QString _databasePath;
|
|
|
|
QIcon folderIcon;
|
|
QIcon folderFinishedIcon;
|
|
|
|
bool showRecent;
|
|
qlonglong recentDays;
|
|
};
|
|
|
|
#endif
|