Prototype migrating old localStorage and cookies from StorageRecord.

b/362296377
youtube · Jan 8, 2025 · 43a0740 · 43a0740
1 parent 6c07cf6
commit 43a0740
Show file tree

Hide file tree

Showing 8 changed files with 493 additions and 2 deletions.
diff --git a/cobalt/BUILD.gn b/cobalt/BUILD.gn
@@ -19,6 +19,7 @@ group("gn_all") {
 
   # TODO(b/371589344): Fix android build configs.
   deps = [
+    "//cobalt/migrate_storage_record",
     "//starboard($starboard_toolchain)",
     "//starboard/nplb",
   ]
@@ -45,6 +46,7 @@ if (!is_android) {
     defines = []
 
     deps = [
+      "//cobalt/migrate_storage_record",
       "//cobalt/user_agent",
       "//content/public/app",
       "//content/shell:content_shell_app",

diff --git a/cobalt/android/BUILD.gn b/cobalt/android/BUILD.gn
@@ -168,6 +168,7 @@ shared_library("libcobalt_content_shell_content_view") {
   # TODO(b/375655377): remove testonly
   testonly = true
   deps = [
+    "//cobalt/migrate_storage_record",
     "//cobalt/user_agent",
 
     # TODO: what can be removed in the dependencies?

diff --git a/cobalt/cobalt_content_browser_client.cc b/cobalt/cobalt_content_browser_client.cc
@@ -16,14 +16,32 @@
 
 #include <string>
 
+#include "base/logging.h"
+#include "cobalt/migrate_storage_record/migrate.h"
 #include "cobalt/user_agent/user_agent_platform_info.h"
 #include "content/public/common/user_agent.h"
 #include "third_party/blink/public/common/web_preferences/web_preferences.h"
 
-#include "base/logging.h"
-
 namespace cobalt {
 
+std::atomic_bool migration_attempted_{false};
+
+void CobaltWebContentsObserver::OnCookiesAccessed(
+    content::RenderFrameHost* render_frame_host,
+    const content::CookieAccessDetails& details) {
+  if (!migration_attempted_.load()) {
+    migration_attempted_.store(true);
+    if (base::CommandLine::ForCurrentProcess()->HasSwitch(
+            "clear-cookies-and-local-storage")) {
+      LOG(INFO) << "Clearing cookies and localStorage.";
+      cobalt::migrate_storage_record::ClearAll(render_frame_host);
+    } else {
+      LOG(INFO) << "Try to migrate cookies and localStorage if needed.";
+      cobalt::migrate_storage_record::Migrate(render_frame_host);
+    }
+  }
+}
+
 #define COBALT_BRAND_NAME "Cobalt"
 #define COBALT_MAJOR_VERSION "26"
 #define COBALT_VERSION "26.lts.0-qa"
@@ -91,4 +109,9 @@ void CobaltContentBrowserClient::OverrideWebkitPrefs(
   content::ShellContentBrowserClient::OverrideWebkitPrefs(web_contents, prefs);
 }
 
+void CobaltContentBrowserClient::OnWebContentsCreated(
+    content::WebContents* web_contents) {
+  web_contents_observer_.reset(new CobaltWebContentsObserver(web_contents));
+}
+
 }  // namespace cobalt
diff --git a/cobalt/cobalt_content_browser_client.h b/cobalt/cobalt_content_browser_client.h
@@ -15,10 +15,20 @@
 #ifndef COBALT_COBALT_CONTENT_BROWSER_CLIENT_H_
 #define COBALT_COBALT_CONTENT_BROWSER_CLIENT_H_
 
+#include "content/public/browser/web_contents_observer.h"
 #include "content/shell/browser/shell_content_browser_client.h"
 
 namespace cobalt {
 
+class CobaltWebContentsObserver : public content::WebContentsObserver {
+ public:
+  explicit CobaltWebContentsObserver(content::WebContents* web_contents)
+      : content::WebContentsObserver(web_contents) {}
+
+  void OnCookiesAccessed(content::RenderFrameHost* render_frame_host,
+                         const content::CookieAccessDetails& details) override;
+};
+
 class CobaltContentBrowserClient : public content::ShellContentBrowserClient {
  public:
   CobaltContentBrowserClient();
@@ -31,6 +41,10 @@ class CobaltContentBrowserClient : public content::ShellContentBrowserClient {
   blink::UserAgentMetadata GetUserAgentMetadata() override;
   void OverrideWebkitPrefs(content::WebContents* web_contents,
                            blink::web_pref::WebPreferences* prefs) override;
+  void OnWebContentsCreated(content::WebContents* web_contents);
+
+ private:
+  std::unique_ptr<CobaltWebContentsObserver> web_contents_observer_;
 };
 
 }  // namespace cobalt

diff --git a/cobalt/migrate_storage_record/BUILD.gn b/cobalt/migrate_storage_record/BUILD.gn
@@ -0,0 +1,27 @@
+import("//testing/test.gni")
+import("//third_party/protobuf/proto_library.gni")
+
+proto_library("storage_proto") {
+  sources = [ "storage.proto" ]
+  generate_python = false
+}
+
+source_set("migrate_storage_record") {
+  # Needed to depend on |//content/shell:content_shell_lib|.
+  testonly = true
+
+  sources = [
+    "migrate.cc",
+    "migrate.h",
+  ]
+  deps = [
+    ":storage_proto",
+    "//base",
+    "//content/shell:content_shell_lib",
+    "//net",
+    "//services/network/public/mojom:cookies_mojom",
+    "//starboard/common",
+    "//url",
+  ]
+  public_deps = [ "//skia" ]
+}