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

#ifndef CHROME_BROWSER_UI_VIEWS_EXTENSIONS_EXTENSION_CONTEXT_MENU_CONTROLLER_H_
#define CHROME_BROWSER_UI_VIEWS_EXTENSIONS_EXTENSION_CONTEXT_MENU_CONTROLLER_H_

#include <memory>

#include "base/memory/raw_ptr.h"
#include "chrome/browser/extensions/extension_context_menu_model.h"
#include "ui/base/mojom/menu_source_type.mojom-forward.h"
#include "ui/views/context_menu_controller.h"

class ToolbarActionViewModel;

namespace views {
class MenuModelAdapter;
class MenuRunner;
}  // namespace views

class ExtensionContextMenuController : public views::ContextMenuController {
 public:
  class Observer {
   public:
    // Called when a context menu is shown.
    virtual void OnContextMenuShown() = 0;

    // Called when a context menu.
    virtual void OnContextMenuClosed() = 0;

   protected:
    virtual ~Observer() = default;
  };

  ExtensionContextMenuController(
      ToolbarActionViewModel* action_model,
      Observer* observer,
      extensions::ExtensionContextMenuModel::ContextMenuSource
          context_menu_source);

  ExtensionContextMenuController(const ExtensionContextMenuController&) =
      delete;
  ExtensionContextMenuController& operator=(
      const ExtensionContextMenuController&) = delete;

  ~ExtensionContextMenuController() override;

  // views::ContextMenuController:
  void ShowContextMenuForViewImpl(
      views::View* source,
      const gfx::Point& point,
      ui::mojom::MenuSourceType source_type) override;

  bool IsMenuRunning() const;

 private:
  // Callback for MenuModelAdapter.
  void OnMenuClosed();

  // Responsible for converting the context menu model into |menu_|.
  std::unique_ptr<views::MenuModelAdapter> menu_adapter_;

  // Responsible for running the menu.
  std::unique_ptr<views::MenuRunner> menu_runner_;

  // This controller contains the data for the extension's context menu.
  const raw_ptr<ToolbarActionViewModel> action_model_;

  // The observer to notify when the context menu opens/closes.
  const raw_ptr<Observer> observer_;

  // Location where the context menu is open from.
  extensions::ExtensionContextMenuModel::ContextMenuSource context_menu_source_;
};

#endif  // CHROME_BROWSER_UI_VIEWS_EXTENSIONS_EXTENSION_CONTEXT_MENU_CONTROLLER_H_