Process Modality (final, open, abstract) and Visibility (public, protected, internal, private)

1 files changed, 47 insertions, 25 deletions

diff --git a/src/Model/DocumentationNodeBuilder.kt b/src/Model/DocumentationNodeBuilder.kt
index 60f9ca0d..1f553f33 100644
--- a/src/Model/DocumentationNodeBuilder.kt
+++ b/src/Model/DocumentationNodeBuilder.kt
@@ -1,41 +1,59 @@
 package org.jetbrains.dokka
 
 import org.jetbrains.jet.lang.resolve.*
-import org.jetbrains.jet.lang.psi.*
 import org.jetbrains.jet.lang.descriptors.*
 import org.jetbrains.jet.lang.descriptors.impl.*
 import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns
 
 class DocumentationNodeBuilder(val context: BindingContext) : DeclarationDescriptorVisitorEmptyBodies<DocumentationNode, DocumentationNode>() {
 
+    fun reference(from: DocumentationNode, to: DocumentationNode, kind: DocumentationReference.Kind) {
+        from.addReferenceTo(to, kind)
+        if (kind == DocumentationReference.Kind.Link)
+            to.addReferenceTo(from, DocumentationReference.Kind.Link)
+        else
+            to.addReferenceTo(from, DocumentationReference.Kind.Owner)
+    }
+
+    fun addModality(descriptor: MemberDescriptor, data: DocumentationNode): DocumentationNode {
+        val modifier = descriptor.getModality().name().toLowerCase()
+        val node = DocumentationNode(descriptor, modifier, DocumentationContent.Empty, DocumentationNode.Kind.Modifier)
+        reference(data, node, DocumentationReference.Kind.Detail)
+        return node
+    }
+
+    fun addVisibility(descriptor: MemberDescriptor, data: DocumentationNode): DocumentationNode {
+        val modifier = descriptor.getVisibility().toString()
+        val node = DocumentationNode(descriptor, modifier, DocumentationContent.Empty, DocumentationNode.Kind.Modifier)
+        reference(data, node, DocumentationReference.Kind.Detail)
+        return node
+    }
+
     override fun visitDeclarationDescriptor(descriptor: DeclarationDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val doc = context.getDocumentation(descriptor!!)
         val node = DocumentationNode(descriptor, descriptor.getName().asString(), doc, DocumentationNode.Kind.Unknown)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Link)
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
+        reference(data!!, node, DocumentationReference.Kind.Link)
         return node
     }
 
     override fun visitReceiverParameterDescriptor(descriptor: ReceiverParameterDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val node = DocumentationNode(descriptor!!, descriptor.getName().asString(), DocumentationContent.Empty, DocumentationNode.Kind.Receiver)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Detail)
+        reference(data!!, node, DocumentationReference.Kind.Detail)
 
         val typeNode = DocumentationNode(descriptor, descriptor.getType().toString(), DocumentationContent.Empty, DocumentationNode.Kind.Type)
-        node.addReferenceTo(typeNode, DocumentationReference.Kind.Detail)
+        reference(node, typeNode, DocumentationReference.Kind.Detail)
 
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
         return node
     }
 
     override fun visitValueParameterDescriptor(descriptor: ValueParameterDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val doc = context.getDocumentation(descriptor!!)
         val node = DocumentationNode(descriptor, descriptor.getName().asString(), doc, DocumentationNode.Kind.Parameter)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Detail)
+        reference(data!!, node, DocumentationReference.Kind.Detail)
 
         val typeNode = DocumentationNode(descriptor, descriptor.getType().toString(), DocumentationContent.Empty, DocumentationNode.Kind.Type)
-        node.addReferenceTo(typeNode, DocumentationReference.Kind.Detail)
+        reference(node, typeNode, DocumentationReference.Kind.Detail)
 
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
         return node
     }
 
