ai.vespa.schemals.schemadocument.resolvers.RankExpression.argument.SymbolArgument 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.441.21

Show newest version

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

import java.util.Optional;

import org.eclipse.lsp4j.Diagnostic;
import org.eclipse.lsp4j.DiagnosticSeverity;

import ai.vespa.schemals.common.SchemaDiagnostic;
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.SchemaNode;
import ai.vespa.schemals.tree.rankingexpression.RankNode;

public class SymbolArgument implements Argument {
    
    private SymbolType symbolType;
    private String displayStr;

    public SymbolArgument(SymbolType symbolType, String displayStr) {
        this.symbolType = symbolType;
        this.displayStr = displayStr;
    }

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

    protected SchemaNode findSymbolNode(RankNode node) {
        SchemaNode symbolNode = node.getSchemaNode();

        while (!symbolNode.hasSymbol() && symbolNode.size() > 0) {
            symbolNode = symbolNode.get(0).getSchemaNode();
        }

        return symbolNode;
    }

    @Override
    public boolean validateArgument(RankNode node) {
        SchemaNode symbolNode = findSymbolNode(node);
        return symbolNode.hasSymbol();
    }

    @Override
    public Optional parseArgument(ParseContext context, RankNode node) {

        SchemaNode symbolNode = findSymbolNode(node);

        if (symbolNode.hasSymbol()) {
            Symbol symbol = symbolNode.getSymbol();

            if (symbol.getStatus() == SymbolStatus.REFERENCE) {
                symbol.setStatus(SymbolStatus.UNRESOLVED);
                context.schemaIndex().deleteSymbolReference(symbol);
            }
            
            if (symbol.getStatus() == SymbolStatus.UNRESOLVED) {
                symbol.setType(symbolType);
            }
        } else {
            return Optional.of(new SchemaDiagnostic.Builder()
                .setRange(node.getRange())
                .setMessage("The argument must be to a symbol of type: " + symbolType)
                .setSeverity(DiagnosticSeverity.Error)
                .build());
        }

        return Optional.empty();
    }

    public String displayString() {
        return displayStr;
    }
}