Search in sources :

Example 6 with SNode

use of org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode in project sts4 by spring-projects.

the class YTypeAssistContext method getKeyCompletions.

public List<ICompletionProposal> getKeyCompletions(YamlDocument doc, SNode node, int offset, String query) throws Exception {
    int queryOffset = offset - query.length();
    DynamicSchemaContext dynamicCtxt = getSchemaContext();
    List<YTypedProperty> allProperties = typeUtil.getProperties(type);
    if (CollectionUtil.hasElements(allProperties)) {
        List<List<YTypedProperty>> tieredProperties = sortIntoTiers(allProperties);
        Set<String> definedProps = dynamicCtxt.getDefinedProperties();
        List<ICompletionProposal> proposals = new ArrayList<>();
        boolean suggestDeprecated = typeUtil.suggestDeprecatedProperties();
        YamlIndentUtil indenter = new YamlIndentUtil(doc);
        for (List<YTypedProperty> thisTier : tieredProperties) {
            List<YTypedProperty> undefinedProps = thisTier.stream().filter(p -> !definedProps.contains(p.getName()) && (suggestDeprecated || !p.isDeprecated())).collect(Collectors.toList());
            if (!undefinedProps.isEmpty()) {
                for (YTypedProperty p : undefinedProps) {
                    String name = p.getName();
                    double score = FuzzyMatcher.matchScore(query, name);
                    if (score != 0) {
                        TypeBasedSnippetProvider snippetProvider = typeUtil.getSnippetProvider();
                        DocumentEdits edits;
                        if (snippetProvider != null) {
                            // Generate edits from snippet
                            Snippet snippet = snippetProvider.getSnippet(p);
                            edits = createEditFromSnippet(doc, node, offset, query, indenter, snippet);
                        } else {
                            // Generate edits the old-fashioned way
                            edits = new DocumentEdits(doc.getDocument());
                            YType YType = p.getType();
                            edits.delete(queryOffset, query);
                            int referenceIndent = doc.getColumn(queryOffset);
                            boolean needNewline = node.getNodeType() == SNodeType.KEY;
                            StringBuilder snippet = new StringBuilder();
                            if (needNewline) {
                                snippet.append("\n");
                                referenceIndent = YamlIndentUtil.getNewChildKeyIndent(node);
                            } else if (queryOffset > 0 && !Character.isWhitespace(doc.getChar(queryOffset - 1))) {
                                // See https://www.pivotaltracker.com/story/show/137722057
                                snippet.append(" ");
                                referenceIndent++;
                            }
                            snippet.append(p.getName());
                            snippet.append(":");
                            snippet.append(appendTextFor(YType));
                            edits.insert(queryOffset, indenter.applyIndentation(snippet.toString(), referenceIndent));
                        }
                        ICompletionProposal completion = completionFactory().beanProperty(doc.getDocument(), contextPath.toPropString(), getType(), query, p, score, edits, typeUtil);
                        if (p.isDeprecated() && completion instanceof ScoreableProposal) {
                            completion.deemphasize(DEEMP_DEPRECATION);
                        }
                        proposals.add(completion);
                    }
                }
            }
            // We should only move on to the next tier if all required properties in this tier are defined.
            if (undefinedProps.stream().anyMatch(p -> p.isRequired())) {
                // stop here, take no more from next tier!
                return proposals;
            }
        }
        return proposals;
    }
    return Collections.emptyList();
}
Also used : LoggerFactory(org.slf4j.LoggerFactory) YTypeUtil(org.springframework.ide.vscode.commons.yaml.schema.YTypeUtil) FuzzyMatcher(org.springframework.ide.vscode.commons.util.FuzzyMatcher) YamlIndentUtil(org.springframework.ide.vscode.commons.yaml.util.YamlIndentUtil) ArrayList(java.util.ArrayList) Strings(com.google.common.base.Strings) YamlPathSegmentType(org.springframework.ide.vscode.commons.yaml.path.YamlPathSegment.YamlPathSegmentType) ImmutableList(com.google.common.collect.ImmutableList) DEEMP_DEPRECATION(org.springframework.ide.vscode.commons.languageserver.completion.ScoreableProposal.DEEMP_DEPRECATION) Map(java.util.Map) ValueParseException(org.springframework.ide.vscode.commons.util.ValueParseException) YamlPath(org.springframework.ide.vscode.commons.yaml.path.YamlPath) DynamicSchemaContext(org.springframework.ide.vscode.commons.yaml.schema.DynamicSchemaContext) SNodeType(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNodeType) YamlPathSegment(org.springframework.ide.vscode.commons.yaml.path.YamlPathSegment) Logger(org.slf4j.Logger) ExceptionUtil(org.springframework.ide.vscode.commons.util.ExceptionUtil) YTypedProperty(org.springframework.ide.vscode.commons.yaml.schema.YTypedProperty) Log(org.springframework.ide.vscode.commons.util.Log) DocumentRegion(org.springframework.ide.vscode.commons.util.text.DocumentRegion) ISubCompletionEngine(org.springframework.ide.vscode.commons.yaml.schema.ISubCompletionEngine) ScoreableProposal(org.springframework.ide.vscode.commons.languageserver.completion.ScoreableProposal) Collection(java.util.Collection) TypeBasedSnippetProvider(org.springframework.ide.vscode.commons.yaml.snippet.TypeBasedSnippetProvider) Set(java.util.Set) PartialCollection(org.springframework.ide.vscode.commons.util.PartialCollection) YamlDocument(org.springframework.ide.vscode.commons.yaml.structure.YamlDocument) DEEMP_DASH_PROPOSAL(org.springframework.ide.vscode.commons.languageserver.completion.ScoreableProposal.DEEMP_DASH_PROPOSAL) Collectors(java.util.stream.Collectors) YType(org.springframework.ide.vscode.commons.yaml.schema.YType) YPropertyInfoTemplates(org.springframework.ide.vscode.commons.yaml.hover.YPropertyInfoTemplates) List(java.util.List) CollectionUtil(org.springframework.ide.vscode.commons.util.CollectionUtil) DocumentEdits(org.springframework.ide.vscode.commons.languageserver.completion.DocumentEdits) ICompletionProposal(org.springframework.ide.vscode.commons.languageserver.completion.ICompletionProposal) SNodeDynamicSchemaContext(org.springframework.ide.vscode.commons.yaml.schema.SNodeDynamicSchemaContext) SNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode) YValueHint(org.springframework.ide.vscode.commons.yaml.schema.YValueHint) Snippet(org.springframework.ide.vscode.commons.yaml.snippet.Snippet) Renderable(org.springframework.ide.vscode.commons.util.Renderable) ValueProposal(org.springframework.ide.vscode.commons.yaml.completion.DefaultCompletionFactory.ValueProposal) Collections(java.util.Collections) DocumentEdits(org.springframework.ide.vscode.commons.languageserver.completion.DocumentEdits) ScoreableProposal(org.springframework.ide.vscode.commons.languageserver.completion.ScoreableProposal) ArrayList(java.util.ArrayList) DynamicSchemaContext(org.springframework.ide.vscode.commons.yaml.schema.DynamicSchemaContext) SNodeDynamicSchemaContext(org.springframework.ide.vscode.commons.yaml.schema.SNodeDynamicSchemaContext) Snippet(org.springframework.ide.vscode.commons.yaml.snippet.Snippet) YValueHint(org.springframework.ide.vscode.commons.yaml.schema.YValueHint) YamlIndentUtil(org.springframework.ide.vscode.commons.yaml.util.YamlIndentUtil) ICompletionProposal(org.springframework.ide.vscode.commons.languageserver.completion.ICompletionProposal) TypeBasedSnippetProvider(org.springframework.ide.vscode.commons.yaml.snippet.TypeBasedSnippetProvider) ArrayList(java.util.ArrayList) ImmutableList(com.google.common.collect.ImmutableList) List(java.util.List) YTypedProperty(org.springframework.ide.vscode.commons.yaml.schema.YTypedProperty) YType(org.springframework.ide.vscode.commons.yaml.schema.YType)

