UNPKG

@graphql-tools/utils

Version:

Common package containing utils and types for GraphQL tools

58 lines (57 loc) • 2.44 kB
import { hasOwnProperty, isSafeObjectKey } from './jsutils.js'; import { mergeDeep } from './mergeDeep.js'; export function mergeIncrementalResult({ incrementalResult, executionResult, }) { const path = ['data', ...(incrementalResult.path ?? [])]; if (incrementalResult.items) { // Reject the whole items batch if any path segment is unsafe, so a later // index increment cannot turn a bad segment into a write at `data.NaN`. if (isSafeKeyPath(path)) { for (const item of incrementalResult.items) { setObjectKeyPath(executionResult, path, item); // Increment the last path segment (the array index) to merge the next item at the next index path[path.length - 1]++; } } } if (incrementalResult.data) { setObjectKeyPath(executionResult, path, incrementalResult.data); } if (incrementalResult.errors) { executionResult.errors = executionResult.errors || []; executionResult.errors.push(...incrementalResult.errors); } if (incrementalResult.extensions) { setObjectKeyPath(executionResult, ['extensions'], incrementalResult.extensions); } if (incrementalResult.incremental) { incrementalResult.incremental.forEach(incrementalSubResult => { mergeIncrementalResult({ incrementalResult: incrementalSubResult, executionResult, }); }); } } function isSafeKeyPath(keyPath) { return keyPath.every(isSafeObjectKey); } function setObjectKeyPath(obj, keyPath, value) { // Validate the full path before creating any parent containers, so a late // unsafe segment cannot leave partial writes on executionResult. if (!isSafeKeyPath(keyPath)) { return; } let current = obj; let i; for (i = 0; i < keyPath.length - 1; i++) { const key = keyPath[i]; if (!hasOwnProperty(current, key) || current[key] == null) { // Determine if the next key is a number to create an array, otherwise create an object current[key] = typeof keyPath[i + 1] === 'number' ? [] : {}; } current = current[key]; } const finalKey = keyPath[i]; const existingValue = hasOwnProperty(current, finalKey) ? current[finalKey] : undefined; current[finalKey] = existingValue != null ? mergeDeep([existingValue, value]) : value; }