database.h

#ifndef DATABASE_H
#define DATABASE_H

#include "model/object.h"

#include <QAbstractItemModel>
#include <QString>

class DatabaseItem
{
public:
    explicit DatabaseItem(const QString &path, DatabaseItem *parentItem = nullptr);
    ~DatabaseItem();

    void appendItem(DatabaseItem *child);

    // GETTERS
    QString getName() const;
    QString getPath() const;
    QImage getIcon() const;
    Object *getObject();
    Qt::ItemFlags getFlags() const;
    bool isDir() const;

    // SETTERS
    void setPath(QString path);
    void setName(const QString &name);

    // BOILERPLATE
    DatabaseItem *getChild(int row);
    int getChildCount() const;
    int getColumnCount() const;
    QVariant getData(int column) const;
    int getRow() const;
    DatabaseItem *getParentItem();

private:
    QString path, name;
    DatabaseItem *parent;
    QVector<DatabaseItem*> childItems;
    CompositeObject *object{};
    mutable QImage *icon{};
};

class Database : public QAbstractItemModel
{
    Q_OBJECT
public:
    explicit Database(const QString &path);
    ~Database();

    QVariant data(const QModelIndex &index, int role) const override;
    void update();
    bool iterateLeaves(const QModelIndex &index,
                 std::function<bool(const QModelIndex&)> fun);

    // BOILERPLATE
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    QModelIndex index(int row, int column, const QModelIndex &parent = {}) const
        override;
    QModelIndex parent(const QModelIndex &index) const override;
    int rowCount(const QModelIndex &parent = {}) const override;
    int columnCount(const QModelIndex &parent = {}) const override;

private:
    // HELPERS
    void iterate(const QString &dir, DatabaseItem *parent);

    // ATTRIBUTES
    DatabaseItem *rootItem{};
    QString path;
};

#endif // DATABASE_H