#pragma once
#include <QFileSystemModel>
#include <QSet>
#include <QString>
#include <QStringList>

namespace whittle {

	class FolderTreeModel : public QFileSystemModel {
		Q_OBJECT
	public:
		explicit FolderTreeModel(QObject* parent = nullptr);

		void setRecursive(bool on);
		bool recursive() const { return m_recursive; }
		QStringList scanRoots() const { return QStringList(m_checked.values()); }
		QStringList excludedPaths() const { return QStringList(m_excluded.values()); }
		void setCheckedRoots(const QStringList& roots,
			const QStringList& excluded = {});
		void clearChecks();

		Qt::ItemFlags flags(const QModelIndex&) const override;
		QVariant data(const QModelIndex&, int role) const override;
		bool setData(const QModelIndex&, const QVariant&, int role) override;

		signals:
		void selectionChanged();

	private:
		Qt::CheckState stateFor(const QString& path) const;
		bool effectivelyChecked(const QString& path) const;
		void check(const QString& path);
		void uncheck(const QString& path);
		void emitSubtree(const QModelIndex&);
		void emitAncestors(const QModelIndex&);

		QSet<QString> m_checked;
		QSet<QString> m_excluded;
		bool m_recursive = true;
	};

}