path: root/src/sessiontreemodel.h

#ifndef VMCHOOSER_SESSIONTREEMODEL_H
#define VMCHOOSER_SESSIONTREEMODEL_H

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

#include "session.h"

class SessionTreeItem;

class SessionTreeModel : public QAbstractItemModel {
    Q_OBJECT

  public:
    explicit SessionTreeModel(QObject *parent = nullptr, bool ignoreSections = false);
    ~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);
    void addLabelItem(const QString& label);
    void removeItem(const QString& name);
    QList<Session*> lookForItem(const QString& label);
    QModelIndex getSection(const SectionType type);

    void updateView();


  private:
    SessionTreeItem* root_;
    bool ignoreSections_;
};

#endif // VMCHOOSER_SESSIONTREEMODEL_H