// Copyright (c) 2013 GitHub, Inc.
// Use of this source code is governed by the MIT license that can be
// found in the LICENSE file.

#ifndef ELECTRON_SHELL_BROWSER_API_ELECTRON_API_MENU_MAC_H_
#define ELECTRON_SHELL_BROWSER_API_ELECTRON_API_MENU_MAC_H_

#include "shell/browser/api/electron_api_menu.h"

#include <map>

#include "gin/weak_cell.h"
#import "shell/browser/ui/cocoa/electron_menu_controller.h"

namespace electron {
class NativeWindow;
class WebFrameMain;

namespace api {

class MenuMac : public Menu {
 public:
  // Make public for cppgc::MakeGarbageCollected.
  explicit MenuMac(gin::Arguments* args);
  ~MenuMac() override;

  void Trace(cppgc::Visitor*) const override;

 protected:
  // Menu
  void PopupAt(BaseWindow* window,
               std::optional<WebFrameMain*> frame,
               int x,
               int y,
               int positioning_item,
               ui::mojom::MenuSourceType source_type,
               base::OnceClosure callback) override;
  void PopupOnUI(const base::WeakPtr<NativeWindow>& native_window,
                 const base::WeakPtr<WebFrameMain>& frame,
                 int32_t window_id,
                 int x,
                 int y,
                 int positioning_item,
                 base::OnceClosure callback);
  void ClosePopupAt(int32_t window_id) override;
  std::u16string GetAcceleratorTextAtForTesting(int index) const override;

 private:
  friend class Menu;

  void ClosePopupOnUI(int32_t window_id);
  void OnClosed(int32_t window_id, base::OnceClosure callback);

  ElectronMenuController* __strong menu_controller_;

  // window ID -> open context menu
  std::map<int32_t, ElectronMenuController*> popup_controllers_;

  gin::WeakCellFactory<MenuMac> weak_cell_factory_{this};
};

}  // namespace api
}  // namespace electron

#endif  // ELECTRON_SHELL_BROWSER_API_ELECTRON_API_MENU_MAC_H_
