#include "GUI.h"
#include "RenderManager.h"
#include <algorithm>
#include "tinyxml2.h"
#include "Menu.h"

CUserCmd* m_cmd;
Vector m_position;
Vector m_size;
IClientEntity* m_player;

CGUI GUI;

CGUI::CGUI() {}

// Draws all windows
void CGUI::Draw() {
	bool ShouldDrawCursor = false;

	for ( auto window : Windows ) {
		if ( window->m_bIsOpen ) {
			ShouldDrawCursor = true;
			if ( window == Windows[1] ) {
				DrawWindow(window, 2);
			}
			else {
				DrawWindow(window, 1);
			}
		}
	}

	if ( ShouldDrawCursor ) {
		if ( ShouldDrawCursor ) {
			Render::Clear(Mouse.x + 1, Mouse.y, 1, 17, Color(3, 6, 26, 255));
			for ( int i = 0; i < 11; i++ )
				Render::Clear(Mouse.x + 2 + i, Mouse.y + 1 + i, 1, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 8, Mouse.y + 12, 5, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 8, Mouse.y + 13, 1, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 9, Mouse.y + 14, 1, 2, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 10, Mouse.y + 16, 1, 2, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 8, Mouse.y + 18, 2, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 7, Mouse.y + 16, 1, 2, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 6, Mouse.y + 14, 1, 2, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 5, Mouse.y + 13, 1, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 4, Mouse.y + 14, 1, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 3, Mouse.y + 15, 1, 1, Color(3, 6, 26, 255));
			Render::Clear(Mouse.x + 2, Mouse.y + 16, 1, 1, Color(3, 6, 26, 255));
			for ( int i = 0; i < 4; i++ )
				Render::Clear(Mouse.x + 2 + i, Mouse.y + 2 + i, 1, 14 - (i * 2), Color(37 - (i * 4), 137 - (i * 4), 255 - (i * 4), 255));
			Render::Clear(Mouse.x + 6, Mouse.y + 6, 1, 8, Color(0, 150, 255, 255));
			Render::Clear(Mouse.x + 7, Mouse.y + 7, 1, 9, Color(0, 150, 255, 255));
			for ( int i = 0; i < 4; i++ )
				Render::Clear(Mouse.x + 8 + i, Mouse.y + 8 + i, 1, 4 - i, Color(37 - (i * 4), 137 - (i * 4), 255 - (i * 4), 255));
			Render::Clear(Mouse.x + 8, Mouse.y + 14, 1, 4, Color(0, 150, 255, 255));
			Render::Clear(Mouse.x + 9, Mouse.y + 16, 1, 2, Color(0, 150, 255, 255));
		}
	}
}

