// Copyright 2026 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/views/toolbar/webui_reload_control.h"

#include "base/time/time.h"
#include "chrome/app/chrome_command_ids.h"
#include "chrome/browser/ui/views/toolbar/webui_toolbar_web_view.h"
#include "chrome/browser/ui/webui/webui_toolbar/webui_toolbar_ui.h"
#include "chrome/grit/generated_resources.h"
#include "components/browser_apis/ui_controllers/toolbar/toolbar_ui_api_data_model.mojom.h"
#include "ui/base/window_open_disposition_utils.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/views/controls/menu/menu_runner.h"
#include "ui/views/widget/widget.h"

WebUIReloadControl::WebUIReloadControl(WebUIToolbarControlDelegate* delegate)
    : delegate_(delegate) {
  menu_model_ = std::make_unique<ui::SimpleMenuModel>(this);
  menu_model_->AddItemWithStringId(IDC_RELOAD,
                                   IDS_RELOAD_MENU_NORMAL_RELOAD_ITEM);
  menu_model_->AddItemWithStringId(IDC_RELOAD_BYPASSING_CACHE,
                                   IDS_RELOAD_MENU_HARD_RELOAD_ITEM);
  menu_model_->AddItemWithStringId(IDC_RELOAD_CLEARING_CACHE,
                                   IDS_RELOAD_MENU_EMPTY_AND_HARD_RELOAD_ITEM);

  menu_runner_ = std::make_unique<views::MenuRunner>(
      menu_model_.get(), views::MenuRunner::HAS_MNEMONICS,
      base::BindRepeating(&WebUIReloadControl::UpdateState,
                          base::Unretained(this)));

  // The accessibility and tooltip attributes are handled by the WebUI.
}

WebUIReloadControl::~WebUIReloadControl() = default;

void WebUIReloadControl::Init() {
  CHECK(!is_initialized_);
  is_initialized_ = true;
  UpdateState();
}

void WebUIReloadControl::ChangeMode(ReloadControl::Mode mode, bool force) {
  mode_ = mode;
  if (force) {
    ++state_token_;
  }
  UpdateState();
}

bool WebUIReloadControl::GetDevToolsStatusForTesting() const {
  return is_dev_tools_connected_;
}

void WebUIReloadControl::SetDevToolsStatus(bool is_dev_tools_connected) {
  is_dev_tools_connected_ = is_dev_tools_connected;
  UpdateState();
}

bool WebUIReloadControl::HandleContextMenu(views::Widget* widget,
                                           const gfx::Rect& screen_rect,
                                           ui::mojom::MenuSourceType source) {
  if (is_dev_tools_connected_) {
    menu_runner_->RunMenuAt(widget, nullptr, screen_rect,
                            views::MenuAnchorPosition::kTopLeft, source);
    UpdateState();
  }
  return true;
}

bool WebUIReloadControl::IsCommandIdChecked(int command_id) const {
  return false;
}

bool WebUIReloadControl::IsCommandIdEnabled(int command_id) const {
  return true;
}

bool WebUIReloadControl::IsCommandIdVisible(int command_id) const {
  return true;
}

bool WebUIReloadControl::GetAcceleratorForCommandId(
    int command_id,
    ui::Accelerator* accelerator) const {
  return delegate_->GetView()->GetWidget()->GetAccelerator(command_id,
                                                           accelerator);
}

void WebUIReloadControl::ExecuteCommand(int command_id, int event_flags) {
  CHECK(delegate_->GetCommandController());
  delegate_->GetCommandController()->ExecuteCommandWithDisposition(
      command_id, ui::DispositionFromEventFlags(event_flags));
}

void WebUIReloadControl::SetDoubleClickIntervalForTesting(
    base::TimeDelta double_click_interval) {
  double_click_interval_ = double_click_interval;
  UpdateState();
}

void WebUIReloadControl::UpdateState() {
  auto state = toolbar_ui_api::mojom::ReloadControlState::New();
  state->double_click_interval = double_click_interval_;
  state->can_show_menu =
      is_dev_tools_connected_ && (mode_ == ReloadControl::Mode::kReload);
  state->is_navigation_loading = (mode_ == ReloadControl::Mode::kStop);
  state->is_context_menu_visible = menu_runner_->IsRunning();
  state->state_token = state_token_;
  delegate_->OnReloadControlStateChanged(std::move(state));
}