com.jacobmountain.graphql.client.visitor.ClientDetailsVisitor Maven / Gradle / Ivy

Go to download
package com.jacobmountain.graphql.client.visitor;

import com.jacobmountain.graphql.client.annotations.GraphQLQuery;
import com.jacobmountain.graphql.client.annotations.GraphQLSubscription;
import com.jacobmountain.graphql.client.modules.ClientDetails;

import javax.lang.model.element.TypeElement;
import javax.lang.model.util.ElementKindVisitor8;

public class ClientDetailsVisitor extends ElementKindVisitor8 {

    @Override
    public ClientDetails visitType(TypeElement type, Void unused) {
        return ClientDetails.builder()
                .requiresSubscriber(requiresSubscriber(type))
                .requiresFetcher(requiresFetcher(type))
                .build();
    }

    private boolean requiresSubscriber(TypeElement element) {
        return element.getEnclosedElements()
                .stream()
                .anyMatch(it -> it.getAnnotation(GraphQLSubscription.class) != null);
    }

    private boolean requiresFetcher(TypeElement element) {
        return element.getEnclosedElements()
                .stream()
                .anyMatch(it -> it.getAnnotation(GraphQLQuery.class) != null);
    }

}