AttributesDescriptorsBuilder DSL for Intellij Plugins

AttributesDescriptorsDSL.kt

import com.intellij.openapi.editor.colors.TextAttributesKey
import com.intellij.openapi.options.colors.AttributesDescriptor
import java.util.*

fun attributesDescriptors(block: AttributesDescriptorsBuilder.() -> Unit): Array<AttributesDescriptor> =
    AttributesDescriptorsBuilder().apply(block).build()

@DslMarker
annotation class DescriptorDsl

@DescriptorDsl
class AttributesDescriptorsBuilder {

    private val attrs = mutableListOf<AttributesDescriptor>()

    fun descriptor(displayName: String, attrKey: TextAttributesKey) =
        attrs.add(AttributesDescriptor(displayName, attrKey))

    fun descriptor(block: DescriptorBuilder.() -> Unit) = attrs.add(DescriptorBuilder().apply(block).build())

    fun descriptors(displayName: String, block: DESCRIPTORS.() -> Unit) =
        attrs.addAll(DESCRIPTORS(displayName).apply(block))

    fun build(): Array<AttributesDescriptor> = attrs.toTypedArray()

}

@DescriptorDsl
class DESCRIPTORS(private val parentDisplayName: String) : ArrayList<AttributesDescriptor>() {

    fun descriptor(displayName: String, attrKey: TextAttributesKey) = add(DescriptorBuilder(parentDisplayName).apply {
        name = displayName
        key = attrKey
    }.build())

    fun descriptor(block: DescriptorBuilder.() -> Unit) = add(DescriptorBuilder(parentDisplayName).apply(block).build())
}

@DescriptorDsl
class DescriptorBuilder(private val parentDisplayName: String = "") {

    lateinit var name: String
    lateinit var key: TextAttributesKey

    fun build() = AttributesDescriptor(
        if (parentDisplayName.isEmpty()) name
        else xBundle["settings.color.attribute.descriptor.separator", parentDisplayName, name], key
    )

}