#pragma once
#include "e2dobject.h"

namespace e2d
{


	class Button
		: public Node
	{
	public:
		Button();

		explicit Button(
			Node * normal,					/* 普通状态 */
			const Function& func = nullptr	/* 按钮点击后的回调函数 */
		);

		explicit Button(
			Node * normal,					/* 普通状态 */
			Node * selected,				/* 鼠标按下状态 */
			const Function& func = nullptr	/* 按钮点击后的回调函数 */
		);

		explicit Button(
			Node * normal,					/* 普通状态 */
			Node * mouseover,				/* 鼠标移入状态 */
			Node * selected,				/* 鼠标按下状态 */
			const Function& func = nullptr	/* 按钮点击后的回调函数 */
		);

		explicit Button(
			Node * normal,					/* 普通状态 */
			Node * mouseover,				/* 鼠标移入状态 */
			Node * selected,				/* 鼠标移入状态 */
			Node * disabled,				/* 按钮禁用状态 */
			const Function& func = nullptr	/* 按钮点击后的回调函数 */
		);

		// 获取按钮状态是启用还是禁用
		bool IsEnable() const;

		// 设置按钮启用或禁用
		void SetEnabled(
			bool enabled
		);

		// 设置一般情况下显示的按钮
		virtual void SetNormal(
			Node * normal
		);

		// 设置鼠标移入按钮时显示的按钮
		virtual void SetMouseOver(
			Node * mouseover
		);

		// 设置鼠标按下按钮时显示的按钮
		virtual void SetSelected(
			Node * selected
		);

		// 设置按钮被禁用时显示的按钮
		virtual void SetDisabled(
			Node * disabled
		);

		// 设置按钮点击后的回调函数
		void SetCallbackOnClick(
			const Function& func
		);

		// 设置锚点位置
		// 默认为 (0, 0), 范围 [0, 1]
		virtual void SetAnchor(
			float anchor_x,
			float anchor_y
		) override;

	protected:
		E2D_DISABLE_COPY(Button);

		// 按钮状态枚举
		enum class Status { Normal, Mouseover, Selected };

		// 设置按钮状态
		virtual void SetStatus(
			Status status
		);

		// 刷新按钮显示
		virtual void UpdateVisible();

		// 分发鼠标消息
		virtual bool Dispatch(
			const MouseEvent& e,
			bool handled
		) override;

		// 遍历节点
		virtual void Visit() override;

	protected:
		Node * normal_;
		Node *		mouseover_;
		Node *		selected_;
		Node *		disabled_;
		bool		enabled_;
		bool		is_selected_;
		Status		status_;
		Function	callback_;
	};


	// 菜单
	class Menu
		: public Node
	{
	public:
		Menu();

		explicit Menu(
			const std::vector<Button*>& buttons	/* 按钮数组 */
		);

		// 获取菜单是否禁用
		bool IsEnable() const;

		// 获取菜单中的按钮数量
		size_t GetButtonCount() const;

		// 设置菜单启用或禁用
		void SetEnabled(
			bool enabled
		);

		// 添加按钮
		void AddButton(
			Button * button
		);

		// 移除按钮
		bool RemoveButton(
			Button * button
		);

		// 获取所有按钮
		const std::vector<Button*>& GetAllButtons() const;

	protected:
		E2D_DISABLE_COPY(Menu);

	protected:
		bool enabled_;
		std::vector<Button*> buttons_;
	};


}