#ifndef __gui__pages__torrents_h
#define __gui__pages__torrents_h

#include "core/torrent.h"
#include <QFrame>
#include <QAbstractListModel>
#include <QSortFilterProxyModel>
#include <QItemSelection>

class QTreeView;

class TorrentsPageListSortFilter final : public QSortFilterProxyModel {
		Q_OBJECT

	public:
		TorrentsPageListSortFilter(QObject* parent = nullptr);

	protected:
		bool lessThan(const QModelIndex& l, const QModelIndex& r) const override;
};

class TorrentsPageListModel final : public QAbstractListModel {
		Q_OBJECT

	public:
		enum columns {
			TL_TITLE,
			TL_EPISODE,
			TL_GROUP,
			TL_SIZE,
			TL_RESOLUTION,
			TL_SEEDERS, 
			TL_LEECHERS,
			TL_DOWNLOADERS,
			TL_DESCRIPTION,
			TL_FILENAME,
			TL_RELEASEDATE,

			NB_COLUMNS
		};

		TorrentsPageListModel(QObject* parent);
		~TorrentsPageListModel() override = default;
		int rowCount(const QModelIndex& parent = QModelIndex()) const override;
		int columnCount(const QModelIndex& parent = QModelIndex()) const override;
		bool setData(const QModelIndex& index, const QVariant& value, int role) override;
		QVariant data(const QModelIndex& index, int role) const override;
		QVariant headerData(const int section, const Qt::Orientation orientation, const int role) const override;
		Qt::ItemFlags flags(const QModelIndex& index) const override;

		QByteArray DownloadTorrentList();
		void DownloadTorrents(QItemSelection selection);
		void ParseFeedDescription(const std::string& description, Torrent& torrent);
		void ParseTorrentList(const QByteArray& ba);
		void RefreshTorrentList();

	private:
		class TorrentModelItem : public Torrent {
			public:
				bool GetChecked() const { return _checked; };
				void SetChecked(bool checked) { _checked = checked; };

			private:
				bool _checked = false;
		};

		std::vector<TorrentModelItem> list;
};

class TorrentsPage final : public QFrame {
		Q_OBJECT

	public:
		TorrentsPage(QWidget* parent = nullptr);
		void DownloadSelection();
		void Refresh();

	private:
		TorrentsPageListModel* model = nullptr;
		TorrentsPageListSortFilter* sort_model = nullptr;
		QTreeView* treeview = nullptr;
};

#endif // __gui__pages__torrents_h
