From 4a018fa175bee9e7223232cbaf6365c81c540239 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Fri, 13 Feb 2015 12:38:32 +0100
Subject: allow multiple builders to provide package nodes; move some
 kotlin-independent logic to DocumentationNode

---
 src/Model/DocumentationNode.kt | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

(limited to 'src/Model')

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)
+    }
+}
-- 
cgit