relay-runtime 9.0.0 → 10.1.0

package/store/RelayStoreReactFlightUtils.js.flow

@@ -0,0 +1,64 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @flow
+ * @format
+ */
+
+// flowlint ambiguous-object-type:error
+
+'use strict';
+
+const invariant = require('invariant');
+
+const {getType} = require('./RelayModernRecord');
+
+import type {ReactFlightPayloadData} from '../network/RelayNetworkTypes';
+import type {ReactFlightClientResponse, Record} from './RelayStoreTypes';
+
+const REACT_FLIGHT_QUERIES_STORAGE_KEY = 'queries';
+const REACT_FLIGHT_TREE_STORAGE_KEY = 'tree';
+const REACT_FLIGHT_TYPE_NAME = 'ReactFlightComponent';
+
+function refineToReactFlightPayloadData(
+  payload: mixed,
+): ?ReactFlightPayloadData {
+  if (
+    payload == null ||
+    typeof payload !== 'object' ||
+    !Array.isArray(payload.tree) ||
+    !Array.isArray(payload.queries)
+  ) {
+    return null;
+  }
+  return (payload: $FlowFixMe);
+}
+
+function getReactFlightClientResponse(
+  record: Record,
+): ?ReactFlightClientResponse {
+  invariant(
+    getType(record) === REACT_FLIGHT_TYPE_NAME,
+    'getReactFlightClientResponse(): Expected a ReactFlightComponentRecord, ' +
+      'got %s.',
+    record,
+  );
+  const response: ?ReactFlightClientResponse = (record[
+    REACT_FLIGHT_TREE_STORAGE_KEY
+  ]: $FlowFixMe);
+  if (response != null) {
+    return response;
+  }
+  return null;
+}
+
+module.exports = {
+  REACT_FLIGHT_QUERIES_STORAGE_KEY,
+  REACT_FLIGHT_TREE_STORAGE_KEY,
+  REACT_FLIGHT_TYPE_NAME,
+  getReactFlightClientResponse,
+  refineToReactFlightPayloadData,
+};