// Handle all input etc
void CGUI::Update() {
	static int bWasntHolding = false;
	static int bGrabbing = false;
	static int iOffsetX = 0, iOffsetY = 0;
	//Key Array
	std::copy(keys, keys + 255, oldKeys);
	for ( int x = 0; x < 255; x++ ) {
		keys[x] = (GetAsyncKeyState(x));
	}

	POINT mp;
	GetCursorPos(&mp);
	ScreenToClient(GetForegroundWindow(), &mp);
	Mouse.x = mp.x;
	Mouse.y = mp.y;

	RECT Screen = Render::GetViewport();

	// Window Binds
	for ( auto& bind : WindowBinds ) {
		if ( GetKeyPress(bind.first) ) {
			bind.second->Toggle();
		}
	}

	// Stop dragging
	if ( IsDraggingWindow && !GetKeyState(VK_LBUTTON) ) {
		IsDraggingWindow = false;
		DraggingWindow = nullptr;
	}

	// If we are in the proccess of dragging a window
	if ( IsDraggingWindow && GetKeyState(VK_LBUTTON) && !GetKeyPress(VK_LBUTTON) ) {
		if ( DraggingWindow ) {
			DraggingWindow->m_x = Mouse.x - DragOffsetX;
			DraggingWindow->m_y = Mouse.y - DragOffsetY;
		}
	}
	// Process some windows
	for ( auto window : Windows ) {
		if ( window->m_bIsOpen ) {
			// Used to tell the widget processing that there could be a click
			bool bCheckWidgetClicks = false;

			// If the user clicks inside the window
			if ( GetKeyPress(VK_LBUTTON) || GetKeyPress(VK_RETURN) ) {
				if ( IsMouseInRegion(window->GetClientArea1()) ) {
					// User is selecting a new tab
					if ( IsMouseInRegion(window->GetTabArea()) ) {
						// Loose focus on the control
						window->IsFocusingControl = false;
						window->FocusedControl = nullptr;

						int iTab = 0;
						int TabCount = window->Tabs.size();
						if ( TabCount ) // If there are some tabs
						{
							int TabSize = (window->m_iWidth - 4 - 12) / TabCount;
							int Dist = Mouse.x - (window->m_x + 8);
							while ( Dist > TabSize ) {
								if ( Dist > TabSize ) {
									iTab++;
									Dist -= TabSize;
								}
							}
							window->SelectedTab = window->Tabs[iTab];
						}
					}
					else
						bCheckWidgetClicks = true;
				}
				else if ( IsMouseInRegion(window->m_x, window->m_y, window->m_x + window->m_iWidth, window->m_y + window->m_iHeight) ) {
					// Must be in the around the title or side of the window
					// So we assume the user is trying to drag the window
					IsDraggingWindow = true;

					DraggingWindow = window;

					DragOffsetX = Mouse.x - window->m_x;
					DragOffsetY = Mouse.y - window->m_y;

					// Loose focus on the control
					window->IsFocusingControl = false;
					window->FocusedControl = nullptr;
				}
			}

			// Controls
			if ( window->SelectedTab != nullptr ) {
				// Focused widget
				bool SkipWidget = false;
				CControl* SkipMe = nullptr;

				// this window is focusing on a widget??
				if ( window->IsFocusingControl ) {
					if ( window->FocusedControl != nullptr ) {
						// We've processed it once, skip it later
						SkipWidget = true;
						SkipMe = window->FocusedControl;

						POINT cAbs = window->FocusedControl->GetAbsolutePos();
						RECT controlRect = {cAbs.x, cAbs.y, window->FocusedControl->m_iWidth, window->FocusedControl->m_iHeight};
						window->FocusedControl->OnUpdate();

						if ( window->FocusedControl->Flag(UI_Clickable) && IsMouseInRegion(controlRect) && bCheckWidgetClicks ) {
							window->FocusedControl->OnClick();

							// If it gets clicked we loose focus
							window->IsFocusingControl = false;
							window->FocusedControl = nullptr;
							bCheckWidgetClicks = false;
						}
					}
				}

				// Itterate over the rest of the control
				for ( auto control : window->SelectedTab->Controls ) {
					if ( control != nullptr ) {
						if ( SkipWidget && SkipMe == control )
							continue;

						POINT cAbs = control->GetAbsolutePos();
						RECT controlRect = {cAbs.x, cAbs.y, control->m_iWidth, control->m_iHeight};
						control->OnUpdate();

						if ( control->Flag(UI_Clickable) && IsMouseInRegion(controlRect) && bCheckWidgetClicks ) {
							control->OnClick();
							bCheckWidgetClicks = false;

							// Change of focus
							if ( control->Flag(UI_Focusable) ) {
								window->IsFocusingControl = true;
								window->FocusedControl = control;
							}
							else {
								window->IsFocusingControl = false;
								window->FocusedControl = nullptr;
							}
						}
					}
				}

				// We must have clicked whitespace
				if ( bCheckWidgetClicks ) {
					// Loose focus on the control
					window->IsFocusingControl = false;
					window->FocusedControl = nullptr;
				}
			}
		}
	}
}

// Returns
bool CGUI::GetKeyPress( unsigned int key ) {
	if ( keys[key] == true && oldKeys[key] == false )
		return true;
	return false;
}

bool CGUI::GetKeyState( unsigned int key ) {
	return keys[key];
}

bool CGUI::IsMouseInRegion( int x, int y, int x2, int y2 ) {
	if ( Mouse.x > x && Mouse.y > y && Mouse.x < x2 && Mouse.y < y2 )
		return true;
	return false;
}

