Merge pull request #165 from graphql/rfc-type-extension

[RFC] Type extension syntax

Author: leebyron

src/language/ast.js

@@ -55,6 +55,7 @@ export type Node = Name
                  | EnumTypeDefinition
                  | EnumValueDefinition
                  | InputObjectTypeDefinition
+                 | TypeExtensionDefinition
 
 // Name
 
@@ -75,6 +76,7 @@ export type Document = {
 export type Definition = OperationDefinition
                        | FragmentDefinition
                        | TypeDefinition
+                       | TypeExtensionDefinition
 
 export type OperationDefinition = {
   kind: 'OperationDefinition';
@@ -323,3 +325,11 @@ export type InputObjectTypeDefinition = {
   name: Name;
   fields: Array<InputValueDefinition>;
 }
+
+// Type Extention
+
+export type TypeExtensionDefinition = {
+  kind: 'TypeExtensionDefinition';
+  loc?: ?Location;
+  definition: ObjectTypeDefinition;
+}

src/language/kinds.js

@@ -59,3 +59,7 @@ export const SCALAR_TYPE_DEFINITION = 'ScalarTypeDefinition';
 export const ENUM_TYPE_DEFINITION = 'EnumTypeDefinition';
 export const ENUM_VALUE_DEFINITION = 'EnumValueDefinition';
 export const INPUT_OBJECT_TYPE_DEFINITION = 'InputObjectTypeDefinition';
+
+// Type Extension Definitions
+
+export const TYPE_EXTENSION_DEFINITION = 'TypeExtensionDefinition';

src/language/parser.js

@@ -48,6 +48,8 @@ import type {
   EnumTypeDefinition,
   EnumValueDefinition,
   InputObjectTypeDefinition,
+
+  TypeExtensionDefinition,
 } from './ast';
 
 import {
@@ -89,6 +91,8 @@ import {
   ENUM_TYPE_DEFINITION,
   ENUM_VALUE_DEFINITION,
   INPUT_OBJECT_TYPE_DEFINITION,
+
+  TYPE_EXTENTION_DEFINITION,
 } from './kinds';
 
 
@@ -177,6 +181,7 @@ function parseDocument(parser): Document {
  *   - OperationDefinition
  *   - FragmentDefinition
  *   - TypeDefinition
+ *   - TypeExtensionDefinition
  */
 function parseDefinition(parser): Definition {
   if (peek(parser, TokenKind.BRACE_L)) {
@@ -196,6 +201,8 @@ function parseDefinition(parser): Definition {
       case 'scalar':
       case 'enum':
       case 'input': return parseTypeDefinition(parser);
+
+      case 'extend': return parseTypeExtensionDefinition(parser);
     }
   }
 
@@ -883,6 +890,23 @@ function parseInputObjectTypeDefinition(parser): InputObjectTypeDefinition {
 }
 
 
+// Implements the parsing rules for Type Extension
+
+/**
+ * TypeExtensionDefinition : extend ObjectTypeDefinition
+ */
+function parseTypeExtensionDefinition(parser): TypeExtensionDefinition {
+  var start = parser.token.start;
+  expectKeyword(parser, 'extend');
+  var definition = parseObjectTypeDefinition(parser);
+  return {
+    kind: TYPE_EXTENTION_DEFINITION,
+    definition,
+    loc: loc(parser, start),
+  };
+}
+
+
 // Core parsing utility functions
 
 /**

src/language/printer.js

@@ -115,6 +115,10 @@ var printDocASTReducer = {
 
   InputObjectTypeDefinition: ({ name, fields }) =>
     `input ${name} ${block(fields)}`,
+
+  // Type Extension Definition
+
+  TypeExtensionDefinition: ({ definition }) => `extend ${definition}`,
 };
 
 /**

src/language/visitor.js

@@ -47,6 +47,8 @@ export var QueryDocumentKeys = {
   EnumTypeDefinition: [ 'name', 'values' ],
   EnumValueDefinition: [ 'name' ],
   InputObjectTypeDefinition: [ 'name', 'fields' ],
+
+  TypeExtensionDefinition: [ 'definition' ],
 };
 
 export const BREAK = {};