#include "cru/ui/controls/button.hpp"
#include <memory>

#include "../helper.hpp"
#include "cru/platform/graph/brush.hpp"
#include "cru/platform/native/cursor.hpp"
#include "cru/platform/native/ui_application.hpp"
#include "cru/ui/render/border_render_object.hpp"
#include "cru/ui/ui_manager.hpp"
#include "cru/ui/window.hpp"

namespace cru::ui::controls {
using cru::platform::native::SystemCursorType;

namespace {
void Set(render::BorderRenderObject* o, const ButtonStateStyle& s) {
  o->SetBorderBrush(s.border_brush);
  o->SetBorderThickness(s.border_thickness);
  o->SetBorderRadius(s.border_radius);
  o->SetForegroundBrush(s.foreground_brush);
  o->SetBackgroundBrush(s.background_brush);
}

std::shared_ptr<platform::native::ICursor> GetSystemCursor(
    SystemCursorType type) {
  return GetUiApplication()->GetCursorManager()->GetSystemCursor(type);
}
}  // namespace

Button::Button() : click_detector_(this) {
  style_ = UiManager::GetInstance()->GetThemeResources()->button_style;

  render_object_ = std::make_unique<render::BorderRenderObject>();
  render_object_->SetAttachedControl(this);
  Set(render_object_.get(), style_.normal);
  render_object_->SetBorderEnabled(true);

  MouseEnterEvent()->Direct()->AddHandler([this](event::MouseEventArgs& args) {
    CRU_UNUSED(args)

    if (click_detector_.GetPressingButton() & trigger_button_) {
      SetState(ButtonState::Press);
    } else {
      SetState(ButtonState::Hover);
    }
  });

  MouseLeaveEvent()->Direct()->AddHandler([this](event::MouseEventArgs& args) {
    CRU_UNUSED(args)

    if (click_detector_.GetPressingButton() & trigger_button_) {
      SetState(ButtonState::Normal);
    } else {
      SetState(ButtonState::PressCancel);
    }
  });

  click_detector_.ClickBeginEvent()->AddHandler([this](MouseButton button) {
    if (button & trigger_button_) {
      SetState(ButtonState::Press);
    }
  });

  click_detector_.ClickEndEvent()->AddHandler([this](MouseButton button) {
    if (button & trigger_button_) {
      SetState(ButtonState::Hover);
    }
  });
}

render::RenderObject* Button::GetRenderObject() const {
  return render_object_.get();
}

void Button::OnChildChanged(Control* old_child, Control* new_child) {
  if (old_child != nullptr) render_object_->RemoveChild(0);
  if (new_child != nullptr)
    render_object_->AddChild(new_child->GetRenderObject(), 0);
}

void Button::OnStateChange(ButtonState oldState, ButtonState newState) {
  CRU_UNUSED(oldState)

  switch (newState) {
    case ButtonState::Normal:
      Set(render_object_.get(), style_.normal);
      SetCursor(GetSystemCursor(SystemCursorType::Arrow));
      break;
    case ButtonState::Hover:
      Set(render_object_.get(), style_.hover);
      SetCursor(GetSystemCursor(SystemCursorType::Hand));
      break;
    case ButtonState::Press:
      Set(render_object_.get(), style_.press);
      SetCursor(GetSystemCursor(SystemCursorType::Hand));
      break;
    case ButtonState::PressCancel:
      Set(render_object_.get(), style_.press_cancel);
      SetCursor(GetSystemCursor(SystemCursorType::Arrow));
      break;
  }
}
}  // namespace cru::ui::controls