// Copyright 2020 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_service_factory.h"

#include "chrome/browser/browser_process.h"
#include "chrome/browser/optimization_guide/model_execution/optimization_guide_global_state.h"
#include "chrome/browser/optimization_guide/optimization_guide_global_state_holder_keyed_service.h"
#include "chrome/browser/optimization_guide/optimization_guide_global_state_holder_keyed_service_factory.h"
#include "chrome/browser/profiles/incognito_helpers.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/safe_browsing/chrome_client_side_detection_service_delegate.h"
#include "chrome/common/chrome_switches.h"
#include "components/keyed_service/content/browser_context_dependency_manager.h"
#include "components/safe_browsing/content/browser/client_side_detection_service.h"
#include "components/safe_browsing/core/common/features.h"
#include "services/network/public/cpp/shared_url_loader_factory.h"

namespace safe_browsing {

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

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

ClientSideDetectionServiceFactory::ClientSideDetectionServiceFactory()
    : ProfileKeyedServiceFactory(
          "ClientSideDetectionService",
          ProfileSelections::Builder()
              .WithRegular(ProfileSelection::kOriginalOnly)
              // ChromeOS creates various profiles (login, lock screen...) that
              // do not display web content and thus do not need the
              // client side phishing detection
              .WithAshInternals(ProfileSelection::kNone)
              .Build()) {
  DependsOn(
      OptimizationGuideGlobalStateHolderKeyedServiceFactory::GetInstance());
}

std::unique_ptr<KeyedService>
ClientSideDetectionServiceFactory::BuildServiceInstanceForBrowserContext(
    content::BrowserContext* context) const {
  if (!g_browser_process || !g_browser_process->safe_browsing_service()) {
    return nullptr;
  }

  Profile* profile = Profile::FromBrowserContext(context);

  auto* holder_service =
      OptimizationGuideGlobalStateHolderKeyedServiceFactory::GetForProfile(
          Profile::FromBrowserContext(context));
  if (!holder_service) {
    return nullptr;
  }
  auto& opt_guide = holder_service->GetGlobalState().model_provider();

  return std::make_unique<ClientSideDetectionService>(
      std::make_unique<ChromeClientSideDetectionServiceDelegate>(profile),
      &opt_guide);
}

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

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

}  // namespace safe_browsing