bool CGUI::IsMouseInRegion( RECT region ) {
	return IsMouseInRegion(region.left, region.top, region.left + region.right, region.top + region.bottom);
}

POINT CGUI::GetMouse() {
	return Mouse;
}

bool CGUI::DrawWindow( CWindow* window, int menu ) { {
		//Inner
		Render::Clear(window->m_x + 8 - 90, window->m_y + 1 + 28 * 2, window->m_iWidth - 4 - 12 + 90, window->m_iHeight - 2 - 8 - 26, Color(28, 28, 28, 255));
		Render::Clear(window->m_x + 8 - 90, window->m_y + 1 + 28 * 2 - 6, window->m_iWidth - 4 - 12 + 90, 6, Color(40, 40, 40, 255)); //top i believe
		Render::Clear(window->m_x + 8 - 90, window->m_y + 1 + 28 * 2 + window->m_iHeight - 2 - 8 - 26, window->m_iWidth - 4 - 12 + 90, 6, Color(40, 40, 40, 255)); //bottom iirc
		Render::Clear(window->m_x + 8 - 90 - 6, window->m_y + 1 + 28 * 2 - 6, 6, window->m_iHeight - 2 - 8 - 26 + 12, Color(40, 40, 40, 255)); // outside left
		Render::Clear(window->m_x + 8 + window->m_iWidth - 4 - 12, window->m_y + 1 + 28 * 2 - 6, 6, window->m_iHeight - 2 - 8 - 26 + 12, Color(40, 40, 40, 255)); //outside right

		//Tab
		Render::Clear(window->m_x + 8 - 90, window->m_y + 1 + 28 * 2, 90, window->m_iHeight - 2 - 8 - 26, Color(12, 12, 12, 255));
		Render::Outline(window->m_x + 8 - 90, window->m_y + 1 + 28 * 2, window->m_iWidth - 4 - 12 + 90, window->m_iHeight - 2 - 8 - 26, Color(48, 48, 48, 255));
		Render::Outline(window->m_x + 8 - 6 - 90, window->m_y + 1 + 28 * 2 - 6, window->m_iWidth - 4 - 12 + 90 + 12, window->m_iHeight - 2 - 8 - 26 + 12, Color(48, 48, 48, 255));
		Render::Outline(window->m_x + 8 - 90, window->m_y + 1 + 28 * 2, 90, window->m_iHeight - 2 - 8 - 26, Color(48, 48, 48, 255));
		int TabCount = window->Tabs.size();
		if ( TabCount ) // If there are some tabs
		{
			bool isOut = true;

			for ( int i = 0; i < TabCount; i++ ) {
				CTab* tab = window->Tabs[i];
				CTab* lBotTab = window->Tabs[0];
				CTab* rBotTab = window->Tabs[1];
				CTab* advancedTab = window->Tabs[2];
				CTab* vTab = window->Tabs[3];
				CTab* miscTab = window->Tabs[4];

				Menu::Window.SetSize(677, 657);

				float xAxis;
				float yAxis;
				float yWinPos = window->m_y + 1 + 28 * 2;
				float yWinHeight = window->m_iHeight - 2 - 8 - 26;

				yAxis = yWinPos + 16 + (i * 66) - 10;

				RECT TabDrawArea = {window->m_x + 8 - 90 + 1, yWinPos + 16 + (i * 100) - 8, 90 - 1, 90};

				RECT TextSize;
				TextSize = Render::GetTextSize(Render::Fonts::Tabs, tab->Title.c_str());

				RECT ClickTabArea = {xAxis,
					yAxis,
					TextSize.right,
					TextSize.bottom};

				if ( GetAsyncKeyState(VK_LBUTTON) ) {
					if ( IsMouseInRegion(TabDrawArea) ) {
						window->SelectedTab = window->Tabs[i];
					}
				}

				if ( window->SelectedTab == tab ) {
					xAxis = window->m_x + 8 - (150 + TextSize.right / 2);
					Render::Clear(TabDrawArea.left, TabDrawArea.top, TabDrawArea.right, TabDrawArea.bottom, Color(28, 28, 28, 255));
					Render::Line(TabDrawArea.left, TabDrawArea.top, TabDrawArea.left + TabDrawArea.right, TabDrawArea.top, Color(48, 48, 48, 255));
					Render::Line(TabDrawArea.left, TabDrawArea.top + TabDrawArea.bottom, TabDrawArea.left + TabDrawArea.right, TabDrawArea.top + TabDrawArea.bottom, Color(48, 48, 48, 255));
					Render::Text(TabDrawArea.left + (TabDrawArea.right / 2) - (TextSize.right / 2), TabDrawArea.top + (TabDrawArea.bottom / 2) - (TextSize.bottom / 2), Color(245, 245, 245, 255), Render::Fonts::Tabs, tab->Title.c_str());
				}
				else {
					xAxis = window->m_x + 8 - (45 + TextSize.right / 2);
					Render::GradientH(window->m_x - 80, window->m_y + 57, window->m_iWidth / 1.5, 1, Color(0, 160, 255, 255), Color(160, 0, 255, 255));
					Render::GradientH(window->m_x + window->m_iWidth / 2 - 10, window->m_y + 57, window->m_iWidth / 2 - 2, 1, Color(160, 0, 255, 255), Color(255, 255, 0, 255));
					Render::Text(TabDrawArea.left + (TabDrawArea.right / 2) - (TextSize.right / 2), TabDrawArea.top + (TabDrawArea.bottom / 2) - (TextSize.bottom / 2), Color(130, 130, 130, 255), Render::Fonts::Tabs, tab->Title.c_str());
				}
			}
		}
	}

	// Controls
	if ( window->SelectedTab != nullptr ) {
		// Focused widget
		bool SkipWidget = false;
		CControl* SkipMe = nullptr;

		// this window is focusing on a widget??
		if ( window->IsFocusingControl ) {
			if ( window->FocusedControl != nullptr ) {
				// We need to draw it last, so skip it in the regular loop
				SkipWidget = true;
				SkipMe = window->FocusedControl;
			}
		}

		// Itterate over all the other controls
		for ( auto control : window->SelectedTab->Controls ) {
			if ( SkipWidget && SkipMe == control )
				continue;

			if ( control != nullptr && control->Flag(UI_Drawable) ) {
				POINT cAbs = control->GetAbsolutePos();
				RECT controlRect = {cAbs.x, cAbs.y, control->m_iWidth, control->m_iHeight};
				bool hover = false;
				if ( IsMouseInRegion(controlRect) ) {
					hover = true;
				}
				control->Draw(hover);
			}
		}

		// Draw the skipped widget last
		if ( SkipWidget ) {
			auto control = window->FocusedControl;

			if ( control != nullptr && control->Flag(UI_Drawable) ) {
				POINT cAbs = control->GetAbsolutePos();
				RECT controlRect = {cAbs.x, cAbs.y, control->m_iWidth, control->m_iHeight};
				bool hover = false;
				if ( IsMouseInRegion(controlRect) ) {
					hover = true;
				}
				control->Draw(hover);
			}
		}
	}

	return true;
}

