yacreader/YACReaderLibrary/db/reading_list_model.h

#ifndef READING_LIST_MODEL_H
#define READING_LIST_MODEL_H

#include <QAbstractItemModel>
#include <QModelIndex>
#include <QVariant>
#include <QSqlQuery>
#include <QSqlDatabase>

class LabelItem;
class SpecialListItem;
class ReadingListItem;

class ReadingListModel : public QAbstractItemModel
{
    Q_OBJECT
public:
    explicit ReadingListModel(QObject *parent = 0);

    //QAbstractItemModel methods
    int rowCount(const QModelIndex &parent = QModelIndex()) const;
    int columnCount(const QModelIndex &parent = QModelIndex()) const;
    QVariant data(const QModelIndex &index, int role) const;
    Qt::ItemFlags flags(const QModelIndex &index) const;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const;
    QModelIndex index(int row, int column,
                      const QModelIndex &parent = QModelIndex()) const;
    QModelIndex parent(const QModelIndex &index) const;

    //Convenience methods
    void setupModelData(QString path);

signals:

public slots:
    void deleteItem(const QModelIndex & mi);

private:
    void setupModelData(QSqlQuery &sqlquery, ReadingListItem *parent);

    //Special lists
    QList<SpecialListItem *> specialLists;

    //Label
    QList<LabelItem *> labels;

    //Reading lists
    ReadingListItem * rootItem; //
    QMap<unsigned long long int, ReadingListItem *> items; //lists relationship

    QString _databasePath;

};

#endif // READING_LIST_MODEL_H