ai.vespa.schemals.schemadocument.resolvers.RankExpression.argument.EnumArgument Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of schema-language-server Show documentation

There is a newer version: 8.458.13

Show newest version

package ai.vespa.schemals.schemadocument.resolvers.RankExpression.argument;

import java.util.List;
import java.util.Optional;

import org.eclipse.lsp4j.Diagnostic;

import ai.vespa.schemals.context.ParseContext;
import ai.vespa.schemals.index.Symbol;
import ai.vespa.schemals.index.Symbol.SymbolStatus;
import ai.vespa.schemals.index.Symbol.SymbolType;
import ai.vespa.schemals.tree.rankingexpression.RankNode;

/**
 * EnumArgument
 */
public class EnumArgument implements Argument {
    String displayString;
    List validArguments;

    public EnumArgument(String displayString, List validArguments) {
        this.validArguments = List.copyOf(validArguments);
        this.displayString = displayString;
    }

    public List getValidArguments() { return validArguments; }

	@Override
	public Optional parseArgument(ParseContext context, RankNode node) {
        if (!validateArgument(node)) return Optional.empty();

        RankNode leaf = node;
        while (leaf.getChildren().size() > 0) {
            leaf = leaf.getChildren().get(0);
        }

        Symbol symbol = leaf.getSymbol();
        if (symbol == null) return Optional.empty();

        if (symbol.getStatus() == SymbolStatus.REFERENCE) {
            context.schemaIndex().deleteSymbolReference(symbol);
        }

        symbol.setType(SymbolType.DIMENSION);
        symbol.setStatus(SymbolStatus.BUILTIN_REFERENCE);
        return Optional.empty();
	}

	@Override
	public boolean validateArgument(RankNode node) {
        return validArguments.contains(node.getSchemaNode().getText());
	}

	@Override
	public int getStrictness() {
        return 5;
	}

	@Override
	public String displayString() {
        return displayString;
	}

    
}