void CGUI::RegisterWindow( CWindow* window ) {
	Windows.push_back(window);

	// Resorting to put groupboxes at the start
	for ( auto tab : window->Tabs ) {
		for ( auto control : tab->Controls ) {
			if ( control->Flag(UI_RenderFirst) ) {
				CControl* c = control;
				tab->Controls.erase(remove(tab->Controls.begin(), tab->Controls.end(), control), tab->Controls.end());
				tab->Controls.insert(tab->Controls.begin(), control);
			}
		}
	}
}

void CGUI::BindWindow( unsigned char Key, CWindow* window ) {
	if ( window )
		WindowBinds[Key] = window;
	else
		WindowBinds.erase(Key);
}

void CGUI::SaveWindowState( CWindow* window, std::string Filename ) {
	// Create a whole new document and we'll just save over top of the old one
	tinyxml2::XMLDocument Doc;

	// Root Element is called "ayy"
	tinyxml2::XMLElement* Root = Doc.NewElement("LegacyConfigs");
	Doc.LinkEndChild(Root);

	// If the window has some tabs..
	if ( Root && window->Tabs.size() > 0 ) {
		for ( auto Tab : window->Tabs ) {
			// Add a new section for this tab
			tinyxml2::XMLElement* TabElement = Doc.NewElement(Tab->Title.c_str());
			Root->LinkEndChild(TabElement);

			// Now we itterate the controls this tab contains
			if ( TabElement && Tab->Controls.size() > 0 ) {
				for ( auto Control : Tab->Controls ) {
					// If the control is ok to be saved
					if ( Control && Control->Flag(UI_SaveFile) && Control->FileIdentifier.length() > 1 && Control->FileControlType ) {
						// Create an element for the control
						tinyxml2::XMLElement* ControlElement = Doc.NewElement(Control->FileIdentifier.c_str());
						TabElement->LinkEndChild(ControlElement);

						if ( !ControlElement ) {
							return;
						}

						CCheckBox* cbx = nullptr;
						CComboBox* cbo = nullptr;
						CKeyBind* key = nullptr;
						CSlider* sld = nullptr;

						// Figure out what kind of control and data this is
						switch ( Control->FileControlType ) {
							case UIC_CheckBox:
								cbx = (CCheckBox*)Control;
								ControlElement->SetText(cbx->GetState());
								break;
							case UIC_ComboBox:
								cbo = (CComboBox*)Control;
								ControlElement->SetText(cbo->GetIndex());
								break;
							case UIC_KeyBind:
								key = (CKeyBind*)Control;
								ControlElement->SetText(key->GetKey());
								break;
							case UIC_Slider:
								sld = (CSlider*)Control;
								ControlElement->SetText(sld->GetValue());
								break;
						}
					}
				}
			}
		}
	}

	//Save the file
	if ( Doc.SaveFile(Filename.c_str()) != tinyxml2::XML_NO_ERROR ) {
		MessageBox(nullptr, "Failed to save config!", "Legacy", MB_OK);
	}
}

