Examples with YamlDocument org.springframework.ide.vscode.commons.yaml.structure.YamlDocument used on opensource projects

Example 1 with YamlDocument

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

the class YamlCompletionEngine method getCompletions.

@Override
public Collection<ICompletionProposal> getCompletions(TextDocument _doc, int offset) throws Exception {
    YamlDocument doc = new YamlDocument(_doc, structureProvider);
    if (!doc.isCommented(offset)) {
        SRootNode root = doc.getStructure();
        SNode current = root.find(offset);
        int cursorIndent = doc.getColumn(offset);
        int nodeIndent = current.getIndent();
        int baseIndent = YamlIndentUtil.minIndent(cursorIndent, nodeIndent);
        List<SNode> contextNodes = getContextNodes(doc, current, offset, baseIndent);
        if (current.getNodeType() == SNodeType.RAW) {
            // relaxed indentation
            List<ICompletionProposal> completions = new ArrayList<>();
            double deempasizeBy = 0.0;
            for (SNode contextNode : contextNodes) {
                completions.addAll(getRelaxedCompletions(offset, doc, current, contextNode, baseIndent, deempasizeBy));
                deempasizeBy += ScoreableProposal.DEEMP_NEXT_CONTEXT;
            }
            return completions;
        } else {
            // precise indentation only
            Assert.isLegal(contextNodes.size() <= 1);
            for (SNode contextNode : contextNodes) {
                return getBaseCompletions(offset, doc, current, contextNode);
            }
        }
    }
    return Collections.emptyList();
}

Example 2 with YamlDocument

use of org.springframework.ide.vscode.commons.yaml.structure.YamlDocument 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();
}

Example 3 with YamlDocument

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

the class YamlHoverInfoProvider method getHoverInfo.

@Override
public Tuple2<Renderable, IRegion> getHoverInfo(IDocument doc, int offset) throws Exception {
    YamlFileAST ast = getAst(doc);
    if (ast != null) {
        IRegion region = getHoverRegion(ast, offset);
        if (region != null) {
            YamlDocument ymlDoc = new YamlDocument(doc, structureProvider);
            YamlAssistContext assistContext = assistContextProvider.getGlobalAssistContext(ymlDoc);
            if (assistContext != null) {
                List<NodeRef<?>> astPath = ast.findPath(offset);
                final YamlPath path = YamlPath.fromASTPath(astPath);
                if (path != null) {
                    YamlPath assistPath = path;
                    if (assistPath.pointsAtKey()) {
                        // When a path points at a key we must tramsform it to a
                        // 'value-terminating path'
                        // to be able to reuse the 'getHoverInfo' method on
                        // YamlAssistContext (as navigation
                        // into 'key' is not defined for YamlAssistContext.
                        String key = path.getLastSegment().toPropString();
                        assistPath = path.dropLast().append(YamlPathSegment.valueAt(key));
                    }
                    assistContext = assistPath.traverse(assistContext);
                    if (assistContext != null) {
                        Renderable info = path.pointsAtValue() ? assistContext.getValueHoverInfo(ymlDoc, new DocumentRegion(doc, region)) : assistContext.getHoverInfo();
                        // Fix for: PT 134914895. If assist context cannot provide an info, then don't return a Tuple.
                        if (info != null) {
                            return Tuples.of(info, region);
                        }
                    }
                }
            }
        }
    }
    return null;
}

Example 4 with YamlDocument

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

the class MockYamlEditor method parseStructure.

public SRootNode parseStructure() throws Exception {
    YamlStructureProvider sp = YamlStructureProvider.DEFAULT;
    TextDocument _doc = new TextDocument(null, getLanguageId());
    _doc.setText(text);
    YamlDocument doc = new YamlDocument(_doc, sp);
    return sp.getStructure(doc);
}