// Copyright 2011 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/download/download_ui_context_menu_view.h"

#include <memory>
#include <utility>

#include "base/check.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/i18n/rtl.h"
#include "base/memory/weak_ptr.h"
#include "base/metrics/histogram_functions.h"
#include "chrome/browser/download/bubble/download_bubble_ui_controller.h"
#include "chrome/browser/download/download_commands.h"
#include "chrome/browser/download/download_stats.h"
#include "chrome/browser/download/download_ui_context_menu.h"
#include "chrome/browser/download/download_ui_model.h"
#include "ui/base/mojom/menu_source_type.mojom-forward.h"
#include "ui/menus/simple_menu_model.h"
#include "ui/views/controls/menu/menu_runner.h"
#include "ui/views/controls/menu/menu_types.h"

DownloadUiContextMenuView::DownloadUiContextMenuView(
    base::WeakPtr<DownloadUIModel> download_ui_model)
    : DownloadUiContextMenu(download_ui_model) {}

DownloadUiContextMenuView::DownloadUiContextMenuView(
    base::WeakPtr<DownloadUIModel> download_ui_model,
    base::WeakPtr<DownloadBubbleUIController> bubble_controller)
    : DownloadUiContextMenu(download_ui_model),
      bubble_controller_(std::move(bubble_controller)) {}

DownloadUiContextMenuView::~DownloadUiContextMenuView() = default;

void DownloadUiContextMenuView::Run(
    views::Widget* parent_widget,
    const gfx::Rect& rect,
    ui::mojom::MenuSourceType source_type,
    base::RepeatingClosure on_menu_closed_callback) {
  using Position = views::MenuAnchorPosition;
  ui::MenuModel* menu_model = GetMenuModel();
  // Run() should not be getting called if the DownloadItem was destroyed.
  DCHECK(menu_model);

  menu_runner_ = std::make_unique<views::MenuRunner>(
      menu_model,
      views::MenuRunner::HAS_MNEMONICS | views::MenuRunner::CONTEXT_MENU,
      base::BindRepeating(&DownloadUiContextMenuView::OnMenuClosed,
                          base::Unretained(this),
                          std::move(on_menu_closed_callback)));

  // The menu's alignment is determined based on the UI layout.
  Position position;
  if (base::i18n::IsRTL()) {
    position = Position::kTopRight;
  } else {
    position = Position::kTopLeft;
  }

  menu_runner_->RunMenuAt(parent_widget, nullptr, rect, position, source_type);
}

void DownloadUiContextMenuView::SetOnMenuWillShowCallback(
    base::OnceClosure on_menu_will_show_callback) {
  on_menu_will_show_callback_ = std::move(on_menu_will_show_callback);
}

void DownloadUiContextMenuView::OnMenuClosed(
    base::RepeatingClosure on_menu_closed_callback) {
  close_time_ = base::TimeTicks::Now();

  // This must be run before clearing |menu_runner_| who owns the reference.
  if (!on_menu_closed_callback.is_null()) {
    on_menu_closed_callback.Run();
  }

  menu_runner_.reset();
}

void DownloadUiContextMenuView::OnMenuWillShow(ui::SimpleMenuModel* source) {
  if (on_menu_will_show_callback_) {
    std::move(on_menu_will_show_callback_).Run();
  }
}

void DownloadUiContextMenuView::ExecuteCommand(int command_id,
                                               int event_flags) {
  if (!download_commands_executed_recorded_[command_id]) {
    base::UmaHistogramEnumeration(
        "Download.ContextMenuAction",
        DownloadCommandToContextMenuAction(
            static_cast<DownloadCommands::Command>(command_id),
            /*clicked=*/true));
    download_commands_executed_recorded_[command_id] = true;
  }

  DownloadUiContextMenu::ExecuteCommand(command_id, event_flags);
  // ExecuteCommand can delete `this`.
}