contenttreemodel.h

#ifndef CONTENTTREEMODEL_H
#define CONTENTTREEMODEL_H

#include "contenttree.h"
#include "Torrent.h"
#include "Client.h"

#include <QAbstractItemModel>

namespace Bittorrent {

class ContentTreeModel : public QAbstractItemModel
{
    Q_OBJECT

public:
    explicit ContentTreeModel(Client* client,
                              std::vector<fileObj>* ptr_trackerList,
                              QObject *parent = nullptr);
    explicit ContentTreeModel(const Torrent& modifiedTorrent,
                              QObject *parent = nullptr);
    ~ContentTreeModel();

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


private:
    void setupModelData(ContentTree *parent);
    void setupModelData(const Torrent& modifiedTorrent,
                        ContentTree *parent);
    int findNode(unsigned int& hash,
                   const QList<ContentTree*>& tList);

    std::vector<fileObj>* fileList;
    Client* ioClient;

    ContentTree *rootItem;
};

}

#endif // CONTENTTREEMODEL_H