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

#include "base/memory/raw_ptr.h"
#include "build/build_config.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/browser/ui/views/media_router/app_menu_test_api.h"
#include "chrome/browser/ui/views/toolbar/app_menu.h"
#include "chrome/browser/ui/views/toolbar/browser_app_menu_button.h"
#include "chrome/browser/ui/views/toolbar/toolbar_view.h"
#include "ui/views/interaction/element_tracker_views.h"

namespace {

class AppMenuTestApiViews : public test::AppMenuTestApi {
 public:
  explicit AppMenuTestApiViews(BrowserWindowInterface* browser);

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

  ~AppMenuTestApiViews() override;

  // AppMenuTestApi:
  bool IsMenuShowing() override;
  void ShowMenu() override;
  void ExecuteCommand(int command) override;

 private:
  raw_ptr<BrowserWindowInterface> browser_;
};

AppMenuTestApiViews::AppMenuTestApiViews(BrowserWindowInterface* browser)
    : browser_(browser) {}
AppMenuTestApiViews::~AppMenuTestApiViews() = default;

bool AppMenuTestApiViews::IsMenuShowing() {
  auto* button = views::AsViewClass<BrowserAppMenuButton>(
      views::ElementTrackerViews::GetInstance()->GetFirstMatchingView(
          kToolbarAppMenuButtonElementId,
          BrowserView::GetBrowserViewForBrowser(browser_)
              ->GetElementContext()));
  return button->IsMenuShowing();
}

void AppMenuTestApiViews::ShowMenu() {
  auto* button = views::AsViewClass<BrowserAppMenuButton>(
      views::ElementTrackerViews::GetInstance()->GetFirstMatchingView(
          kToolbarAppMenuButtonElementId,
          BrowserView::GetBrowserViewForBrowser(browser_)
              ->GetElementContext()));
  button->ShowMenu(views::MenuRunner::NO_FLAGS);
}

void AppMenuTestApiViews::ExecuteCommand(int command) {
  // TODO(ellyjones): This doesn't behave properly for nested menus.
  auto* button = views::AsViewClass<BrowserAppMenuButton>(
      views::ElementTrackerViews::GetInstance()->GetFirstMatchingView(
          kToolbarAppMenuButtonElementId,
          BrowserView::GetBrowserViewForBrowser(browser_)
              ->GetElementContext()));
  button->app_menu()->ExecuteCommand(command, 0);
}

}  // namespace

namespace test {

std::unique_ptr<AppMenuTestApi> AppMenuTestApi::Create(
    BrowserWindowInterface* browser) {
  return std::make_unique<AppMenuTestApiViews>(browser);
}

}  // namespace test