#pragma once

#include <list>
#include <src/layout/IHyprLayout.hpp>

class Hy3Layout;
struct Hy3Node;

enum class Hy3GroupLayout {
	SplitH,
	SplitV,
	Tabbed,
};

enum class ShiftDirection {
	Left,
	Up,
	Down,
	Right,
};

struct Hy3GroupData {
	Hy3GroupLayout layout = Hy3GroupLayout::SplitH;
	std::list<Hy3Node*> children;

	bool hasChild(Hy3Node* child);

	Hy3GroupData(Hy3GroupLayout layout);

private:
	Hy3GroupData(Hy3GroupData&&) = default;
	Hy3GroupData(const Hy3GroupData&) = default;

	friend class Hy3NodeData;
};

void swapNodeData(Hy3Node& a, Hy3Node& b);

class Hy3NodeData {
public:
	enum { Group, Window } type;
	union {
		Hy3GroupData as_group;
		CWindow* as_window;
	};

	bool operator==(const Hy3NodeData&) const;

	Hy3NodeData();
	~Hy3NodeData();
	Hy3NodeData(CWindow*);
	Hy3NodeData(Hy3GroupLayout);
	Hy3NodeData& operator=(CWindow*);
	Hy3NodeData& operator=(Hy3GroupLayout);

	//private: - I give up, C++ wins
	Hy3NodeData(Hy3GroupData);
	Hy3NodeData(const Hy3NodeData&);
	Hy3NodeData(Hy3NodeData&&);
	Hy3NodeData& operator=(const Hy3NodeData&);

	friend void swapNodeData(Hy3Node&, Hy3Node&);
};

struct Hy3Node {
	Hy3Node* parent = nullptr;
	Hy3NodeData data;
	Vector2D position;
	Vector2D size;
	float size_ratio = 1.0;
	int workspace_id = -1;
	bool valid = true;
	Hy3Layout* layout = nullptr;

	void recalcSizePosRecursive(bool force = false);
	// remove a child node, returns true on success.
	// fails if not a group
	// if only a single child node remains && childSwallows, replace this group with said child.
	// if no children remain, remove this node from its parent.
	bool removeChild(Hy3Node* child, bool childSwallows = false);

	bool operator==(const Hy3Node&) const;
};

class Hy3Layout: public IHyprLayout {
public:
	virtual void onWindowCreatedTiling(CWindow*);
	virtual void onWindowRemovedTiling(CWindow*);
	virtual void onWindowFocusChange(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 replaceWindowDataWith(CWindow* from, CWindow* to);

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

	void makeGroupOn(CWindow*, Hy3GroupLayout);
	void shiftWindow(CWindow*, ShiftDirection);
	void shiftFocus(CWindow*, ShiftDirection);

private:
	std::list<Hy3Node> nodes;
	CWindow* lastActiveWindow = nullptr;

	int getWorkspaceNodeCount(const int&);
	Hy3Node* getNodeFromWindow(CWindow*);
	Hy3Node* getWorkspaceRootGroup(const int&);
	void applyNodeDataToWindow(Hy3Node*, bool force = false);

	friend struct Hy3Node;
};