···
-
import type { ASTNode } from './ast';
-
export function printString(string: string) {
return JSON.stringify(string);
-
export function printBlockString(string: string) {
return '"""\n' + string.replace(/"""/g, '\\"""') + '\n"""';
-
const hasItems = <T>(array: ReadonlyArray<T> | undefined | null): array is ReadonlyArray<T> =>
-
!!(array && array.length);
const MAX_LINE_LENGTH = 80;
-
[NodeT in ASTNode as NodeT['kind']]?: (node: NodeT) => string;
-
OperationDefinition(node) {
-
node.operation === 'query' &&
-
!hasItems(node.variableDefinitions) &&
-
!hasItems(node.directives)
-
return nodes.SelectionSet!(node.selectionSet);
let out: string = node.operation;
if (node.name) out += ' ' + node.name.value;
-
if (hasItems(node.variableDefinitions)) {
if (!node.name) out += ' ';
-
out += '(' + node.variableDefinitions.map(nodes.VariableDefinition!).join(', ') + ')';
-
if (hasItems(node.directives)) out += ' ' + node.directives.map(nodes.Directive!).join(' ');
-
return out + ' ' + nodes.SelectionSet!(node.selectionSet);
-
VariableDefinition(node) {
-
let out = nodes.Variable!(node.variable) + ': ' + print(node.type);
-
if (node.defaultValue) out += ' = ' + print(node.defaultValue);
-
if (hasItems(node.directives)) out += ' ' + node.directives.map(nodes.Directive!).join(' ');
-
let out = (node.alias ? node.alias.value + ': ' : '') + node.name.value;
-
if (hasItems(node.arguments)) {
-
const args = node.arguments.map(nodes.Argument!);
-
const argsLine = out + '(' + args.join(', ') + ')';
-
argsLine.length > MAX_LINE_LENGTH
-
? out + '(\n ' + args.join('\n').replace(/\n/g, '\n ') + '\n)'
-
if (hasItems(node.directives)) out += ' ' + node.directives.map(nodes.Directive!).join(' ');
-
return node.selectionSet ? out + ' ' + nodes.SelectionSet!(node.selectionSet) : out;
-
return node.block ? printBlockString(node.value) : printString(node.value);
return '$' + node.name.value;
-
return '[' + node.values.map(print).join(', ') + ']';
-
return '{' + node.fields.map(nodes.ObjectField!).join(', ') + '}';
-
return node.name.value + ': ' + print(node.value);
-
return hasItems(node.definitions) ? node.definitions.map(print).join('\n\n') : '';
-
return '{\n ' + node.selections.map(print).join('\n').replace(/\n/g, '\n ') + '\n}';
-
return node.name.value + ': ' + print(node.value);
let out = '...' + node.name.value;
-
if (hasItems(node.directives)) out += ' ' + node.directives.map(nodes.Directive!).join(' ');
if (node.typeCondition) out += ' on ' + node.typeCondition.name.value;
-
if (hasItems(node.directives)) out += ' ' + node.directives.map(nodes.Directive!).join(' ');
-
return out + ' ' + print(node.selectionSet);
-
FragmentDefinition(node) {
let out = 'fragment ' + node.name.value;
out += ' on ' + node.typeCondition.name.value;
-
if (hasItems(node.directives)) out += ' ' + node.directives.map(nodes.Directive!).join(' ');
-
return out + ' ' + print(node.selectionSet);
let out = '@' + node.name.value;
-
if (hasItems(node.arguments)) out += '(' + node.arguments.map(nodes.Argument!).join(', ') + ')';
-
return '[' + print(node.type) + ']';
-
return print(node.type) + '!';
-
export function print(node: ASTNode): string {
-
return nodes[node.kind] ? (nodes as any)[node.kind]!(node) : '';
···
+
VariableDefinitionNode,
+
OperationDefinitionNode,
+
FragmentDefinitionNode,
+
function mapJoin<T>(value: readonly T[], joiner: string, mapper: (value: T) => string): string {
+
for (let index = 0; index < value.length; index++) {
+
if (index) out += joiner;
+
out += mapper(value[index]);
+
function printString(string: string) {
return JSON.stringify(string);
+
function printBlockString(string: string) {
return '"""\n' + string.replace(/"""/g, '\\"""') + '\n"""';
const MAX_LINE_LENGTH = 80;
+
OperationDefinition(node: OperationDefinitionNode): string {
let out: string = node.operation;
if (node.name) out += ' ' + node.name.value;
+
if (node.variableDefinitions && node.variableDefinitions.length) {
if (!node.name) out += ' ';
+
out += '(' + mapJoin(node.variableDefinitions, ', ', nodes.VariableDefinition) + ')';
+
if (node.directives && node.directives.length)
+
out += ' ' + mapJoin(node.directives, ' ', nodes.Directive);
+
? out + ' ' + nodes.SelectionSet(node.selectionSet)
+
: nodes.SelectionSet(node.selectionSet);
+
VariableDefinition(node: VariableDefinitionNode): string {
+
let out = nodes.Variable!(node.variable) + ': ' + _print(node.type);
+
if (node.defaultValue) out += ' = ' + _print(node.defaultValue);
+
if (node.directives && node.directives.length)
+
out += ' ' + mapJoin(node.directives, ' ', nodes.Directive);
+
Field(node: FieldNode): string {
+
let out = node.alias ? node.alias.value + ': ' + node.name.value : node.name.value;
+
if (node.arguments && node.arguments.length) {
+
const args = mapJoin(node.arguments, ', ', nodes.Argument);
+
if (out.length + args.length + 2 > MAX_LINE_LENGTH) {
+
mapJoin(node.arguments, LF, nodes.Argument) +
+
(LF = LF.slice(0, -2)) +
+
out += '(' + args + ')';
+
if (node.directives && node.directives.length)
+
out += ' ' + mapJoin(node.directives, ' ', nodes.Directive);
+
if (node.selectionSet) out += ' ' + nodes.SelectionSet(node.selectionSet);
+
StringValue(node: StringValueNode): string {
+
return printBlockString(node.value).replace(/\n/g, LF);
+
return printString(node.value);
+
BooleanValue(node: BooleanValueNode): string {
+
NullValue(_node: NullValueNode): string {
+
IntValue(node: IntValueNode): string {
+
FloatValue(node: FloatValueNode): string {
+
EnumValue(node: EnumValueNode): string {
+
Name(node: NameNode): string {
+
Variable(node: VariableNode): string {
return '$' + node.name.value;
+
ListValue(node: ListValueNode): string {
+
return '[' + mapJoin(node.values, ', ', _print) + ']';
+
ObjectValue(node: ObjectValueNode): string {
+
return '{' + mapJoin(node.fields, ', ', nodes.ObjectField) + '}';
+
ObjectField(node: ObjectFieldNode): string {
+
return node.name.value + ': ' + _print(node.value);
+
Document(node: DocumentNode): string {
+
if (!node.definitions || !node.definitions.length) return '';
+
return mapJoin(node.definitions, '\n\n', _print);
+
SelectionSet(node: SelectionSetNode): string {
+
return '{' + (LF += ' ') + mapJoin(node.selections, LF, _print) + (LF = LF.slice(0, -2)) + '}';
+
Argument(node: ArgumentNode): string {
+
return node.name.value + ': ' + _print(node.value);
+
FragmentSpread(node: FragmentSpreadNode): string {
let out = '...' + node.name.value;
+
if (node.directives && node.directives.length)
+
out += ' ' + mapJoin(node.directives, ' ', nodes.Directive);
+
InlineFragment(node: InlineFragmentNode): string {
if (node.typeCondition) out += ' on ' + node.typeCondition.name.value;
+
if (node.directives && node.directives.length)
+
out += ' ' + mapJoin(node.directives, ' ', nodes.Directive);
+
out += ' ' + nodes.SelectionSet(node.selectionSet);
+
FragmentDefinition(node: FragmentDefinitionNode): string {
let out = 'fragment ' + node.name.value;
out += ' on ' + node.typeCondition.name.value;
+
if (node.directives && node.directives.length)
+
out += ' ' + mapJoin(node.directives, ' ', nodes.Directive);
+
return out + ' ' + nodes.SelectionSet(node.selectionSet);
+
Directive(node: DirectiveNode): string {
let out = '@' + node.name.value;
+
if (node.arguments && node.arguments.length)
+
out += '(' + mapJoin(node.arguments, ', ', nodes.Argument) + ')';
+
NamedType(node: NamedTypeNode): string {
+
ListType(node: ListTypeNode): string {
+
return '[' + _print(node.type) + ']';
+
NonNullType(node: NonNullTypeNode): string {
+
return _print(node.type) + '!';
+
const _print = (node: ASTNode): string => nodes[node.kind](node);
+
function print(node: ASTNode): string {
+
return nodes[node.kind] ? nodes[node.kind](node) : '';
+
export { print, printString, printBlockString };
kitten.sh