// 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 "chrome/browser/ui/ash/boca/chrome_tab_strip_delegate.h"

#include "ash/constants/ash_features.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/test_future.h"
#include "chrome/browser/apps/platform_apps/app_browsertest_util.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_tabstrip.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/browser_window/public/create_browser_window.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"
#include "chrome/test/base/ash/util/ash_test_util.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/browser/navigation_entry.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/test_navigation_observer.h"
#include "ui/base/base_window.h"

namespace {
constexpr char kTabUrl1[] = "https://foo/1";
constexpr char kTabUrl2[] = "https://foo/2";
constexpr char kTabUrl3[] = "https://foo/3";

constexpr char kDefaultTitle[] = "foo";

}  // namespace

class ChromeTabStripDelegateBrowserTest
    : public extensions::PlatformAppBrowserTest {
 public:
  ChromeTabStripDelegateBrowserTest() = default;
  ChromeTabStripDelegateBrowserTest(const ChromeTabStripDelegateBrowserTest&) =
      delete;
  ChromeTabStripDelegateBrowserTest& operator=(
      const ChromeTabStripDelegateBrowserTest&) = delete;
  ~ChromeTabStripDelegateBrowserTest() override = default;

  void SetUp() override {
    scoped_feature_list_.InitWithFeatures(
        {ash::features::kBoca, ash::features::kBocaConsumer,
         ash::features::kOnDeviceSpeechRecognition},
        /*disabled_features=*/{});
    extensions::PlatformAppBrowserTest::SetUp();
  }

  // extensions::PlatformAppBrowserTest:
  void SetUpOnMainThread() override {
    extensions::PlatformAppBrowserTest::SetUpOnMainThread();
    delegate_ = std::make_unique<ChromeTabStripDelegate>();
  }

  BrowserWindowInterface* CreateBrowser(
      const std::vector<GURL>& urls,
      std::optional<size_t> active_url_index) {
    BrowserWindowCreateParams params(BrowserWindowInterface::TYPE_NORMAL,
                                     profile(), /*from_user_gesture=*/false);
    BrowserWindowInterface* browser = CreateBrowserWindow(std::move(params));
    // Create a new tab and make sure the urls have loaded.
    for (const auto& url : urls) {
      // content::TestNavigationObserver navigation_observer(urls[i]);
      ui_test_utils::NavigateToURLWithDisposition(
          browser, url, WindowOpenDisposition::NEW_FOREGROUND_TAB,
          ui_test_utils::BROWSER_TEST_WAIT_FOR_LOAD_STOP);
    }
    return browser;
  }
  ChromeTabStripDelegate* delegate() { return delegate_.get(); }

  void ExpectTab(GURL url, std::string title, ash::TabInfo tab) {
    // Tab in test setup doesn't include title info.
    EXPECT_EQ(base::UTF8ToUTF16(title), tab.title);
    EXPECT_EQ(url, tab.url);
    EXPECT_GT(tab.last_access_timetick, base::TimeTicks());
    EXPECT_GT(tab.id, 0);
  }

 private:
  std::unique_ptr<ChromeTabStripDelegate> delegate_;
  base::test::ScopedFeatureList scoped_feature_list_;
};

IN_PROC_BROWSER_TEST_F(ChromeTabStripDelegateBrowserTest, GetTabListForWindow) {
  BrowserWindowInterface* browser =
      CreateBrowser({GURL(kTabUrl1), GURL(kTabUrl2)}, /*active_url_index=*/0);

  // Add tab in a new browser.
  CreateBrowser({GURL(kTabUrl3)}, /*active_url_index=*/1);

  auto* aura_window = browser->GetWindow()->GetNativeWindow();

  base::test::TestFuture<std::vector<ash::TabInfo>> future;
  auto tab_list = delegate()->GetTabsListForWindow(aura_window);

  ASSERT_EQ(2u, tab_list.size());
  ExpectTab(GURL(kTabUrl1), kDefaultTitle, tab_list[0]);
  ExpectTab(GURL(kTabUrl2), kDefaultTitle, tab_list[1]);

  // Verify last access time reflects the access order.
  EXPECT_GT(tab_list[1].last_access_timetick, tab_list[0].last_access_timetick);
}

IN_PROC_BROWSER_TEST_F(ChromeTabStripDelegateBrowserTest,
                       GetTabListForSWANonEmptyWindow) {
  ASSERT_EQ(1u, GlobalBrowserCollection::GetInstance()->GetSize());

  // Create browser 1 and navigate to url1 and then url2
  CreateBrowser({GURL(kTabUrl1), GURL(kTabUrl2)}, /*active_url_index=*/1);
  // Create a SWA
  ui_test_utils::BrowserCreatedObserver browser_created_observer;
  ash::test::InstallSystemAppsForTesting(profile());
  ash::test::CreateSystemWebApp(profile(), ash::SystemWebAppType::BOCA);
  BrowserWindowInterface* const swa_browser = browser_created_observer.Wait();

  chrome::AddTabAt(swa_browser, GURL(kTabUrl3),
                   /*index=*/0,
                   /*foreground=*/false);
  EXPECT_EQ(3u, GlobalBrowserCollection::GetInstance()->GetSize());

  auto tab_list = delegate()->GetTabsListForWindow(
      swa_browser->GetWindow()->GetNativeWindow());

  // Contains the new tab and webui itself.
  ASSERT_EQ(2u, tab_list.size());

  // Verify tab is listed in non-ascending order inside window based on last
  // access time.
  EXPECT_EQ(kTabUrl3, tab_list[1].url);
}

IN_PROC_BROWSER_TEST_F(ChromeTabStripDelegateBrowserTest,
                       GetTabListForSWAEmptyWindow) {
  ui_test_utils::BrowserCreatedObserver browser_created_observer;
  ash::test::InstallSystemAppsForTesting(profile());
  ash::test::CreateSystemWebApp(profile(), ash::SystemWebAppType::BOCA);
  BrowserWindowInterface* const swa_browser = browser_created_observer.Wait();

  auto tab_list = delegate()->GetTabsListForWindow(
      swa_browser->GetWindow()->GetNativeWindow());
  // Contains the webui itself.
  EXPECT_EQ(1u, tab_list.size());
}