es.iti.wakamiti.lsp.internal.CompletionContextMap Maven / Gradle / Ivy

Go to download
/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

package es.iti.wakamiti.lsp.internal;

import java.util.*;

import es.iti.wakamiti.core.gherkin.parser.*;




class CompletionContextMap {


    private enum Section {
        MAIN, COMMENT, STEP
    }

    public static class CompletionContext {

        public final Node node;
        public final Section section;

        public CompletionContext(Node node, Section section) {
            this.node = node;
            this.section = section;
        }
    }


    public static CompletionContextMap empty() {
        return new CompletionContextMap(null);
    }


    private final Map> map = new HashMap<>();


    public CompletionContextMap(GherkinDocument document) {
        if (document != null && document.getFeature() != null) {
            buildContextMap(document.getFeature());
        }
    }


    public Optional getContext(int line) {
        return map.getOrDefault(line, Optional.empty());
    }

    private void buildContextMap(CommentedNode node) {
        put(node.getLocation(), node, Section.MAIN);
        for (Comment comment : node.getComments()) {
            put(comment.getLocation(), node, Section.COMMENT);
        }
        if (node instanceof Feature) {
            ((Feature)node).getChildren().forEach(this::buildContextMap);
        }
        if (node instanceof ScenarioDefinition) {
            for (Step step : ((ScenarioDefinition)node).getSteps()) {
                put(step.getLocation(), node, Section.STEP);
            }
        }
    }




    private void put(Location location, CommentedNode node, Section section) {
        map.put(location.getLine(),Optional.of(new CompletionContext(node, section)));
    }

}