graphql.language.ObjectTypeDefinition Java Examples

/**
 * Extract GraphQL Types and Fields from given schema
 *
 * @param schema GraphQL Schema
 * @return list of all types and fields
 */
public List<GraphqlSchemaType> extractGraphQLTypeList(String schema) {
    List<GraphqlSchemaType> typeList = new ArrayList<>();
    SchemaParser schemaParser = new SchemaParser();
    TypeDefinitionRegistry typeRegistry = schemaParser.parse(schema);
    Map<java.lang.String, TypeDefinition> list = typeRegistry.types();
    for (Map.Entry<String, TypeDefinition> entry : list.entrySet()) {
        if (entry.getValue() instanceof ObjectTypeDefinition) {
            GraphqlSchemaType graphqlSchemaType = new GraphqlSchemaType();
            List<String> fieldList = new ArrayList<>();
            graphqlSchemaType.setType(entry.getValue().getName());
            for (FieldDefinition fieldDef : ((ObjectTypeDefinition) entry.getValue()).getFieldDefinitions()) {
                fieldList.add(fieldDef.getName());
            }
            graphqlSchemaType.setFieldList(fieldList);
            typeList.add(graphqlSchemaType);
        }
    }
    return typeList;
}
 

@SuppressWarnings("unchecked")
ObjectTypeDefinition createObject(Map<String, Object> input) {
    assertTrue(input.get("kind").equals("OBJECT"), () -> "wrong input");

    ObjectTypeDefinition.Builder builder = ObjectTypeDefinition.newObjectTypeDefinition()
            .name((String) input.get("name"))
            .description(getDescription(input));
    if (input.containsKey("interfaces")) {
        builder.implementz(
                ((List<Map<String, Object>>) input.get("interfaces")).stream()
                        .map(this::createTypeIndirection)
                        .collect(Collectors.toList())
        );
    }
    List<Map<String, Object>> fields = (List<Map<String, Object>>) input.get("fields");

    builder.fieldDefinitions(createFields(fields));

    return builder.build();
}
 

private List<? extends NamedNode> getDefinitions( Definition def )
{
  if( def instanceof OperationDefinition )
  {
    return ((OperationDefinition)def).getVariableDefinitions();
  }
  if( def instanceof InputObjectTypeDefinition )
  {
    return ((InputObjectTypeDefinition)def).getInputValueDefinitions();
  }
  if( def instanceof ObjectTypeDefinition )
  {
    return ((ObjectTypeDefinition)def).getFieldDefinitions();
  }
  throw new IllegalStateException();
}
 

private void addObjectExtensions( ObjectTypeDefinition type, SrcLinkedClass srcClass )
{
  List<FieldDefinition> baseFieldDefinitions = type.getFieldDefinitions();
  List<ObjectTypeExtensionDefinition> objectExtensions = _registry.objectTypeExtensions().get( type.getName() );
  if( objectExtensions != null )
  {
    for( ObjectTypeExtensionDefinition ext: objectExtensions )
    {
      List<FieldDefinition> extFieldDefinitions = ext.getFieldDefinitions();
      for( FieldDefinition member: extFieldDefinitions )
      {
        addMember( srcClass, member,
          name ->
            baseFieldDefinitions.stream().anyMatch( f -> f.getName().equals( name ) ) ||
            extFieldDefinitions.stream().anyMatch( f -> f.getName().equals( name ) )
        );
      }
    }
  }
}
 

public synchronized List<Field> getFields() {
    if ( null == fields ) {
        Definition def = typeEntry.getDefinition();
        if ( isObjectType() ) {
            fields = getFields((ObjectTypeDefinition)def);
        } else if ( isInterfaceType() ) {
            fields = getFields((InterfaceTypeDefinition)def);
        } else if ( isInputObjectType() ) {
            fields = getFields((InputObjectTypeDefinition)def);
        } else if ( isUnionType() ) {
            fields = getFields((UnionTypeDefinition)def);
        } else if ( isEnumType() ) {
            fields = getFields((EnumTypeDefinition)def);
        } else if ( isSchemaType() ) {
            fields = getFields((SchemaDefinition)def);
        } else {
            fields = Collections.emptyList();
        }
    }
    return fields;
}
 

