@apollo/client 4.3.0-alpha.4 → 4.3.0-alpha.6

package/__cjs/cache/inmemory/readFromStore.cjs

@@ -96,12 +96,10 @@ class StoreReader {
             },
         });
     }
-    diffQueryAgainstStore({ store, query, rootId = "ROOT_QUERY", variables, returnPartialData = true, [internal_1.handleIncrementalSymbol]: handleIncremental, }) {
+    diffQueryAgainstStore({ store, query, rootId = "ROOT_QUERY", variables, returnPartialData = true, ...options }) {
+        const returnIncremental = Object.hasOwn(options, internal_1.handleIncrementalSymbol);
         const policies = this.config.cache.policies;
-        variables = {
-            ...(0, internal_1.getDefaultValues)((0, internal_1.getQueryDefinition)(query)),
-            ...variables,
-        };
+        variables = (0, internal_1.compact)((0, internal_1.getDefaultValues)((0, internal_1.getQueryDefinition)(query)), variables);
         const rootRef = (0, internal_1.makeReference)(rootId);
         const context = {
             store,
@@ -110,7 +108,7 @@ class StoreReader {
             variables,
             varString: (0, utilities_1.canonicalStringify)(variables),
             ...(0, helpers_js_1.extractFragmentContext)(query, this.config.fragments),
-            ...(typeof handleIncremental === "object" ? handleIncremental : (undefined)),
+            ...options[internal_1.handleIncrementalSymbol],
         };
         let execResult = this.executeSelectionSet({
             selectionSet: (0, internal_1.getMainDefinition)(query).selectionSet,
@@ -123,17 +121,7 @@ class StoreReader {
         // partial defer boundaries. The "deferPartial" data state tells us that the
         // only part of the result that contributed to its partiality is data inside
         // a defer boundary.
-        if (handleIncremental &&
-            (execResult.dataState === "deferPartial" ||
-                execResult.dataState === "streamPartial" ||
-                // If the last cache write repaired a partial @stream array to a
-                // complete array, the stream array might contain stale entries after
-                // the last written value. We only want to deliver the results up to
-                // the index the network wrote so we need to prune it too.
-                context.streamInfo ||
-                // The network hasn't delivered these @defer boundaries yet, so prune
-                // the (possibly complete) cached data sitting at them.
-                context.deferInfo)) {
+        if (returnIncremental && shouldPrune(execResult, context)) {
             const pruned = this.prunePartialBoundaries({
                 selectionSet: (0, internal_1.getMainDefinition)(query).selectionSet,
                 data: execResult.result,
@@ -141,7 +129,7 @@ class StoreReader {
                 context,
                 path: [],
             });
-            const changed = execResult.result !== pruned;
+            const changed = execResult.result !== pruned.result;
             // It's possible that pruning didn't actually change the result which can
             // happen if a defer boundary is misclassified as "deferPartial" instead
             // of "streaming" (sibling defer boundaries with overlapping selection
@@ -151,19 +139,12 @@ class StoreReader {
             // we want to keep the original execResult which contains the partial
             // data.
             if (!changed || !returnPartialData) {
-                const { dataState } = execResult;
                 // Omit `missing` property since pruning puts it in a state that doesn't
                 // report missing fields.
                 execResult = {
-                    result: pruned,
+                    result: pruned.result,
                     partialBoundaries: execResult.partialBoundaries,
-                    dataState: dataState === "deferPartial" ? "streaming"
-                        : dataState === "streamPartial" ? "complete"
-                            // A cached @defer boundary the network hasn't delivered was
-                            // pruned from an otherwise complete result, so we're still
-                            // streaming.
-                            : context.deferInfo && changed ? "streaming"
-                                : dataState,
+                    dataState: pruned.dataState,
                 };
             }
         }
@@ -174,7 +155,7 @@ class StoreReader {
             // We don't need to report missing fields inside defer boundaries since
             // the "streaming" dataState tells us that the only missing fields in
             // the object is inside a defer boundary.
-            (dataState !== "streaming" || !handleIncremental);
+            (dataState !== "streaming" || !returnIncremental);
         // If we get all root @defer boundaries with an empty result, report it as
         // empty instead of streaming.
         if (dataState === "streaming" && Object.keys(result).length === 0) {
@@ -183,7 +164,7 @@ class StoreReader {
         let missingError;
         if (dataState === "deferPartial" ||
             dataState === "streamPartial" ||
-            (dataState === "streaming" && !handleIncremental)) {
+            (dataState === "streaming" && !returnIncremental)) {
             dataState = "partial";
         }
         if (dataState === "partial" && !returnPartialData) {
@@ -203,7 +184,7 @@ class StoreReader {
                 return missingError;
             },
         };
-        if (handleIncremental) {
+        if (returnIncremental) {
             diffResult.dataState =
                 dataState;
         }
@@ -308,7 +289,7 @@ class StoreReader {
                         if (typename) {
                             const policy = policies["getFieldPolicy"](typename, fieldName);
                             if (policy?.scalar) {
-                                __DEV__ && invariant_1.invariant.warn(112, `${typename}.${fieldName}`, policy.scalar);
+                                __DEV__ && invariant_1.invariant.warn(114, `${typename}.${fieldName}`, policy.scalar);
                             }
                         }
                     }
@@ -345,7 +326,7 @@ class StoreReader {
             else {
                 const fragment = (0, internal_1.getFragmentFromSelection)(selection, context.lookupFragment);
                 if (!fragment && selection.kind === graphql_1.Kind.FRAGMENT_SPREAD) {
-                    throw (0, invariant_1.newInvariantError)(113, selection.name.value);
+                    throw (0, invariant_1.newInvariantError)(115, selection.name.value);
                 }
                 if (fragment && policies.fragmentMatches(fragment, typename)) {
                     const isDeferBoundary = (0, internal_1.isDeferredFragment)(selection, context.variables);
@@ -379,7 +360,8 @@ class StoreReader {
                     if (execResult.missing) {
                         missing = missingMerger.merge(missing, execResult.missing);
                     }
-                    if (isDeferBoundary && nextDataState === "partial") {
+                    if (isDeferBoundary &&
+                        (nextDataState === "partial" || nextDataState === "empty")) {
                         partialBoundaries.add(selection);
                     }
                     dataState = mergeDataState(dataState, isDeferBoundary ?
@@ -472,10 +454,12 @@ class StoreReader {
     }
     prunePartialBoundariesImpl({ boundaries, context, data, path, selectionSet, }) {
         const { variables, lookupFragment, policies } = context;
-        if (data == null || !boundaries)
-            return data;
+        if (data == null || !boundaries) {
+            return { result: data, dataState: "complete" };
+        }
         const merger = new internal_1.DeepMerger();
         let changed = false;
+        let dataState = "complete";
         const result = {};
         // __typename might not be part of the selection set, so preserve it when
         // available, otherwise it gets removed since it's never visited when
@@ -501,8 +485,9 @@ class StoreReader {
                         context,
                         path: path.concat(resultName),
                     });
-                    changed ||= pruned !== fieldValue;
-                    result[resultName] = pruned;
+                    changed ||= pruned.result !== fieldValue;
+                    result[resultName] = pruned.result;
+                    dataState = mergeDataState(dataState, pruned.dataState);
                 }
                 else if (!selection.selectionSet) {
                     result[resultName] = fieldValue;
@@ -515,13 +500,14 @@ class StoreReader {
                         context,
                         path: path.concat(resultName),
                     });
-                    changed ||= pruned !== fieldValue;
+                    changed ||= pruned.result !== fieldValue;
+                    dataState = mergeDataState(dataState, pruned.dataState);
                     // A response key can be selected by more than one selection (e.g. a
                     // field and an overlapping fragment), so merge their kept fields.
                     result[resultName] =
                         Object.hasOwn(result, resultName) ?
-                            merger.merge(result[resultName], pruned)
-                            : pruned;
+                            merger.merge(result[resultName], pruned.result)
+                            : pruned.result;
                 }
                 return;
             }
@@ -533,11 +519,19 @@ class StoreReader {
             // Object.keys(result).length check actually detects whether keys were
             // dropped or not.
             const fragment = (0, internal_1.getFragmentFromSelection)(selection, lookupFragment);
-            if (fragment &&
-                policies.fragmentMatches(fragment, data.__typename) &&
-                !boundaries.has(selection) &&
-                !context.deferInfo?.peekArray(path)) {
-                fragment.selectionSet.selections.forEach(workSet.add, workSet);
+            let prune = false;
+            if (context.deferInfo && (0, internal_1.isDeferredFragment)(selection, variables)) {
+                const directive = selection.directives?.find((d) => d.name.value === "defer");
+                const label = directive && (0, internal_1.getDirectiveArgValue)(directive, "label", graphql_1.Kind.STRING);
+                prune = !!context.deferInfo.peekArray(path.concat(label || []));
+            }
+            if (fragment && policies.fragmentMatches(fragment, data.__typename)) {
+                if (boundaries.has(selection) || prune) {
+                    dataState = mergeDataState(dataState, "streaming");
+                }
+                else {
+                    fragment.selectionSet.selections.forEach(workSet.add, workSet);
+                }
             }
         });
         if (Object.keys(result).length !== Object.keys(data).length) {
@@ -550,18 +544,22 @@ class StoreReader {
             // actually dropped by the partial @defer fragment.
             changed = !(0, equality_1.equal)(result, data);
         }
-        return changed ? result : data;
+        return { result: changed ? result : data, dataState };
     }
     prunePartialStreamArrayImpl({ field, array, boundaries, context, path, }) {
         if (!boundaries)
-            return array;
+            return { result: array, dataState: "complete" };
         let changed = false;
+        let dataState = "complete";
         let pruned = [];
         const state = context.streamInfo?.peekArray(path)?.state;
         const length = Math.min(array.length, state?.truncate ? state.streamPosition : Number.MAX_SAFE_INTEGER);
         for (let i = 0; i < length; i++) {
             const item = array[i];
-            let prunedItem = item;
+            let prunedResult = {
+                result: item,
+                dataState: "complete",
+            };
             const boundary = boundaries.getChild(i);
             if (boundary?.has(field)) {
                 // The presence of streamInfo determines how we truncate partial
@@ -576,11 +574,12 @@ class StoreReader {
                 }
                 else {
                     pruned = [];
+                    dataState = "complete";
                 }
                 break;
             }
             if (Array.isArray(item)) {
-                prunedItem = this.prunePartialStreamArray({
+                prunedResult = this.prunePartialStreamArray({
                     field,
                     array: item,
                     boundaries: boundaries.getChild(i),
@@ -589,7 +588,7 @@ class StoreReader {
                 });
             }
             else if (field.selectionSet) {
-                prunedItem = this.prunePartialBoundaries({
+                prunedResult = this.prunePartialBoundaries({
                     data: item,
                     selectionSet: field.selectionSet,
                     boundaries: boundaries.getChild(i),
@@ -597,11 +596,12 @@ class StoreReader {
                     path: path.concat(i),
                 });
             }
-            pruned.push(prunedItem);
-            changed ||= prunedItem !== item;
+            pruned.push(prunedResult.result);
+            changed ||= prunedResult.result !== item;
+            dataState = mergeDataState(dataState, prunedResult.dataState);
         }
         changed ||= pruned.length !== array.length;
-        return changed ? pruned : array;
+        return { result: changed ? pruned : array, dataState };
     }
 }
 exports.StoreReader = StoreReader;
@@ -624,7 +624,7 @@ function assertSelectionSetForIdValue(store, field, fieldValue) {
             if ((0, internal_1.isNonNullObject)(value)) {
                 (0, invariant_1.invariant)(
                     !(0, utilities_1.isReference)(value),
-                    114,
+                    116,
                     (0, helpers_js_1.getTypenameFromStoreObject)(store, value),
                     field.name.value
                 );
@@ -675,6 +675,23 @@ class PartialBoundaries {
         return this;
     }
 }
+function shouldPrune({ dataState }, context) {
+    if (dataState === "deferPartial" || dataState === "streamPartial") {
+        return true;
+    }
+    if (dataState === "complete" || dataState === "streaming") {
+        return !!(
+        // If the last cache write repaired a partial @stream array to a
+        // complete array, the stream array might contain stale entries after
+        // the last written value. We only want to deliver the results up to
+        // the index the network wrote so we need to prune it too.
+        (context.streamInfo ||
+            // The network hasn't delivered these @defer boundaries yet, so prune
+            // the (possibly complete) cached data sitting at them.
+            context.deferInfo));
+    }
+    return false;
+}
 // Describes the data state transitions that change the running state when it's
 // combined with the next data state. Omitted object values represent
 // "impossible" merges where the data state should remain the same.