// Copyright 2023 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/apps/app_service/metrics/website_metrics_browser_test_mixin.h"

#include <utility>

#include "base/time/default_clock.h"
#include "base/time/default_tick_clock.h"
#include "chrome/browser/apps/app_service/app_service_proxy_ash.h"
#include "chrome/browser/apps/app_service/app_service_proxy_factory.h"
#include "chrome/browser/apps/app_service/metrics/app_platform_metrics_service.h"
#include "chrome/browser/apps/app_service/metrics/website_metrics.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_manager.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/browser/ui/navigator/browser_navigator.h"
#include "chrome/browser/ui/navigator/browser_navigator_params.h"
#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/test_navigation_observer.h"
#include "ui/base/base_window.h"
#include "ui/base/window_open_disposition.h"
#include "ui/wm/public/activation_client.h"
#include "url/gurl.h"

namespace apps {

WebsiteMetricsBrowserTestMixin::WebsiteMetricsBrowserTestMixin(
    InProcessBrowserTestMixinHost* host)
    : InProcessBrowserTestMixin(host) {}

WebsiteMetricsBrowserTestMixin::~WebsiteMetricsBrowserTestMixin() = default;

void WebsiteMetricsBrowserTestMixin::SetUpOnMainThread() {
  auto* const profile = ProfileManager::GetPrimaryUserProfile();
  if (!::apps::AppServiceProxyFactory::IsAppServiceAvailableForProfile(
          profile)) {
    // Cannot initialize website metrics for the profile. Return.
    return;
  }

  auto* const app_service_proxy =
      ::apps::AppServiceProxyFactory::GetForProfile(profile);
  CHECK(app_service_proxy);
  // Instantiate app platform metrics service only if one does not exist
  // already. This ensures observers continue observing the same source while
  // retrieving the initialized website metrics component.
  app_platform_metrics_service_ =
      app_service_proxy->AppPlatformMetricsService();
  if (!app_platform_metrics_service_) {
    auto metrics_service = std::make_unique<AppPlatformMetricsService>(
        profile, base::DefaultClock::GetInstance(),
        base::DefaultTickClock::GetInstance(),
        base::SequencedTaskRunner::GetCurrentDefault());
    app_platform_metrics_service_ = metrics_service.get();
    app_service_proxy->SetAppPlatformMetricsServiceForTesting(
        std::move(metrics_service));
  }
  app_platform_metrics_service_->Start(
      app_service_proxy->AppRegistryCache(),
      app_service_proxy->InstanceRegistry(),
      app_service_proxy->AppCapabilityAccessCache());
}

BrowserWindowInterface* WebsiteMetricsBrowserTestMixin::CreateBrowser() {
  DCHECK_CURRENTLY_ON(::content::BrowserThread::UI);
  auto* const profile = ProfileManager::GetPrimaryUserProfile();
  CHECK(profile);
  BrowserWindowCreateParams params(profile, /*from_user_gesture=*/true);

  // Create a new browser instance. The subsequent `BrowserWindow` that was
  // created as part of this instantiation will own the browser instance.
  BrowserWindowInterface* const browser =
      CreateBrowserWindow(std::move(params));
  browser->GetWindow()->Show();
  auto* const window = browser->GetWindow()->GetNativeWindow();
  wm::GetActivationClient(window->GetRootWindow())->ActivateWindow(window);
  return browser;
}

::content::WebContents* WebsiteMetricsBrowserTestMixin::NavigateAndWait(
    BrowserWindowInterface* browser,
    const std::string& url,
    WindowOpenDisposition disposition) {
  NavigateParams params(browser, GURL(url),
                        ::ui::PAGE_TRANSITION_AUTO_TOPLEVEL);
  params.disposition = disposition;
  Navigate(&params);
  auto* const contents = params.navigated_or_inserted_contents.get();
  CHECK_EQ(GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
               params.navigated_or_inserted_contents),
           browser);
  ::content::TestNavigationObserver observer(contents);
  observer.Wait();
  return contents;
}

void WebsiteMetricsBrowserTestMixin::NavigateActiveTab(
    BrowserWindowInterface* browser,
    const std::string& url) {
  NavigateAndWait(browser, url, WindowOpenDisposition::CURRENT_TAB);
}

::content::WebContents* WebsiteMetricsBrowserTestMixin::InsertForegroundTab(
    BrowserWindowInterface* browser,
    const std::string& url) {
  return NavigateAndWait(browser, url,
                         WindowOpenDisposition::NEW_FOREGROUND_TAB);
}

::content::WebContents* WebsiteMetricsBrowserTestMixin::InsertBackgroundTab(
    BrowserWindowInterface* browser,
    const std::string& url) {
  return NavigateAndWait(browser, url,
                         WindowOpenDisposition::NEW_BACKGROUND_TAB);
}

AppPlatformMetricsService* WebsiteMetricsBrowserTestMixin::metrics_service() {
  return app_platform_metrics_service_;
}

WebsiteMetrics* WebsiteMetricsBrowserTestMixin::website_metrics() {
  CHECK(app_platform_metrics_service_);
  return app_platform_metrics_service_->WebsiteMetrics();
}

}  // namespace apps