// 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/notification_telemetry/notification_telemetry_service_factory.h"

#include "base/task/sequenced_task_runner.h"
#include "base/task/thread_pool.h"
#include "build/build_config.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/push_messaging/push_messaging_service_factory.h"
#include "chrome/browser/safe_browsing/notification_telemetry/notification_telemetry_service.h"
#include "chrome/browser/safe_browsing/safe_browsing_service.h"
#include "components/safe_browsing/core/browser/db/database_manager.h"
#include "components/safe_browsing/core/common/safe_browsing_prefs.h"
#include "content/public/browser/browser_context.h"

namespace safe_browsing {

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

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

NotificationTelemetryServiceFactory::NotificationTelemetryServiceFactory()
    : ProfileKeyedServiceFactory(
          "NotificationTelemetryService",
          ProfileSelections::Builder()
              .WithRegular(ProfileSelection::kOriginalOnly)
              .Build()) {
  DependsOn(PushMessagingServiceFactory::GetInstance());
}

std::unique_ptr<KeyedService>
NotificationTelemetryServiceFactory::BuildServiceInstanceForBrowserContext(
    content::BrowserContext* context) const {
// Exclude Android arm32 devices for performance and memory reasons.
// The ClientIncidentReport proto used to send these reports increases the
// Android binary size by more than the arm32 threshold.
#if !BUILDFLAG(IS_ANDROID) || (BUILDFLAG(IS_ANDROID) && defined(ARCH_CPU_ARM64))
  if (!g_browser_process || !g_browser_process->safe_browsing_service()) {
    return nullptr;
  }

  Profile* profile = Profile::FromBrowserContext(context);
  bool isEsb = IsEnhancedProtectionEnabled(*profile->GetPrefs());

  if (base::FeatureList::IsEnabled(
          kGlobalCacheListForGatingNotificationProtections)) {
    return std::make_unique<NotificationTelemetryService>(
        Profile::FromBrowserContext(context), nullptr,
        isEsb ? g_browser_process->safe_browsing_service()->ui_manager()
              : nullptr);
  }

  // TODO(crbug.com/433543634): Clean up the use of `database_manager` post
  // GlobalCacheListForGatingNotificationProtections launch.
  if (!g_browser_process->safe_browsing_service()->database_manager()) {
    return nullptr;
  }

  return std::make_unique<NotificationTelemetryService>(
      profile, g_browser_process->safe_browsing_service()->database_manager(),
      isEsb ? g_browser_process->safe_browsing_service()->ui_manager()
            : nullptr);
#else
  return nullptr;
#endif  // !(!BUILDFLAG(IS_ANDROID) || (BUILDFLAG(IS_ANDROID) &&
        // defined(ARCH_CPU_ARM64)))
}

// Create a telemetry service instance at profile creation so that
// it can register as an observer for service worker registration events.
bool NotificationTelemetryServiceFactory::ServiceIsCreatedWithBrowserContext()
    const {
  return true;
}

}  // namespace safe_browsing