generate all types index

1 files changed, 14 insertions, 0 deletions

diff --git a/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt b/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt
index 3c4c0c39..1bc41f54 100644
--- a/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt
+++ b/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt
@@ -293,6 +293,7 @@ class DocumentationBuilder
         }
 
         propagateExtensionFunctionsToSubclasses(fragments)
+        generateAllTypesNode()
     }
 
     private fun propagateExtensionFunctionsToSubclasses(fragments: Collection<PackageFragmentDescriptor>) {
@@ -356,6 +357,19 @@ class DocumentationBuilder
         return false
     }
 
+    private fun DocumentationNode.generateAllTypesNode() {
+        val allTypes = members(NodeKind.Package)
+                .flatMap { it.members.filter { it.kind in NodeKind.classLike || it.kind == NodeKind.ExternalClass } }
+                .sortedBy { if (it.kind == NodeKind.ExternalClass) it.name.substringAfterLast('.') else it.name }
+
+        val allTypesNode = DocumentationNode("alltypes", Content.Empty, NodeKind.AllTypes)
+        for (typeNode in allTypes) {
+            allTypesNode.addReferenceTo(typeNode, RefKind.Member)
+        }
+
+        append(allTypesNode, RefKind.Member)
+    }
+
     fun DeclarationDescriptor.build(): DocumentationNode = when (this) {
         is ClassDescriptor -> build()
         is ConstructorDescriptor -> build()