src/main/kotlin/nbt/lang/format/NbttFoldingBuilder.kt diff

← → edited src/main/kotlin/nbt/lang/format/NbttFoldingBuilder.kt

Ignore whitespaces Open in IDE

⁠ joe: 2025

Copy Copy

/*
* Minecraft Development for IntelliJ
*
* https://mcdev.io/
*
* Copyright (C) 2024 minecraft-dev
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, version 3.0 only.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/

package com.demonwav.mcdev.nbt.lang.format

import com.demonwav.mcdev.nbt.lang.gen.psi.NbttByteArray
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttCompound
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttIntArray
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttList
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttLongArray
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttTypes
import com.intellij.lang.ASTNode
import com.intellij.lang.folding.FoldingBuilder
import com.intellij.lang.folding.FoldingDescriptor
import com.intellij.openapi.editor.Document
import com.intellij.openapi.util.TextRange

class NbttFoldingBuilder : FoldingBuilder {
override fun getPlaceholderText(node: ASTNode): String? {
return when (node.elementType) {
NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LIST -> "..."
NbttTypes.COMPOUND -> {
val tagList = (node.psi as NbttCompound).getNamedTagList()
if (tagList.isEmpty()) {
return null
}
val tag = tagList[0].tag
if (tagList.size == 1 && tag?.getList() == null && tag?.getCompound() == null &&
tag?.getIntArray() == null && tag?.getByteArray() == null
) {
tagList[0].text
} else {
"..."
}
}
else -> null
}
}

override fun buildFoldRegions(node: ASTNode, document: Document): Array<FoldingDescriptor> {
val list = mutableListOf<FoldingDescriptor>()
foldChildren(node, list)
return list.toTypedArray()
}

private fun foldChildren(node: ASTNode, list: MutableList<FoldingDescriptor>) {
when (node.elementType) {
NbttTypes.COMPOUND -> {
val lbrace = node.findChildByType(NbttTypes.LBRACE)
val rbrace = node.findChildByType(NbttTypes.RBRACE)
if (lbrace != null && rbrace != null) {
if (lbrace.textRange.endOffset != rbrace.textRange.startOffset) {
list.add(
FoldingDescriptor(
node,
TextRange(lbrace.textRange.endOffset, rbrace.textRange.startOffset),
),
)
}
}
}
NbttTypes.LIST -> {
val lbracket = node.findChildByType(NbttTypes.LBRACKET)
val rbracket = node.findChildByType(NbttTypes.RBRACKET)
if (lbracket != null && rbracket != null) {
if (lbracket.textRange.endOffset != rbracket.textRange.startOffset) {
list.add(
FoldingDescriptor(
node,
TextRange(lbracket.textRange.endOffset, rbracket.textRange.startOffset),
),
)
}
}
}
NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LONG_ARRAY -> {
val lparen = node.findChildByType(NbttTypes.LPAREN)
val rparen = node.findChildByType(NbttTypes.RPAREN)
if (lparen != null && rparen != null) {
if (lparen.textRange.endOffset != rparen.textRange.startOffset) {
list.add(
FoldingDescriptor(
node,
TextRange(lparen.textRange.endOffset, rparen.textRange.startOffset),
),
)
}
}
}
}

node.getChildren(null).forEach { foldChildren(it, list) }
}

override fun isCollapsedByDefault(node: ASTNode): Boolean {
val size = when (val psi = node.psi) {
is NbttByteArray -> psi.getByteList().size
is NbttIntArray -> psi.getIntList().size
is NbttLongArray -> psi.getLongList().size
is NbttList -> psi.getTagList().size
is NbttCompound -> {
if (psi.getNamedTagList().size == 1) {
val tag = psi.getNamedTagList()[0].tag
if (
tag?.getList() == null &&
tag?.getCompound() == null &&
tag?.getIntArray() == null &&
tag?.getByteArray() == null
) {
return true
}
}
psi.getNamedTagList().size
}
else -> 0
}

return size > 50 // TODO arbitrary? make a setting?
}
}

/*
* Minecraft Development for IntelliJ
*
* https://mcdev.io/
*
* Copyright (C) 2025 minecraft-dev
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, version 3.0 only.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/

package com.demonwav.mcdev.nbt.lang.format

import com.demonwav.mcdev.nbt.lang.gen.psi.NbttByteArray
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttCompound
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttIntArray
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttList
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttLongArray
import com.demonwav.mcdev.nbt.lang.gen.psi.NbttTypes
import com.intellij.lang.ASTNode
import com.intellij.lang.folding.FoldingBuilder
import com.intellij.lang.folding.FoldingDescriptor
import com.intellij.openapi.editor.Document
import com.intellij.openapi.util.TextRange

class NbttFoldingBuilder : FoldingBuilder {
override fun getPlaceholderText(node: ASTNode): String? {
return when (node.elementType) {
NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LIST -> "..."
NbttTypes.COMPOUND -> {
val tagList = (node.psi as NbttCompound).getNamedTagList()
if (tagList.isEmpty()) {
return null
}
val tag = tagList[0].tag
if (tagList.size == 1 && tag?.getList() == null && tag?.getCompound() == null &&
tag?.getIntArray() == null && tag?.getByteArray() == null
) {
tagList[0].text
} else {
"..."
}
}
else -> null
}
}

override fun buildFoldRegions(node: ASTNode, document: Document): Array<FoldingDescriptor> {
val list = mutableListOf<FoldingDescriptor>()
foldChildren(node, list)
return list.toTypedArray()
}

private fun foldChildren(node: ASTNode, list: MutableList<FoldingDescriptor>) {
when (node.elementType) {
NbttTypes.COMPOUND -> {
val lbrace = node.findChildByType(NbttTypes.LBRACE)
val rbrace = node.findChildByType(NbttTypes.RBRACE)
if (lbrace != null && rbrace != null) {
if (lbrace.textRange.endOffset != rbrace.textRange.startOffset) {
list.add(
FoldingDescriptor(
node,
TextRange(lbrace.textRange.endOffset, rbrace.textRange.startOffset),
),
)
}
}
}
NbttTypes.LIST -> {
val lbracket = node.findChildByType(NbttTypes.LBRACKET)
val rbracket = node.findChildByType(NbttTypes.RBRACKET)
if (lbracket != null && rbracket != null) {
if (lbracket.textRange.endOffset != rbracket.textRange.startOffset) {
list.add(
FoldingDescriptor(
node,
TextRange(lbracket.textRange.endOffset, rbracket.textRange.startOffset),
),
)
}
}
}
NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LONG_ARRAY -> {
val lparen = node.findChildByType(NbttTypes.LPAREN)
val rparen = node.findChildByType(NbttTypes.RPAREN)
if (lparen != null && rparen != null) {
if (lparen.textRange.endOffset != rparen.textRange.startOffset) {
list.add(
FoldingDescriptor(
node,
TextRange(lparen.textRange.endOffset, rparen.textRange.startOffset),
),
)
}
}
}
}

node.getChildren(null).forEach { foldChildren(it, list) }
}

override fun isCollapsedByDefault(node: ASTNode): Boolean {
val size = when (val psi = node.psi) {
is NbttByteArray -> psi.getByteList().size
is NbttIntArray -> psi.getIntList().size
is NbttLongArray -> psi.getLongList().size
is NbttList -> psi.getTagList().size
is NbttCompound -> {
if (psi.getNamedTagList().size == 1) {
val tag = psi.getNamedTagList()[0].tag
if (
tag?.getList() == null &&
tag?.getCompound() == null &&
tag?.getIntArray() == null &&
tag?.getByteArray() == null
) {
return true
}
}
psi.getNamedTagList().size
}
else -> 0
}

return size > 50 // TODO arbitrary? make a setting?
}
}

/* * Minecraft Development for IntelliJ * * https://mcdev.io/ * * Copyright (C) 2024 minecraft-dev * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published * by the Free Software Foundation, version 3.0 only. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this program. If not, see <https://www.gnu.org/licenses/>. */ package com.demonwav.mcdev.nbt.lang.format import com.demonwav.mcdev.nbt.lang.gen.psi.NbttByteArray import com.demonwav.mcdev.nbt.lang.gen.psi.NbttCompound import com.demonwav.mcdev.nbt.lang.gen.psi.NbttIntArray import com.demonwav.mcdev.nbt.lang.gen.psi.NbttList import com.demonwav.mcdev.nbt.lang.gen.psi.NbttLongArray import com.demonwav.mcdev.nbt.lang.gen.psi.NbttTypes import com.intellij.lang.ASTNode import com.intellij.lang.folding.FoldingBuilder import com.intellij.lang.folding.FoldingDescriptor import com.intellij.openapi.editor.Document import com.intellij.openapi.util.TextRange class NbttFoldingBuilder : FoldingBuilder { override fun getPlaceholderText(node: ASTNode): String? { return when (node.elementType) { NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LIST -> "..." NbttTypes.COMPOUND -> { val tagList = (node.psi as NbttCompound).getNamedTagList() if (tagList.isEmpty()) { return null } val tag = tagList[0].tag if (tagList.size == 1 && tag?.getList() == null && tag?.getCompound() == null && tag?.getIntArray() == null && tag?.getByteArray() == null ) { tagList[0].text } else { "..." } } else -> null } } override fun buildFoldRegions(node: ASTNode, document: Document): Array<FoldingDescriptor> { val list = mutableListOf<FoldingDescriptor>() foldChildren(node, list) return list.toTypedArray() } private fun foldChildren(node: ASTNode, list: MutableList<FoldingDescriptor>) { when (node.elementType) { NbttTypes.COMPOUND -> { val lbrace = node.findChildByType(NbttTypes.LBRACE) val rbrace = node.findChildByType(NbttTypes.RBRACE) if (lbrace != null && rbrace != null) { if (lbrace.textRange.endOffset != rbrace.textRange.startOffset) { list.add( FoldingDescriptor( node, TextRange(lbrace.textRange.endOffset, rbrace.textRange.startOffset), ), ) } } } NbttTypes.LIST -> { val lbracket = node.findChildByType(NbttTypes.LBRACKET) val rbracket = node.findChildByType(NbttTypes.RBRACKET) if (lbracket != null && rbracket != null) { if (lbracket.textRange.endOffset != rbracket.textRange.startOffset) { list.add( FoldingDescriptor( node, TextRange(lbracket.textRange.endOffset, rbracket.textRange.startOffset), ), ) } } } NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LONG_ARRAY -> { val lparen = node.findChildByType(NbttTypes.LPAREN) val rparen = node.findChildByType(NbttTypes.RPAREN) if (lparen != null && rparen != null) { if (lparen.textRange.endOffset != rparen.textRange.startOffset) { list.add( FoldingDescriptor( node, TextRange(lparen.textRange.endOffset, rparen.textRange.startOffset), ), ) } } } } node.getChildren(null).forEach { foldChildren(it, list) } } override fun isCollapsedByDefault(node: ASTNode): Boolean { val size = when (val psi = node.psi) { is NbttByteArray -> psi.getByteList().size is NbttIntArray -> psi.getIntList().size is NbttLongArray -> psi.getLongList().size is NbttList -> psi.getTagList().size is NbttCompound -> { if (psi.getNamedTagList().size == 1) { val tag = psi.getNamedTagList()[0].tag if ( tag?.getList() == null && tag?.getCompound() == null && tag?.getIntArray() == null && tag?.getByteArray() == null ) { return true } } psi.getNamedTagList().size } else -> 0 } return size > 50 // TODO arbitrary? make a setting? } } /* * Minecraft Development for IntelliJ * * https://mcdev.io/ * * Copyright (C) 2025 minecraft-dev * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published * by the Free Software Foundation, version 3.0 only. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this program. If not, see <https://www.gnu.org/licenses/>. */ package com.demonwav.mcdev.nbt.lang.format import com.demonwav.mcdev.nbt.lang.gen.psi.NbttByteArray import com.demonwav.mcdev.nbt.lang.gen.psi.NbttCompound import com.demonwav.mcdev.nbt.lang.gen.psi.NbttIntArray import com.demonwav.mcdev.nbt.lang.gen.psi.NbttList import com.demonwav.mcdev.nbt.lang.gen.psi.NbttLongArray import com.demonwav.mcdev.nbt.lang.gen.psi.NbttTypes import com.intellij.lang.ASTNode import com.intellij.lang.folding.FoldingBuilder import com.intellij.lang.folding.FoldingDescriptor import com.intellij.openapi.editor.Document import com.intellij.openapi.util.TextRange class NbttFoldingBuilder : FoldingBuilder { override fun getPlaceholderText(node: ASTNode): String? { return when (node.elementType) { NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LIST -> "..." NbttTypes.COMPOUND -> { val tagList = (node.psi as NbttCompound).getNamedTagList() if (tagList.isEmpty()) { return null } val tag = tagList[0].tag if (tagList.size == 1 && tag?.getList() == null && tag?.getCompound() == null && tag?.getIntArray() == null && tag?.getByteArray() == null ) { tagList[0].text } else { "..." } } else -> null } } override fun buildFoldRegions(node: ASTNode, document: Document): Array<FoldingDescriptor> { val list = mutableListOf<FoldingDescriptor>() foldChildren(node, list) return list.toTypedArray() } private fun foldChildren(node: ASTNode, list: MutableList<FoldingDescriptor>) { when (node.elementType) { NbttTypes.COMPOUND -> { val lbrace = node.findChildByType(NbttTypes.LBRACE) val rbrace = node.findChildByType(NbttTypes.RBRACE) if (lbrace != null && rbrace != null) { if (lbrace.textRange.endOffset != rbrace.textRange.startOffset) { list.add( FoldingDescriptor( node, TextRange(lbrace.textRange.endOffset, rbrace.textRange.startOffset), ), ) } } } NbttTypes.LIST -> { val lbracket = node.findChildByType(NbttTypes.LBRACKET) val rbracket = node.findChildByType(NbttTypes.RBRACKET) if (lbracket != null && rbracket != null) { if (lbracket.textRange.endOffset != rbracket.textRange.startOffset) { list.add( FoldingDescriptor( node, TextRange(lbracket.textRange.endOffset, rbracket.textRange.startOffset), ), ) } } } NbttTypes.BYTE_ARRAY, NbttTypes.INT_ARRAY, NbttTypes.LONG_ARRAY -> { val lparen = node.findChildByType(NbttTypes.LPAREN) val rparen = node.findChildByType(NbttTypes.RPAREN) if (lparen != null && rparen != null) { if (lparen.textRange.endOffset != rparen.textRange.startOffset) { list.add( FoldingDescriptor( node, TextRange(lparen.textRange.endOffset, rparen.textRange.startOffset), ), ) } } } } node.getChildren(null).forEach { foldChildren(it, list) } } override fun isCollapsedByDefault(node: ASTNode): Boolean { val size = when (val psi = node.psi) { is NbttByteArray -> psi.getByteList().size is NbttIntArray -> psi.getIntList().size is NbttLongArray -> psi.getLongList().size is NbttList -> psi.getTagList().size is NbttCompound -> { if (psi.getNamedTagList().size == 1) { val tag = psi.getNamedTagList()[0].tag if ( tag?.getList() == null && tag?.getCompound() == null && tag?.getIntArray() == null && tag?.getByteArray() == null ) { return true } } psi.getNamedTagList().size } else -> 0 } return size > 50 // TODO arbitrary? make a setting? } }

Add build comment

Mute test

Server communication failure

Run Custom Build

TODO

Responsibility

Edit tags

Loading...

Are you sure?