@apollographql/graphql-language-service-utils/dist/getASTNodeAtPosition.js.flow

Utilities to support the GraphQL Language Service

/**
 *  Copyright (c) Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the license found in the
 *  LICENSE file in the root directory of this source tree.
 *
 *  @flow
 */

import type {ASTNode} from 'graphql/language';

import {Position} from './Range';
import {visit} from 'graphql';

export function getASTNodeAtPosition(
  query: string,
  ast: ASTNode,
  point: Position,
): ?ASTNode {
  const offset = pointToOffset(query, point);
  let nodeContainingPosition: ?ASTNode;
  visit(ast, {
    enter(node) {
      if (
        node.kind !== 'Name' && // We're usually interested in their parents
        node.loc &&
        node.loc.start <= offset &&
        offset <= node.loc.end
      ) {
        nodeContainingPosition = node;
      } else {
        return false;
      }
    },
    leave(node) {
      if (node.loc && node.loc.start <= offset && offset <= node.loc.end) {
        return false;
      }
    },
  });
  return nodeContainingPosition;
}

export function pointToOffset(text: string, point: Position): number {
  const linesUntilPosition = text.split('\n').slice(0, point.line);
  return (
    point.character +
    linesUntilPosition
      .map(
        line => line.length + 1, // count EOL
      )
      .reduce((a, b) => a + b, 0)
  );
}