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

#include "ash/wm/overview/birch/tab_app_selection_view.h"

#include "ash/birch/birch_item_remover.h"
#include "ash/birch/birch_model.h"
#include "ash/birch/test_birch_client.h"
#include "ash/constants/ash_features.h"
#include "ash/constants/ash_pref_names.h"
#include "ash/shell.h"
#include "ash/test/ash_test_base.h"
#include "ash/wm/coral/coral_test_util.h"
#include "ash/wm/overview/birch/coral_chip_button.h"
#include "ash/wm/overview/birch/tab_app_selection_host.h"
#include "ash/wm/overview/birch/tab_app_selection_view.h"
#include "ash/wm/overview/overview_utils.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "ui/views/test/ax_event_counter.h"
#include "ui/views/view_utils.h"

namespace ash {

class TabAppSelectionViewTest : public AshTestBase {
 public:
  void SetUp() override {
    AshTestBase::SetUp();

    // Create test birch client.
    auto* birch_model = Shell::Get()->birch_model();
    birch_client_ = std::make_unique<TestBirchClient>(birch_model);
    birch_model->SetClientAndInit(birch_client_.get());

    base::RunLoop run_loop;
    birch_model->GetItemRemoverForTest()->SetProtoInitCallbackForTest(
        run_loop.QuitClosure());
    run_loop.Run();

    // Prepare a coral response so we have a coral glanceable to click.
    std::vector<coral::mojom::GroupPtr> test_groups;
    test_groups.push_back(CreateDefaultTestGroup());
    OverrideTestResponse(std::move(test_groups));
  }

  void TearDown() override {
    Shell::Get()->birch_model()->SetClientAndInit(nullptr);
    birch_client_.reset();
    AshTestBase::TearDown();
  }

 private:
  std::unique_ptr<TestBirchClient> birch_client_;

