LoneAnonymousOperation.java

package graphql.validation.rules;

import graphql.language.Document;
import graphql.language.OperationDefinition;
import graphql.validation.AbstractRule;
import graphql.validation.ValidationContext;
import graphql.validation.ValidationErrorCollector;
import graphql.validation.ValidationError;
import graphql.validation.ValidationErrorType;

/**
 * Created by Exbeinfo on 2/7/2017.
 */
public class LoneAnonymousOperation extends AbstractRule {

    boolean hasAnonymousOp = false;
    int count = 0;

    public LoneAnonymousOperation(ValidationContext validationContext, ValidationErrorCollector validationErrorCollector) {
        super(validationContext,validationErrorCollector);
    }

    @Override
    public void checkOperationDefinition(OperationDefinition operationDefinition) {
        super.checkOperationDefinition(operationDefinition);
        String name = operationDefinition.getName();
        String message = null;

        if(name == null) {
            hasAnonymousOp = true;
            if(count > 0) {
                message = "Anonymous operation with other operations.";
            }
        } else {
            if(hasAnonymousOp) {
              message = "Operation " + name + " is following anonymous operation.";
            }
        }
        count++;
        if(message != null) {
            addError(new ValidationError(ValidationErrorType.LoneAnonymousOperationViolation, operationDefinition.getSourceLocation(), message));
        }
    }

    @Override
    public void documentFinished(Document document) {
        super.documentFinished(document);
        hasAnonymousOp = false;
    }
}