// 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_TEST_VERTICAL_TABS_BROWSER_TEST_MIXIN_H_
#define CHROME_BROWSER_UI_VIEWS_TEST_VERTICAL_TABS_BROWSER_TEST_MIXIN_H_

#include <concepts>

#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser_window/public/browser_window_features.h"
#include "chrome/browser/ui/tabs/features.h"
#include "chrome/browser/ui/tabs/split_tab_metrics.h"
#include "chrome/browser/ui/tabs/tab_strip_api/tab_strip_service_feature.h"
#include "chrome/browser/ui/tabs/vertical_tab_strip_state_controller.h"
#include "chrome/browser/ui/views/frame/base_tab_strip_region_view.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/browser/ui/views/frame/vertical_tab_strip_region_view.h"
#include "chrome/browser/ui/views/interaction/browser_elements_views.h"
#include "chrome/browser/ui/views/tabs/common/root_tab_collection_node.h"
#include "chrome/browser/ui/views/tabs/common/tab_collection_node.h"
#include "chrome/browser/ui/views/tabs/common/tab_strip_collection_controller.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/prefs/pref_service.h"

// Template to be used as a mixin class for vertical tabs tests extending
// InProcessBrowserTest.
template <typename T>
  requires(std::derived_from<T, InProcessBrowserTest>)
class VerticalTabsBrowserTestMixin : public T {
 public:
  template <class... Args>
  explicit VerticalTabsBrowserTestMixin(Args&&... args)
      : T(std::forward<Args>(args)...) {}

  ~VerticalTabsBrowserTestMixin() override = default;

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

  void SetUpCommandLine(base::CommandLine* command_line) override {
    T::SetUpCommandLine(command_line);
    scoped_feature_list_.InitWithFeaturesAndParameters(GetEnabledFeatures(),
                                                       GetDisabledFeatures());
  }

  void SetUpOnMainThread() override {
    EnterVerticalTabsMode();
    T::SetUpOnMainThread();
  }

  TabStripModel* tab_strip_model() { return T::browser()->GetTabStripModel(); }

  tabs::VerticalTabStripStateController* vertical_tab_strip_state_controller() {
    return tabs::VerticalTabStripStateController::From(T::browser());
  }

  TabStripCollectionController* vertical_tab_strip_controller() {
    auto* region_view = views::AsViewClass<BaseTabStripRegionView>(
        BrowserView::GetBrowserViewForBrowser(T::browser())->tab_strip_view());
    return region_view ? region_view->GetTabStripCollectionController()
                       : nullptr;
  }

  TabHoverCardController* hover_card_controller() {
    if (TabStripCollectionController* controller =
            vertical_tab_strip_controller()) {
      return controller->GetHoverCardController();
    }
    return nullptr;
  }

  void EnterVerticalTabsMode() {
    vertical_tab_strip_state_controller()->SetVerticalTabsEnabled(true);
    T::RunScheduledLayouts();
  }

  void ExitVerticalTabsMode() {
    vertical_tab_strip_state_controller()->SetVerticalTabsEnabled(false);
    T::RunScheduledLayouts();
  }

  tabs_api::TabStripService* tab_strip_service() {
    return TabStripServiceFeature::From(T::browser())->GetTabStripService();
  }

  virtual const std::vector<base::test::FeatureRefAndParams>
  GetEnabledFeatures() {
    return {{tabs::kVerticalTabsExpandOnHover, {}}};
  }

  virtual const std::vector<base::test::FeatureRef> GetDisabledFeatures() {
    return {};
  }

  RootTabCollectionNode* root_node() {
    auto* region_view = views::AsViewClass<BaseTabStripRegionView>(
        BrowserView::GetBrowserViewForBrowser(T::browser())->tab_strip_view());
    return region_view ? region_view->root_node_for_testing() : nullptr;
  }

  TabCollectionNode* unpinned_collection_node() {
    return root_node()->GetChildNodeOfType(TabCollectionNode::Type::UNPINNED);
  }

  TabCollectionNode* pinned_collection_node() {
    return root_node()->GetChildNodeOfType(TabCollectionNode::Type::PINNED);
  }

  void AppendPinnedTab() {
    std::unique_ptr<content::WebContents> contents =
        content::WebContents::Create(
            content::WebContents::CreateParams(T::browser()->GetProfile()));
    tab_strip_model()->InsertWebContentsAt(
        tab_strip_model()->count(), std::move(contents),
        ADD_INHERIT_OPENER | ADD_ACTIVE | ADD_PINNED);
  }

  void AppendTab() {
    NavigateToURLWithDisposition(
        T::browser(), GURL(url::kAboutBlankURL),
        WindowOpenDisposition::NEW_FOREGROUND_TAB,
        ui_test_utils::BROWSER_TEST_WAIT_FOR_LOAD_STOP);
  }

  void AppendSplitTab() {
    AppendTab();
    const int index1 = tab_strip_model()->active_index();
    AppendTab();
    const int index2 = tab_strip_model()->active_index();

    tab_strip_model()->ActivateTabAt(
        index1, TabStripUserGestureDetails(
                    TabStripUserGestureDetails::GestureType::kOther));

    tab_strip_model()->AddToNewSplit(
        {index2}, {}, split_tabs::SplitTabCreatedSource::kTabContextMenu);
  }

 private:
  base::test::ScopedFeatureList scoped_feature_list_;
};

#endif  // CHROME_BROWSER_UI_VIEWS_TEST_VERTICAL_TABS_BROWSER_TEST_MIXIN_H_