chrome/browser/extensions/extension_checkup.cc

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

#include "chrome/browser/extensions/extension_checkup.h"

#include "base/metrics/field_trial_params.h"
#include "chrome/browser/extensions/extension_service.h"
#include "extensions/browser/extension_prefs.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/extension_system.h"
#include "extensions/browser/management_policy.h"
#include "extensions/common/extension_features.h"
#include "extensions/common/extension_set.h"

using content::BrowserContext;

namespace {

bool ShouldShowExtensionsCheckup(content::BrowserContext* context) {
  // Don't show the promo if the extensions checkup experiment isn't enabled.
  if (!base::FeatureList::IsEnabled(
          extensions_features::kExtensionsCheckupTool)) {
    return false;
  }

  // Don't show promo if extensions are not enabled.
  extensions::ExtensionService* extension_service =
      extensions::ExtensionSystem::Get(context)->extension_service();
  if (!extension_service || !extension_service->extensions_enabled())
    return false;

  // Don't show the promo if there are no extensions installed.
  std::unique_ptr<extensions::ExtensionSet> extension_set =
      extensions::ExtensionRegistry::Get(context)
          ->GenerateInstalledExtensionsSet();
  if (extension_set->is_empty())
    return false;

  // Check if users have non policy-installed extensions. If all the extensions
  // are policy-installed (even if they can be disabled), then do not show the
  // extensions checkup experiment.
  for (const auto& extension : *extension_set) {
    if (!extensions::Manifest::IsPolicyLocation(extension->location())) {
      return true;
    }
  }
  return false;
}

}  // namespace

namespace extensions {

bool ShouldShowExtensionsCheckupOnStartup(content::BrowserContext* context) {
  ExtensionPrefs* prefs = ExtensionPrefs::Get(context);
  if (ShouldShowExtensionsCheckup(context) &&
      base::GetFieldTrialParamValueByFeature(
          extensions_features::kExtensionsCheckupTool,
          extensions_features::kExtensionsCheckupToolEntryPointParameter) ==
          "startup" &&
      !prefs->HasUserSeenExtensionsCheckupOnStartup()) {
    // Stores a boolean in ExtensionPrefs so we can make sure that the user is
    // redirected to the extensions page upon startup once.
    prefs->SetUserHasSeenExtensionsCheckupOnStartup(true);
    return true;
  }
  return false;
}

bool ShouldShowExtensionsCheckupPromo(content::BrowserContext* context) {
  return ShouldShowExtensionsCheckup(context) &&
         base::GetFieldTrialParamValueByFeature(
             extensions_features::kExtensionsCheckupTool,
             extensions_features::kExtensionsCheckupToolEntryPointParameter) ==
             "promo";
}

}  // namespace extensions