chrome/browser/ash/system_extensions/system_extensions_provider_factory.cc

// Copyright 2021 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/ash/system_extensions/system_extensions_provider_factory.h"

#include "base/no_destructor.h"
#include "chrome/browser/ash/profiles/profile_helper.h"
#include "chrome/browser/ash/system_extensions/system_extensions_provider.h"
#include "chrome/browser/profiles/incognito_helpers.h"
#include "chrome/browser/profiles/profile.h"
#include "components/keyed_service/content/browser_context_dependency_manager.h"
#include "components/user_manager/user_manager.h"

// static
SystemExtensionsProvider*
SystemExtensionsProviderFactory::GetForProfileIfExists(Profile* profile) {
  return static_cast<SystemExtensionsProvider*>(
      SystemExtensionsProviderFactory::GetInstance()
          .GetServiceForBrowserContext(profile, /*create=*/false));
}

// static
SystemExtensionsProviderFactory&
SystemExtensionsProviderFactory::GetInstance() {
  static base::NoDestructor<SystemExtensionsProviderFactory> instance;
  return *instance;
}

SystemExtensionsProviderFactory::SystemExtensionsProviderFactory()
    : BrowserContextKeyedServiceFactory(
          "SystemExtensionsProvider",
          BrowserContextDependencyManager::GetInstance()) {}

SystemExtensionsProviderFactory::~SystemExtensionsProviderFactory() = default;

KeyedService* SystemExtensionsProviderFactory::BuildServiceInstanceFor(
    content::BrowserContext* context) const {
  return new SystemExtensionsProvider();
}

bool SystemExtensionsProviderFactory::ServiceIsCreatedWithBrowserContext()
    const {
  return true;
}

content::BrowserContext*
SystemExtensionsProviderFactory::GetBrowserContextToUse(
    content::BrowserContext* context) const {
  DCHECK(context);
  // Enable System Extensions on the primary profile only for now. As we
  // implement new System Extension types we will enable the provider on other
  // profiles.
  Profile* const profile = Profile::FromBrowserContext(context);
  if (profile->IsSystemProfile())
    return nullptr;

  if (!chromeos::ProfileHelper::IsRegularProfile(profile))
    return nullptr;

  if (!chromeos::ProfileHelper::IsPrimaryProfile(profile))
    return nullptr;

  auto* user_manager = user_manager::UserManager::Get();
  if (user_manager && user_manager->IsLoggedInAsAnyKioskApp())
    return nullptr;

  if (profile->IsGuestSession())
    return nullptr;

  return BrowserContextKeyedServiceFactory::GetBrowserContextToUse(context);
}