#include "cru/ui/components/Menu.h"
#include "cru/platform/gui/Window.h"
#include "cru/ui/ThemeManager.h"
#include "cru/ui/controls/Button.h"
#include "cru/ui/controls/Control.h"
#include "cru/ui/controls/ControlHost.h"
#include "cru/ui/controls/FlexLayout.h"
#include "cru/ui/controls/TextBlock.h"
#include "cru/ui/controls/Window.h"
#include "cru/ui/helper/ClickDetector.h"
#include "cru/ui/style/StyleRuleSet.h"

namespace cru::ui::components {
MenuItem::MenuItem() {
  container_.SetChild(&text_);
  container_.GetStyleRuleSet()->SetParent(
      ThemeManager::GetInstance()->GetResourceStyleRuleSet("menuitem.style"));
  container_.ClickEvent()->AddHandler([this](const helper::ClickEventArgs&) {
    if (this->on_click_) this->on_click_();
  });
}

MenuItem::MenuItem(std::string text) : MenuItem() { SetText(std::move(text)); }

void MenuItem::SetText(std::string text) { text_.SetText(std::move(text)); }

Menu::Menu() {
  container_.SetFlexDirection(controls::FlexDirection::Vertical);
  container_.SetItemCrossAlign(controls::FlexCrossAlignment::Stretch);
}

Menu::~Menu() {
  for (auto item : items_) {
    item->DeleteIfDeleteByParent();
  }
}

void Menu::AddItemAt(Component* item, Index index) {
  Expects(index >= 0 && index <= GetItemCount());

  items_.insert(items_.cbegin() + index, item);
  container_.InsertChildAt(item->GetRootControl(), index);
}

Component* Menu::RemoveItemAt(Index index) {
  Expects(index >= 0 && index < GetItemCount());

  Component* item = items_[index];

  items_.erase(items_.cbegin() + index);
  container_.RemoveChildAt(index);

  return item;
}

void Menu::ClearItems() {
  container_.RemoveAllChild();

  for (auto item : items_) {
    item->DeleteIfDeleteByParent();
  }

  items_.clear();
}

void Menu::AddTextItemAt(std::string text, Index index,
                         std::function<void()> on_click) {
  MenuItem* item = new MenuItem(std::move(text));
  item->SetOnClick([this, index, on_click = std::move(on_click)] {
    auto on_item_click = on_item_click_;
    on_click();
    if (on_item_click) on_item_click(index);
  });
  item->SetDeleteByParent(true);
  AddItemAt(item, index);
}

PopupMenu::PopupMenu(controls::Control* attached_control)
    : attached_control_(attached_control) {
  menu_.SetOnItemClick([this](Index) { popup_->GetNativeWindow()->Close(); });
  popup_.reset(controls::Window::CreatePopup(attached_control));
  popup_->InsertChildAt(menu_.GetRootControl(), 0);
}

PopupMenu::~PopupMenu() {}

controls::Control* PopupMenu::GetRootControl() { return popup_.get(); }

void PopupMenu::SetPosition(const Point& position) {
  auto window = popup_->GetNativeWindow();
  window->SetClientRect(Rect{position, window->GetClientSize()});
}

void PopupMenu::Show() {
  auto native_window = popup_->GetNativeWindow();
  native_window->SetVisibility(platform::gui::WindowVisibilityType::Show);
  popup_->GetControlHost()->RelayoutWithSize(Size::Infinite(), true);
  native_window->RequestFocus();
  native_window->SetToForeground();
}

void PopupMenu::Close() { popup_->GetNativeWindow()->Close(); }
}  // namespace cru::ui::components