// 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.

#ifndef CHROME_BROWSER_UI_VIEWS_TOOLBAR_WEBUI_HOME_CONTROL_H_
#define CHROME_BROWSER_UI_VIEWS_TOOLBAR_WEBUI_HOME_CONTROL_H_

#include "base/gtest_prod_util.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/ui/views/toolbar/home_button.h"
#include "chrome/browser/ui/views/toolbar/pinned_action_toolbar_button_menu_model.h"
#include "components/browser_apis/browser_controls/browser_controls_api.mojom.h"
#include "components/browser_apis/ui_controllers/toolbar/toolbar_ui_api_data_model.mojom.h"
#include "components/prefs/pref_member.h"
#include "ui/base/models/menu_model.h"
#include "ui/base/mojom/menu_source_type.mojom.h"
#include "ui/gfx/geometry/point.h"
#include "ui/views/controls/menu/menu_runner.h"
#include "url/gurl.h"

class WebUIToolbarControlDelegate;

// WebUIHomeControl implements C++-side functionality for the WebUI-based
// implementation of the home button in the toolbar.
class WebUIHomeControl {
 public:
  explicit WebUIHomeControl(WebUIToolbarControlDelegate* delegate);
  WebUIHomeControl(const WebUIHomeControl&) = delete;
  WebUIHomeControl& operator=(const WebUIHomeControl&) = delete;
  ~WebUIHomeControl();

  // Initializes the control. Should be called when the parent view is added to
  // the widget.
  void Init();

  // Returns true if the home button is pinned, and so should be shown if
  // there's enough room for it on the toolbar.
  bool IsPinned() const;

  // Sets whether or not the button has overflowed - that is, not displayed on
  // the toolbar because there's no space for it. When not pinned, overflowed
  // should be set to false. Updates state if needed.
  //
  // May only be set to true if features::IsWebUIToolbarFullyEnabled() is false.
  // Otherwise, overflow is managed in Javascript.
  void SetIsOverflowed(bool is_overflowed);
  bool is_overflowed() const { return is_overflowed_; }

  // Handles context menu requests from the WebUI.
  void HandleContextMenu(const gfx::Rect& screen_rect,
                         ui::mojom::MenuSourceType source);

  // Called when a URL or file is dropped on the home button. Sets the homepage
  // and displays the undo bubble.
  void OnHomeButtonDropUrl(const GURL& url);

  ui::MenuModel* GetMenuModelForTesting() { return &home_menu_; }

 private:
  FRIEND_TEST_ALL_PREFIXES(WebUIToolbarWebViewPixelBrowserTest,
                           CheckHomeButtonColor);
  FRIEND_TEST_ALL_PREFIXES(WebUIToolbarRightClickContextMenuTest,
                           RightClickShowsContextMenu);
  FRIEND_TEST_ALL_PREFIXES(WebUIHomeControlInteractiveUiTest,
                           LongPressHomeButton);
  FRIEND_TEST_ALL_PREFIXES(WebUIToolbarButtonPressAndDragTest,
                           PressAndDragDown);

  void OnIsPinnedChanged();

  void UpdateState();

  // Displays the bubble confirming the home page was set.
  void ShowSetHomePageBubble(const GURL& undo_url, bool undo_is_ntp);

  raw_ptr<WebUIToolbarControlDelegate> delegate_;
  BooleanPrefMember pin_state_;
  bool is_pinned_ = false;
  bool is_context_menu_visible_ = false;
  bool is_overflowed_ = false;

  ui::mojom::MenuSourceType last_source_type_for_testing_ =
      ui::mojom::MenuSourceType::kNone;

  PinnedActionToolbarButtonMenuModel home_menu_;
  std::unique_ptr<views::MenuRunner> menu_runner_;
  std::unique_ptr<HomePageUndoBubbleCoordinator> undo_bubble_coordinator_;
};

#endif  // CHROME_BROWSER_UI_VIEWS_TOOLBAR_WEBUI_HOME_CONTROL_H_