public synchronized List<String> getImports() {
    if ( null == imports ) {
        Definition def = typeEntry.getDefinition();
        Set<String> names = new TreeSet<String>();
        if ( isObjectType() ) {
            addImports(names, (ObjectTypeDefinition)def);
        } else if ( isInterfaceType() ) {
            addImports(names, (InterfaceTypeDefinition)def);
        } else if ( isInputObjectType() ) {
            addImports(names, (InputObjectTypeDefinition)def);
        } else if ( isUnionType() ) {
            addImports(names, (UnionTypeDefinition)def);
        } else if ( isEnumType() ) {
            addImports(names, (EnumTypeDefinition)def);
        } else if ( isSchemaType() ) {
            addImports(names, (SchemaDefinition)def);
        }
        imports = new ArrayList<>(names);
    }
    return imports;
}
 

public List<Interface> getInterfaces() {

        interfaces = new ArrayList<>();

        if (!isObjectType()) {
            return interfaces;
        }

        ObjectTypeDefinition objectTypeDefinition = (ObjectTypeDefinition) typeEntry.getDefinition();

        List<Type> interfaceTypes = objectTypeDefinition.getImplements();

        for (Type anInterfaceType : interfaceTypes) {
            Interface anInterface = new Interface();
            anInterface.type = toJavaTypeName(anInterfaceType);
            interfaces.add(anInterface);
        }

        return interfaces;
    }
 

private static List<Directive> getDirectives(Definition def) {
    if ( def instanceof ObjectTypeDefinition ) {
        return ((ObjectTypeDefinition)def).getDirectives();
    } if ( def instanceof InterfaceTypeDefinition ) {
        return ((InterfaceTypeDefinition)def).getDirectives();
    } if ( def instanceof EnumTypeDefinition ) {
        return ((EnumTypeDefinition)def).getDirectives();
    } if ( def instanceof ScalarTypeDefinition ) {
        return ((ScalarTypeDefinition)def).getDirectives();
    } if ( def instanceof UnionTypeDefinition ) {
        return ((UnionTypeDefinition)def).getDirectives();
    } if ( def instanceof InputObjectTypeDefinition ) {
        return ((InputObjectTypeDefinition)def).getDirectives();
    } if ( def instanceof SchemaDefinition ) {
        return ((SchemaDefinition)def).getDirectives();
    }
    return Collections.emptyList();
}
 

public boolean hasIdField() {
    if ( definition instanceof ObjectTypeDefinition) {
        return ((ObjectTypeDefinition)definition).getFieldDefinitions()
            .stream()
            .anyMatch((field) -> "id".equals(field.getName()));
    }
    return false;
}
 

/**
 *
 * @param entry Entry
 * @param operationArray operationArray
 */
private void addOperations(Map.Entry<String, TypeDefinition> entry, List<URITemplate> operationArray) {
    for (FieldDefinition fieldDef : ((ObjectTypeDefinition) entry.getValue()).getFieldDefinitions()) {
        URITemplate operation = new URITemplate();
        operation.setHTTPVerb(entry.getKey());
        operation.setUriTemplate(fieldDef.getName());
        operationArray.add(operation);
    }
}
 

private List<Field> getFields(ObjectTypeDefinition def) {
    List<Field> fields = new ArrayList<Field>();
    for ( FieldDefinition fieldDef : def.getFieldDefinitions() ) {
        Field field = new Field(fieldDef.getName(), toJavaTypeName(fieldDef.getType()));
        field.graphQLType = toGraphQLType(fieldDef.getType());
        field.dataResolver = toDataResolver(fieldDef.getType());
        field.args = toArgs(fieldDef.getInputValueDefinitions());
        fields.add(field);
    }
    return fields;
}
 

