Examples with ConstantNode com.intellij.codeInsight.template.impl.ConstantNode used on opensource projects

Example 6 with ConstantNode

use of com.intellij.codeInsight.template.impl.ConstantNode in project intellij-community by JetBrains.

the class TestIntegrationUtils method createTestMethodTemplate.

public static Template createTestMethodTemplate(@NotNull MethodKind methodKind, TestFramework descriptor, @NotNull PsiClass targetClass, @Nullable PsiClass sourceClass, @Nullable String name, boolean automatic, Set<String> existingNames) {
    FileTemplateDescriptor templateDesc = methodKind.getFileTemplateDescriptor(descriptor);
    String templateName = templateDesc.getFileName();
    FileTemplate fileTemplate = FileTemplateManager.getInstance(targetClass.getProject()).getCodeTemplate(templateName);
    Template template = TemplateManager.getInstance(targetClass.getProject()).createTemplate("", "");
    String templateText;
    try {
        Properties properties = new Properties();
        if (sourceClass != null && sourceClass.isValid()) {
            properties.setProperty(FileTemplate.ATTRIBUTE_CLASS_NAME, sourceClass.getQualifiedName());
        }
        templateText = fileTemplate.getText(properties);
    } catch (IOException e) {
        LOG.warn(e);
        templateText = fileTemplate.getText();
    }
    if (name == null)
        name = methodKind.getDefaultName();
    if (existingNames != null && !existingNames.add(name)) {
        int idx = 1;
        while (existingNames.contains(name)) {
            final String newName = name + (idx++);
            if (existingNames.add(newName)) {
                name = newName;
                break;
            }
        }
    }
    templateText = StringUtil.replace(templateText, "${BODY}\n", "");
    int from = 0;
    while (true) {
        int index = templateText.indexOf("${NAME}", from);
        if (index == -1)
            break;
        template.addTextSegment(templateText.substring(from, index));
        if (index > 0 && !Character.isWhitespace(templateText.charAt(index - 1))) {
            name = StringUtil.capitalize(name);
        } else {
            name = StringUtil.decapitalize(name);
        }
        if (from == 0) {
            Expression nameExpr = new ConstantNode(name);
            template.addVariable("name", nameExpr, nameExpr, !automatic);
        } else {
            template.addVariableSegment("name");
        }
        from = index + "${NAME}".length();
    }
    template.addTextSegment(templateText.substring(from, templateText.length()));
    template.setToIndent(true);
    template.setToReformat(true);
    template.setToShortenLongNames(true);
    return template;
}

Example 7 with ConstantNode

use of com.intellij.codeInsight.template.impl.ConstantNode in project intellij-community by JetBrains.

the class TemplateBuilderImpl method replaceElement.

@Override
public void replaceElement(@NotNull PsiElement element, TextRange rangeWithinElement, String replacementText) {
    final RangeMarker key = myDocument.createRangeMarker(rangeWithinElement.shiftRight(element.getTextRange().getStartOffset()));
    ConstantNode value = new ConstantNode(replacementText);
    replaceElement(key, value);
}