Add variable content to PageNodes and extras to DocumentationNodes

I'm just sorry

3 files changed, 54 insertions, 12 deletions

diff --git a/core/src/main/kotlin/Model/DocumentationNode.kt b/core/src/main/kotlin/Model/DocumentationNode.kt
index fc81db6d..55ae5902 100644
--- a/core/src/main/kotlin/Model/DocumentationNode.kt
+++ b/core/src/main/kotlin/Model/DocumentationNode.kt
@@ -8,13 +8,15 @@ import org.jetbrains.kotlin.kdoc.psi.impl.KDocTag
 class Module(val packages: List<Package>) : DocumentationNode<Nothing>() {
     override val dri: DRI = DRI.topLevel
     override val children: List<Package> = packages
+    override val extra: MutableSet<Extra> = mutableSetOf()
 }
 
 class Package(
     override val dri: DRI,
     override val functions: List<Function>,
     override val properties: List<Property>,
-    override val classes: List<Class>
+    override val classes: List<Class>,
+    override val extra: MutableSet<Extra> = mutableSetOf()
 ) : ScopeNode<Nothing>() {
     val name = dri.packageName.orEmpty()
 }
@@ -27,7 +29,8 @@ class Class(
     override val properties: List<Property>,
     override val classes: List<Class>,
     override val expectDescriptor: Descriptor<ClassDescriptor>?,
-    override val actualDescriptors: List<Descriptor<ClassDescriptor>>
+    override val actualDescriptors: List<Descriptor<ClassDescriptor>>,
+    override val extra: MutableSet<Extra> = mutableSetOf()
 ) : ScopeNode<ClassDescriptor>()
 
 class Function(
@@ -36,7 +39,8 @@ class Function(
     override val receiver: Parameter?,
     val parameters: List<Parameter>,
     override val expectDescriptor: Descriptor<FunctionDescriptor>?,
-    override val actualDescriptors: List<Descriptor<FunctionDescriptor>>
+    override val actualDescriptors: List<Descriptor<FunctionDescriptor>>,
+    override val extra: MutableSet<Extra> = mutableSetOf()
 ) : CallableNode<FunctionDescriptor>() {
     override val children: List<Parameter>
         get() = listOfNotNull(receiver) + parameters
@@ -47,7 +51,8 @@ class Property(
     val name: String,
     override val receiver: Parameter?,
     override val expectDescriptor: Descriptor<PropertyDescriptor>?,
-    override val actualDescriptors: List<Descriptor<PropertyDescriptor>>
+    override val actualDescriptors: List<Descriptor<PropertyDescriptor>>,
+    override val extra: MutableSet<Extra> = mutableSetOf()
 ) : CallableNode<PropertyDescriptor>() {
     override val children: List<Parameter>
         get() = listOfNotNull(receiver)
@@ -57,7 +62,8 @@ class Property(
 class Parameter(
     override val dri: DRI,
     val name: String?,
-    override val actualDescriptors: List<Descriptor<ParameterDescriptor>>
+    override val actualDescriptors: List<Descriptor<ParameterDescriptor>>,
+    override val extra: MutableSet<Extra> = mutableSetOf()
 ) : DocumentationNode<ParameterDescriptor>() {
     override val children: List<DocumentationNode<*>>
         get() = emptyList()
@@ -103,6 +109,8 @@ abstract class DocumentationNode<out T : DeclarationDescriptor> {
 
     val briefDocstring: String
         get() = descriptors.firstOrNull()?.docTag?.getContent().orEmpty().shorten(40)
+
+    open val extra: MutableSet<Extra> = mutableSetOf()
 }
 
 abstract class ScopeNode<out T : ClassOrPackageFragmentDescriptor> : DocumentationNode<T>() {
@@ -120,4 +128,28 @@ abstract class CallableNode<out T : CallableDescriptor> : DocumentationNode<T>()
 
 private fun String.shorten(maxLength: Int) = lineSequence().first().let {
     if (it.length != length || it.length > maxLength) it.take(maxLength - 3) + "..." else it
-}
\ No newline at end of file
+}
+
+fun DocumentationNode<*>.walk(process: DocumentationNode<*>.() -> Unit) {
+    this.process()
+    this.children.forEach { it.process() }
+}
+
+fun DocumentationNode<*>.dfs(predicate: (DocumentationNode<*>) -> Boolean): DocumentationNode<*>? =
+    if (predicate(this)) {
+        this
+    } else {
+        this.children.asSequence().mapNotNull { it.dfs(predicate) }.firstOrNull()
+    }
+
+fun DocumentationNode<*>.findAll(predicate: (DocumentationNode<*>) -> Boolean): Set<DocumentationNode<*>> {
+    val found = mutableSetOf<DocumentationNode<*>>()
+    if (predicate(this)) {
+        found.add(this)
+    } else {
+        this.children.asSequence().mapNotNull { it.findAll(predicate) }.forEach { found.addAll(it) }
+    }
+    return found
+}
+
+interface Extra
\ No newline at end of file
diff --git a/core/src/main/kotlin/pages/ContentNodes.kt b/core/src/main/kotlin/pages/ContentNodes.kt
index 40b5e114..dcc511cc 100644
--- a/core/src/main/kotlin/pages/ContentNodes.kt
+++ b/core/src/main/kotlin/pages/ContentNodes.kt
@@ -132,4 +132,14 @@ interface HTMLMetadata: Extra {
 }
 
 data class HTMLSimpleAttr(override val key: String, override val value: String): HTMLMetadata
-data class HTMLTableMetadata(val item: String, override val key: String, override val value: String): HTMLMetadata
\ No newline at end of file
+data class HTMLTableMetadata(val item: String, override val key: String, override val value: String): HTMLMetadata
+
+fun ContentNode.dfs(predicate: (ContentNode) -> Boolean): ContentNode? = if (predicate(this)) {
+    this
+} else {
+    if (this is ContentComposite) {
+        this.children.asSequence().mapNotNull { it.dfs(predicate) }.firstOrNull()
+    } else {
+        null
+    }
+}
diff --git a/core/src/main/kotlin/pages/PageNodes.kt b/core/src/main/kotlin/pages/PageNodes.kt
index 810cd776..6c56d5fe 100644
--- a/core/src/main/kotlin/pages/PageNodes.kt
+++ b/core/src/main/kotlin/pages/PageNodes.kt
@@ -6,7 +6,7 @@ import org.jetbrains.dokka.links.DRI
 
 interface PageNode {
     val name: String
-    val content: ContentNode
+    var content: ContentNode
     val parent: PageNode?
     val dri: DRI
     val documentationNode: DocumentationNode<*>?
@@ -42,7 +42,7 @@ abstract class BasicPageNode(children: List<PageNode>) : PageNode {
 
 class ModulePageNode(
     override val name: String,
-    override val content: ContentNode,
+    override var content: ContentNode,
     override val documentationNode: DocumentationNode<*>?,
     children: List<PageNode>,
     override val embeddedResources: List<String> = listOf()
@@ -62,7 +62,7 @@ class ModulePageNode(
 
 class PackagePageNode(
     override val name: String,
-    override val content: ContentNode,
+    override var content: ContentNode,
     override val dri: DRI,
     override val documentationNode: DocumentationNode<*>?,
     children: List<PageNode>,
@@ -81,7 +81,7 @@ class PackagePageNode(
 
 class ClassPageNode(
     override val name: String,
-    override val content: ContentNode,
+    override var content: ContentNode,
     override val dri: DRI,
     override val documentationNode: DocumentationNode<*>?,
     children: List<PageNode>,
@@ -100,7 +100,7 @@ class ClassPageNode(
 
 class MemberPageNode(
     override val name: String,
-    override val content: ContentNode,
+    override var content: ContentNode,
     override val dri: DRI,
     override val documentationNode: DocumentationNode<*>?,
     children: List<PageNode> = emptyList(),