private static void ensureQueryTypeExists(TypeDefinitionRegistry typeRegistry) {
    final String queryName = typeRegistry.schemaDefinition()
            .flatMap(sdef -> sdef.getOperationTypeDefinitions()
                    .stream()
                    .filter(op -> "query".equals(op.getName()))
                    .findFirst()
                    .map(def -> def.getTypeName().getName()))
            .orElse("Query");
    if (!typeRegistry.getType(queryName).isPresent()) {
        typeRegistry.add(ObjectTypeDefinition.newObjectTypeDefinition().name(queryName).build());
    }
}
 

@SuppressWarnings("unused")
private String getStartSymbol( Node node )
{
  String start = "";
  if( node instanceof TypeDefinition )
  {
    if( node instanceof ObjectTypeDefinition )
    {
      start = "type";
    }
    else if( node instanceof EnumTypeDefinition )
    {
      start = "enum";
    }
    else if( node instanceof InputObjectTypeDefinition )
    {
      start = "input";
    }
    else if( node instanceof InterfaceTypeDefinition )
    {
      start = "interface";
    }
    else if( node instanceof ScalarTypeDefinition )
    {
      start = "scalar";
    }
    else if( node instanceof UnionTypeDefinition )
    {
      start = "union";
    }
  }
  else if( node instanceof OperationDefinition )
  {
    start = ((OperationDefinition)node).getOperation().name().toLowerCase();
  }
  return start;
}
 

private String findLub( UnionTypeDefinition typeDef )
{
  Set<Set<InterfaceTypeDefinition>> ifaces = new HashSet<>();
  for( Type t: typeDef.getMemberTypes() )
  {
    TypeDefinition td = findTypeDefinition( t );
    if( td instanceof ObjectTypeDefinition )
    {
      ifaces.add( ((ObjectTypeDefinition)td).getImplements().stream()
        .map( type -> (InterfaceTypeDefinition)findTypeDefinition( type ) )
        .collect( Collectors.toSet() ) );
    }
    else if( td instanceof InterfaceTypeDefinition )
    {
      ifaces.add( Collections.singleton( (InterfaceTypeDefinition)td ) );
    }
  }
  //noinspection OptionalGetWithoutIsPresent
  Set<InterfaceTypeDefinition> intersection =
    ifaces.stream().reduce( ( p1, p2 ) -> {
      p1.retainAll( p2 );
      return p1;
    } ).get();
  if( intersection.isEmpty() )
  {
    // no common interface
    return null;
  }
  else
  {
    // can only use one
    return intersection.iterator().next().getName();
  }
}
 

private void addIntersectionMethods( SrcLinkedClass srcClass, UnionTypeDefinition union )
{
  Set<Pair<String, String>> fieldDefs = new HashSet<>();
  Map<String, FieldDefinition> nameToFieldDef = new HashMap<>();
  for( Type memberType: union.getMemberTypes() )
  {
    TypeDefinition typeDef = findTypeDefinition( memberType );
    if( typeDef instanceof ObjectTypeDefinition )
    {
      if( fieldDefs.isEmpty() )
      {
        fieldDefs.addAll( ((ObjectTypeDefinition)typeDef).getFieldDefinitions()
          .stream().map( e -> {
            nameToFieldDef.put( e.getName(), e );
            return new Pair<>( e.getName(), e.getType().toString() );
          } ).collect( Collectors.toSet() ) );
      }
      else
      {
        fieldDefs.retainAll( ((ObjectTypeDefinition)typeDef).getFieldDefinitions()
          .stream().map( e -> new Pair<>( e.getName(), e.getType().toString() ) ).collect( Collectors.toSet() ) );
      }
    }
  }
  fieldDefs.forEach(
    fieldDef -> addMember( srcClass, null, nameToFieldDef.get( fieldDef.getFirst() ).getType(), fieldDef.getFirst(),
      name -> fieldDefs.stream().anyMatch( f -> f.getFirst().equals( name ) ) ) );
}
 

