Extra2D/Easy2D/include/easy2d/ui/button.h

#pragma once

#include <easy2d/ui/widget.h>
#include <easy2d/graphics/font.h>
#include <easy2d/graphics/texture.h>
#include <easy2d/platform/window.h>

namespace easy2d {

// 图片缩放模式
enum class ImageScaleMode {
    Original,       // 使用原图大小
    Stretch,        // 拉伸填充
    ScaleFit,       // 等比缩放，保持完整显示
    ScaleFill       // 等比缩放，填充整个区域（可能裁剪）
};

// ============================================================================
// 基础按钮类
// ============================================================================
class Button : public Widget {
public:
    Button();
    ~Button() override = default;

    static Ptr<Button> create();

    void setText(const String& text);
    const String& getText() const { return text_; }

    void setFont(Ptr<FontAtlas> font);
    Ptr<FontAtlas> getFont() const { return font_; }

    void setPadding(const Vec2& padding);
    Vec2 getPadding() const { return padding_; }

    void setTextColor(const Color& color);
    Color getTextColor() const { return textColor_; }

    // 纯色背景设置
    void setBackgroundColor(const Color& normal, const Color& hover, const Color& pressed);
    void setBorder(const Color& color, float width);

    // 图片背景设置
    void setBackgroundImage(Ptr<Texture> normal, Ptr<Texture> hover = nullptr, Ptr<Texture> pressed = nullptr);
    void setBackgroundImageScaleMode(ImageScaleMode mode);
    void setCustomSize(const Vec2& size);
    void setCustomSize(float width, float height);

    // 圆角矩形设置
    void setCornerRadius(float radius);
    float getCornerRadius() const { return cornerRadius_; }
    void setRoundedCornersEnabled(bool enabled);
    bool isRoundedCornersEnabled() const { return roundedCornersEnabled_; }

    // 鼠标光标设置（悬停时显示的光标形状）
    void setHoverCursor(CursorShape cursor);
    CursorShape getHoverCursor() const { return hoverCursor_; }
    
    // Alpha遮罩点击检测（用于不规则形状按钮）
    void setUseAlphaMaskForHitTest(bool enabled);
    bool isUseAlphaMaskForHitTest() const { return useAlphaMaskForHitTest_; }

    void setOnClick(Function<void()> callback);

protected:
    void onDraw(RenderBackend& renderer) override;
    void drawBackgroundImage(RenderBackend& renderer, const Rect& rect);
    void drawRoundedRect(RenderBackend& renderer, const Rect& rect, const Color& color, float radius);
    void fillRoundedRect(RenderBackend& renderer, const Rect& rect, const Color& color, float radius);
    Vec2 calculateImageSize(const Vec2& buttonSize, const Vec2& imageSize);

    // 状态访问（供子类使用）
    bool isHovered() const { return hovered_; }
    bool isPressed() const { return pressed_; }

private:
    String text_;
    Ptr<FontAtlas> font_;
    Vec2 padding_ = Vec2(10.0f, 6.0f);

    // 文字颜色
    Color textColor_ = Colors::White;

    // 纯色背景
    Color bgNormal_ = Color(0.2f, 0.2f, 0.2f, 1.0f);
    Color bgHover_ = Color(0.28f, 0.28f, 0.28f, 1.0f);
    Color bgPressed_ = Color(0.15f, 0.15f, 0.15f, 1.0f);

    // 图片背景
    Ptr<Texture> imgNormal_;
    Ptr<Texture> imgHover_;
    Ptr<Texture> imgPressed_;
    ImageScaleMode scaleMode_ = ImageScaleMode::Original;
    bool useImageBackground_ = false;

    // 边框
    Color borderColor_ = Color(0.6f, 0.6f, 0.6f, 1.0f);
    float borderWidth_ = 1.0f;

    // 圆角矩形
    float cornerRadius_ = 8.0f;           // 圆角半径
    bool roundedCornersEnabled_ = false;  // 默认关闭

    // 鼠标光标
    CursorShape hoverCursor_ = CursorShape::Hand;  // 悬停时默认显示手型光标
    bool cursorChanged_ = false;           // 标记是否已改变光标
    
    // Alpha遮罩点击检测（用于不规则形状按钮）
    bool useAlphaMaskForHitTest_ = false;  // 默认关闭

    bool hovered_ = false;
    bool pressed_ = false;

    Function<void()> onClick_;
};

// ============================================================================
// 切换按钮 - 点击切换两种状态（支持图片和文字）
// ============================================================================
class ToggleImageButton : public Button {
public:
    ToggleImageButton();
    ~ToggleImageButton() override = default;

    static Ptr<ToggleImageButton> create();

    // 设置两种状态的图片
    void setStateImages(Ptr<Texture> stateOffNormal, Ptr<Texture> stateOnNormal,
                        Ptr<Texture> stateOffHover = nullptr, Ptr<Texture> stateOnHover = nullptr,
                        Ptr<Texture> stateOffPressed = nullptr, Ptr<Texture> stateOnPressed = nullptr);

    // 设置两种状态的文字
    void setStateText(const String& textOff, const String& textOn);

    // 设置两种状态的文字颜色
    void setStateTextColor(const Color& colorOff, const Color& colorOn);

    // 获取/设置当前状态
    bool isOn() const { return isOn_; }
    void setOn(bool on);
    void toggle();

    // 设置状态改变回调
    void setOnStateChange(Function<void(bool)> callback);

protected:
    void onDraw(RenderBackend& renderer) override;

private:
    // 状态图片
    Ptr<Texture> imgOffNormal_, imgOnNormal_;
    Ptr<Texture> imgOffHover_, imgOnHover_;
    Ptr<Texture> imgOffPressed_, imgOnPressed_;

    // 状态文字
    String textOff_, textOn_;
    bool useStateText_ = false;

    // 状态文字颜色
    Color textColorOff_ = Colors::White;
    Color textColorOn_ = Colors::White;
    bool useStateTextColor_ = false;

    bool isOn_ = false;
    Function<void(bool)> onStateChange_;
};

} // namespace easy2d