#include "GraphicApplication.hpp"

#include "RootWidget.hpp"
#include "AnimationTestWidget.hpp"
#include "FloatingWidget.hpp"
#include "DockWidget.hpp"

using namespace tp;


/*
 * adjust child widgets in the layout view
 * make scrollable area
 * change Widget::clampMinMax function to include child enclosure
 * refactor all sizing in Widget -> move to separate files
 * refactor root widget -> reorganize code
*/


class WidgetApplication : public Application {
public:
	WidgetApplication() {
		setup1();
	}

	void setupLay() {
		mRootWidget.setRootWidget(&mWidget);

		mWidget.addChild(&mWidgets[1]);

		mWidgets[1].addChild(&mButtons[1]);
		mWidgets[1].addChild(&mButtons[2]);
		mWidgets[1].addChild(&mWidgets[2]);

		RootWidget::setWidgetArea(mWidgets[1], { 300, 100, 350, 800 });
		RootWidget::setWidgetArea(mButtons[2], { 100, 100, 150, 300 });

		mWidgets[2].addChild(&mButtons[3]);
		mWidgets[2].addChild(&mButtons[4]);
		mWidgets[2].addChild(&mButtons[5]);
		mWidgets[2].addChild(&mButtons[6]);

		mButtons[1].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);
		mButtons[2].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);
		mButtons[3].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);
		mButtons[4].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);
		mButtons[5].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);
		mButtons[6].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);

		mWidgets[2].setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);
		((BasicLayout*)mWidgets[1].getLayout())->setLayoutPolicy(LayoutPolicy::Horizontal);
	}

	void setup1() {
		mRootWidget.setRootWidget(&mDockLayout);

		mDockLayout.addChild(&mFloatingMenu2);
		mDockLayout.addChild(&mFloatingMenu);

		mDockLayout.setCenterWidget(&mButton5);
		mDockLayout.dockWidget(&mButton6, DockLayout::RIGHT);

		mFloatingMenu.addToMenu(&mButton);
		mFloatingMenu2.addToMenu(&mButton2);
		//mFloatingMenu2.addToMenu(&mFloatingMenu);
		mFloatingMenu2.addToMenu(&mButton3);
		mFloatingMenu2.addToMenu(&mButton4);

		mButton.setAction([this]() { mButton2.setColor(RGBA::random()); });
		mButton2.setAction([this]() { mButton.setColor(RGBA::random()); });

		mButton5.setAction([this]() { mDockLayout.dockWidget(&mFloatingMenu, DockLayout::LEFT); });
		mButton6.setAction([this]() { mDockLayout.undockWidget(DockLayout::LEFT); });

		mButton3.setAction([this]() { mDockLayout.toggleWidgetVisibility(DockLayout::LEFT); });
		// mButton4.setAction([this]() { mDockLayout.undockWidget(DockLayout::LEFT); });

		mButton3.setText("toggle");
		mButton5.setText("dock");
		mButton6.setText("undock");

		// mLayoutWidget.addChild(&mButton2);

		// mWidgetFloating.addChild(&mButton);

		// mRootWidget.setRootWidget(&mAnimationTestWidget);

		// mAnimationTestWidget.addChild(&mWidgetFloating);
		// mAnimationTestWidget.addChild(&mLayoutWidget);

		RootWidget::setWidgetArea(mFloatingMenu2, { 300, 100, 150, 500 });
		RootWidget::setWidgetArea(mFloatingMenu, { 100, 100, 150, 300 });

		// mWidgetFloating.addChild(&mLayoutWidget);

		// mLayoutWidget.addChild(&mButton);
		// mLayoutWidget.addChild(&mLabel);
	}

	void setup3() {
		mRootWidget.setRootWidget(&mWidget);
		mWidget.addChild(&mFloatingWidget);

		mFloatingWidget.addChild(&mButton);
		mFloatingWidget.addChild(&mButton2);
		mFloatingWidget.addChild(&mButton3);

		mButton.setSizePolicy(SizePolicy::Minimal, SizePolicy::Expanding);
		mButton2.setSizePolicy(SizePolicy::Minimal, SizePolicy::Expanding);
		mButton3.setSizePolicy(SizePolicy::Expanding, SizePolicy::Expanding);

		RootWidget::setWidgetArea(mFloatingWidget, { 100, 100, 400, 400 });
	}

	void setup2() {
		mRootWidget.setRootWidget(&mWidget);
		// mDesktopLayout.addChild(&mWidget);

		mWidget.addChild(&mButton);
		mWidget.addChild(&mButton2);
		mWidget.addChild(&mButton3);

		mButton.setSizePolicy(SizePolicy::Expanding, SizePolicy::Minimal);
		mButton2.setSizePolicy(SizePolicy::Expanding, SizePolicy::Minimal);
		mButton3.setSizePolicy(SizePolicy::Expanding, SizePolicy::Minimal);

		mWidget.setSizePolicy(SizePolicy::Minimal, SizePolicy::Minimal);
		// mWidget.setLayoutPolicy(LayoutPolicy::Horizontal);

		// mButton.setAction([this]() { mButton2.setMinSize(mButton2.getMinSize() + 10); });
		// mButton2.setAction([this]() { mButton.setMinSize(mButton.getMinSize() + 10); } );

		RootWidget::setWidgetArea(mWidget, { 400, 100, 150, 200 });
	}

	void processFrame(EventHandler* eventHandler, halnf deltaTime) override {
		const auto rec = RectF({ 0, 0 }, mWindow->getSize());
		mRootWidget.processFrame(eventHandler, rec);
	}

	void drawFrame(Canvas* canvas) override {
		mRootWidget.drawFrame(*canvas);

		drawDebug();
	}

	bool forceNewFrame() override {
		return mRootWidget.needsUpdate();
	}

private:
	Widget mWidgets[15];
	ButtonWidget mButtons[15];

	Widget mWidget;

	ButtonWidget mButton;
	ButtonWidget mButton2;
	ButtonWidget mButton3;
	ButtonWidget mButton4;
	ButtonWidget mButton5;
	ButtonWidget mButton6;

	FloatingWidget mFloatingWidget;
	FloatingMenu mFloatingMenu;
	FloatingMenu mFloatingMenu2;

	LabelWidget mLabel;
	FloatingWidget mWidgetFloating;
	AnimationTestWidget mAnimationTestWidget;

	DockWidget mDockLayout;

	RootWidget mRootWidget;
};

int main() {
	{
		WidgetApplication gui;
		gui.run();
	}
}