z-bar-qt/Plugins/ZShell/Services/desktopmodel.hpp

#pragma once

#include <QAbstractListModel>
#include <QList>
#include <QString>
#include <QQmlEngine>
#include <QFileSystemWatcher>

namespace ZShell::services {

struct DesktopItem {
	QString fileName;
	QString filePath;
	bool isDir;
	int gridX;
	int gridY;
};

class DesktopModel : public QAbstractListModel {
	Q_OBJECT
	QML_ELEMENT

	public:
	enum DesktopRoles {
		FileNameRole = Qt::UserRole + 1,
		FilePathRole,
		IsDirRole,
		GridXRole,
		GridYRole
	};

	explicit DesktopModel(QObject* parent = nullptr);

	[[nodiscard]] int rowCount(
		const QModelIndex& parent = QModelIndex()) const override;
	[[nodiscard]] QVariant data(
		const QModelIndex& index, int role = Qt::DisplayRole) const override;
	[[nodiscard]] QHash<int, QByteArray> roleNames() const override;

	Q_INVOKABLE void loadDirectory(const QString& path);
	Q_INVOKABLE void moveIcon(int index, int newX, int newY);
	Q_INVOKABLE void massMove(
		const QVariantList& selectedPathsList,
		const QString& leaderPath,
		int targetX,
		int targetY,
		int maxCol,
		int maxRow);

	Q_PROPERTY(int rows READ rows WRITE setRows NOTIFY rowsChanged)

	[[nodiscard]] int rows() const { return m_rows; }
	void setRows(int r) {
		if (m_rows != r) {
			m_rows = r;
			emit rowsChanged();
		}
	}

	signals:
	void rowsChanged();

	private:
	int m_rows = 1;
	QList<DesktopItem> m_items;
	QString m_watchedPath;
	QFileSystemWatcher m_watcher;
	void saveCurrentLayout();
	[[nodiscard]] QPoint getEmptySpot(const QSet<QString>& occupied) const;
	void onDirectoryChanged();
};

}; // namespace ZShell::services