chylex/IntelliJ-Rainbow-Brackets
1
0
Fork 0
mirror of https://github.com/chylex/IntelliJ-Rainbow-Brackets.git synced 2025-04-11 03:15:49 +02:00
Code

Support for XML

Browse Source
This commit is contained in:
Yii.Guxing 2019-09-28 12:11:07 +08:00
parent 6c3c7d6d4f
commit edf872f949
1 changed files with 25 additions and 9 deletions

34
src/main/kotlin/com/github/izhangzhihao/rainbow/brackets/indents/RainbowIndentsPass.kt
View File

@ -20,12 +20,17 @@ import com.intellij.openapi.progress.ProgressIndicator
import com.intellij.openapi.progress.ProgressManager
import com.intellij.openapi.project.DumbAware
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.Condition
import com.intellij.openapi.util.Key
import com.intellij.openapi.util.Segment
import com.intellij.openapi.util.TextRange
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiFile
import com.intellij.psi.PsiManager
import com.intellij.psi.tree.TokenSet
import com.intellij.psi.util.PsiTreeUtil
import com.intellij.psi.xml.XmlFile
import com.intellij.psi.xml.XmlTag
import com.intellij.util.DocumentUtil
import com.intellij.util.containers.IntStack
import com.intellij.util.text.CharArrayUtil
@ -335,18 +340,12 @@ class RainbowIndentsPass internal constructor(
private val INDENT_HIGHLIGHTERS_IN_EDITOR_KEY = Key.create<MutableList<RangeHighlighter>>("_INDENT_HIGHLIGHTERS_IN_EDITOR_KEY_")
private val LAST_TIME_INDENTS_BUILT = Key.create<Long>("_LAST_TIME_INDENTS_BUILT_")
private val XML_TAG_PARENT_CONDITION = Condition<PsiElement> { it is XmlTag }
private val RENDERER: CustomHighlighterRenderer = CustomHighlighterRenderer renderer@{ editor, highlighter, g ->
if (editor !is EditorEx) return@renderer
val document = editor.document
val project = editor.project ?: return@renderer
val psiFile = PsiManager.getInstance(project).findFile(editor.virtualFile) ?: return@renderer
val element = psiFile.findElementAt(highlighter.endOffset)?.parent ?: return@renderer
if (document.getLineNumber(element.startOffset) < document.getLineNumber(highlighter.startOffset)) {
return@renderer
}
val rainbowInfo = RainbowInfo.RAINBOW_INFO_KEY[element] ?: return@renderer
val rainbowInfo = getRainbowInfo(editor, highlighter) ?: return@renderer
val startOffset = highlighter.startOffset
val doc = highlighter.document
@ -469,6 +468,23 @@ class RainbowIndentsPass internal constructor(
}
}
private fun getRainbowInfo(editor: EditorEx, highlighter: RangeHighlighter): RainbowInfo? {
val document = editor.document
val project = editor.project ?: return null
val psiFile = PsiManager.getInstance(project).findFile(editor.virtualFile) ?: return null
var element = psiFile.findElementAt(highlighter.endOffset)?.parent ?: return null
if (psiFile is XmlFile && element !is XmlTag) {
element = PsiTreeUtil.findFirstParent(element, true, XML_TAG_PARENT_CONDITION) ?: return null
}
if (document.getLineNumber(element.startOffset) < document.getLineNumber(highlighter.startOffset)) {
return null
}
return RainbowInfo.RAINBOW_INFO_KEY[element]
}
private fun createHighlighter(mm: MarkupModel, range: TextRange): RangeHighlighter {
return mm.addRangeHighlighter(
range.startOffset,