#ifndef __TOOLBAR_HPP__
#define __TOOLBAR_HPP__

#include <windows.h>
#include <objidl.h>
#include <gdiplus.h>

#include <string>
#include <vector>

#include "PaintUtil.hpp"
#include "Theme.hpp"

namespace jdi
{

	/**
	 * @brief 工具栏按钮触发的动作。
	 */
	enum class ToolbarAction : int
	{
		NONE = 0,		/**< 未触发任何动作。 */
		OPEN_FILE,		/**< 打开 CSV 文件。 */
		FIT_ALL,		/**< 视图自适应到可见信号范围（同时退出缩放模式）。 */
		ZOOM_TIME,		/**< 切换「在时间轴缩放」模式。 */
		ZOOM_VALUE,		/**< 切换「在 Y 轴缩放」模式。 */
		CURSOR,			/**< 切换时间光标。 */
		MERGE_LANES,	/**< 合并为单一网格。 */
		SPLIT_LANES,		/**< 按信号拆分为多网格。 */
		SELECT_ALL,		/**< 勾选当前标签页全部信号。 */
		CLEAR_ALL		/**< 取消当前标签页全部信号勾选。 */
	};

	/**
	 * @brief 顶部工具栏：文件操作与视图控制的入口。
	 */
	class Toolbar
	{
	public:
		Toolbar();

		/**
		 * @brief 绑定字体与主题资源。
		 * @param [IN] fonts 字体集合，需在工具栏生命周期内保持有效。
		 * @param [IN] theme 主题配色，需在工具栏生命周期内保持有效。
		 */
		void Setup(const UiFonts* fonts, const Theme* theme);

		/**
		 * @brief 设置 DPI 缩放比例。
		 * @param [IN] scale 缩放比例（1.0 对应 96 DPI）。
		 */
		void SetScale(float scale);

		/**
		 * @brief 设置是否已加载数据（未加载时大部分按钮被禁用）。
		 * @param [IN] hasDocument 是否已加载数据。
		 */
		void SetDocumentLoaded(bool hasDocument);

		/**
		 * @brief 设置工具栏右侧的状态文本。
		 * @param [IN] text 状态文本。
		 */
		void SetStatusText(const std::wstring& text);

		/**
		 * @brief 设置可切换按钮的选中状态。
		 * @param [IN] action 目标按钮动作。
		 * @param [IN] checked 是否处于选中（生效）状态。
		 */
		void SetToggleState(ToolbarAction action, bool checked);

		/**
		 * @brief 计算工具栏在指定缩放下的期望高度。
		 * @return 期望高度（像素）。
		 */
		float DesiredHeight() const;

		/**
		 * @brief 按给定区域重新排布按钮。
		 * @param [IN] area 工具栏可用区域。
		 */
		void Layout(const gd::RectF& area);

		/**
		 * @brief 绘制工具栏。
		 * @param [IN,OUT] g 绘图上下文。
		 */
		void Paint(gd::Graphics& g);

		/**
		 * @brief 处理鼠标移动，更新悬停状态。
		 * @param [IN] x 客户区横坐标。
		 * @param [IN] y 客户区纵坐标。
		 * @return 悬停状态发生变化返回 true。
		 */
		bool OnMouseMove(int x, int y);

		/**
		 * @brief 处理鼠标左键按下。
		 * @param [IN] x 客户区横坐标。
		 * @param [IN] y 客户区纵坐标。
		 * @param [OUT] action 命中的按钮动作。
		 * @return 命中按钮返回 true。
		 */
		bool OnLButtonDown(int x, int y, ToolbarAction& action);

		/**
		 * @brief 处理鼠标左键抬起。
		 * @param [IN] x 客户区横坐标。
		 * @param [IN] y 客户区纵坐标。
		 * @param [OUT] action 抬起时命中的按钮动作（仅在按钮上抬起时有效）。
		 * @return 视觉状态发生变化返回 true。
		 */
		bool OnLButtonUp(int x, int y, ToolbarAction& action);

		/**
		 * @brief 处理鼠标离开工具栏。
		 * @return 视觉状态发生变化返回 true。
		 */
		bool OnMouseLeave();

		/**
		 * @brief 判断坐标是否落在工具栏区域内。
		 * @param [IN] x 客户区横坐标。
		 * @param [IN] y 客户区纵坐标。
		 * @return 落在区域内返回 true。
		 */
		bool HitTest(int x, int y) const;

	private:
		/**
		 * @brief 单个工具栏按钮的运行时状态。
		 */
		struct Button
		{
			ToolbarAction action = ToolbarAction::NONE;	/**< 按钮动作。 */
			std::wstring label;							/**< 按钮文本。 */
			gd::RectF rect;								/**< 按钮矩形。 */
			float width = 0.0f;							/**< 按钮宽度。 */
			bool primary = false;						/**< 是否为主按钮（强调色填充）。 */
			bool needsDocument = true;					/**< 是否依赖已加载数据。 */			bool toggle = false;						/**< 是否为可切换按钮。 */
			bool checked = false;						/**< 可切换按钮的选中状态。 */			bool hover = false;							/**< 悬停状态。 */
			bool pressed = false;						/**< 按下状态。 */
		};

		/**
		 * @brief 判断按钮当前是否可用。
		 * @param [IN] button 目标按钮。
		 * @return 可用返回 true。
		 */
		bool IsEnabled(const Button& button) const;

		const UiFonts* m_fonts = nullptr;	/**< 字体资源。 */
		const Theme* m_theme = nullptr;		/**< 主题资源。 */
		float m_scale = 1.0f;				/**< DPI 缩放比例。 */
		bool m_documentLoaded = false;		/**< 是否已加载数据。 */
		std::wstring m_statusText;			/**< 右侧状态文本。 */
		std::vector<Button> m_buttons;		/**< 按钮集合。 */
		gd::RectF m_area;					/**< 工具栏整体区域。 */
		int m_hoverIndex = -1;				/**< 当前悬停按钮下标。 */
		int m_pressedIndex = -1;			/**< 当前按下按钮下标。 */
	};

} // namespace jdi

#endif // __TOOLBAR_HPP__