Add edge objects to schema.

TODO: Properly split out behavior between different config flags.

Author: rcapshaw

graphql-api-generator/generator.py

@@ -86,9 +86,6 @@ def run(schema: GraphQLSchema, config: dict):
             schema = add_scalar_filters(schema)
             schema = add_type_filters(schema)
 
-        # remove field arguments for edges (should not be in the API schema)
-        schema = remove_field_arguments_for_types(schema)
-
         if config.get('generation').get('query_type_filter'):
             schema = add_object_type_filters(schema)
 
@@ -108,7 +105,7 @@ def run(schema: GraphQLSchema, config: dict):
 
         # add edge input types
         if config.get('generation').get('input_to_create_edge_objects'):
-            raise UnsupportedOperation('{0} is currently not supported'.format('input_to_create_edge_objects'))
+            schema = add_create_edge_objects(schema)
         if config.get('generation').get('input_to_update_edge_objects'):
             raise UnsupportedOperation('{0} is currently not supported'.format('input_to_update_edge_objects'))
 
@@ -128,10 +125,13 @@ def run(schema: GraphQLSchema, config: dict):
         if config.get('generation').get('delete_edge_objects'):
             raise UnsupportedOperation('{0} is currently not supported'.format('delete_edge_objects'))
 
+        # remove field arguments for edges (should not be in the API schema)
+        schema = remove_field_arguments_for_types(schema)
+
     return schema
 
 
-def validate_names(schema:GraphQLSchema, validate):
+def validate_names(schema: GraphQLSchema, validate):
     # types and interfaces
     if validate.get('type_names'):
         # type names

graphql-api-generator/resources/config.yml

@@ -26,7 +26,7 @@ generation:
     input_to_create_objects: true
     input_to_update_objects: true
     # add edge input types (not supported)
-    input_to_create_edge_objects: false
+    input_to_create_edge_objects: true
     input_to_update_edge_objects: false
     # add mutations
     create_objects: true

graphql-api-generator/utils/utils.py

@@ -516,12 +516,64 @@ def add_object_type_filters(schema: GraphQLSchema):
     return schema
 
 
+def add_create_edge_objects(schema: GraphQLSchema):
+    make = ''
+    for _type in schema.type_map.values():
+        if not is_schema_defined_type(_type) or is_interface_type(_type):
+            continue
+
+        for field_name, field in _type.fields.items():
+            if field_name[0] == '_':
+                continue
+            field_type = field.type
+            inner_field_type = get_named_type(field_type)
+            if is_scalar_type(inner_field_type) or is_enum_type(inner_field_type):
+                continue
+
+            if is_interface_type(_type):
+                connected_types = schema.get_possible_types(_type)
+            else:
+                connected_types = [_type]
+
+            for t in connected_types:
+                edge_from = f'{capitalize(field_name)}EdgeFrom{t.name}'
+                edge_create = f'create{edge_from}'
+                edge_input = f'_InputToCreate{edge_from}'
+                annotate_input = f'_InputToAnnotate{edge_from}'
+
+                annotation_fields = []
+                for arg, arg_type in field.args.items():
+                    if arg == 'filter':
+                        continue
+                    # if not is_scalar_type(arg_type.type.name) and not is_enum_type(arg_type.type.name):
+                    #     raise Exception("Fields cannot be object types")
+                    annotation_fields.append(f'{arg}: {arg_type.type}')
+                annotations = " ".join(annotation_fields)
+
+                if len(annotations) > 0:
+                    make += f'input {edge_input} {{sourceID: ID! targetID: ID! annotations: {annotate_input} }}\n'
+                    make += f'input {annotate_input}{{{annotations}}}\n'
+                    make += f'type _{edge_from} {{id:ID! source: {t.name}! target: {inner_field_type}! {annotations}}}\n'
+                else:
+                    make += f'input {edge_input} {{sourceID: ID! targetID: ID!}}\n'
+                    make += f'type _{edge_from} {{id: ID! source: {t.name}! target: {inner_field_type}!}}\n'
+                make += f'extend type Mutation{{{edge_create}(data: {edge_input}):_{edge_from}}}\n'
+
+    schema = add_to_schema(schema, make)
+    return schema
+
+
 def remove_field_arguments_for_types(schema: GraphQLSchema):
+    keep_args = ['filter']
     for _type in schema.type_map.values():
         if not is_schema_defined_type(_type):
             continue
         for field_name, field in _type.fields.items():
-            field.args = {}
+            args = {}
+            for arg in field.args:
+                if arg in keep_args:
+                    args[arg] = field.args[arg]
+            field.args = args
     return schema