private void addInnerObjectType( ObjectTypeDefinition type, SrcLinkedClass enclosingType )
{
  String identifier = makeIdentifier( type.getName(), false );
  String fqn = getFqn() + '.' + identifier;
  SrcLinkedClass srcClass = new SrcLinkedClass( fqn, enclosingType, Interface )
    .addInterface( IJsonBindingsBacked.class.getSimpleName() )
    .addAnnotation( new SrcAnnotationExpression( Structural.class.getSimpleName() )
      .addArgument( "factoryClass", Class.class, identifier + ".ProxyFactory.class" ) )
    .modifiers( Modifier.PUBLIC );
  addUnionInterfaces( type, srcClass );
  addActualNameAnnotation( srcClass, type.getName(), false );
  addSourcePositionAnnotation( srcClass, type, srcClass );
  List<Type> interfaces = type.getImplements();
  addInterfaces( srcClass, interfaces );
  addProxyFactory( srcClass );
  addBuilder( srcClass, type );
  addCreateMethod( srcClass, type );
  addBuilderMethod( srcClass, type );
  addLoadMethod( srcClass );
  addCopier( srcClass, type );
  addCopierMethod( srcClass );
  addCopyMethod( srcClass );

  List<FieldDefinition> fieldDefinitions = type.getFieldDefinitions();
  for( FieldDefinition member: fieldDefinitions )
  {
    addMember( srcClass, member, name -> fieldDefinitions.stream().anyMatch( f -> f.getName().equals( name ) ) );
  }
  addObjectExtensions( type, srcClass );
  enclosingType.addInnerClass( srcClass );
}
 

private void addInnerTypes( SrcLinkedClass srcClass )
{
  mapUnionMemberToUnions();

  for( TypeDefinition type: _registry.types().values() )
  {
    if( type instanceof ObjectTypeDefinition )
    {
      addInnerObjectType( (ObjectTypeDefinition)type, srcClass );
    }
    else if( type instanceof InterfaceTypeDefinition )
    {
      addInnerInterfaceType( (InterfaceTypeDefinition)type, srcClass );
    }
    else if( type instanceof EnumTypeDefinition )
    {
      addInnerEnumType( (EnumTypeDefinition)type, srcClass );
    }
    else if( type instanceof InputObjectTypeDefinition )
    {
      addInnerInputType( (InputObjectTypeDefinition)type, srcClass );
    }
    else if( type instanceof UnionTypeDefinition )
    {
      addInnerUnionType( (UnionTypeDefinition)type, srcClass );
    }
  }
}
 

@Override
protected String generateBody() {
    ObjectTypeDefinition typeDefinition = (ObjectTypeDefinition) schema.getType(typeName)
            .orElseThrow(() -> new GraphQlGeneratorException("type '" + typeName + "' not found in schema"));
    return typeDefinition.getFieldDefinitions().stream()
            .filter(field -> fieldNames.contains(field.getName()))
            .map(this::toJava)
            .collect(joining(";\n    ", "    ", ";\n"));
}
 

private JavaType returnType() {
    ObjectTypeDefinition query = (ObjectTypeDefinition) schema.getType("Query")
            .orElseThrow(() -> new GraphQlGeneratorException("'Query' type not found in schema"));
    FieldDefinition fieldDefinition = query.getFieldDefinitions().stream()
            .filter(f -> f.getName().equals(method.getName()))
            .findAny()
            .orElseThrow(() -> new GraphQlGeneratorException("field (method) '" + method.getName()
                    + "' not found in "
                    + query.getFieldDefinitions().stream().map(FieldDefinition::getName).collect(listString())));
    List<String> selections = operation.getSelectionSet().getSelectionsOfType(Field.class).stream()
            .flatMap(this::selectedFields)
            .map(Field::getName)
            .collect(toList());
    return new JavaType(fieldDefinition.getType(), selections);
}
 

public boolean isObjectType() {
    return typeEntry.getDefinition() instanceof ObjectTypeDefinition;
}
 

private void addImports(Collection<String> imports, ObjectTypeDefinition def) {
    for ( FieldDefinition fieldDef : def.getFieldDefinitions() ) {
        addImports(imports, fieldDef.getType());
    }
}