Native re-implementation of Web Discovery (part 1)

browser/brave_profile_prefs.cc

@@ -48,6 +48,7 @@
 #include "brave/components/search_engines/brave_prepopulated_engines.h"
 #include "brave/components/speedreader/common/buildflags/buildflags.h"
 #include "brave/components/tor/buildflags/buildflags.h"
+#include "brave/components/web_discovery/browser/wdp_service.h"
 #include "build/build_config.h"
 #include "chrome/browser/prefetch/pref_names.h"
 #include "chrome/browser/prefs/session_startup_pref.h"
@@ -508,6 +509,8 @@ void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry) {
 
   registry->SetDefaultPrefValue(prefs::kSearchSuggestEnabled,
                                 base::Value(false));
+
+  web_discovery::WDPService::RegisterProfilePrefs(registry);
 }
 
 }  // namespace brave

browser/browser_context_keyed_service_factories.cc

@@ -30,6 +30,7 @@
 #include "brave/browser/search_engines/search_engine_tracker.h"
 #include "brave/browser/sync/brave_sync_alerts_service_factory.h"
 #include "brave/browser/url_sanitizer/url_sanitizer_service_factory.h"
+#include "brave/browser/web_discovery/wdp_service_factory.h"
 #include "brave/components/ai_chat/core/common/buildflags/buildflags.h"
 #include "brave/components/brave_perf_predictor/browser/named_third_party_registry_factory.h"
 #include "brave/components/brave_vpn/common/buildflags/buildflags.h"
@@ -216,6 +217,8 @@ void EnsureBrowserContextKeyedServiceFactoriesBuilt() {
 #if BUILDFLAG(ENABLE_AI_CHAT)
   ai_chat::ModelServiceFactory::GetInstance();
 #endif
+
+  web_discovery::WDPServiceFactory::GetInstance();
 }
 
 }  // namespace brave

browser/sources.gni

