// Copyright (c) 2016-2018 Easy2D - Nomango
// 
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#include "..\e2dcomponent.h"
#include "..\e2dmodule.h"

#define SAFE_SET(pointer, func, ...) if (pointer) { pointer->##func(__VA_ARGS__); }

#define SET_BUTTON_NODE(Old, New)								\
	if (New != Old)												\
	{															\
		if (Old) this->RemoveChild(Old);						\
		if (New)												\
		{														\
			New->SetPivot(GetPivotX(), GetPivotY());			\
			this->AddChild(New);								\
		}														\
		Old = New;												\
		UpdateVisible();										\
	}															\


e2d::Button::Button()
	: callback_(nullptr)
	, status_(Status::Normal)
	, enabled_(true)
	, is_selected_(false)
	, normal_(nullptr)
	, mouseover_(nullptr)
	, selected_(nullptr)
	, disabled_(nullptr)
{
}

e2d::Button::Button(Node * normal, const Function& func)
	: callback_(nullptr)
	, status_(Status::Normal)
	, enabled_(true)
	, is_selected_(false)
	, normal_(nullptr)
	, mouseover_(nullptr)
	, selected_(nullptr)
	, disabled_(nullptr)
{
	this->SetNormal(normal);
	this->SetCallbackOnClick(func);
}

e2d::Button::Button(Node * normal, Node * selected, const Function& func)
	: callback_(nullptr)
	, status_(Status::Normal)
	, enabled_(true)
	, is_selected_(false)
	, normal_(nullptr)
	, mouseover_(nullptr)
	, selected_(nullptr)
	, disabled_(nullptr)
{
	this->SetNormal(normal);
	this->SetSelected(selected);
	this->SetCallbackOnClick(func);
}

e2d::Button::Button(Node * normal, Node * mouseover, Node * selected, const Function& func)
	: callback_(nullptr)
	, status_(Status::Normal)
	, enabled_(true)
	, is_selected_(false)
	, normal_(nullptr)
	, mouseover_(nullptr)
	, selected_(nullptr)
	, disabled_(nullptr)
{
	this->SetNormal(normal);
	this->SetMouseOver(mouseover);
	this->SetSelected(selected);
	this->SetCallbackOnClick(func);
}

e2d::Button::Button(Node * normal, Node * mouseover, Node * selected, Node * disabled, const Function& func)
	: callback_(nullptr)
	, status_(Status::Normal)
	, enabled_(true)
	, is_selected_(false)
	, normal_(nullptr)
	, mouseover_(nullptr)
	, selected_(nullptr)
	, disabled_(nullptr)
{
	this->SetNormal(normal);
	this->SetMouseOver(mouseover);
	this->SetSelected(selected);
	this->SetDisabled(disabled);
	this->SetCallbackOnClick(func);
}

bool e2d::Button::IsEnable() const
{
	return enabled_;
}

void e2d::Button::SetNormal(Node * normal)
{
	SET_BUTTON_NODE(normal_, normal);
	if (normal)
	{
		this->SetSize(normal->GetWidth(), normal->GetHeight());
	}
}

void e2d::Button::SetMouseOver(Node * mouseover)
{
	SET_BUTTON_NODE(mouseover_, mouseover);
}

void e2d::Button::SetSelected(Node * selected)
{
	SET_BUTTON_NODE(selected_, selected);
}

void e2d::Button::SetDisabled(Node * disabled)
{
	SET_BUTTON_NODE(disabled_, disabled);
}

void e2d::Button::SetEnabled(bool enabled)
{
	if (enabled_ != enabled)
	{
		enabled_ = enabled;
		UpdateVisible();
	}
}

void e2d::Button::SetCallbackOnClick(const Function& func)
{
	callback_ = func;
}

void e2d::Button::SetPivot(float pivot_x, float pivot_y)
{
	Node::SetPivot(pivot_x, pivot_y);
	SAFE_SET(normal_, SetPivot, pivot_x, pivot_y);
	SAFE_SET(mouseover_, SetPivot, pivot_x, pivot_y);
	SAFE_SET(selected_, SetPivot, pivot_x, pivot_y);
	SAFE_SET(disabled_, SetPivot, pivot_x, pivot_y);
}

bool e2d::Button::Dispatch(const MouseEvent & e, bool handled)
{
	if (!handled && enabled_ && visible_ && normal_)
	{
		bool contains = normal_->ContainsPoint(e.GetPosition());
		if (e.GetType() == MouseEvent::Type::LeftUp && is_selected_ && contains)
		{
			if (callback_)
			{
				callback_();
			}
			is_selected_ = false;
			SetStatus(Status::Normal);
			return true;
		}
		else if (e.GetType() == MouseEvent::Type::LeftDown)
		{
			is_selected_ = contains;
			SetStatus(contains ? Status::Selected : Status::Normal);
			
			if (contains)
				return true;
		}
		else if (e.GetType() == MouseEvent::Type::LeftUp)
		{
			is_selected_ = false;
		}
		else if (e.GetType() == MouseEvent::Type::MoveBy && is_selected_ && contains)
		{
			SetStatus(Status::Selected);
			return true;
		}
		else
		{
			if (!e.IsLButtonDown() && is_selected_)
			{
				is_selected_ = false;
			}

			SetStatus(contains ? Status::Mouseover : Status::Normal);

			if (contains)
				return true;
		}
	}

	return Node::Dispatch(e, handled);
}

void e2d::Button::Visit()
{
	Node::Visit();

	if (visible_ &&
		!enabled_ &&
		normal_ &&
		normal_->ContainsPoint(Device::GetInput()->GetMousePos()))
	{
		HCURSOR hcursor = ::LoadCursor(nullptr, IDC_NO);
		if (hcursor)
		{
			::SetCursor(hcursor);
		}
	}
	else if (status_ == Status::Mouseover || status_ == Status::Selected)
	{
		HCURSOR hcursor = ::LoadCursor(nullptr, IDC_HAND);
		if (hcursor)
		{
			::SetCursor(hcursor);
		}
	}
}

void e2d::Button::SetStatus(Status status)
{
	if (status_ != status)
	{
		status_ = status;
		UpdateVisible();
	}
}

void e2d::Button::UpdateVisible()
{
	SAFE_SET(normal_, SetVisible, false);
	SAFE_SET(mouseover_, SetVisible, false);
	SAFE_SET(selected_, SetVisible, false);
	SAFE_SET(disabled_, SetVisible, false);

	if (enabled_)
	{
		if (status_ == Status::Selected && selected_)
		{
			selected_->SetVisible(true);
		}
		else if (status_ == Status::Mouseover && mouseover_)
		{
			mouseover_->SetVisible(true);
		}
		else
		{
			if (normal_) normal_->SetVisible(true);
		}
	}
	else
	{
		if (disabled_)
		{
			disabled_->SetVisible(true);
		}
		else
		{
			if (normal_) normal_->SetVisible(true);
		}
	}
}