// Copyright 2023 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/hash_realtime_service_factory.h"

#include "chrome/browser/browser_process.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/safe_browsing/network_context_service_factory.h"
#include "chrome/browser/safe_browsing/ohttp_key_service_factory.h"
#include "chrome/browser/safe_browsing/safe_browsing_service.h"
#include "chrome/browser/safe_browsing/v5_search_hashes_cache_factory.h"
#include "components/safe_browsing/content/browser/web_ui/web_ui_content_info_singleton.h"
#include "components/safe_browsing/core/browser/db/v5_search_hashes_cache.h"
#include "components/safe_browsing/core/browser/hashprefix_realtime/hash_realtime_service.h"
#include "content/public/browser/browser_context.h"

namespace safe_browsing {

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

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

HashRealTimeServiceFactory::HashRealTimeServiceFactory()
    : ProfileKeyedServiceFactory(
          "HashRealTimeService",
          ProfileSelections::Builder()
              .WithRegular(ProfileSelection::kOriginalOnly)
              // TODO(crbug.com/41488885): Check if this service is needed for
              // Ash Internals.
              .WithAshInternals(ProfileSelection::kOriginalOnly)
              .Build()) {
  DependsOn(V5SearchHashesCacheFactory::GetInstance());
  DependsOn(NetworkContextServiceFactory::GetInstance());
  DependsOn(OhttpKeyServiceFactory::GetInstance());
}

std::unique_ptr<KeyedService>
HashRealTimeServiceFactory::BuildServiceInstanceForBrowserContext(
    content::BrowserContext* context) const {
  if (!g_browser_process->safe_browsing_service()) {
    return nullptr;
  }
  Profile* profile = Profile::FromBrowserContext(context);
  return std::make_unique<HashRealTimeService>(
      base::BindRepeating(&HashRealTimeServiceFactory::GetNetworkContext,
                          profile),
      V5SearchHashesCacheFactory::GetForProfile(profile),
      OhttpKeyServiceFactory::GetForProfile(profile),
      WebUIContentInfoSingleton::GetInstance());
}

// static
network::mojom::NetworkContext* HashRealTimeServiceFactory::GetNetworkContext(
    Profile* profile) {
  return g_browser_process->safe_browsing_service()->GetNetworkContext(profile);
}

}  // namespace safe_browsing