@@ -135,6 +135,7 @@ brave_chrome_browser_deps = [
   "//brave/browser/sync",
   "//brave/browser/themes",
   "//brave/browser/ui",
+  "//brave/browser/web_discovery",
   "//brave/common",
   "//brave/components/ai_chat/core/common/buildflags",
   "//brave/components/ai_rewriter/common/buildflags",
@@ -204,6 +205,7 @@ brave_chrome_browser_deps = [
   "//brave/components/speedreader/common/buildflags",
   "//brave/components/tor/buildflags",
   "//brave/components/version_info",
+  "//brave/components/web_discovery/browser",
   "//brave/components/webcompat:core",
   "//brave/services/network/public/cpp",
   "//brave/third_party/blink/renderer",

browser/web_discovery/BUILD.gn

@@ -5,6 +5,20 @@
 
 import("//extensions/buildflags/buildflags.gni")
 
+source_set("web_discovery") {
+  sources = [
+    "wdp_service_factory.cc",
+    "wdp_service_factory.h",
+  ]
+  deps = [
+    "//base",
+    "//brave/components/web_discovery/browser",
+    "//components/keyed_service/content",
+    "//components/user_prefs",
+    "//content/public/browser",
+  ]
+}
+
 source_set("unit_tests") {
   if (enable_extensions) {
     testonly = true

browser/web_discovery/wdp_service_factory.cc

@@ -0,0 +1,47 @@
+/* Copyright (c) 2024 The Brave Authors. All rights reserved.
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+#include "brave/browser/web_discovery/wdp_service_factory.h"
+
+#include "brave/components/web_discovery/browser/wdp_service.h"
+#include "components/keyed_service/content/browser_context_dependency_manager.h"
+#include "components/user_prefs/user_prefs.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/storage_partition.h"
+
+namespace web_discovery {
+
+WDPServiceFactory* WDPServiceFactory::GetInstance() {
+  static base::NoDestructor<WDPServiceFactory> instance;
+  return instance.get();
+}
+
+WDPServiceFactory::WDPServiceFactory()
+    : BrowserContextKeyedServiceFactory(
+          "WDPService",
+          BrowserContextDependencyManager::GetInstance()) {}
+
+WDPServiceFactory::~WDPServiceFactory() = default;
+
+KeyedService* WDPServiceFactory::BuildServiceInstanceFor(
+    content::BrowserContext* context) const {
+  auto* default_storage_partition = context->GetDefaultStoragePartition();
+  auto shared_url_loader_factory =
+      default_storage_partition->GetURLLoaderFactoryForBrowserProcess();
+  return new WDPService(user_prefs::UserPrefs::Get(context),
+                        shared_url_loader_factory);
+}
+
+content::BrowserContext* WDPServiceFactory::GetBrowserContextToUse(
+    content::BrowserContext* context) const {
+  // Prevents creation of service instance for incognito/OTR profiles
+  return context->IsOffTheRecord() ? nullptr : context;
+}
+
+bool WDPServiceFactory::ServiceIsCreatedWithBrowserContext() const {
+  return true;
+}
+
+}  // namespace web_discovery

browser/web_discovery/wdp_service_factory.h

@@ -0,0 +1,36 @@
+/* Copyright (c) 2024 The Brave Authors. All rights reserved.
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+#ifndef BRAVE_BROWSER_WEB_DISCOVERY_WDP_SERVICE_FACTORY_H_
+#define BRAVE_BROWSER_WEB_DISCOVERY_WDP_SERVICE_FACTORY_H_
+
+#include "base/no_destructor.h"
+#include "components/keyed_service/content/browser_context_keyed_service_factory.h"
+
+namespace web_discovery {
+
+class WDPServiceFactory : public BrowserContextKeyedServiceFactory {
+ public:
+  static WDPServiceFactory* GetInstance();
+
+ private:
+  friend base::NoDestructor<WDPServiceFactory>;
+
+  WDPServiceFactory();
+  ~WDPServiceFactory() override;
+
+  WDPServiceFactory(const WDPServiceFactory&) = delete;
+  WDPServiceFactory& operator=(const WDPServiceFactory&) = delete;
+
+  KeyedService* BuildServiceInstanceFor(
+      content::BrowserContext* context) const override;
+  content::BrowserContext* GetBrowserContextToUse(
+      content::BrowserContext* context) const override;
+  bool ServiceIsCreatedWithBrowserContext() const override;
+};
+
+}  // namespace web_discovery
+
+#endif  // BRAVE_BROWSER_WEB_DISCOVERY_WDP_SERVICE_FACTORY_H_

components/web_discovery/browser/BUILD.gn

@@ -0,0 +1,32 @@
+# Copyright (c) 2024 The Brave Authors. All rights reserved.
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at https://mozilla.org/MPL/2.0/.
+
+static_library("browser") {
+  sources = [
+    "credential_manager.cc",
+    "credential_manager.h",
+    "pref_names.h",
+    "rsa.cc",
+    "rsa.h",
+    "server_config_loader.cc",
+    "server_config_loader.h",
+    "util.cc",
+    "util.h",
+    "wdp_service.cc",
+    "wdp_service.h",
+  ]
+  deps = [
+    "anonymous_credentials/rs:rust_lib",
+    "//base",
+    "//brave/brave_domains",
+    "//brave/components/constants",
+    "//components/keyed_service/core",
+    "//components/prefs",
+    "//net",
+    "//services/network/public/cpp",
+    "//third_party/boringssl",
+    "//url",
+  ]
+}

components/web_discovery/browser/DEPS

@@ -0,0 +1,4 @@
+include_rules = [
+  "+services/network/public",
+  "+third_party/boringssl/src/include",
+]

components/web_discovery/browser/anonymous_credentials/rs/.gitignore

@@ -0,0 +1,3 @@
+target
+example/main
+Cargo.lock

components/web_discovery/browser/anonymous_credentials/rs/BUILD.gn

@@ -0,0 +1,21 @@
+# Copyright (c) 2024 The Brave Authors. All rights reserved.
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at https://mozilla.org/MPL/2.0/.
+
+import("//build/rust/rust_static_library.gni")
+
+rust_static_library("rust_lib") {
+  crate_name = "anonymous_credentials_cxx"
+  crate_root = "src/lib.rs"
+  allow_unsafe = true
+
+  edition = "2021"
+  sources = [ "src/lib.rs" ]
+
+  visibility = [ "//brave/components/web_discovery/browser:*" ]
+
+  cxx_bindings = [ "src/lib.rs" ]
+
+  deps = [ "//brave/third_party/rust/anonymous_credentials/v0_1:lib" ]
+}