// Copyright 2025 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_TABS_VERTICAL_VERTICAL_TAB_STRIP_TOP_CONTAINER_H_
#define CHROME_BROWSER_UI_VIEWS_TABS_VERTICAL_VERTICAL_TAB_STRIP_TOP_CONTAINER_H_

#include "ui/menus/simple_menu_model.h"
#include "ui/views/context_menu_controller.h"
#include "ui/views/layout/delegating_layout_manager.h"
#include "ui/views/layout/layout_types.h"
#include "ui/views/view.h"

class TabStripComboButton;
class TabStripFlatEdgeButton;
class BrowserWindowInterface;
class ExpandOnHoverLock;

namespace gfx {
class Point;
}  // namespace gfx

namespace tabs {
class VerticalTabStripStateController;
}  // namespace tabs

namespace views {
class ActionViewController;
class LabelButton;
class MenuRunner;
}  // namespace views

// Top container of the vertical tab strip, manages the collapse and tab search
// buttons, accounting for space that might be needed for caption buttons.
class VerticalTabStripTopContainer : public views::View,
                                     public views::LayoutDelegate,
                                     public views::ContextMenuController,
                                     public ui::SimpleMenuModel::Delegate {
  METADATA_HEADER(VerticalTabStripTopContainer, views::View)

 public:
  DECLARE_CLASS_ELEMENT_IDENTIFIER_VALUE(
      kToggleVerticalTabsExpandOnHoverElementId);
  VerticalTabStripTopContainer(
      tabs::VerticalTabStripStateController* state_controller,
      actions::ActionItem* root_action_item,
      BrowserWindowInterface* browser);
  ~VerticalTabStripTopContainer() override;

  // View:
  void Layout(PassKey) override;

  // LayoutDelegate:
  views::ProposedLayout CalculateProposedLayout(
      const views::SizeBounds& size_bounds) const override;

  // Creates a TopContainerButton (Collapse Button).
  views::LabelButton* AddChildButtonFor(actions::ActionId action_id);

  TabStripComboButton* GetComboButton();
  TabStripFlatEdgeButton* GetTabSearchButton();
  views::LabelButton* GetCollapseButton() { return collapse_button_; }
  views::LabelButton* GetUnfocusButton() { return unfocus_button_; }

  bool IsPositionInWindowCaption(const gfx::Point& point);

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

  // ui::SimpleMenuModel::Delegate:
  bool IsCommandIdChecked(int command_id) const override;
  bool IsCommandIdEnabled(int command_id) const override;
  void ExecuteCommand(int command_id, int event_flags) override;

  // These methods provide the toolbar height and exclusion width, before the
  // layout of this view, for use in calculating positioning of child views. If
  // an exclusion width is provided, nothing can be rendered within the
  // rectangle defined by `(caption_button_width, toolbar_height)` that is
  // aligned to the leading, top corner.
  void SetToolbarHeightForLayout(int toolbar_height);
  void SetCaptionButtonWidthForLayout(int caption_button_width);
  void SetIsExitingExpandOnHoverForLayout(bool is_exiting_expand_on_hover);
  bool WillWrapDueToOverflow(int available_width) const;

 private:
  void OnCollapseButtonContextMenuClosed();

  // Calculates the width of the visible buttons and returns the sum along with
  // the padding between them.
  int GetPreferredWidth() const;

  // Returns the minimum height required for the top container to display all
  // its buttons at their baseline positions.
  int GetBaselineMinHeight() const;

  raw_ptr<tabs::VerticalTabStripStateController> state_controller_ = nullptr;
  raw_ptr<actions::ActionItem> root_action_item_ = nullptr;
  raw_ptr<BrowserWindowInterface> browser_ = nullptr;
  raw_ptr<TabStripComboButton> combo_button_ = nullptr;
  raw_ptr<views::LabelButton> collapse_button_ = nullptr;
  raw_ptr<views::LabelButton> unfocus_button_ = nullptr;

  std::unique_ptr<views::ActionViewController> action_view_controller_;

  std::unique_ptr<ui::SimpleMenuModel> context_menu_model_;
  std::unique_ptr<views::MenuRunner> context_menu_runner_;
  std::unique_ptr<ExpandOnHoverLock> expand_on_hover_lock_;

  // This represents the toolbar (element containing toolbar buttons, omnibox,
  // app menu, etc) height.
  int toolbar_height_ = 0;
  // When provided, a rectangle formed by the caption button width is anchored
  // to the top leading edge of this container and cannot have any elements
  // rendered inside of it becaused that area is reserved for outside UI
  // elements.
  int caption_button_width_ = 0;
  // This is true when the width of the tab strip needs to shrink while exiting
  // the expand-on-hover animation. This is because the final state needs to be
  // within the wrapped state bounds when leading caption buttons are present.
  bool is_exiting_expand_on_hover_ = false;

  // This is updated during layout calculation and then applied during layout.
  mutable views::LayoutOrientation combo_button_orientation_ =
      views::LayoutOrientation::kHorizontal;
};

#endif  // CHROME_BROWSER_UI_VIEWS_TABS_VERTICAL_VERTICAL_TAB_STRIP_TOP_CONTAINER_H_