void CGUI::LoadWindowState( CWindow* window, std::string Filename ) {
	// Lets load our meme
	tinyxml2::XMLDocument Doc;
	if ( Doc.LoadFile(Filename.c_str()) == tinyxml2::XML_NO_ERROR ) {
		tinyxml2::XMLElement* Root = Doc.RootElement();

		// The root "ayy" element
		if ( Root ) {
			// If the window has some tabs..
			if ( Root && window->Tabs.size() > 0 ) {
				for ( auto Tab : window->Tabs ) {
					// We find the corresponding element for this tab
					tinyxml2::XMLElement* TabElement = Root->FirstChildElement(Tab->Title.c_str());
					if ( TabElement ) {
						// Now we itterate the controls this tab contains
						if ( TabElement && Tab->Controls.size() > 0 ) {
							for ( auto Control : Tab->Controls ) {
								// If the control is ok to be saved
								if ( Control && Control->Flag(UI_SaveFile) && Control->FileIdentifier.length() > 1 && Control->FileControlType ) {
									// Get the controls element
									tinyxml2::XMLElement* ControlElement = TabElement->FirstChildElement(Control->FileIdentifier.c_str());

									if ( ControlElement ) {
										CCheckBox* cbx = nullptr;
										CComboBox* cbo = nullptr;
										CKeyBind* key = nullptr;
										CSlider* sld = nullptr;

										// Figure out what kind of control and data this is
										switch ( Control->FileControlType ) {
											case UIC_CheckBox:
												cbx = (CCheckBox*)Control;
												cbx->SetState(ControlElement->GetText()[0] == '1' ? true : false);
												break;
											case UIC_ComboBox:
												cbo = (CComboBox*)Control;
												cbo->SelectIndex(atoi(ControlElement->GetText()));
												break;
											case UIC_KeyBind:
												key = (CKeyBind*)Control;
												key->SetKey(atoi(ControlElement->GetText()));
												break;
											case UIC_Slider:
												sld = (CSlider*)Control;
												sld->SetValue(atof(ControlElement->GetText()));
												break;
										}
									}
								}
							}
						}
					}
				}
			}
		}
	}
}