From 4a53e468fe4bddc3bafa73625ced86c337d7ef9e Mon Sep 17 00:00:00 2001
From: James Strachan <james.strachan@gmail.com>
Date: Sat, 3 Mar 2012 06:44:42 +0000
Subject: [PATCH] render extension functions nicely in kdoc too

---
 .../org/jetbrains/kotlin/doc/templates/KDocTemplate.kt     | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/kdoc/src/main/kotlin/org/jetbrains/kotlin/doc/templates/KDocTemplate.kt b/kdoc/src/main/kotlin/org/jetbrains/kotlin/doc/templates/KDocTemplate.kt
index d4a9cb7dca2..8f4237e40e3 100644
--- a/kdoc/src/main/kotlin/org/jetbrains/kotlin/doc/templates/KDocTemplate.kt
+++ b/kdoc/src/main/kotlin/org/jetbrains/kotlin/doc/templates/KDocTemplate.kt
@@ -74,19 +74,20 @@ abstract class KDocTemplate() : TextTemplate() {
         val arguments = t.arguments
         return if (c != null) {
             val prefix = if (c.isAnnotation()) "@" else ""
-            if (c.name.startsWith("jet.Function") && arguments.notEmpty()) {
+            val cname = c.name
+            if ((cname.startsWith("jet.Function") || cname.startsWith("jet.ExtensionFunction")) && arguments.notEmpty()) {
                 val rt = arguments.last()
                 // TODO use drop()
                 val rest = arguments.subList(0, arguments.size - 1).notNull()
                 "${typeArguments(rest, "(", ")", "()")}&nbsp;<A HREF=\"${href(c)}\" title=\"${c.kind} in ${c.packageName}\">-&gt;</a>&nbsp;${link(rt)}"
-            } else if (c.name.startsWith("jet.Tuple")) {
+            } else if (cname.startsWith("jet.Tuple")) {
                 if (arguments.isEmpty()) {
                     "Unit"
                 } else {
                     "<A HREF=\"${href(c)}\" title=\"${c.kind} in ${c.packageName}\">#</a>${typeArguments(arguments, "(", ")", "()")}"
                 }
             } else {
-                val name = if (fullName) c.name else c.simpleName
+                val name = if (fullName) cname else c.simpleName
                 "<A HREF=\"${href(c)}\" title=\"${c.kind} in ${c.packageName}\">$prefix$name</A>${typeArguments(arguments)}"
             }
         } else {