Example 7 with SNode

use of org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode in project sts4 by spring-projects.

the class YTypeAssistContext method getSchemaContext.

protected DynamicSchemaContext getSchemaContext() {
    try {
        SNode contextNode = getContextNode();
        YamlPath fullContextPath = contextPath.prepend(YamlPathSegment.valueAt(documentSelector));
        return new SNodeDynamicSchemaContext(contextNode, fullContextPath);
    } catch (Exception e) {
        Log.log(e);
        return DynamicSchemaContext.NULL;
    }
}
Also used : YamlPath(org.springframework.ide.vscode.commons.yaml.path.YamlPath) SNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode) SNodeDynamicSchemaContext(org.springframework.ide.vscode.commons.yaml.schema.SNodeDynamicSchemaContext) ValueParseException(org.springframework.ide.vscode.commons.util.ValueParseException)

Example 8 with SNode

use of org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode in project sts4 by spring-projects.

the class YamlCompletionEngine method isLesserIndentRelaxable.

protected final boolean isLesserIndentRelaxable(final SNode currentNode, final SNode contextNode) {
    if (!options.includeDeindentedProposals()) {
        return false;
    }
    SChildBearingNode parent = currentNode.getParent();
    while (parent != null && parent != contextNode) {
        SNode lastChild = parent.getLastRealChild();
        if (lastChild != null && lastChild.getStart() >= currentNode.getNodeEnd()) {
            return false;
        }
        parent = parent.getParent();
    }
    return true;
}
Also used : SNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode) SChildBearingNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SChildBearingNode)

