File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed
src/main/kotlin/com/github/xepozz/caddy/language/injection Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -6,8 +6,8 @@ import com.intellij.lang.ASTNode
66import com.intellij.lang.Language
77import com.intellij.lang.injection.MultiHostInjector
88import com.intellij.lang.injection.MultiHostRegistrar
9- import com.intellij.lang.tree.util.children
109import com.intellij.psi.PsiElement
10+ import com.intellij.psi.tree.TokenSet
1111
1212class CaddyHeredocLanguageInjector : MultiHostInjector {
1313 override fun elementsToInjectIn () = mutableListOf (CaddyHeredoc ::class .java)
@@ -18,10 +18,9 @@ class CaddyHeredocLanguageInjector : MultiHostInjector {
1818// println("PsiElement is not a CaddyHeredoc")
1919 return
2020 }
21- val childrenNodes = context.node.children().toList()
2221// println("children = $childrenNodes")
2322
24- val heredocStart = childrenNodes.first()
23+ val heredocStart = context.node.firstChildNode
2524
2625 val marker = extractMarkerFromHeredocStart(heredocStart)
2726 val language = findLanguageByMarker(marker)
@@ -59,8 +58,7 @@ class CaddyHeredocLanguageInjector : MultiHostInjector {
5958
6059 private fun collectHeredocContent (heredocStart : CaddyHeredoc ): Collection <PsiElement > =
6160 heredocStart.node
62- .children()
63- .filter { it.elementType == = CaddyTypes .HEREDOC_CONTENT }
61+ .getChildren(TokenSet .create(CaddyTypes .HEREDOC_CONTENT ))
6462 .mapNotNull { it.psi }
6563 .toList()
6664}
You can’t perform that action at this time.
0 commit comments