relay-runtime 11.0.0-rc.0 → 12.0.0

package/lib/store/RelayReferenceMarker.js

@@ -28,12 +28,17 @@ var cloneRelayHandleSourceField = require('./cloneRelayHandleSourceField');
 
 var getOperation = require('../util/getOperation');
 
-var invariant = require("fbjs/lib/invariant");
+var invariant = require('invariant');
 
-var _require = require('./TypeID'),
-    generateTypeID = _require.generateTypeID;
+var _require = require('./RelayConcreteVariables'),
+    getLocalVariables = _require.getLocalVariables;
 
-var CONDITION = RelayConcreteNode.CONDITION,
+var _require2 = require('./TypeID'),
+    generateTypeID = _require2.generateTypeID;
+
+var ACTOR_CHANGE = RelayConcreteNode.ACTOR_CHANGE,
+    CONDITION = RelayConcreteNode.CONDITION,
+    CLIENT_COMPONENT = RelayConcreteNode.CLIENT_COMPONENT,
     CLIENT_EXTENSION = RelayConcreteNode.CLIENT_EXTENSION,
     DEFER = RelayConcreteNode.DEFER,
     FLIGHT_FIELD = RelayConcreteNode.FLIGHT_FIELD,
@@ -50,11 +55,11 @@ var ROOT_ID = RelayStoreUtils.ROOT_ID,
     getStorageKey = RelayStoreUtils.getStorageKey,
     getModuleOperationKey = RelayStoreUtils.getModuleOperationKey;
 
