SnapState.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 fuddlesworth
// SPDX-License-Identifier: LGPL-2.1-or-later
 
#pragma once
 
#include <phosphorsnapengine_export.h>
#include <PhosphorEngine/IPlacementState.h>
#include <PhosphorEngine/IWindowRegistry.h>
 
#include <QHash>
#include <QJsonObject>
#include <QObject>
#include <QRect>
#include <QSet>
#include <QString>
#include <QStringList>
 
#include <optional>
 
namespace PhosphorSnapEngine {
 
class PHOSPHORSNAPENGINE_EXPORT SnapState : public QObject, public PhosphorEngine::IPlacementState
{
    Q_OBJECT
 
public:
    explicit SnapState(const QString& screenId, QObject* parent = nullptr);
    ~SnapState() override;
 
    SnapState(const SnapState&) = delete;
    SnapState& operator=(const SnapState&) = delete;
 
    void setWindowRegistry(PhosphorEngine::IWindowRegistry* registry)
    {
        m_windowRegistry = registry;
    }
 
    // ═══════════════════════════════════════════════════════════════════════════
    // IPlacementState
    // ═══════════════════════════════════════════════════════════════════════════
 
    QString screenId() const override;
    int windowCount() const override;
    QStringList managedWindows() const override;
    bool containsWindow(const QString& windowId) const override;
    bool isFloating(const QString& windowId) const override;
    QStringList floatingWindows() const override;
    QString placementIdForWindow(const QString& windowId) const override;
    QJsonObject toJson() const override;
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Zone Assignment CRUD
    // ═══════════════════════════════════════════════════════════════════════════
 
    void assignWindowToZone(const QString& windowId, const QString& zoneId, const QString& screenId,
                            int virtualDesktop);
    void assignWindowToZones(const QString& windowId, const QStringList& zoneIds, const QString& screenId,
                             int virtualDesktop);
    struct UnassignResult
    {
        bool wasAssigned = false;
        bool lastUsedZoneCleared = false;
    };
    UnassignResult unassignWindow(const QString& windowId);
 
    QString zoneForWindow(const QString& windowId) const;
    QStringList zonesForWindow(const QString& windowId) const;
    QStringList windowsInZone(const QString& zoneId) const;
    QStringList snappedWindows() const;
    bool isWindowSnapped(const QString& windowId) const;
 
    QString screenForWindow(const QString& windowId) const;
    int desktopForWindow(const QString& windowId) const;
 
    bool reassignDesktop(const QString& windowId, int virtualDesktop);
 
    QStringList windowsOnScreenAndDesktop(const QString& screenId, int virtualDesktop) const;
 
    const QHash<QString, QString>& screenAssignments() const
    {
        return m_windowScreenAssignments;
    }
    const QHash<QString, int>& desktopAssignments() const
    {
        return m_windowDesktopAssignments;
    }
    void setScreenAssignments(const QHash<QString, QString>& s)
    {
        if (m_windowScreenAssignments == s) {
            return;
        }
        m_windowScreenAssignments = s;
        Q_EMIT stateChanged();
    }
    void setDesktopAssignments(const QHash<QString, int>& d)
    {
        if (m_windowDesktopAssignments == d) {
            return;
        }
        m_windowDesktopAssignments = d;
        Q_EMIT stateChanged();
    }
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Floating State
    // ═══════════════════════════════════════════════════════════════════════════
 
    void setFloating(const QString& windowId, bool floating);
 
    void setFloatingOnScreen(const QString& windowId, const QString& screenId, int virtualDesktop);
 
    UnassignResult unsnapForFloat(const QString& windowId);
    QString preFloatZone(const QString& windowId) const;
    QStringList preFloatZones(const QString& windowId) const;
    QString preFloatScreen(const QString& windowId) const;
    void clearPreFloatZone(const QString& windowId);
    void addPreFloatZone(const QString& windowId, const QStringList& zoneIds);
    void addPreFloatScreen(const QString& windowId, const QString& screenId);
 
    const QHash<QString, QStringList>& preFloatZoneAssignments() const
    {
        return m_preFloatZoneAssignments;
    }
    const QHash<QString, QString>& preFloatScreenAssignments() const
    {
        return m_preFloatScreenAssignments;
    }
    void setPreFloatScreenAssignments(const QHash<QString, QString>& a)
    {
        if (m_preFloatScreenAssignments == a) {
            return;
        }
        m_preFloatScreenAssignments = a;
        Q_EMIT stateChanged();
    }
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Window Lifecycle
    // ═══════════════════════════════════════════════════════════════════════════
 
    void windowClosed(const QString& windowId);
    bool isEmpty() const;
    void clear();
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Rotation
    // ═══════════════════════════════════════════════════════════════════════════
 
    QStringList rotateAssignments(bool clockwise);
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Last-Used Zone Tracking
    // ═══════════════════════════════════════════════════════════════════════════
 
