chrome/browser/ui/hats/hats_helper.cc

// Copyright 2018 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/hats/hats_helper.h"

#include "base/functional/callback_helpers.h"
#include "base/metrics/field_trial_params.h"
#include "chrome/browser/page_info/merchant_trust_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/hats/hats_service.h"
#include "chrome/browser/ui/hats/hats_service_factory.h"
#include "chrome/browser/ui/hats/trust_safety_sentiment_service.h"
#include "chrome/browser/ui/hats/trust_safety_sentiment_service_factory.h"
#include "chrome/browser/ui/performance_controls/performance_controls_hats_service.h"
#include "chrome/browser/ui/performance_controls/performance_controls_hats_service_factory.h"
#include "chrome/common/chrome_features.h"
#include "chrome/common/webui_url_constants.h"
#include "components/page_info/core/merchant_trust_service.h"
#include "components/safe_browsing/core/common/features.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"

HatsHelper::~HatsHelper() = default;

HatsHelper::HatsHelper(content::WebContents* web_contents)
    : WebContentsObserver(web_contents),
      content::WebContentsUserData<HatsHelper>(*web_contents) {
  // Construct PerformanceControlsHatsService immediately to register pref
  // listeners.
  performance_controls_hats_service_ =
      PerformanceControlsHatsServiceFactory::GetForProfile(profile());
}

void HatsHelper::PrimaryPageChanged(content::Page& page) {
  // Ignore everything except NTP opens.
  if (web_contents()->GetLastCommittedURL() != chrome::kChromeUINewTabURL) {
    return;
  }

  if (auto* sentiment_service =
          TrustSafetySentimentServiceFactory::GetForProfile(profile())) {
    sentiment_service->OpenedNewTabPage();
  }

  if (performance_controls_hats_service_) {
    performance_controls_hats_service_->OpenedNewTabPage();
  }

  if (auto* merchant_trust_service =
          MerchantTrustServiceFactory::GetForProfile(profile())) {
    merchant_trust_service->MaybeShowEvaluationSurvey();
  }

  // If the demo HaTS feature is enabled display a test survey on every NTP
  // load unless the "auto_prompt" parameter is explicitly set to false. The
  // demo feature also disables client-side HaTS rate limiting, thus setting
  // "auto_prompt" to false allows testing of non-demo surveys without
  // triggering a demo survey on every page load.
  const bool demo_enabled =
      base::FeatureList::IsEnabled(
          features::kHappinessTrackingSurveysForDesktopDemo) &&
      base::FeatureParam<bool>(
          &features::kHappinessTrackingSurveysForDesktopDemo, "auto_prompt",
          true)
          .Get();

  if (demo_enabled) {
    HatsService* hats_service = HatsServiceFactory::GetForProfile(
        profile(), /*create_if_necessary=*/true);

    if (hats_service) {
      hats_service->LaunchSurvey(
          kHatsSurveyTriggerTesting, base::DoNothing(), base::DoNothing(),
          {{"Test Field 1", true}, {"Test Field 2", false}},
          {{"Test Field 3", "Test value"}});
    }
  }
}

Profile* HatsHelper::profile() const {
  return Profile::FromBrowserContext(web_contents()->GetBrowserContext());
}

WEB_CONTENTS_USER_DATA_KEY_IMPL(HatsHelper);