-function mark(recordSource, selector, references, operationLoader) {
+function mark(recordSource, selector, references, operationLoader, shouldProcessClientComponents) {
   var dataID = selector.dataID,
       node = selector.node,
       variables = selector.variables;
-  var marker = new RelayReferenceMarker(recordSource, variables, references, operationLoader);
+  var marker = new RelayReferenceMarker(recordSource, variables, references, operationLoader, shouldProcessClientComponents);
   marker.mark(node, dataID);
 }
 /**
@@ -63,12 +68,13 @@ function mark(recordSource, selector, references, operationLoader) {
 
 
 var RelayReferenceMarker = /*#__PURE__*/function () {
-  function RelayReferenceMarker(recordSource, variables, references, operationLoader) {
+  function RelayReferenceMarker(recordSource, variables, references, operationLoader, shouldProcessClientComponents) {
     this._operationLoader = operationLoader !== null && operationLoader !== void 0 ? operationLoader : null;
     this._operationName = null;
     this._recordSource = recordSource;
     this._references = references;
     this._variables = variables;
+    this._shouldProcessClientComponents = shouldProcessClientComponents;
   }
 
   var _proto = RelayReferenceMarker.prototype;
@@ -94,7 +100,8 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
   };
 
   _proto._getVariableValue = function _getVariableValue(name) {
-    !this._variables.hasOwnProperty(name) ? process.env.NODE_ENV !== "production" ? invariant(false, 'RelayReferenceMarker(): Undefined variable `%s`.', name) : invariant(false) : void 0;
+    !this._variables.hasOwnProperty(name) ? process.env.NODE_ENV !== "production" ? invariant(false, 'RelayReferenceMarker(): Undefined variable `%s`.', name) : invariant(false) : void 0; // $FlowFixMe[cannot-write]
+
     return this._variables[name];
   };
 
@@ -104,6 +111,12 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
     selections.forEach(function (selection) {
       /* eslint-disable no-fallthrough */
       switch (selection.kind) {
+        case ACTOR_CHANGE:
+          // TODO: T89695151 Support multi-actor record sources in RelayReferenceMarker.js
+          _this._traverseLink(selection.linkedField, record);
+
+          break;
+
         case LINKED_FIELD:
           if (selection.plural) {
             _this._traversePluralLink(selection, record);
@@ -114,7 +127,7 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
           break;
 
         case CONDITION:
-          var conditionValue = _this._getVariableValue(selection.condition);
+          var conditionValue = Boolean(_this._getVariableValue(selection.condition));
 
           if (conditionValue === selection.passingValue) {
             _this._traverseSelections(selection.selections, record);
@@ -129,24 +142,25 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
             if (typeName != null && typeName === selection.type) {
               _this._traverseSelections(selection.selections, record);
             }
-          } else if (RelayFeatureFlags.ENABLE_PRECISE_TYPE_REFINEMENT) {
+          } else {
             var _typeName = RelayModernRecord.getType(record);
 
             var typeID = generateTypeID(_typeName);
 
             _this._references.add(typeID);
 
-            _this._traverseSelections(selection.selections, record);
-          } else {
             _this._traverseSelections(selection.selections, record);
           }
 
           break;
-        // $FlowFixMe[incompatible-type]
 
         case FRAGMENT_SPREAD:
+          var prevVariables = _this._variables;
+          _this._variables = getLocalVariables(_this._variables, selection.fragment.argumentDefinitions, selection.args);
+
           _this._traverseSelections(selection.fragment.selections, record);
 
+          _this._variables = prevVariables;
           break;
 
         case LINKED_HANDLE:
@@ -181,13 +195,11 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
 
         case TYPE_DISCRIMINATOR:
           {
-            if (RelayFeatureFlags.ENABLE_PRECISE_TYPE_REFINEMENT) {
-              var _typeName2 = RelayModernRecord.getType(record);
+            var _typeName2 = RelayModernRecord.getType(record);
 
-              var _typeID = generateTypeID(_typeName2);
+            var _typeID = generateTypeID(_typeName2);
 
-              _this._references.add(_typeID);
-            }
+            _this._references.add(_typeID);
 
             break;
           }
@@ -211,6 +223,15 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
 
           break;
 
+        case CLIENT_COMPONENT:
+          if (_this._shouldProcessClientComponents === false) {
+            break;
+          }
+
+          _this._traverseSelections(selection.fragment.selections, record);
+
+          break;
+
         default:
           selection;
           !false ? process.env.NODE_ENV !== "production" ? invariant(false, 'RelayReferenceMarker: Unknown AST node `%s`.', selection) : invariant(false) : void 0;
@@ -233,9 +254,13 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
     var normalizationRootNode = operationLoader.get(operationReference);
 
     if (normalizationRootNode != null) {
-      var selections = getOperation(normalizationRootNode).selections;
+      var operation = getOperation(normalizationRootNode);
+      var prevVariables = this._variables;
+      this._variables = getLocalVariables(this._variables, operation.argumentDefinitions, moduleImport.args);
+
+      this._traverseSelections(operation.selections, record);
 
-      this._traverseSelections(selections, record);
+      this._variables = prevVariables;
     } // Otherwise, if the operation is not available, we assume that the data
     // cannot have been processed yet and therefore isn't in the store to
     // begin with.
@@ -286,26 +311,26 @@ var RelayReferenceMarker = /*#__PURE__*/function () {
       return;
     }
 
-    var reachableQueries = RelayModernRecord.getValue(reactFlightClientResponseRecord, RelayStoreReactFlightUtils.REACT_FLIGHT_QUERIES_STORAGE_KEY);
+    var reachableExecutableDefinitions = RelayModernRecord.getValue(reactFlightClientResponseRecord, RelayStoreReactFlightUtils.REACT_FLIGHT_EXECUTABLE_DEFINITIONS_STORAGE_KEY);
 
-    if (!Array.isArray(reachableQueries)) {
+    if (!Array.isArray(reachableExecutableDefinitions)) {
       return;
     }
 
     var operationLoader = this._operationLoader;
-    !(operationLoader !== null) ? process.env.NODE_ENV !== "production" ? invariant(false, 'DataChecker: Expected an operationLoader to be configured when using ' + 'React Flight') : invariant(false) : void 0; // In Flight, the variables that are in scope for reachable queries aren't
-    // the same as what's in scope for the outer query.
+    !(operationLoader !== null) ? process.env.NODE_ENV !== "production" ? invariant(false, 'DataChecker: Expected an operationLoader to be configured when using ' + 'React Flight') : invariant(false) : void 0; // In Flight, the variables that are in scope for reachable executable
+    // definitions aren't the same as what's in scope for the outer query.
 
     var prevVariables = this._variables; // $FlowFixMe[incompatible-cast]
 
-    var _iterator = (0, _createForOfIteratorHelper2["default"])(reachableQueries),
+    var _iterator = (0, _createForOfIteratorHelper2["default"])(reachableExecutableDefinitions),
         _step;
 
     try {
       for (_iterator.s(); !(_step = _iterator.n()).done;) {
-        var query = _step.value;
-        this._variables = query.variables;
-        var operationReference = query.module;
+        var definition = _step.value;
+        this._variables = definition.variables;
+        var operationReference = definition.module;
         var normalizationRootNode = operationLoader.get(operationReference);
 
         if (normalizationRootNode != null) {