// Copyright 2025 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/safe_browsing/client_side_detection_intelligent_scan_delegate_factory.h"

#include "build/buildflag.h"
#include "chrome/browser/enterprise/browser_management/management_service_factory.h"
#include "chrome/browser/optimization_guide/optimization_guide_keyed_service.h"
#include "chrome/browser/optimization_guide/optimization_guide_keyed_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "components/keyed_service/core/keyed_service.h"
#include "components/policy/core/common/management/management_service.h"
#include "components/safe_browsing/core/browser/intelligent_scan_delegate.h"
#if BUILDFLAG(IS_ANDROID)
#include "chrome/browser/safe_browsing/android/client_side_detection_intelligent_scan_delegate_android.h"
#else
#include "chrome/browser/safe_browsing/client_side_detection_intelligent_scan_delegate_desktop.h"
#endif

namespace safe_browsing {

// static
IntelligentScanDelegate*
ClientSideDetectionIntelligentScanDelegateFactory::GetForProfile(
    Profile* profile) {
  return static_cast<IntelligentScanDelegate*>(
      GetInstance()->GetServiceForBrowserContext(profile, /*create=*/true));
}

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

ClientSideDetectionIntelligentScanDelegateFactory::
    ClientSideDetectionIntelligentScanDelegateFactory()
    : ProfileKeyedServiceFactory(
          "IntelligentScanDelegate",
          ProfileSelections::Builder()
              .WithRegular(ProfileSelection::kOriginalOnly)
              .WithGuest(ProfileSelection::kNone)
              .WithAshInternals(ProfileSelection::kOriginalOnly)
              .Build()) {
  DependsOn(OptimizationGuideKeyedServiceFactory::GetInstance());
  DependsOn(policy::ManagementServiceFactory::GetInstance());
}

std::unique_ptr<KeyedService>
ClientSideDetectionIntelligentScanDelegateFactory::
    BuildServiceInstanceForBrowserContext(
        content::BrowserContext* context) const {
  Profile* profile = Profile::FromBrowserContext(context);
  auto* opt_guide =
      OptimizationGuideKeyedServiceFactory::GetForProfile(profile);
  if (!opt_guide) {
    return nullptr;
  }
#if BUILDFLAG(IS_ANDROID)
  return std::make_unique<ClientSideDetectionIntelligentScanDelegateAndroid>(
      *profile->GetPrefs(), opt_guide->CreateModelBrokerClient(), opt_guide);
#else
  return std::make_unique<ClientSideDetectionIntelligentScanDelegateDesktop>(
      *profile->GetPrefs(), opt_guide,
      policy::ManagementServiceFactory::GetForProfile(profile));
#endif
}

bool ClientSideDetectionIntelligentScanDelegateFactory::
    ServiceIsCreatedWithBrowserContext() const {
  // The service is created early to start listening to on-device model updates.
  return true;
}

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

}  // namespace safe_browsing