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

#import "ios/chrome/browser/optimization_guide/model/optimization_guide_service_factory.h"

#import "base/feature_list.h"
#import "base/path_service.h"
#import "components/application_locale_storage/application_locale_storage.h"
#import "components/optimization_guide/core/delivery/prediction_manager.h"
#import "components/optimization_guide/core/delivery/prediction_model_store.h"
#import "components/optimization_guide/core/hints/optimization_guide_store.h"
#import "components/optimization_guide/core/optimization_guide_features.h"
#import "ios/chrome/browser/optimization_guide/model/ios_chrome_hints_manager.h"
#import "ios/chrome/browser/optimization_guide/model/optimization_guide_service.h"
#if BUILDFLAG(BUILD_WITH_MODEL_EXECUTION)
#import "ios/chrome/browser/optimization_guide/model/ios_model_execution_manager_delegate.h"
#import "ios/chrome/browser/private_ai/model/private_ai_service_factory.h"
#endif
#import "ios/chrome/browser/shared/model/application_context/application_context.h"
#import "ios/chrome/browser/shared/model/browser/browser_list_factory.h"
#import "ios/chrome/browser/shared/model/paths/paths.h"
#import "ios/chrome/browser/shared/model/profile/profile_ios.h"
#import "ios/chrome/browser/signin/model/identity_manager_factory.h"
#import "services/network/public/cpp/shared_url_loader_factory.h"

namespace {

std::unique_ptr<KeyedService> BuildOptimizationGuideService(
    ProfileIOS* profile) {
  if (!optimization_guide::features::IsOptimizationHintsEnabled()) {
    return nullptr;
  }

  ProfileIOS* original_profile = profile->GetOriginalProfile();

  // Regardless of whether the profile is off the record or not, initialize the
  // Optimization Guide with the database associated with the original profile.
  auto* proto_db_provider = original_profile->GetProtoDatabaseProvider();
  base::FilePath profile_path = original_profile->GetStatePath();

  base::WeakPtr<optimization_guide::OptimizationGuideStore> hint_store;
  if (profile->IsOffTheRecord()) {
    OptimizationGuideService* original_ogs =
        OptimizationGuideServiceFactory::GetForProfile(original_profile);
    DCHECK(original_ogs);
    hint_store = original_ogs->GetHintsManager()->hint_store();
  }

  std::unique_ptr<optimization_guide::ModelExecutionManager::Delegate>
      delegate = nullptr;
#if BUILDFLAG(BUILD_WITH_MODEL_EXECUTION)
  if (!profile->IsOffTheRecord()) {
    private_ai::PrivateAiService* private_ai_service =
        PrivateAiServiceFactory::GetForProfile(profile);
    delegate =
        std::make_unique<IOSModelExecutionManagerDelegate>(private_ai_service);
  }
#endif

  auto service = std::make_unique<OptimizationGuideService>(
      proto_db_provider, profile_path, profile->IsOffTheRecord(),
      GetApplicationContext()->GetApplicationLocaleStorage()->Get(), hint_store,
      profile->GetPrefs(), BrowserListFactory::GetForProfile(profile),
      GetApplicationContext()->GetSharedURLLoaderFactory(),
      IdentityManagerFactory::GetForProfile(profile), std::move(delegate));

  service->DoFinalInit(
      BackgroundDownloadServiceFactory::GetForProfile(profile));
  return service;
}

}  // namespace

// static
OptimizationGuideService* OptimizationGuideServiceFactory::GetForProfile(
    ProfileIOS* profile) {
  return GetInstance()->GetServiceForProfileAs<OptimizationGuideService>(
      profile, /*create=*/true);
}

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

// static
void OptimizationGuideServiceFactory::InitializePredictionModelStore() {
  base::FilePath model_downloads_dir;
  base::PathService::Get(ios::DIR_USER_DATA, &model_downloads_dir);
  model_downloads_dir = model_downloads_dir.Append(
      optimization_guide::kOptimizationGuideModelStoreDirPrefix);
  GetApplicationContext()
      ->GetOptimizationGuideGlobalState()
      ->prediction_model_store()
      .Initialize(model_downloads_dir);
}

OptimizationGuideServiceFactory::OptimizationGuideServiceFactory()
    : ProfileKeyedServiceFactoryIOS("OptimizationGuideService",
                                    ServiceCreation::kCreateWithProfile,
                                    TestingCreation::kNoServiceForTests,
                                    ProfileSelection::kOwnInstanceInIncognito) {
  DependsOn(BackgroundDownloadServiceFactory::GetInstance());
  DependsOn(BrowserListFactory::GetInstance());
  DependsOn(IdentityManagerFactory::GetInstance());
#if BUILDFLAG(BUILD_WITH_MODEL_EXECUTION)
  DependsOn(PrivateAiServiceFactory::GetInstance());
#endif
}

OptimizationGuideServiceFactory::~OptimizationGuideServiceFactory() = default;

// static
OptimizationGuideServiceFactory::TestingFactory
OptimizationGuideServiceFactory::GetDefaultFactory() {
  return base::BindOnce(&BuildOptimizationGuideService);
}

std::unique_ptr<KeyedService>
OptimizationGuideServiceFactory::BuildServiceInstanceFor(
    ProfileIOS* profile) const {
  return BuildOptimizationGuideService(profile);
}