graphql.validation.rules.UniqueArgumentNamesRule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graphql-java Show documentation

GraphqL Java

There is a newer version: 230521-nf-execution

package graphql.validation.rules;

import com.google.common.collect.Sets;
import graphql.Internal;
import graphql.language.Argument;
import graphql.language.Directive;
import graphql.language.Field;
import graphql.language.Node;
import graphql.validation.AbstractRule;
import graphql.validation.ValidationContext;
import graphql.validation.ValidationErrorCollector;
import graphql.validation.ValidationErrorType;

import java.util.List;
import java.util.Set;

/**
 * Unique argument names
 *
 * A GraphQL field or directive is only valid if all supplied arguments are uniquely named.
 */
@Internal
public class UniqueArgumentNamesRule extends AbstractRule {
    public UniqueArgumentNamesRule(ValidationContext validationContext, ValidationErrorCollector validationErrorCollector) {
        super(validationContext, validationErrorCollector);
    }

    @Override
    public void checkField(Field field) {
        if (field.getArguments() == null || field.getArguments().size() <= 1) {
            return;
        }

        Set arguments = Sets.newHashSetWithExpectedSize(field.getArguments().size());

        for (Argument argument : field.getArguments()) {
            if (arguments.contains(argument.getName())) {
                addError(ValidationErrorType.DuplicateArgumentNames, field.getSourceLocation(), duplicateArgumentNameMessage(argument.getName()));
            } else {
                arguments.add(argument.getName());
            }
        }
    }

    @Override
    public void checkDirective(Directive directive, List ancestors) {
        if (directive.getArguments() == null || directive.getArguments().size() <= 1) {
            return;
        }

        Set arguments = Sets.newHashSetWithExpectedSize(directive.getArguments().size());

        for (Argument argument : directive.getArguments()) {
            if (arguments.contains(argument.getName())) {
                addError(ValidationErrorType.DuplicateArgumentNames, directive.getSourceLocation(), duplicateArgumentNameMessage(argument.getName()));
            } else {
                arguments.add(argument.getName());
            }
        }

    }

    static String duplicateArgumentNameMessage(String argumentName) {
        return String.format("There can be only one argument named '%s'", argumentName);
    }
}