#ifndef VMCHOOSER_SESSIONTREEMODEL_H
#define VMCHOOSER_SESSIONTREEMODEL_H

#include <QAbstractItemModel>
#include <QModelIndex>
#include <QVariant>

class SessionTreeItem;
class Session;

class SessionTreeModel : public QAbstractItemModel {
    Q_OBJECT

  public:
    explicit SessionTreeModel(QObject *parent = 0);
    ~SessionTreeModel();

    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;
    int rowCount(const QModelIndex &parent = QModelIndex()) const;
    int columnCount(const QModelIndex &parent = QModelIndex()) const;

    void addItems(const QList<Session*>& sessions, const QString& section);


  private:
    SessionTreeItem* root_;
};

#endif // VMCHOOSER_SESSIONTREEMODEL_H