hxuanyu/intellij-sdk-code-samples
1
0
Fork 0
mirror of https://github.com/JetBrains/intellij-sdk-code-samples.git synced 2025-07-27 16:57:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

simple language: SimpleFoldingBuilder use visitor, cleanup

Browse Source
This commit is contained in:
Yann Cébron 2023-08-10 14:04:28 +02:00
parent e52fcb44e6
commit f483e89218
1 changed files with 52 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
simple_language_plugin/src/main/java/org/intellij/sdk/language/SimpleFoldingBuilder.java
View File

@ -11,16 +11,16 @@ import com.intellij.openapi.project.DumbAware;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.JavaRecursiveElementWalkingVisitor;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiLiteralExpression;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.psi.util.PsiLiteralUtil;
import com.intellij.util.containers.ContainerUtil;
import org.intellij.sdk.language.psi.SimpleProperty;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
@ -32,12 +32,14 @@ public class SimpleFoldingBuilder extends FoldingBuilderEx implements DumbAware
FoldingGroup group = FoldingGroup.newGroup(SimpleAnnotator.SIMPLE_PREFIX_STR);
// Initialize the list of folding regions
List<FoldingDescriptor> descriptors = new ArrayList<>();
// Get a collection of the literal expressions in the document below root
Collection<PsiLiteralExpression> literalExpressions =
PsiTreeUtil.findChildrenOfType(root, PsiLiteralExpression.class);
// Evaluate the collection
for (final PsiLiteralExpression literalExpression : literalExpressions) {
String value = literalExpression.getValue() instanceof String ? (String) literalExpression.getValue() : null;
root.accept(new JavaRecursiveElementWalkingVisitor() {
@Override
public void visitLiteralExpression(@NotNull PsiLiteralExpression literalExpression) {
super.visitLiteralExpression(literalExpression);
String value = PsiLiteralUtil.getStringLiteralContent(literalExpression);
if (value != null && value.startsWith(SimpleAnnotator.SIMPLE_PREFIX_STR + SimpleAnnotator.SIMPLE_SEPARATOR_STR)) {
Project project = literalExpression.getProject();
String key = value.substring(
@ -54,6 +56,8 @@ public class SimpleFoldingBuilder extends FoldingBuilderEx implements DumbAware
}
}
}
});
return descriptors.toArray(FoldingDescriptor.EMPTY);
}
@ -67,22 +71,34 @@ public class SimpleFoldingBuilder extends FoldingBuilderEx implements DumbAware
@Nullable
@Override
public String getPlaceholderText(@NotNull ASTNode node) {
if (!(node.getPsi() instanceof PsiLiteralExpression nodeElement)) {
return null ;
if (node.getPsi() instanceof PsiLiteralExpression psiLiteralExpression) {
String text = PsiLiteralUtil.getStringLiteralContent(psiLiteralExpression);
if (text == null) {
return null;
}
String key = ((String) nodeElement.getValue()).substring(
SimpleAnnotator.SIMPLE_PREFIX_STR.length() + SimpleAnnotator.SIMPLE_SEPARATOR_STR.length()
);
String key = text.substring(SimpleAnnotator.SIMPLE_PREFIX_STR.length() +
SimpleAnnotator.SIMPLE_SEPARATOR_STR.length());
SimpleProperty simpleProperty = ContainerUtil.getOnlyItem(SimpleUtil.findProperties(nodeElement.getProject(), key));
if (simpleProperty == null) return StringUtil.THREE_DOTS;
SimpleProperty simpleProperty = ContainerUtil.getOnlyItem(
SimpleUtil.findProperties(psiLiteralExpression.getProject(), key)
);
if (simpleProperty == null) {
return StringUtil.THREE_DOTS;
}
String propertyValue = simpleProperty.getValue();
// IMPORTANT: keys can come with no values, so a test for null is needed
// IMPORTANT: Convert embedded \n to backslash n, so that the string will look
// like it has LF embedded in it and embedded " to escaped "
return propertyValue == null ? StringUtil.THREE_DOTS : propertyValue.replaceAll("\n", "\\n").replaceAll("\"", "\\\\\"");
if (propertyValue == null) {
return StringUtil.THREE_DOTS;
}
return propertyValue.replaceAll("\n", "\\n").replaceAll("\"", "\\\\\"");
}
return null;
}
@Override