// 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_TAB_STRIP_INTERACTIVE_TEST_MIXIN_H_
#define CHROME_BROWSER_UI_VIEWS_TEST_TAB_STRIP_INTERACTIVE_TEST_MIXIN_H_

#include <concepts>

#include "base/test/bind.h"
#include "chrome/browser/ui/browser_element_identifiers.h"
#include "chrome/browser/ui/tabs/features.h"
#include "chrome/browser/ui/ui_features.h"
#include "chrome/browser/ui/views/tabs/common/tab_group_header_view.h"
#include "chrome/browser/ui/views/tabs/common/tab_view.h"
#include "chrome/browser/ui/views/tabs/tab_strip.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/interaction/interaction_test_util_browser.h"
#include "chrome/test/interaction/interactive_browser_test.h"
#include "ui/base/interaction/interactive_test.h"

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

  ~TabStripInteractiveTestMixin() override = default;

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

  auto FinishTabstripAnimations() {
    if (base::FeatureList::IsEnabled(tabs::kTabStripUnification)) {
      return T::Steps(T::WaitForShow(kTabStripRegionElementId));
    }
    return T::Steps(T::WaitForShow(kTabStripElementId),
                    T::WithView(kTabStripElementId, [](TabStrip* tab_strip) {
                      tab_strip->StopAnimating();
                    }));
  }

  auto FocusTabAt(int index) {
    const char kTabToFocus[] = "Tab to focus";
    if (base::FeatureList::IsEnabled(tabs::kTabStripUnification)) {
      return T::Steps(FinishTabstripAnimations(),
                      T::template NameDescendantViewByType<TabView>(
                          kBrowserViewElementId, kTabToFocus, index),
                      T::FocusElement(kTabToFocus));
    }
    return T::Steps(FinishTabstripAnimations(),
                    T::template NameDescendantViewByType<Tab>(
                        kTabStripElementId, kTabToFocus, index),
                    T::FocusElement(kTabToFocus));
  }

  auto HoverTabAt(int index) {
    const char kTabToHover[] = "Tab to hover";
    if (base::FeatureList::IsEnabled(tabs::kTabStripUnification)) {
      return T::Steps(FinishTabstripAnimations(),
                      T::template NameDescendantViewByType<TabView>(
                          kBrowserViewElementId, kTabToHover, index),
                      T::MoveMouseTo(kTabToHover));
    }
    return T::Steps(FinishTabstripAnimations(),
                    T::template NameDescendantViewByType<Tab>(
                        kTabStripElementId, kTabToHover, index),
                    T::MoveMouseTo(kTabToHover));
  }

  auto HoverTabGroupHeader(tab_groups::TabGroupId group_id) {
    const char kTabGroupHeaderToHover[] = "Tab group header to hover";
    return T::Steps(
        FinishTabstripAnimations(),
        T::NameDescendantView(
            kBrowserViewElementId, kTabGroupHeaderToHover,
            base::BindRepeating(
                [](tab_groups::TabGroupId group_id, const views::View* view) {
                  if (const TabGroupHeader* header =
                          views::AsViewClass<TabGroupHeader>(view)) {
                    return header->group().value() == group_id;
                  }
                  if (const TabGroupHeaderView* header =
                          views::AsViewClass<TabGroupHeaderView>(view)) {
                    return header->group() == group_id;
                  }
                  return false;
                },
                group_id)),
        T::MoveMouseTo(kTabGroupHeaderToHover));
  }

  auto WaitForActiveTabChange(int index) {
    DEFINE_LOCAL_STATE_IDENTIFIER_VALUE(ui::test::PollingStateObserver<int>,
                                        kTabActiveChangeObserver);
    return T::Steps(
        T::PollState(kTabActiveChangeObserver,
                     base::BindRepeating(
                         &TabStripModel::active_index,
                         base::Unretained(T::browser()->GetTabStripModel()))),
        T::WaitForState(kTabActiveChangeObserver, index),
        T::StopObservingState(kTabActiveChangeObserver));
  }
};

#endif  // CHROME_BROWSER_UI_VIEWS_TEST_TAB_STRIP_INTERACTIVE_TEST_MIXIN_H_