78 lines
2.3 KiB
C++
78 lines
2.3 KiB
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include <QAbstractItemModel>
|
||
|
|
#include <QModelIndex>
|
||
|
|
#include <QVariant>
|
||
|
|
#include <QVector>
|
||
|
|
#include <QDateTime>
|
||
|
|
#include <QDate>
|
||
|
|
#include "core/mailitem.h"
|
||
|
|
|
||
|
|
class EmailTreeModel : public QAbstractItemModel
|
||
|
|
{
|
||
|
|
Q_OBJECT
|
||
|
|
|
||
|
|
public:
|
||
|
|
enum Columns {
|
||
|
|
ColSubject = 0,
|
||
|
|
ColSender = 1,
|
||
|
|
ColDate = 2,
|
||
|
|
ColCount = 3
|
||
|
|
};
|
||
|
|
|
||
|
|
enum Roles {
|
||
|
|
MailIdRole = Qt::UserRole + 1,
|
||
|
|
IsGroupRole = Qt::UserRole + 2,
|
||
|
|
GroupNameRole = Qt::UserRole + 3,
|
||
|
|
GroupIndexRole = Qt::UserRole + 4,
|
||
|
|
MailIndexRole = Qt::UserRole + 5
|
||
|
|
};
|
||
|
|
|
||
|
|
explicit EmailTreeModel(QObject *parent = nullptr);
|
||
|
|
~EmailTreeModel() override;
|
||
|
|
|
||
|
|
// Basic model interface
|
||
|
|
QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const override;
|
||
|
|
QModelIndex parent(const QModelIndex &child) const override;
|
||
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
|
||
|
|
int columnCount(const QModelIndex &parent = QModelIndex()) const override;
|
||
|
|
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
|
||
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
|
||
|
|
Qt::ItemFlags flags(const QModelIndex &index) const override;
|
||
|
|
|
||
|
|
// Email data management
|
||
|
|
void setEmails(const QVector<MailItem> &emails);
|
||
|
|
void clear();
|
||
|
|
|
||
|
|
// Grouping
|
||
|
|
enum GroupMode {
|
||
|
|
NoGrouping = 0,
|
||
|
|
GroupByDate = 1
|
||
|
|
};
|
||
|
|
void setGroupMode(GroupMode mode);
|
||
|
|
GroupMode groupMode() const { return m_groupMode; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
struct TreeItem {
|
||
|
|
enum Type { Group, Mail } type;
|
||
|
|
QString groupName; // for Group type
|
||
|
|
int groupIndex = -1; // for Group type
|
||
|
|
int mailIndex = -1; // for Mail type (index in m_emails)
|
||
|
|
QVector<TreeItem*> children;
|
||
|
|
TreeItem* parent = nullptr;
|
||
|
|
int row = -1;
|
||
|
|
|
||
|
|
TreeItem(Type t, TreeItem* p = nullptr) : type(t), parent(p) {}
|
||
|
|
~TreeItem() { qDeleteAll(children); }
|
||
|
|
};
|
||
|
|
|
||
|
|
GroupMode m_groupMode = NoGrouping;
|
||
|
|
QVector<MailItem> m_emails;
|
||
|
|
TreeItem* m_rootItem = nullptr;
|
||
|
|
|
||
|
|
void setupModelData();
|
||
|
|
void buildGroups();
|
||
|
|
QString groupNameForDate(const QDateTime &date) const;
|
||
|
|
int groupIndexForDate(const QDateTime &date) const;
|
||
|
|
void deleteTree(TreeItem* item);
|
||
|
|
};
|