@salesforce/lds-drafts 1.108.0 → 1.109.0

package/dist/ldsDrafts.js

@@ -1648,7 +1648,7 @@ class DraftManager {
 }
 
 function makeEnvironmentDraftAware(luvio, env, durableStore, handlers, draftQueue) {
-    let draftMetadata = null;
+    const draftMetadata = {};
     // setup existing store redirects when bootstrapping the environment
     (async () => {
         const mappings = await getDraftIdMappings(durableStore);
@@ -1678,7 +1678,6 @@ function makeEnvironmentDraftAware(luvio, env, durableStore, handlers, draftQueu
         if (queue.length === 0) {
             return env.handleSuccessResponse(ingestAndBroadcastFunc, getResponseCacheKeysFunc);
         }
-        const ingestMetadata = {};
         const cacheKeySet = getResponseCacheKeysFunc();
         for (const possibleKey of cacheKeySet.keys()) {
             const key = typeof possibleKey === 'string' ? possibleKey : '';
@@ -1691,37 +1690,44 @@ function makeEnvironmentDraftAware(luvio, env, durableStore, handlers, draftQueu
                         // if this key is related to a draft then mark it mergeable so
                         // base environment revives it to staging store before ingestion
                         cacheKey.mergeable = true;
-                        ingestMetadata[key] = metadata;
+                        const existing = draftMetadata[key];
+                        if (existing === undefined) {
+                            draftMetadata[key] = { metadata, refCount: 1 };
+                        }
+                        else {
+                            draftMetadata[key] = { metadata, refCount: existing.refCount + 1 };
+                        }
                     }
                     break;
                 }
             }
         }
-        draftMetadata = keys(ingestMetadata).length > 0 ? ingestMetadata : null;
         return env.handleSuccessResponse(ingestAndBroadcastFunc, () => cacheKeySet);
     };
     const storePublish = function (key, data) {
-        if (draftMetadata === null) {
-            // no drafts for this response
-            return env.storePublish(key, data);
-        }
+        const draftMetadataForKey = draftMetadata[key];
         for (const handler of handlers) {
             if (handler.canHandlePublish(key)) {
-                handler.applyDraftsToIncomingData(key, data, draftMetadata[key], env.storePublish);
+                handler.applyDraftsToIncomingData(key, data, draftMetadataForKey && draftMetadataForKey.metadata, env.storePublish);
                 return;
             }
         }
+        if (draftMetadataForKey !== undefined) {
+            // WARNING: this code depends on the fact that RecordRepresentation
+            // fields get published before the root record.
+            if (draftMetadataForKey.refCount === 1) {
+                delete draftMetadata[key];
+            }
+            else {
+                draftMetadataForKey.refCount--;
+            }
+        }
         // no handler could handle it so publish
         env.storePublish(key, data);
     };
-    const storeBroadcast = function (rebuildSnapshot, snapshotAvailable) {
-        draftMetadata = create(null);
-        return env.storeBroadcast(rebuildSnapshot, snapshotAvailable);
-    };
     // note the makeEnvironmentUiApiRecordDraftAware will eventually go away once the adapters become draft aware
     return create(env, {
         storePublish: { value: storePublish },
-        storeBroadcast: { value: storeBroadcast },
         handleSuccessResponse: { value: handleSuccessResponse },
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@salesforce/lds-drafts",
-    "version": "1.108.0",
+    "version": "1.109.0",
     "license": "SEE LICENSE IN LICENSE.txt",
     "description": "LDS Drafts",
     "main": "dist/ldsDrafts.js",
@@ -26,6 +26,6 @@
     "dependencies": {
         "@luvio/engine": "0.135.4",
         "@luvio/environments": "0.135.4",
-        "@salesforce/lds-utils-adapters": "^1.108.0"
+        "@salesforce/lds-utils-adapters": "^1.109.0"
     }
 }