@@ -46,68 +64,72 @@ class DocumentationNodeBuilder(val context: BindingContext) : DeclarationDescrip
             ClassKind.TRAIT -> DocumentationNode.Kind.Interface
             else -> DocumentationNode.Kind.Class
         })
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Member)
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
+        reference(data!!, node, DocumentationReference.Kind.Member)
+        addModality(descriptor, node)
+        addVisibility(descriptor, node)
         return node
     }
 
     override fun visitFunctionDescriptor(descriptor: FunctionDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val doc = context.getDocumentation(descriptor!!)
         val node = DocumentationNode(descriptor, descriptor.getName().asString(), doc, DocumentationNode.Kind.Function)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Member)
+        reference(data!!, node, DocumentationReference.Kind.Member)
 
         val typeNode = DocumentationNode(descriptor, descriptor.getReturnType().toString(), DocumentationContent.Empty, DocumentationNode.Kind.Type)
-        node.addReferenceTo(typeNode, DocumentationReference.Kind.Detail)
+        reference(node, typeNode, DocumentationReference.Kind.Detail)
+
+        addModality(descriptor, node)
+        addVisibility(descriptor, node)
 
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
         return node
     }
 
     override fun visitTypeParameterDescriptor(descriptor: TypeParameterDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val doc = context.getDocumentation(descriptor!!)
         val node = DocumentationNode(descriptor, descriptor.getName().asString(), doc, DocumentationNode.Kind.TypeParameter)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Detail)
+        reference(data!!, node, DocumentationReference.Kind.Detail)
         val builtIns = KotlinBuiltIns.getInstance()
         for (constraint in descriptor.getUpperBounds()) {
             if (constraint == builtIns.getDefaultBound())
                 continue
             val constraintNode = DocumentationNode(descriptor, constraint.toString(), DocumentationContent.Empty, DocumentationNode.Kind.UpperBound)
-            node.addReferenceTo(constraintNode, DocumentationReference.Kind.Detail)
+            reference(node, constraintNode, DocumentationReference.Kind.Detail)
         }
         for (constraint in descriptor.getLowerBounds()) {
             if (builtIns.isNothing(constraint))
                 continue
             val constraintNode = DocumentationNode(descriptor, constraint.toString(), DocumentationContent.Empty, DocumentationNode.Kind.LowerBound)
-            node.addReferenceTo(constraintNode, DocumentationReference.Kind.Detail)
+            reference(node, constraintNode, DocumentationReference.Kind.Detail)
         }
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
         return node
     }
 
     override fun visitPropertyDescriptor(descriptor: PropertyDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val doc = context.getDocumentation(descriptor!!)
         val node = DocumentationNode(descriptor, descriptor.getName().asString(), doc, DocumentationNode.Kind.Property)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Member)
+        reference(data!!, node, DocumentationReference.Kind.Member)
 
         val typeNode = DocumentationNode(descriptor, descriptor.getType().toString(), DocumentationContent.Empty, DocumentationNode.Kind.Type)
-        node.addReferenceTo(typeNode, DocumentationReference.Kind.Detail)
+        reference(node, typeNode, DocumentationReference.Kind.Detail)
 
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
+        addModality(descriptor, node)
+        addVisibility(descriptor, node)
         return node
     }
 
     override fun visitConstructorDescriptor(descriptor: ConstructorDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val doc = context.getDocumentation(descriptor!!)
         val node = DocumentationNode(descriptor, descriptor.getName().asString(), doc, DocumentationNode.Kind.Constructor)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Member)
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
+        reference(data!!, node, DocumentationReference.Kind.Member)
+
+        addVisibility(descriptor, node)
+
         return node
     }
 
     override fun visitPackageFragmentDescriptor(descriptor: PackageFragmentDescriptor?, data: DocumentationNode?): DocumentationNode? {
         val node = DocumentationNode(descriptor!!, descriptor.fqName.asString(), DocumentationContent.Empty, DocumentationNode.Kind.Package)
-        data!!.addReferenceTo(node, DocumentationReference.Kind.Member)
-        node.addReferenceTo(data, DocumentationReference.Kind.Owner)
+        reference(data!!, node, DocumentationReference.Kind.Member)
         return node
     }
 }