    void updateLastUsedZone(const QString& zoneId, const QString& screenId, const QString& windowClass,
                            int virtualDesktop);
 
    void restoreLastUsedZone(const QString& zoneId, const QString& screenId, const QString& zoneClass, int desktop);
 
    QString lastUsedZoneId() const
    {
        return m_lastUsedZoneId;
    }
    QString lastUsedScreenId() const
    {
        return m_lastUsedScreenId;
    }
    QString lastUsedZoneClass() const
    {
        return m_lastUsedZoneClass;
    }
    int lastUsedDesktop() const
    {
        return m_lastUsedDesktop;
    }
    void retagLastUsedZoneClass(const QString& newClass)
    {
        if (m_lastUsedZoneClass == newClass) {
            return;
        }
        m_lastUsedZoneClass = newClass;
        Q_EMIT stateChanged();
    }
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Auto-Snap Bookkeeping
    // ═══════════════════════════════════════════════════════════════════════════
 
    void recordSnapIntent(const QString& windowClass, bool wasUserInitiated);
    const QSet<QString>& userSnappedClasses() const
    {
        return m_userSnappedClasses;
    }
    void setUserSnappedClasses(const QSet<QString>& classes)
    {
        if (m_userSnappedClasses == classes) {
            return;
        }
        m_userSnappedClasses = classes;
        Q_EMIT stateChanged();
    }
 
    void markAsAutoSnapped(const QString& windowId);
    bool isAutoSnapped(const QString& windowId) const;
    bool clearAutoSnapped(const QString& windowId);
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Occupied Zone Queries
    // ═══════════════════════════════════════════════════════════════════════════
 
    QSet<QString> buildOccupiedZoneSet(const QString& screenFilter = {}, int desktopFilter = 0) const;
 
    int pruneStaleAssignments(const QSet<QString>& aliveWindowIds);
 
    // ═══════════════════════════════════════════════════════════════════════════
    // Deserialization
    // ═══════════════════════════════════════════════════════════════════════════
 
    static SnapState* fromJson(const QJsonObject& json, QObject* parent = nullptr);
 
    // ═══════════════════════════════════════════════════════════════════════════
    // State Access (for persistence layer)
    // ═══════════════════════════════════════════════════════════════════════════
 
    const QHash<QString, QStringList>& zoneAssignments() const
    {
        return m_windowZoneAssignments;
    }
 
    void setZoneAssignments(const QHash<QString, QStringList>& zones)
    {
        if (m_windowZoneAssignments == zones) {
            return;
        }
        m_windowZoneAssignments = zones;
        Q_EMIT stateChanged();
    }
    void setFloatingWindows(const QSet<QString>& windows)
    {
        if (m_floatingWindows == windows) {
            return;
        }
        m_floatingWindows = windows;
        Q_EMIT stateChanged();
    }
    void setPreFloatZoneAssignments(const QHash<QString, QStringList>& a)
    {
        if (m_preFloatZoneAssignments == a) {
            return;
        }
        m_preFloatZoneAssignments = a;
        Q_EMIT stateChanged();
    }
 
Q_SIGNALS:
    void windowAssigned(const QString& windowId, const QString& zoneId);
    void windowUnassigned(const QString& windowId);
    void floatingChanged(const QString& windowId, bool floating);
    void stateChanged();
 
private:
    QString canonicalizeForLookup(const QString& rawWindowId) const;
 
    bool removeWindowData(const QString& windowId);
 
    UnassignResult clearZoneAssignment(const QString& windowId, bool preserveScreenAndDesktop);
 
    QSet<QString> allManagedWindowIds() const
    {
        QSet<QString> all;
        all.reserve(m_windowZoneAssignments.size() + m_floatingWindows.size() + m_autoSnappedWindows.size());
        for (auto it = m_windowZoneAssignments.constBegin(); it != m_windowZoneAssignments.constEnd(); ++it) {
            all.insert(it.key());
        }
        all.unite(m_floatingWindows);
        all.unite(m_autoSnappedWindows);
        return all;
    }
 
    QString m_screenId;
 
    PhosphorEngine::IWindowRegistry* m_windowRegistry = nullptr;
 
    QHash<QString, QStringList> m_windowZoneAssignments;
    QHash<QString, QString> m_windowScreenAssignments;
    QHash<QString, int> m_windowDesktopAssignments;
    QSet<QString> m_floatingWindows;
    QHash<QString, QStringList> m_preFloatZoneAssignments;
    QHash<QString, QString> m_preFloatScreenAssignments;
 
    QString m_lastUsedZoneId;
    QString m_lastUsedScreenId;
    QString m_lastUsedZoneClass;
    int m_lastUsedDesktop = 0;
 
    QSet<QString> m_userSnappedClasses;
    QSet<QString> m_autoSnappedWindows;
};
 
} // namespace PhosphorSnapEngine