From dd6bbdb88b0e4aaefde6598fee605714a3a516d8 Mon Sep 17 00:00:00 2001
From: Paweł Marks <pmarks@virtuslab.com>
Date: Wed, 5 Aug 2020 15:37:54 +0200
Subject: Add tools for easier content test debugging

---
 core/src/main/kotlin/model/WithChildren.kt | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'core/src/main/kotlin/model')

diff --git a/core/src/main/kotlin/model/WithChildren.kt b/core/src/main/kotlin/model/WithChildren.kt
index 06ff55b0..4aba51c6 100644
--- a/core/src/main/kotlin/model/WithChildren.kt
+++ b/core/src/main/kotlin/model/WithChildren.kt
@@ -65,3 +65,25 @@ fun <T> T.dfs(predicate: (T) -> Boolean): T? where T : WithChildren<T> = if (pre
 } else {
     children.asSequence().mapNotNull { it.dfs(predicate) }.firstOrNull()
 }
+
+fun <T: WithChildren<T>> T.asPrintableTree(
+    nodeNameBuilder: Appendable.(T) -> Unit = { append(it.toString()) }
+): String {
+    fun Appendable.append(element: T, ownPrefix: String, childPrefix: String) {
+        append(ownPrefix)
+        nodeNameBuilder(element)
+        appendLine()
+        element.children.takeIf(Collection<*>::isNotEmpty)?.also { children ->
+            val newOwnPrefix = childPrefix + '\u251c' + '\u2500' + ' '
+            val lastOwnPrefix = childPrefix + '\u2514' + '\u2500' + ' '
+            val newChildPrefix = childPrefix + '\u2502' + ' ' + ' '
+            val lastChildPrefix = childPrefix + ' ' + ' ' + ' '
+            children.forEachIndexed { n, e ->
+                if (n != children.lastIndex) append(e, newOwnPrefix, newChildPrefix)
+                else append(e, lastOwnPrefix, lastChildPrefix)
+            }
+        }
+    }
+
+    return buildString { append(this@asPrintableTree, "", "") }
+}
\ No newline at end of file
-- 
cgit