chrome/browser/user_notes/user_note_service_factory.cc

// Copyright 2022 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/user_notes/user_note_service_factory.h"

#include <utility>

#include "base/check_is_test.h"
#include "base/no_destructor.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/user_notes/user_note_service_delegate_impl.h"
#include "components/user_notes/browser/user_note_service.h"
#include "components/user_notes/storage/user_note_storage_impl.h"
#include "components/user_notes/user_notes_features.h"
#include "content/public/browser/browser_context.h"

namespace user_notes {

// static
UserNoteService* UserNoteServiceFactory::GetForContext(
    content::BrowserContext* context) {
  auto* instance = GetInstance();
  if (instance->service_for_testing_) {
    CHECK_IS_TEST();
    return instance->service_for_testing_;
  }
  return static_cast<UserNoteService*>(
      instance->GetServiceForBrowserContext(context,
                                            /*create=*/true));
}

// static
UserNoteServiceFactory* UserNoteServiceFactory::GetInstance() {
  static base::NoDestructor<UserNoteServiceFactory> instance;
  return instance.get();
}

// static
void UserNoteServiceFactory::SetServiceForTesting(UserNoteService* service) {
  GetInstance()->service_for_testing_ = service;
}

UserNoteServiceFactory::UserNoteServiceFactory()
    // For now, the feature is not supported in Incognito mode.
    // TODO(crbug.com/1313967): This will need to be changed if User Notes are
    // to be available in Incognito.
    : ProfileKeyedServiceFactory(
          "UserNoteService",
          ProfileSelections::Builder()
              .WithRegular(ProfileSelection::kOriginalOnly)
              // TODO(crbug.com/1418376): Check if this service is needed in
              // Guest mode.
              .WithGuest(ProfileSelection::kOriginalOnly)
              .Build()) {}

UserNoteServiceFactory::~UserNoteServiceFactory() = default;

std::unique_ptr<KeyedService>
UserNoteServiceFactory::BuildServiceInstanceForBrowserContext(
    content::BrowserContext* context) const {
  DCHECK(IsUserNotesEnabled());
  return std::make_unique<UserNoteService>(
      std::make_unique<UserNoteServiceDelegateImpl>(
          Profile::FromBrowserContext(context)),
      std::make_unique<UserNoteStorageImpl>(context->GetPath()));
}

// static
void UserNoteServiceFactory::EnsureFactoryBuilt() {
  GetInstance();
}

}  // namespace user_notes