From e9fd8b7bc00491b50e4822acc82e5615ab0bde3b Mon Sep 17 00:00:00 2001
From: "sebastian.sellmair" <sebastian.sellmair@jetbrains.com>
Date: Fri, 5 Jun 2020 09:07:20 +0200
Subject: Implement `reportUndocumented` option to report undocumented code

---
 core/src/main/kotlin/model/Documentable.kt | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'core/src/main/kotlin/model/Documentable.kt')

diff --git a/core/src/main/kotlin/model/Documentable.kt b/core/src/main/kotlin/model/Documentable.kt
index a7dd0919..c4dabb3b 100644
--- a/core/src/main/kotlin/model/Documentable.kt
+++ b/core/src/main/kotlin/model/Documentable.kt
@@ -1,8 +1,6 @@
 package org.jetbrains.dokka.model
 
 import com.intellij.psi.PsiNamedElement
-import org.jetbrains.dokka.DokkaConfiguration
-import org.jetbrains.dokka.Platform
 import org.jetbrains.dokka.links.DRI
 import org.jetbrains.dokka.model.doc.DocumentationNode
 import org.jetbrains.dokka.model.properties.PropertyContainer
@@ -378,6 +376,15 @@ fun Documentable.dfs(predicate: (Documentable) -> Boolean): Documentable? =
         this.children.asSequence().mapNotNull { it.dfs(predicate) }.firstOrNull()
     }
 
+fun Documentable.withDescendants(): Sequence<Documentable> {
+    return sequence {
+        yield(this@withDescendants)
+        children.forEach { child ->
+            yieldAll(child.withDescendants())
+        }
+    }
+}
+
 sealed class Visibility(val name: String)
 sealed class KotlinVisibility(name: String) : Visibility(name) {
     object Public : KotlinVisibility("public")
-- 
cgit