  base::test::ScopedFeatureList feature_list_{features::kCoralFeature};
};

// Tests that the menu can be toggled to show and hide.
TEST_F(TabAppSelectionViewTest, ToggleMenu) {
  TabAppSelectionHost* menu = ShowAndGetSelectorMenu(GetEventGenerator());
  ASSERT_TRUE(menu);
  EXPECT_TRUE(menu->IsVisible());

  LeftClickOn(menu->owner_for_testing()->addon_view());
  EXPECT_FALSE(menu->IsVisible());

  LeftClickOn(menu->owner_for_testing()->addon_view());
  EXPECT_TRUE(menu->IsVisible());
}

TEST_F(TabAppSelectionViewTest, EscapeHidesMenu) {
  TabAppSelectionHost* menu = ShowAndGetSelectorMenu(GetEventGenerator());
  ASSERT_TRUE(menu);
  EXPECT_TRUE(menu->IsVisible());

  PressAndReleaseKey(ui::VKEY_ESCAPE);
  EXPECT_FALSE(menu->IsVisible());
  EXPECT_TRUE(IsInOverviewSession());
}

// Tests clicking the close buttons on the selector menu.
TEST_F(TabAppSelectionViewTest, CloseSelectorItems) {
  TabAppSelectionHost* menu = ShowAndGetSelectorMenu(GetEventGenerator());
  ASSERT_TRUE(menu);

  auto* selection_view =
      views::AsViewClass<TabAppSelectionView>(menu->GetContentsView());
  ASSERT_TRUE(selection_view);

  // Currently there are 3 tabs and 2 apps and they are hardcoded in
  // `TabAppSelectionViewTest`. There should be 2 subtitles.
  ASSERT_TRUE(selection_view->GetViewByID(TabAppSelectionView::kTabSubtitleID));
  ASSERT_TRUE(selection_view->GetViewByID(TabAppSelectionView::kAppSubtitleID));
  EXPECT_EQ(5u, selection_view->item_views_.size());

  // Simulate clicking the close button on the 3 tab items. We do this since
  // `TabAppSelectionItemView` is not exposed. Verify that the tab items are
  // gone, the tab subtitle is also gone, and all the close buttons are gone
  // since we need at least 2 items.
  selection_view->OnCloseButtonPressed(selection_view->item_views_.front());
  selection_view->OnCloseButtonPressed(selection_view->item_views_.front());
  selection_view->OnCloseButtonPressed(selection_view->item_views_.front());
  EXPECT_EQ(2u, selection_view->item_views_.size());
  EXPECT_FALSE(
      selection_view->GetViewByID(TabAppSelectionView::kTabSubtitleID));
  EXPECT_FALSE(
      selection_view->GetViewByID(TabAppSelectionView::kCloseButtonID));
}

// Tests clicking outside the selector view closes it.
TEST_F(TabAppSelectionViewTest, PressToHideMenu) {
  TabAppSelectionHost* menu = ShowAndGetSelectorMenu(GetEventGenerator());
  ASSERT_TRUE(menu);

  // Clicks on the selector itself should not hide it.
  LeftClickOn(menu->GetContentsView());
  EXPECT_TRUE(menu->IsVisible());

  // Test clicking outside the selector.
  GetEventGenerator()->MoveMouseTo(gfx::Point(1, 1));
  GetEventGenerator()->ClickLeftButton();
  EXPECT_TRUE(!menu->IsVisible());

  // Test tapping outside the selector.
  menu = ShowAndGetSelectorMenu(GetEventGenerator());
  ASSERT_TRUE(menu);
  GetEventGenerator()->GestureTapAt(gfx::Point(1, 1));
  EXPECT_TRUE(!menu->IsVisible());
}

// Tests that corresponding metrics are recorded when showing the menu, removing
// the item, and clicking on the thumbs up/down buttons.
TEST_F(TabAppSelectionViewTest, RecordsHistogram) {
  base::HistogramTester histograms;

  TabAppSelectionHost* menu = ShowAndGetSelectorMenu(GetEventGenerator());
  ASSERT_TRUE(menu);
  EXPECT_TRUE(menu->IsVisible());

  // One menu expanded is recorded.
  histograms.ExpectBucketCount("Ash.Birch.Coral.ClusterExpanded", true, 1);

  auto* selection_view =
      views::AsViewClass<TabAppSelectionView>(menu->GetContentsView());
  ASSERT_TRUE(selection_view);
  // There are 5 items.
  ASSERT_EQ(selection_view->item_views_.size(), 5u);

  // Close two items.
  selection_view->OnCloseButtonPressed(selection_view->item_views_.front());
  selection_view->OnCloseButtonPressed(selection_view->item_views_.front());

  // Click on thumbs up button.
  LeftClickOn(
      selection_view->GetViewByID(TabAppSelectionView::ViewID::kThumbsUpID));
  // One thumbs up is recorded.
  histograms.ExpectBucketCount("Ash.Birch.Coral.UserFeedback", true, 1);

  LeftClickOn(menu->owner_for_testing()->addon_view());
  EXPECT_FALSE(menu->IsVisible());

  LeftClickOn(menu->owner_for_testing()->addon_view());
  EXPECT_TRUE(menu->IsVisible());

  // Another menu expanded is recorded.
  histograms.ExpectBucketCount("Ash.Birch.Coral.ClusterExpanded", true, 2);

  // There are 3 items.
  ASSERT_EQ(selection_view->item_views_.size(), 3u);

  // Close one item.
  selection_view->OnCloseButtonPressed(selection_view->item_views_.front());

  // Click on the thumbs down button.
  LeftClickOn(
      selection_view->GetViewByID(TabAppSelectionView::ViewID::kThumbsDownID));
  // One thumbs down is recorded.
  histograms.ExpectBucketCount("Ash.Birch.Coral.UserFeedback", false, 1);

  ExitOverview();
  // After exiting Overview, the total number of removed items is recorded.
  histograms.ExpectBucketCount("Ash.Birch.Coral.ClusterItemRemoved", 3, 1);
}

class TabAppKeyboardNavigationTest : public TabAppSelectionViewTest {
 public:
  TabAppKeyboardNavigationTest() = default;
  TabAppKeyboardNavigationTest(const TabAppKeyboardNavigationTest&) = delete;
  TabAppKeyboardNavigationTest& operator=(const TabAppKeyboardNavigationTest&) =
      delete;
  ~TabAppKeyboardNavigationTest() override = default;

  // TabAppSelectionViewTest:
  void SetUp() override {
    TabAppSelectionViewTest::SetUp();
    TabAppSelectionHost* menu = ShowAndGetSelectorMenu(GetEventGenerator());
    selector_view_ =
        views::AsViewClass<TabAppSelectionView>(menu->GetContentsView());
    ax_counter_ = std::make_unique<views::test::AXEventCounter>(
        views::AXUpdateNotifier::Get());
    ax_counter_->ResetAllCounts();
  }

