// Copyright 2019 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/optimization_guide/optimization_guide_keyed_service_factory.h"

#include "base/no_destructor.h"
#include "build/build_config.h"
#include "build/chromeos_buildflags.h"
#include "chrome/browser/download/background_download_service_factory.h"
#include "chrome/browser/optimization_guide/optimization_guide_keyed_service.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/signin/identity_manager_factory.h"
#if !BUILDFLAG(IS_ANDROID)
#include "chrome/browser/private_ai/private_ai_service_factory.h"
#endif
#include "components/optimization_guide/core/optimization_guide_features.h"
#include "content/public/browser/browser_context.h"

#if BUILDFLAG(IS_ANDROID)
#include "base/android/scoped_java_ref.h"

// Must come after other includes, because FromJniType() uses Profile.
#include "chrome/browser/optimization_guide/android/jni_headers/OptimizationGuideBridgeFactory_jni.h"
#endif  // BUILDFLAG(IS_ANDROID)

// static
OptimizationGuideKeyedService*
OptimizationGuideKeyedServiceFactory::GetForProfile(Profile* profile) {
  if (optimization_guide::features::IsOptimizationHintsEnabled()) {
    return static_cast<OptimizationGuideKeyedService*>(
        GetInstance()->GetServiceForBrowserContext(profile, true));
  }
  return nullptr;
}

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

OptimizationGuideKeyedServiceFactory::OptimizationGuideKeyedServiceFactory()
    : ProfileKeyedServiceFactory(
          "OptimizationGuideKeyedService",
          ProfileSelections::Builder()
              .WithRegular(ProfileSelection::kOwnInstance)
              // Guest Profile follows Regular Profile selection mode.
              .WithGuest(ProfileSelection::kOwnInstance)
              // Do not build the OptimizationGuideKeyedService if it's a
              // sign-in or lockscreen profile since it basically is an
              // ephemeral profile anyway and we cannot provide hints or models
              // to it anyway.
              .WithAshInternals(ProfileSelection::kNone)
              .Build()) {
  DependsOn(BackgroundDownloadServiceFactory::GetInstance());
  DependsOn(IdentityManagerFactory::GetInstance());
#if !BUILDFLAG(IS_ANDROID)
  DependsOn(private_ai::PrivateAiServiceFactory::GetInstance());
#endif
}

OptimizationGuideKeyedServiceFactory::~OptimizationGuideKeyedServiceFactory() =
    default;

std::unique_ptr<KeyedService>
OptimizationGuideKeyedServiceFactory::BuildServiceInstanceForBrowserContext(
    content::BrowserContext* context) const {
  return std::make_unique<OptimizationGuideKeyedService>(context);
}

bool OptimizationGuideKeyedServiceFactory::ServiceIsCreatedWithBrowserContext()
    const {
  return optimization_guide::features::IsOptimizationHintsEnabled();
}

bool OptimizationGuideKeyedServiceFactory::ServiceIsNULLWhileTesting() const {
  return true;
}

#if BUILDFLAG(IS_ANDROID)
static base::android::ScopedJavaLocalRef<jobject>
JNI_OptimizationGuideBridgeFactory_GetForProfile(JNIEnv* env,
                                                 Profile* profile) {
  DCHECK(profile);

  OptimizationGuideKeyedService* service =
      OptimizationGuideKeyedServiceFactory::GetForProfile(profile);
  if (!service) {
    return base::android::ScopedJavaLocalRef<jobject>();
  }
  return service->GetJavaObject();
}
#endif  // BUILDFLAG(IS_ANDROID)

#if BUILDFLAG(IS_ANDROID)
DEFINE_JNI(OptimizationGuideBridgeFactory)
#endif