blob: 8e2a7079b747e3f02f245f9fdc21768676ca05e2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
package org.jetbrains.dokka.Utilities
import org.jetbrains.dokka.Model.DocumentationNode
import org.jetbrains.dokka.pages.*
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
const val DOWN = '\u2503'
const val BRANCH = '\u2523'
const val LAST = '\u2517'
fun <T : DeclarationDescriptor> DocumentationNode<T>.pretty(prefix: String = "", isLast: Boolean = true): String {
val nextPrefix = prefix + (if (isLast) ' ' else DOWN) + ' '
return prefix + (if (isLast) LAST else BRANCH) + this.toString() +
children.dropLast(1)
.map { it.pretty(nextPrefix, false) }
.plus(children.lastOrNull()?.pretty(nextPrefix))
.filterNotNull()
.takeIf { it.isNotEmpty() }
?.joinToString(prefix = "\n", separator = "")
.orEmpty() + if (children.isEmpty()) "\n" else ""
}
fun Any.genericPretty(prefix: String = "", isLast: Boolean = true): String {
val nextPrefix = prefix + (if (isLast) ' ' else DOWN) + ' '
return prefix + (if (isLast) LAST else BRANCH) + this.stringify() +
allChildren().dropLast(1)
.map { it.genericPretty(nextPrefix, false) }
.plus(allChildren().lastOrNull()?.genericPretty(nextPrefix))
.filterNotNull()
.takeIf { it.isNotEmpty() }
?.joinToString(prefix = "\n", separator = "")
.orEmpty() + if (allChildren().isEmpty()) "\n" else ""
}
private fun Any.stringify() = when(this) {
is ContentNode -> toString() + this.dci
is PageNode -> this.name + this::class.simpleName
else -> toString()
}
private fun Any.allChildren() = when(this){
is PageNode -> children + content
is ContentBlock -> this.children
is ContentHeader -> this.items
is ContentStyle -> this.items
is ContentSymbol -> this.parts
is ContentComment -> this.parts
is ContentGroup -> this.children
is ContentList -> this.items
else -> emptyList()
}
|