  void TearDown() override {
    selector_view_ = nullptr;
    TabAppSelectionViewTest::TearDown();
  }

 protected:
  views::View* GetItemViewAt(int i) {
    return selector_view_->GetItemViewAtForTesting(i);
  }

  views::View* GetItemCloseButtonAt(int i) {
    return GetItemViewAt(i)->GetViewByID(TabAppSelectionView::kCloseButtonID);
  }

  int GetItemNum() const { return selector_view_->item_views_.size(); }

  bool IsViewFocusedOnSelector(views::View* view) const {
    return selector_view_->focus_view() == view &&
           ax_counter_->GetCount(ax::mojom::Event::kSelection, view);
  }

  bool IsViewFocusedOnChip(views::View* view) const {
    return view->HasFocus() &&
           ax_counter_->GetCount(ax::mojom::Event::kFocus, view);
  }

  raw_ptr<TabAppSelectionView> selector_view_ = nullptr;
  std::unique_ptr<views::test::AXEventCounter> ax_counter_;
};

// Tests the forward Tab navigation works as expected.
TEST_F(TabAppKeyboardNavigationTest, TabForward) {
  // The tab navigation should go through the user feedback, thumbs up button,
  // thumbs down button, first item, first item's close button, coral chip, and
  // coral chip add-on button in forward order.
  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnSelector(
      selector_view_->GetViewByID(TabAppSelectionView::kUserFeedbackID)));

  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnSelector(
      selector_view_->GetViewByID(TabAppSelectionView::kThumbsUpID)));

  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnSelector(
      selector_view_->GetViewByID(TabAppSelectionView::kThumbsDownID)));
  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(0)));

  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemCloseButtonAt(0)));

  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnChip(GetFirstCoralButton()));

  PressAndReleaseKey(ui::VKEY_TAB);
  EXPECT_TRUE(IsViewFocusedOnChip(GetFirstCoralButton()->addon_view()));
}

// Tests the reverse Tab navigation works as expected.
TEST_F(TabAppKeyboardNavigationTest, TabBackward) {
  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemCloseButtonAt(0)));

  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(0)));

  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(
      selector_view_->GetViewByID(TabAppSelectionView::kThumbsDownID)));

  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(
      selector_view_->GetViewByID(TabAppSelectionView::kThumbsUpID)));

  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(
      selector_view_->GetViewByID(TabAppSelectionView::kUserFeedbackID)));

  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnChip(GetFirstCoralButton()->addon_view()));

  PressAndReleaseKey(ui::VKEY_TAB, ui::EF_SHIFT_DOWN);
  EXPECT_TRUE(IsViewFocusedOnChip(GetFirstCoralButton()));
}

// Tests the UP key works as expected.
TEST_F(TabAppKeyboardNavigationTest, UpNavigation) {
  const int item_num = GetItemNum();

  PressAndReleaseKey(ui::VKEY_UP);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(item_num - 1)));

  PressAndReleaseKey(ui::VKEY_UP);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(item_num - 2)));

  // Pressing Up until the first item is focused.
  while (!IsViewFocusedOnSelector(GetItemViewAt(0))) {
    PressAndReleaseKey(ui::VKEY_UP);
  }

  // Pressing Up again will start from the last item.
  PressAndReleaseKey(ui::VKEY_UP);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(item_num - 1)));
}

// Tests the Down key works as expected.
TEST_F(TabAppKeyboardNavigationTest, DownNavigation) {
  const int item_num = GetItemNum();

  PressAndReleaseKey(ui::VKEY_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(0)));

  PressAndReleaseKey(ui::VKEY_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(1)));

  // Pressing Down until the last item is focused.
  while (!IsViewFocusedOnSelector(GetItemViewAt(item_num - 1))) {
    PressAndReleaseKey(ui::VKEY_DOWN);
  }

  // Pressing Down again will start from the first item.
  PressAndReleaseKey(ui::VKEY_DOWN);
  EXPECT_TRUE(IsViewFocusedOnSelector(GetItemViewAt(0)));
}

}  // namespace ash