relay-compiler 0.0.0-main-bfcb5c59 → 0.0.0-main-4467ae84

package/codegen/NormalizationCodeGenerator.js.flow

@@ -1,565 +0,0 @@
-/**
- * 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 strict-local
- * @format
- */
-
-// flowlint ambiguous-object-type:error
-
-'use strict';
-
-import type {
-  Argument,
-  ArgumentValue,
-  ClientExtension,
-  Condition,
-  Defer,
-  InlineFragment,
-  LinkedField,
-  LocalArgumentDefinition,
-  Metadata,
-  ModuleImport,
-  Root,
-  Selection,
-  SplitOperation,
-  Stream,
-} from '../core/IR';
-import type {Schema, TypeID} from '../core/Schema';
-import type {
-  NormalizationArgument,
-  NormalizationDefer,
-  NormalizationField,
-  NormalizationLinkedField,
-  NormalizationLinkedHandle,
-  NormalizationLocalArgumentDefinition,
-  NormalizationModuleImport,
-  NormalizationOperation,
-  NormalizationSelection,
-  NormalizationSplitOperation,
-  NormalizationStream,
-  NormalizationTypeDiscriminator,
-} from 'relay-runtime';
-
-const {createCompilerError, createUserError} = require('../core/CompilerError');
-const argumentContainsVariables = require('../util/argumentContainsVariables');
-const generateAbstractTypeRefinementKey = require('../util/generateAbstractTypeRefinementKey');
-const partitionArray = require('../util/partitionArray');
-const sortObjectByKey = require('./sortObjectByKey');
-const {getStorageKey, stableCopy} = require('relay-runtime');
-
-/**
- * @public
- *
- * Converts an IR node into a plain JS object representation that can be
- * used at runtime.
- */
-declare function generate(schema: Schema, node: Root): NormalizationOperation;
-declare function generate(
-  schema: Schema,
-  node: SplitOperation,
-): NormalizationSplitOperation;
-function generate(
-  schema: Schema,
-  node: Root | SplitOperation,
-): NormalizationOperation | NormalizationSplitOperation {
-  switch (node.kind) {
-    case 'Root':
-      return generateRoot(schema, node);
-    case 'SplitOperation':
-      return generateSplitOperation(schema, node);
-    default:
-      throw createCompilerError(
-        `NormalizationCodeGenerator: Unsupported AST kind '${node.kind}'.`,
-        [node.loc],
-      );
-  }
-}
-
-function generateRoot(schema: Schema, node: Root): NormalizationOperation {
-  return {
-    argumentDefinitions: generateArgumentDefinitions(
-      schema,
-      node.argumentDefinitions,
-    ),
-    kind: 'Operation',
-    name: node.name,
-    selections: generateSelections(schema, node.selections),
-  };
-}
-
-function generateSplitOperation(
-  schema: Schema,
-  node: SplitOperation,
-): NormalizationSplitOperation {
-  return {
-    kind: 'SplitOperation',
-    metadata: sortObjectByKey(node.metadata),
-    name: node.name,
-    selections: generateSelections(schema, node.selections),
-  };
-}
-
-function generateSelections(
-  schema: Schema,
-  selections: $ReadOnlyArray<Selection>,
-): $ReadOnlyArray<NormalizationSelection> {
-  const normalizationSelections: Array<NormalizationSelection> = [];
-  selections.forEach(selection => {
-    switch (selection.kind) {
-      case 'Condition':
-        normalizationSelections.push(generateCondition(schema, selection));
-        break;
-      case 'ClientExtension':
-        normalizationSelections.push(
-          generateClientExtension(schema, selection),
-        );
-        break;
-      case 'ScalarField':
-        // NOTE: Inline fragments in normalization ast have the abstractKey
-        // but we skip the corresponding ScalarField for the type discriminator
-        // selection, since it's guaranteed to be a duplicate of a parent __typename
-        // selection.
-        const abstractKey = selection.metadata?.abstractKey;
-        if (typeof abstractKey === 'string') {
-          normalizationSelections.push(generateTypeDiscriminator(abstractKey));
-        } else {
-          normalizationSelections.push(...generateScalarField(selection));
-        }
-        break;
-      case 'ModuleImport':
-        normalizationSelections.push(generateModuleImport(selection));
-        break;
-      case 'InlineFragment':
-        normalizationSelections.push(generateInlineFragment(schema, selection));
-        break;
-      case 'LinkedField':
-        normalizationSelections.push(...generateLinkedField(schema, selection));
-        break;
-      case 'Defer':
-        normalizationSelections.push(generateDefer(schema, selection));
-        break;
-      case 'Stream':
-        normalizationSelections.push(generateStream(schema, selection));
-        break;
-      case 'InlineDataFragmentSpread':
-      case 'FragmentSpread':
-        throw new createCompilerError(
-          `NormalizationCodeGenerator: Unexpected IR node ${selection.kind}.`,
-          [selection.loc],
-        );
-      default:
-        (selection: empty);
-        throw new Error();
-    }
-  });
-  return normalizationSelections;
-}
-
-function generateArgumentDefinitions(
-  schema: Schema,
-  nodes: $ReadOnlyArray<LocalArgumentDefinition>,
-): $ReadOnlyArray<NormalizationLocalArgumentDefinition> {
-  return nodes.map(node => {
-    return {
-      defaultValue: stableCopy(node.defaultValue),
-      kind: 'LocalArgument',
-      name: node.name,
-    };
-  });
-}
-
-function generateClientExtension(
-  schema: Schema,
-  node: ClientExtension,
-): NormalizationSelection {
-  return {
-    kind: 'ClientExtension',
-    selections: generateSelections(schema, node.selections),
-  };
-}
-
-function generateCondition(
-  schema: Schema,
-  node: Condition,
-): NormalizationSelection {
-  if (node.condition.kind !== 'Variable') {
-    throw createCompilerError(
-      "NormalizationCodeGenerator: Expected 'Condition' with static " +
-        'value to be pruned or inlined',
-      [node.condition.loc],
-    );
-  }
-  return {
-    condition: node.condition.variableName,
-    kind: 'Condition',
-    passingValue: node.passingValue,
-    selections: generateSelections(schema, node.selections),
-  };
-}
-
-function generateDefer(schema: Schema, node: Defer): NormalizationDefer {
-  if (
-    !(
-      node.if == null ||
-      node.if.kind === 'Variable' ||
-      (node.if.kind === 'Literal' && node.if.value === true)
-    )
-  ) {
-    throw createCompilerError(
-      'NormalizationCodeGenerator: Expected @defer `if` condition to be ' +
-        'a variable, unspecified, or the literal `true`.',
-      [node.if?.loc ?? node.loc],
-    );
-  }
-  return {
-    if:
-      node.if != null && node.if.kind === 'Variable'
-        ? node.if.variableName
-        : null,
-    kind: 'Defer',
-    label: node.label,
-    selections: generateSelections(schema, node.selections),
-  };
-}
-
-function generateInlineFragment(
-  schema: Schema,
-  node: InlineFragment,
-): NormalizationSelection {
-  const rawType = schema.getRawType(node.typeCondition);
-  const isAbstractType = schema.isAbstractType(rawType);
-  const abstractKey = isAbstractType
-    ? generateAbstractTypeRefinementKey(schema, rawType)
-    : null;
-  let selections = generateSelections(schema, node.selections);
-
-  if (isAbstractType) {
-    // Maintain a few invariants:
-    // - InlineFragment (and `selections` arrays generally) cannot be empty
-    // - Don't emit a TypeDiscriminator under an InlineFragment unless it has
-    //   a different abstractKey
-    // This means we have to handle two cases:
-    // - The inline fragment only contains a TypeDiscriminator with the same
-    //   abstractKey: replace the Fragment w the Discriminator
-    // - The inline fragment contains other selections: return all the selections
-    //   minus any Discriminators w the same key
-    const [discriminators, otherSelections] = partitionArray(
-      selections,
-      selection =>
-        selection.kind === 'TypeDiscriminator' &&
-        selection.abstractKey === abstractKey,
-    );
-    const discriminator = discriminators[0];
-    if (discriminator != null && otherSelections.length === 0) {
-      return discriminator;
-    } else {
-      selections = otherSelections;
-    }
-  }
-
-  return {
-    kind: 'InlineFragment',
-    selections,
-    type: schema.getTypeString(rawType),
-    abstractKey,
-  };
-}
-
-function generateLinkedField(
-  schema: Schema,
-  node: LinkedField,
-): $ReadOnlyArray<NormalizationSelection> {
-  // Note: it is important that the arguments of this field be sorted to
-  // ensure stable generation of storage keys for equivalent arguments
-  // which may have originally appeared in different orders across an app.
-  const handles =
-    (node.handles &&
-      node.handles.map(handle => {
-        let handleNode: NormalizationLinkedHandle = {
-          alias: node.alias === node.name ? null : node.alias,
-          args: generateArgs(node.args),
-          filters: handle.filters,
-          handle: handle.name,
-          key: handle.key,
-          kind: 'LinkedHandle',
-          name: node.name,
-        };
-        // NOTE: this intentionally adds a dynamic key in order to avoid
-        // triggering updates to existing queries that do not use dynamic
-        // keys.
-        if (handle.dynamicKey != null) {
-          const dynamicKeyArgName = '__dynamicKey';
-          handleNode = {
-            ...handleNode,
-            dynamicKey: {
-              kind: 'Variable',
-              name: dynamicKeyArgName,
-              variableName: handle.dynamicKey.variableName,
-            },
-          };
-        }
-        if (handle.handleArgs != null) {
-          const handleArgs = generateArgs(handle.handleArgs);
-          if (handleArgs != null) {
-            handleNode = {
-              ...handleNode,
-              handleArgs,
-            };
-          }
-        }
-        return handleNode;
-      })) ||
-    [];
-  const type = schema.getRawType(node.type);
-  let field: NormalizationLinkedField = {
-    alias: node.alias === node.name ? null : node.alias,
-    args: generateArgs(node.args),
-    concreteType: !schema.isAbstractType(type)
-      ? schema.getTypeString(type)
-      : null,
-    kind: 'LinkedField',
-    name: node.name,
-    plural: isPlural(schema, node.type),
-    selections: generateSelections(schema, node.selections),
-    storageKey: null,
-  };
-  // Precompute storageKey if possible
-  const storageKey = getStaticStorageKey(field, node.metadata);
-  if (storageKey != null) {
-    field = {...field, storageKey};
-  }
-  return [field].concat(handles);
-}
-
-function generateModuleImport(node: ModuleImport): NormalizationModuleImport {
-  const fragmentName = node.name;
-  const regExpMatch = fragmentName.match(
-    /^([a-zA-Z][a-zA-Z0-9]*)(?:_([a-zA-Z][_a-zA-Z0-9]*))?$/,
-  );
-  if (!regExpMatch) {
-    throw createCompilerError(
-      'NormalizationCodeGenerator: @module fragments should be named ' +
-        `'FragmentName_propName', got '${fragmentName}'.`,
-      [node.loc],
-    );
-  }
-  const fragmentPropName = regExpMatch[2];
-  if (typeof fragmentPropName !== 'string') {
-    throw createCompilerError(
-      'NormalizationCodeGenerator: @module fragments should be named ' +
-        `'FragmentName_propName', got '${fragmentName}'.`,
-      [node.loc],
-    );
-  }
-  return {
-    args: null,
-    documentName: node.key,
-    fragmentName,
-    fragmentPropName,
-    kind: 'ModuleImport',
-  };
-}
-
-function generateTypeDiscriminator(
-  abstractKey: string,
-): NormalizationTypeDiscriminator {
-  return {
-    kind: 'TypeDiscriminator',
-    abstractKey,
-  };
-}
-
-function generateScalarField(node): Array<NormalizationSelection> {
-  // flowlint-next-line sketchy-null-mixed:off
-  if (node.metadata?.skipNormalizationNode) {
-    return [];
-  }
-  // Note: it is important that the arguments of this field be sorted to
-  // ensure stable generation of storage keys for equivalent arguments
-  // which may have originally appeared in different orders across an app.
-  const handles =
-    (node.handles &&
-      node.handles.map(handle => {
-        if (handle.dynamicKey != null) {
-          throw createUserError(
-            'Dynamic key values are not supported on scalar fields.',
-            [handle.dynamicKey.loc],
-          );
-        }
-        const nodeHandle = {
-          alias: node.alias === node.name ? null : node.alias,
-          args: generateArgs(node.args),
-          filters: handle.filters,
-          handle: handle.name,
-          key: handle.key,
-          kind: 'ScalarHandle',
-          name: node.name,
-        };
-
-        if (handle.handleArgs != null) {
-          // $FlowFixMe handleArgs exists in Handle
-          nodeHandle.handleArgs = generateArgs(handle.handleArgs);
-        }
-
-        return nodeHandle;
-      })) ||
-    [];
-  let field = {
-    alias: node.alias === node.name ? null : node.alias,
-    args: generateArgs(node.args),
-    kind: 'ScalarField',
-    name: node.name,
-    storageKey: null,
-  };
-  // Precompute storageKey if possible
-  const storageKey = getStaticStorageKey(field, node.metadata);
-  if (storageKey != null) {
-    field = {...field, storageKey};
-  }
-  if (node.metadata?.flight === true) {
-    field = {...field, kind: 'FlightField'};
-  }
-  return [field].concat(handles);
-}
-
-function generateStream(schema: Schema, node: Stream): NormalizationStream {
-  if (
-    !(
-      node.if == null ||
-      node.if.kind === 'Variable' ||
-      (node.if.kind === 'Literal' && node.if.value === true)
-    )
-  ) {
-    throw createCompilerError(
-      'NormalizationCodeGenerator: Expected @stream `if` condition to be ' +
-        'a variable, unspecified, or the literal `true`.',
-      [node.if?.loc ?? node.loc],
-    );
-  }
-  return {
-    if:
-      node.if != null && node.if.kind === 'Variable'
-        ? node.if.variableName
-        : null,
-    kind: 'Stream',
-    label: node.label,
-    selections: generateSelections(schema, node.selections),
-  };
-}
-
-function generateArgumentValue(
-  name: string,
-  value: ArgumentValue,
-): NormalizationArgument | null {
-  switch (value.kind) {
-    case 'Variable':
-      return {
-        kind: 'Variable',
-        name: name,
-        variableName: value.variableName,
-      };
-    case 'Literal':
-      return value.value === null
-        ? null
-        : {
-            kind: 'Literal',
-            name: name,
-            value: stableCopy(value.value),
-          };
-    case 'ObjectValue': {
-      const objectKeys = value.fields.map(field => field.name).sort();
-      const objectValues = new Map(
-        value.fields.map(field => {
-          return [field.name, field.value];
-        }),
-      );
-      return {
-        fields: objectKeys.map(fieldName => {
-          const fieldValue = objectValues.get(fieldName);
-          if (fieldValue == null) {
-            throw createCompilerError('Expected to have object field value');
-          }
-          return (
-            generateArgumentValue(fieldName, fieldValue) ?? {
-              kind: 'Literal',
-              name: fieldName,
-              value: null,
-            }
-          );
-        }),
-        kind: 'ObjectValue',
-        name: name,
-      };
-    }
-    case 'ListValue': {
-      return {
-        items: value.items.map((item, index) => {
-          return generateArgumentValue(`${name}.${index}`, item);
-        }),
-        kind: 'ListValue',
-        name: name,
-      };
-    }
-    default:
-      throw createUserError(
-        'NormalizationCodeGenerator: Complex argument values (Lists or ' +
-          'InputObjects with nested variables) are not supported.',
-        [value.loc],
-      );
-  }
-}
-
-function generateArgs(
-  args: $ReadOnlyArray<Argument>,
-): ?$ReadOnlyArray<NormalizationArgument> {
-  const concreteArguments = [];
-  args.forEach(arg => {
-    const concreteArgument = generateArgumentValue(arg.name, arg.value);
-    if (concreteArgument !== null) {
-      concreteArguments.push(concreteArgument);
-    }
-  });
-  return concreteArguments.length === 0
-    ? null
-    : concreteArguments.sort(nameComparator);
-}
-
-function nameComparator(
-  a: {+name: string, ...},
-  b: {+name: string, ...},
-): number {
-  return a.name < b.name ? -1 : a.name > b.name ? 1 : 0;
-}
-
-/**
- * Pre-computes storage key if possible and advantageous. Storage keys are
- * generated for fields with supplied arguments that are all statically known
- * (ie. literals, no variables) at build time.
- */
-function getStaticStorageKey(
-  field: NormalizationField,
-  metadata: Metadata,
-): ?string {
-  const metadataStorageKey = metadata?.storageKey;
-  if (typeof metadataStorageKey === 'string') {
-    return metadataStorageKey;
-  }
-  if (
-    !field.args ||
-    field.args.length === 0 ||
-    field.args.some(argumentContainsVariables)
-  ) {
-    return null;
-  }
-  return getStorageKey(field, {});
-}
-
-function isPlural(schema: Schema, type: TypeID): boolean {
-  return schema.isList(schema.getNullableType(type));
-}
-
-module.exports = {generate};