#pragma once

#include "IHyprLayout.hpp"
#include <list>
#include <deque>
#include "../render/decorations/CHyprGroupBarDecoration.hpp"
#include <array>

class CHyprDwindleLayout;
enum eFullscreenMode : uint8_t;

struct SDwindleNodeData {
    SDwindleNodeData*                pParent = nullptr;
    bool                             isNode  = false;

    CWindow*                         pWindow = nullptr;

    std::array<SDwindleNodeData*, 2> children = {nullptr, nullptr};

    bool                             splitTop = false; // for preserve_split

    bool                             groupHead            = false;
    SDwindleNodeData*                pNextGroupMember     = nullptr;
    SDwindleNodeData*                pPreviousGroupMember = nullptr;

    Vector2D                         position;
    Vector2D                         size;

    int                              workspaceID = -1;

    float                            splitRatio = 1.f;

    bool                             valid = true;

    // For list lookup
    bool operator==(const SDwindleNodeData& rhs) {
        return pWindow == rhs.pWindow && workspaceID == rhs.workspaceID && position == rhs.position && size == rhs.size && pParent == rhs.pParent &&
            children[0] == rhs.children[0] && children[1] == rhs.children[1];
    }

    void                recalcSizePosRecursive(bool force = false);
    void                getAllChildrenRecursive(std::deque<SDwindleNodeData*>*);
    bool                isGroupMember();
    SDwindleNodeData*   getGroupHead();
    SDwindleNodeData*   getGroupVisible();
    int                 getGroupMemberCount();
    void                setGroupFocusedNode(SDwindleNodeData*);
    CHyprDwindleLayout* layout = nullptr;
};

class CHyprDwindleLayout : public IHyprLayout {
  public:
    virtual void                     onWindowCreatedTiling(CWindow*);
    virtual void                     onWindowRemovedTiling(CWindow*);
    virtual bool                     isWindowTiled(CWindow*);
    virtual void                     recalculateMonitor(const int&);
    virtual void                     recalculateWindow(CWindow*);
    virtual void                     resizeActiveWindow(const Vector2D&, CWindow* pWindow = nullptr);
    virtual void                     fullscreenRequestForWindow(CWindow*, eFullscreenMode, bool);
    virtual std::any                 layoutMessage(SLayoutMessageHeader, std::string);
    virtual SWindowRenderLayoutHints requestRenderHints(CWindow*);
    virtual void                     switchWindows(CWindow*, CWindow*);
    virtual void                     alterSplitRatio(CWindow*, float, bool);
    virtual std::string              getLayoutName();

    virtual void                     onEnable();
    virtual void                     onDisable();

  private:
    std::list<SDwindleNodeData> m_lDwindleNodesData;

    int                         getNodesOnWorkspace(const int&);
    void                        applyNodeDataToWindow(SDwindleNodeData*, bool force = false);
    SDwindleNodeData*           getNodeFromWindow(CWindow*);
    SDwindleNodeData*           getFirstNodeOnWorkspace(const int&);
    SDwindleNodeData*           getMasterNodeOnWorkspace(const int&);

    void                        toggleWindowGroup(CWindow*);
    void                        switchGroupWindow(CWindow*, bool forward, CWindow* to = nullptr);
    void                        toggleSplit(CWindow*);
    std::deque<CWindow*>        getGroupMembers(CWindow*);

    friend struct SDwindleNodeData;
};