Example 9 with SNode

use of org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode in project sts4 by spring-projects.

the class YamlStructureParserTest method testTraverseSeq.

@Test
public void testTraverseSeq() throws Exception {
    MockYamlEditor editor = new MockYamlEditor("foo:\n" + "- - - - a\n" + "    - c\n" + "- e");
    SRootNode root = editor.parseStructure();
    YamlPath path;
    path = pathWith(0, "foo", 0, 0, 0, 0);
    assertEquals("SEQ(6): - a\n", path.traverse((SNode) root).toString());
    path = pathWith(0, "foo", -1);
    assertNull(path.traverse((SNode) root));
    path = pathWith(0, "foo", 1);
    assertEquals("SEQ(0): - e\n", path.traverse((SNode) root).toString());
    path = pathWith(0, "foo", 2);
    assertNull(path.traverse((SNode) root));
}
Also used : YamlPath(org.springframework.ide.vscode.commons.yaml.path.YamlPath) SRootNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SRootNode) SNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode) Test(org.junit.Test)

Example 10 with SNode

use of org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode in project sts4 by spring-projects.

the class YamlStructureParserTest method assertFirstRealChild.

private void assertFirstRealChild(MockYamlEditor editor, String testNodeName, String expectedNodeSnippet) throws Exception {
    SDocNode doc = getOnlyDocument(editor.parseStructure());
    SKeyNode testNode = doc.getChildWithKey(testNodeName);
    assertNotNull(testNode);
    SNode expected = null;
    if (expectedNodeSnippet != null) {
        int offset = editor.getRawText().indexOf(expectedNodeSnippet);
        expected = doc.find(offset);
        assertTrue(editor.textUnder(expected).contains(expectedNodeSnippet));
    }
    assertEquals(expected, testNode.getFirstRealChild());
}
Also used : SKeyNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SKeyNode) SNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode) SDocNode(org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SDocNode)

Aggregations

SNode (org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SNode)13 SRootNode (org.springframework.ide.vscode.commons.yaml.structure.YamlStructureParser.SRootNode)6 YamlPath (org.springframework.ide.vscode.commons.yaml.path.YamlPath)5 Test (org.junit.Test)4 ArrayList (java.util.ArrayList)2 ICompletionProposal (org.springframework.ide.vscode.commons.languageserver.completion.ICompletionProposal)2 ValueParseException (org.springframework.ide.vscode.commons.util.ValueParseException)2 YamlPathSegment (org.springframework.ide.vscode.commons.yaml.path.YamlPathSegment)2 SNodeDynamicSchemaContext (org.springframework.ide.vscode.commons.yaml.schema.SNodeDynamicSchemaContext)2 Strings (com.google.common.base.Strings)1 ImmutableList (com.google.common.collect.ImmutableList)1 Collection (java.util.Collection)1 Collections (java.util.Collections)1 List (java.util.List)1 Map (java.util.Map)1 Set (java.util.Set)1 Collectors (java.util.stream.Collectors)1 Logger (org.slf4j.Logger)1 LoggerFactory (org.slf4j.LoggerFactory)1 DocumentEdits (org.springframework.ide.vscode.commons.languageserver.completion.DocumentEdits)1