Type alias support

4 files changed, 80 insertions, 55 deletions

diff --git a/core/src/main/kotlin/Formats/StructuredFormatService.kt b/core/src/main/kotlin/Formats/StructuredFormatService.kt
index f24c9c2d..b7306486 100644
--- a/core/src/main/kotlin/Formats/StructuredFormatService.kt
+++ b/core/src/main/kotlin/Formats/StructuredFormatService.kt
@@ -23,11 +23,9 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
         body()
         if (checkEndsWith && to.endsWith(suffix)) {
             to.setLength(to.length - suffix.length)
-        }
-        else if (to.length > startLength + prefix.length) {
+        } else if (to.length > startLength + prefix.length) {
             to.append(suffix)
-        }
-        else {
+        } else {
             to.setLength(startLength)
         }
     }
@@ -124,8 +122,7 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
                 val child = content.children.singleOrNull()
                 if (child is ContentParagraph) {
                     appendContent(child.children)
-                }
-                else {
+                } else {
                     appendContent(content.children)
                 }
             }
@@ -361,7 +358,7 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
     }
 
     inner class SingleNodePageBuilder(val node: DocumentationNode)
-            : PageBuilder(listOf(node)) {
+        : PageBuilder(listOf(node)) {
 
         override fun build() {
             super.build()
@@ -372,14 +369,15 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
             }
 
             appendSection("Packages", node.members(NodeKind.Package))
-            appendSection("Types", node.members.filter { it.kind in NodeKind.classLike && it.kind != NodeKind.AnnotationClass && it.kind != NodeKind.Exception })
+            appendSection("Types", node.members.filter { it.kind in NodeKind.classLike && it.kind != NodeKind.TypeAlias && it.kind != NodeKind.AnnotationClass && it.kind != NodeKind.Exception })
             appendSection("Annotations", node.members(NodeKind.AnnotationClass))
             appendSection("Exceptions", node.members(NodeKind.Exception))
+            appendSection("Type aliases", node.members(NodeKind.TypeAlias))
             appendSection("Extensions for External Classes", node.members(NodeKind.ExternalClass))
             appendSection("Enum Values", node.members(NodeKind.EnumItem), sortMembers = false)
             appendSection("Constructors", node.members(NodeKind.Constructor))
             appendSection("Properties", node.members(NodeKind.Property))
-            appendSection("Inherited Properties",  node.inheritedMembers(NodeKind.Property))
+            appendSection("Inherited Properties", node.inheritedMembers(NodeKind.Property))
             appendSection("Functions", node.members(NodeKind.Function))
             appendSection("Inherited Functions", node.inheritedMembers(NodeKind.Function))
             appendSection("Companion Object Properties", node.members(NodeKind.CompanionObjectProperty))
@@ -394,6 +392,7 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
                         NodeKind.Object,
                         NodeKind.AnnotationClass,
                         NodeKind.Exception,
+                        NodeKind.TypeAlias,
                         NodeKind.Constructor,
                         NodeKind.Property,
                         NodeKind.Package,
@@ -472,7 +471,7 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
     }
 
     inner class AllTypesNodeBuilder(val node: DocumentationNode)
-           : PageBuilder(listOf(node)) {
+        : PageBuilder(listOf(node)) {
 
         override fun build() {
             appendContent(node.owner!!.summary)
@@ -508,12 +507,10 @@ abstract class StructuredOutputBuilder(val to: StringBuilder,
         if (singleNode != null) {
             if (singleNode.kind == NodeKind.AllTypes) {
                 AllTypesNodeBuilder(singleNode).build()
-            }
-            else {
+            } else {
                 SingleNodePageBuilder(singleNode).build()
             }
-        }
-        else {
+        } else {
             PageBuilder(nodes).build()
         }
     }
diff --git a/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt b/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt
index ecb6edf2..5f2d08f9 100644
--- a/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt
+++ b/core/src/main/kotlin/Kotlin/DocumentationBuilder.kt
@@ -25,10 +25,7 @@ import org.jetbrains.kotlin.resolve.findTopMostOverriddenDescriptors
 import org.jetbrains.kotlin.resolve.jvm.JavaDescriptorResolver
 import org.jetbrains.kotlin.resolve.source.PsiSourceElement
 import org.jetbrains.kotlin.resolve.source.getPsi
-import org.jetbrains.kotlin.types.ErrorUtils
-import org.jetbrains.kotlin.types.KotlinType
-import org.jetbrains.kotlin.types.TypeProjection
-import org.jetbrains.kotlin.types.TypeUtils
+import org.jetbrains.kotlin.types.*
 import org.jetbrains.kotlin.types.typeUtil.isSubtypeOf
 import org.jetbrains.kotlin.types.typeUtil.supertypes
 
@@ -61,13 +58,12 @@ interface PackageDocumentationBuilder {
 }
 
 class DocumentationBuilder
-        @Inject constructor(val resolutionFacade: DokkaResolutionFacade,
-                            val descriptorDocumentationParser: DescriptorDocumentationParser,
-                            val options: DocumentationOptions,
-                            val refGraph: NodeReferenceGraph,
-                            val logger: DokkaLogger,
-                            val linkResolver: DeclarationLinkResolver)
-{
+@Inject constructor(val resolutionFacade: DokkaResolutionFacade,
+                    val descriptorDocumentationParser: DescriptorDocumentationParser,
+                    val options: DocumentationOptions,
+                    val refGraph: NodeReferenceGraph,
+                    val logger: DokkaLogger,
+                    val linkResolver: DeclarationLinkResolver) {
     val boringBuiltinClasses = setOf(
             "kotlin.Unit", "kotlin.Byte", "kotlin.Short", "kotlin.Int", "kotlin.Long", "kotlin.Char", "kotlin.Boolean",
             "kotlin.Float", "kotlin.Double", "kotlin.String", "kotlin.Array", "kotlin.Any")
@@ -97,7 +93,7 @@ class DocumentationBuilder
         return node
     }
 
-    private fun DocumentationNode.withModifiers(descriptor: DeclarationDescriptor) : DocumentationNode{
+    private fun DocumentationNode.withModifiers(descriptor: DeclarationDescriptor): DocumentationNode {
         if (descriptor is MemberDescriptor) {
             appendVisibility(descriptor)
             if (descriptor !is ConstructorDescriptor) {
@@ -124,15 +120,15 @@ class DocumentationBuilder
         appendTextNode(modifier, NodeKind.Modifier)
     }
 
-    fun DocumentationNode.appendSupertypes(descriptor: ClassDescriptor) {
-        val superTypes = descriptor.typeConstructor.supertypes
-        for (superType in superTypes) {
-            if (!ignoreSupertype(superType)) {
-                appendType(superType, NodeKind.Supertype)
-                val superclass = superType?.constructor?.declarationDescriptor
-                link(superclass, descriptor, RefKind.Inheritor)
-                link(descriptor, superclass, RefKind.Superclass)
-            }
+    fun DocumentationNode.appendSupertype(descriptor: ClassDescriptor, superType: KotlinType) {
+        val unwrappedType = superType.unwrap()
+        if (unwrappedType is AbbreviatedType) {
+            appendSupertype(descriptor, unwrappedType.abbreviation)
+        } else if (!ignoreSupertype(unwrappedType)) {
+            appendType(unwrappedType, NodeKind.Supertype)
+            val superclass = unwrappedType.constructor.declarationDescriptor
+            link(superclass, descriptor, RefKind.Inheritor)
+            link(descriptor, superclass, RefKind.Superclass)
         }
     }
 
@@ -148,8 +144,7 @@ class DocumentationBuilder
     fun DocumentationNode.appendProjection(projection: TypeProjection, kind: NodeKind = NodeKind.Type) {
         if (projection.isStarProjection) {
             appendTextNode("*", NodeKind.Type)
-        }
-        else {
+        } else {
             appendType(projection.type, kind, projection.projectionKind.label)
         }
     }
@@ -157,14 +152,18 @@ class DocumentationBuilder
     fun DocumentationNode.appendType(kotlinType: KotlinType?, kind: NodeKind = NodeKind.Type, prefix: String = "") {
         if (kotlinType == null)
             return
+        if (kotlinType is WrappedType)
+            return appendType(kotlinType.unwrap())
+        if (kotlinType is AbbreviatedType)
+            return appendType(kotlinType.abbreviation)
+
         val classifierDescriptor = kotlinType.constructor.declarationDescriptor
         val name = when (classifierDescriptor) {
             is ClassDescriptor -> {
                 if (classifierDescriptor.isCompanionObject) {
                     classifierDescriptor.containingDeclaration.name.asString() +
                             "." + classifierDescriptor.name.asString()
-                }
-                else {
+                } else {
                     classifierDescriptor.name.asString()
                 }
             }
@@ -182,8 +181,7 @@ class DocumentationBuilder
             val jdkLink = linkResolver.buildJdkLink(classifierDescriptor)
             if (jdkLink != null) {
                 node.append(DocumentationNode(jdkLink, Content.Empty, NodeKind.ExternalLink), RefKind.Link)
-            }
-            else {
+            } else {
                 link(node, classifierDescriptor,
                         if (classifierDescriptor.isBoringBuiltinClass()) RefKind.HiddenLink else RefKind.Link)
             }
@@ -197,7 +195,7 @@ class DocumentationBuilder
     }
 
     fun ClassifierDescriptor.isBoringBuiltinClass(): Boolean =
-        DescriptorUtils.getFqName(this).asString() in boringBuiltinClasses
+            DescriptorUtils.getFqName(this).asString() in boringBuiltinClasses
 
     fun DocumentationNode.appendAnnotations(annotated: Annotated) {
         annotated.annotations.filter { it.isDocumented() }.forEach {
@@ -250,8 +248,7 @@ class DocumentationBuilder
                     link(this, baseDescriptor, inheritedLinkKind)
                 }
                 null
-            }
-            else {
+            } else {
                 val descriptorToUse = if (descriptor is ConstructorDescriptor) descriptor else descriptor.original
                 appendChild(descriptorToUse, RefKind.Member)
             }
@@ -378,9 +375,25 @@ class DocumentationBuilder
         is TypeParameterDescriptor -> build()
         is ValueParameterDescriptor -> build()
         is ReceiverParameterDescriptor -> build()
+        is TypeAliasDescriptor -> build()
         else -> throw IllegalStateException("Descriptor $this is not known")
     }
 
+    fun TypeAliasDescriptor.build(): DocumentationNode {
+        val node = nodeForDescriptor(this, NodeKind.TypeAlias)
+
+        node.appendAnnotations(this)
+        node.appendModifiers(this)
+        node.appendInPageChildren(typeConstructor.parameters, RefKind.Detail)
+
+        node.appendType(underlyingType, NodeKind.TypeAliasUnderlyingType)
+
+        node.appendSourceLink(source)
+
+        register(this, node)
+        return node
+    }
+
     fun ClassDescriptor.build(): DocumentationNode {
         val kind = when {
             kind == ClassKind.OBJECT -> NodeKind.Object
@@ -392,7 +405,9 @@ class DocumentationBuilder
             else -> NodeKind.Class
         }
         val node = nodeForDescriptor(this, kind)
-        node.appendSupertypes(this)
+        typeConstructor.supertypes.forEach {
+            node.appendSupertype(this, it)
+        }
         if (getKind() != ClassKind.OBJECT && getKind() != ClassKind.ENUM_ENTRY) {
             node.appendInPageChildren(typeConstructor.parameters, RefKind.Detail)
             val constructorsToDocument = if (getKind() == ClassKind.ENUM_CLASS)
@@ -572,8 +587,7 @@ class DocumentationBuilder
         var receiverClass: DeclarationDescriptor = type.constructor.declarationDescriptor!!
         if ((receiverClass as? ClassDescriptor)?.isCompanionObject ?: false) {
             receiverClass = receiverClass.containingDeclaration!!
-        }
-        else if (receiverClass is TypeParameterDescriptor) {
+        } else if (receiverClass is TypeParameterDescriptor) {
             val upperBoundClass = receiverClass.upperBounds.singleOrNull()?.constructor?.declarationDescriptor
             if (upperBoundClass != null) {
                 receiverClass = upperBoundClass
@@ -648,11 +662,10 @@ class KotlinPackageDocumentationBuilder : PackageDocumentationBuilder {
 }
 
 class KotlinJavaDocumentationBuilder
-        @Inject constructor(val resolutionFacade: DokkaResolutionFacade,
-                            val documentationBuilder: DocumentationBuilder,
-                            val options: DocumentationOptions,
-                            val logger: DokkaLogger) : JavaDocumentationBuilder
-{
+@Inject constructor(val resolutionFacade: DokkaResolutionFacade,
+                    val documentationBuilder: DocumentationBuilder,
+                    val options: DocumentationOptions,
+                    val logger: DokkaLogger) : JavaDocumentationBuilder {
     override fun appendFile(file: PsiJavaFile, module: DocumentationModule, packageContent: Map<String, Content>) {
         val classDescriptors = file.classes.map {
             val javaDescriptorResolver = resolutionFacade.getFrontendService(JavaDescriptorResolver::class.java)
@@ -731,8 +744,12 @@ fun CallableMemberDescriptor.getExtensionClassDescriptor(): ClassifierDescriptor
     return null
 }
 
-fun DeclarationDescriptor.signature(): String = when(this) {
-    is ClassDescriptor, is PackageFragmentDescriptor, is PackageViewDescriptor  -> DescriptorUtils.getFqName(this).asString()
+fun DeclarationDescriptor.signature(): String = when (this) {
+    is ClassDescriptor,
+    is PackageFragmentDescriptor,
+    is PackageViewDescriptor,
+    is TypeAliasDescriptor -> DescriptorUtils.getFqName(this).asString()
+
     is PropertyDescriptor -> containingDeclaration.signature() + "$" + name + receiverSignature()
     is FunctionDescriptor -> containingDeclaration.signature() + "$" + name + parameterSignature()
     is ValueParameterDescriptor -> containingDeclaration.signature() + "/" + name
diff --git a/core/src/main/kotlin/Kotlin/KotlinLanguageService.kt b/core/src/main/kotlin/Kotlin/KotlinLanguageService.kt
index a0d8f95d..6fcf3772 100644
--- a/core/src/main/kotlin/Kotlin/KotlinLanguageService.kt
+++ b/core/src/main/kotlin/Kotlin/KotlinLanguageService.kt
@@ -315,6 +315,7 @@ class KotlinLanguageService : LanguageService {
             NodeKind.Interface -> keyword("interface ")
             NodeKind.EnumItem -> keyword("enum val ")
             NodeKind.Object -> keyword("object ")
+            NodeKind.TypeAlias -> keyword("typealias ")
             else -> throw IllegalArgumentException("Node $node is not a class-like object")
         }
 
@@ -322,6 +323,13 @@ class KotlinLanguageService : LanguageService {
         renderTypeParametersForNode(node, renderMode)
         renderSupertypesForNode(node, renderMode)
         renderExtraTypeParameterConstraints(node, renderMode)
+
+        if (node.kind == NodeKind.TypeAlias) {
+            nbsp()
+            symbol("=")
+            nbsp()
+            renderType(node.detail(NodeKind.TypeAliasUnderlyingType), renderMode)
+        }
     }
 
     private fun ContentBlock.renderFunction(node: DocumentationNode,
diff --git a/core/src/main/kotlin/Model/DocumentationNode.kt b/core/src/main/kotlin/Model/DocumentationNode.kt
index 9aacbca0..def0f626 100644
--- a/core/src/main/kotlin/Model/DocumentationNode.kt
+++ b/core/src/main/kotlin/Model/DocumentationNode.kt
@@ -13,6 +13,7 @@ enum class NodeKind {
     Exception,
     EnumItem,
     Object,
+    TypeAlias,
 
     Constructor,
     Function,
@@ -30,6 +31,8 @@ enum class NodeKind {
     UpperBound,
     LowerBound,
 
+    TypeAliasUnderlyingType,
+
     Modifier,
     NullabilityModifier,
 
@@ -55,7 +58,7 @@ enum class NodeKind {
     OverloadGroupNote;
 
     companion object {
-        val classLike = setOf(Class, Interface, Enum, AnnotationClass, Exception, Object)
+        val classLike = setOf(Class, Interface, Enum, AnnotationClass, Exception, Object, TypeAlias)
     }
 }