Refactor Ancestry Graphs (#2326)

1 files changed, 14 insertions, 0 deletions

diff --git a/core/src/main/kotlin/model/ancestryNode.kt b/core/src/main/kotlin/model/ancestryNode.kt
new file mode 100644
index 00000000..5c3c077b
--- /dev/null
+++ b/core/src/main/kotlin/model/ancestryNode.kt
@@ -0,0 +1,14 @@
+package org.jetbrains.dokka.model
+
+data class AncestryNode(
+    val typeConstructor: TypeConstructor,
+    val superclass: AncestryNode?,
+    val interfaces: List<AncestryNode>,
+) {
+    fun allImplementedInterfaces(): List<TypeConstructor> {
+        fun traverseInterfaces(ancestry: AncestryNode): List<TypeConstructor> =
+            ancestry.interfaces.flatMap { listOf(it.typeConstructor) + traverseInterfaces(it) } +
+                    (ancestry.superclass?.let(::traverseInterfaces) ?: emptyList())
+        return traverseInterfaces(this).distinct()
+    }
+}