#pragma once
#include "../content_control.hpp"

#include "../click_detector.hpp"
#include "../render/border_render_object.hpp"
#include "cru/platform/graph/brush.hpp"
#include "cru/platform/native/basic_types.hpp"

#include <memory>

namespace cru::ui::render {
class BorderRenderObject;
}

namespace cru::ui::controls {
using render::CornerRadius;

struct ButtonStateStyle {
  std::shared_ptr<platform::graph::Brush> border_brush;
  Thickness border_thickness;
  CornerRadius corner_radius;
  std::shared_ptr<platform::graph::Brush> foreground_brush;
  std::shared_ptr<platform::graph::Brush> background_brush;
};

struct ButtonStyle {
  // corresponds to ButtonState::Normal
  ButtonStateStyle normal;
  // corresponds to ButtonState::Hover
  ButtonStateStyle hover;
  // corresponds to ButtonState::Press
  ButtonStateStyle press;
  // corresponds to ButtonState::PressCancel
  ButtonStateStyle press_cancel;
};

enum class ButtonState {
  // mouse is not in it
  Normal,
  // mouse is in it and not pressed
  Hover,
  // mouse is pressed in it
  Press,
  // mouse is pressed outside button
  PressCancel,
};

class Button : public ContentControl {
 public:
  static constexpr auto control_type = L"Button";

  static Button* Create() { return new Button(); }

 protected:
  Button();

 public:
  Button(const Button& other) = delete;
  Button(Button&& other) = delete;
  Button& operator=(const Button& other) = delete;
  Button& operator=(Button&& other) = delete;
  ~Button() override = default;

  std::wstring_view GetControlType() const override final {
    return control_type;
  }

  render::RenderObject* GetRenderObject() const override;

 public:
  const ButtonStyle& GetStyle() const { return style_; }
  void SetStyle(ButtonStyle style);

  ButtonState GetState() const { return state_; }

  // Get the trigger mouse button(s). Return value might be a union of Left,
  // Middle and Right.
  MouseButton GetTriggerButton() const { return trigger_button_; }
  // Set the trigger mouse button(s). You can set multiple buttons by passing a
  // union of Left, Middle and Right. If you disable a button when user is
  // pressing the same one, click will be stopped.
  // Default is only Left.
  void SetTriggerButton(MouseButton newTrigger);

 protected:
  void OnChildChanged(Control* old_child, Control* new_child) override;

  virtual void OnStateChange(ButtonState oldState, ButtonState newState);

 private:
  void SetState(ButtonState newState) {
    if (state_ == newState) return;
    const auto oldState = state_;
    state_ = newState;
    OnStateChange(oldState, newState);
  }

 private:
  std::shared_ptr<render::BorderRenderObject> render_object_{};

  ButtonState state_ = ButtonState::Normal;

  MouseButton trigger_button_ = MouseButton::Left;

  ButtonStyle style_;

  ClickDetector click_detector_;
};
}  // namespace cru::ui::controls