feat: support hasCrossSiteAncestor

GoogleChromeLabs · Jul 26, 2024 · bd80a3f · bd80a3f
1 parent bed13dc
commit bd80a3f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/bidiMapper/modules/network/NetworkUtils.ts b/src/bidiMapper/modules/network/NetworkUtils.ts
@@ -227,7 +227,9 @@ export function bidiToCdpCookie(
     httpOnly: params.cookie.httpOnly ?? false,
     ...(partitionKey.sourceOrigin !== undefined && {
       partitionKey: {
-        hasCrossSiteAncestor: false,
+        hasCrossSiteAncestor: Boolean(
+          partitionKey['goog:hasCrossSiteAncestor']
+        ),
         // CDP's `partitionKey.topLevelSite` is the BiDi's `partition.sourceOrigin`.
         topLevelSite: partitionKey.sourceOrigin,
       },

diff --git a/src/bidiMapper/modules/storage/StorageProcessor.ts b/src/bidiMapper/modules/storage/StorageProcessor.ts
@@ -77,7 +77,9 @@ export class StorageProcessor {
         // are returned.
         (c) =>
           partitionKey.sourceOrigin === undefined ||
-          c.partitionKey?.topLevelSite === partitionKey.sourceOrigin
+          (c.partitionKey?.topLevelSite === partitionKey.sourceOrigin &&
+            c.partitionKey?.hasCrossSiteAncestor ===
+              partitionKey['goog:hasCrossSiteAncestor'])
       )
       .filter((cdpCookie) => {
         const bidiCookie = cdpToBiDiCookie(cdpCookie);
@@ -212,7 +214,12 @@ export class StorageProcessor {
       if (
         key !== undefined &&
         value !== undefined &&
-        !['type', 'sourceOrigin', 'userContext'].includes(key)
+        ![
+          'type',
+          'sourceOrigin',
+          'userContext',
+          'goog:hasCrossSiteAncestor',
+        ].includes(key)
       ) {
         unsupportedPartitionKeys.set(key, value);
       }
@@ -229,10 +236,13 @@ export class StorageProcessor {
 
     // Set `userContext` to `default` if not provided, as it's required in Chromium.
     const userContext = descriptor.userContext ?? 'default';
+    const hasCrossSiteAncestor =
+      descriptor['goog:hasCrossSiteAncestor'] ?? false;
 
     return {
       userContext,
       ...(sourceOrigin === undefined ? {} : {sourceOrigin}),
+      ...(sourceOrigin === undefined ? {} : {'goog:hasCrossSiteAncestor': hasCrossSiteAncestor}),
     };
   }