Fix #44 - Handle schema extensions

Author: wravery

GraphQLTree.cpp

@@ -525,6 +525,12 @@ struct ast_selector<schema_extension>
 {
 };
 
+template <>
+struct ast_selector<operation_type_definition>
+	: std::true_type
+{
+};
+
 template <>
 struct ast_selector<scalar_type_extension>
 	: std::true_type

SchemaGenerator.cpp

@@ -316,11 +316,14 @@ bool Generator::fixupInputFieldList(InputFieldList& fields)
 
 void Generator::visitDefinition(const peg::ast_node& definition)
 {
-	if (definition.is<peg::schema_definition>()
-		|| definition.is<peg::schema_extension>())
+	if (definition.is<peg::schema_definition>())
 	{
 		visitSchemaDefinition(definition);
 	}
+	else if (definition.is<peg::schema_extension>())
+	{
+		visitSchemaExtension(definition);
+	}
 	else if (definition.is<peg::scalar_type_definition>())
 	{
 		visitScalarTypeDefinition(definition);
@@ -383,6 +386,18 @@ void Generator::visitSchemaDefinition(const peg::ast_node& schemaDefinition)
 		});
 }
 
+void Generator::visitSchemaExtension(const peg::ast_node& schemaExtension)
+{
+	peg::for_each_child<peg::operation_type_definition>(schemaExtension,
+		[this](const peg::ast_node & child)
+	{
+		std::string operation(child.children.front()->content());
+		std::string name(child.children.back()->content());
+
+		_operationTypes.push_back({ std::move(name), std::move(operation) });
+	});
+}
+
 void Generator::visitObjectTypeDefinition(const peg::ast_node& objectTypeDefinition)
 {
 	std::string name;

include/SchemaGenerator.h

@@ -209,6 +209,7 @@ class Generator
 	void visitDefinition(const peg::ast_node& definition);
 
 	void visitSchemaDefinition(const peg::ast_node& schemaDefinition);
+	void visitSchemaExtension(const peg::ast_node& schemaExtension);
 	void visitScalarTypeDefinition(const peg::ast_node& scalarTypeDefinition);
 	void visitEnumTypeDefinition(const peg::ast_node& enumTypeDefinition);
 	void visitEnumTypeExtension(const peg::ast_node& enumTypeExtension);