Adds transformation from descriptors to documentation graph

1 files changed, 24 insertions, 0 deletions

diff --git a/core/src/main/kotlin/Utilities/nodeDebug.kt b/core/src/main/kotlin/Utilities/nodeDebug.kt
new file mode 100644
index 00000000..423e3e5f
--- /dev/null
+++ b/core/src/main/kotlin/Utilities/nodeDebug.kt
@@ -0,0 +1,24 @@
+package org.jetbrains.dokka.Utilities
+
+import org.jetbrains.dokka.Model.DocumentationNode
+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 ""
+}
\ No newline at end of file