allow multiple builders to provide package nodes; move some kotlin-independent logic to DocumentationNode

1 files changed, 20 insertions, 1 deletions

diff --git a/src/Model/DocumentationNode.kt b/src/Model/DocumentationNode.kt
index c3700069..64a854a8 100644
--- a/src/Model/DocumentationNode.kt
+++ b/src/Model/DocumentationNode.kt
@@ -101,4 +101,23 @@ val DocumentationNode.path: List<DocumentationNode>
         if (parent == null)
             return listOf(this)
         return parent.path + this
-    }
\ No newline at end of file
+    }
+
+fun DocumentationNode.findOrCreatePackageNode(packageName: String): DocumentationNode {
+    val existingNode = members(DocumentationNode.Kind.Package).firstOrNull { it.name  == packageName }
+    if (existingNode != null) {
+        return existingNode
+    }
+    val newNode = DocumentationNode(packageName, Content.Empty, DocumentationNode.Kind.Package)
+    append(newNode, DocumentationReference.Kind.Member)
+    return newNode
+}
+
+fun DocumentationNode.append(child: DocumentationNode, kind: DocumentationReference.Kind) {
+    addReferenceTo(child, kind)
+    when (kind) {
+        DocumentationReference.Kind.Detail -> child.addReferenceTo(this, DocumentationReference.Kind.Owner)
+        DocumentationReference.Kind.Member -> child.addReferenceTo(this, DocumentationReference.Kind.Owner)
+        DocumentationReference.Kind.Owner -> child.addReferenceTo(this, DocumentationReference.Kind.Member)
+    }
+}