[scaladoc] JavaScript template search uses a st...

[scaladoc] JavaScript template search uses a statically built index to
considerably increase performance. Contributed by Kato Kazuyoshi. Review
by dubochet.

1 files changed, 60 insertions, 16 deletions

diff --git a/src/compiler/scala/tools/nsc/doc/html/page/Index.scala b/src/compiler/scala/tools/nsc/doc/html/page/Index.scala
index 09da71fc1f..9a964ee30f 100644
--- a/src/compiler/scala/tools/nsc/doc/html/page/Index.scala
+++ b/src/compiler/scala/tools/nsc/doc/html/page/Index.scala
@@ -47,6 +47,21 @@ class Index(universe: Universe, indexModel: IndexModelFactory#IndexModel) extend
       </div>
     </body>
 
+
+  def isExcluded(dtpl: DocTemplateEntity) = {
+    val qname = dtpl.qualifiedName
+    ( ( qname.startsWith("scala.Tuple") || qname.startsWith("scala.Product") ||
+       qname.startsWith("scala.Function") || qname.startsWith("scala.runtime.AbstractFunction")
+     ) && !(
+      qname == "scala.Tuple1" || qname == "scala.Tuple2" ||
+      qname == "scala.Product" || qname == "scala.Product1" || qname == "scala.Product2" ||
+      qname == "scala.Function" || qname == "scala.Function1" || qname == "scala.Function2" ||
+      qname == "scala.runtime.AbstractFunction0" || qname == "scala.runtime.AbstractFunction1" ||
+      qname == "scala.runtime.AbstractFunction2"
+    )
+   )
+  }
+
   def browser =
 	<xml:group>
     <div id="browser" class="ui-layout-west">
@@ -61,19 +76,6 @@ class Index(universe: Universe, indexModel: IndexModelFactory#IndexModel) extend
       <div class="ui-west-center">
       <div id="filter"></div>
       <div class="pack" id="tpl">{
-        def isExcluded(dtpl: DocTemplateEntity) = {
-          val qname = dtpl.qualifiedName
-          ( ( qname.startsWith("scala.Tuple") || qname.startsWith("scala.Product") ||
-              qname.startsWith("scala.Function") || qname.startsWith("scala.runtime.AbstractFunction")
-            ) && !(
-              qname == "scala.Tuple1" || qname == "scala.Tuple2" ||
-              qname == "scala.Product" || qname == "scala.Product1" || qname == "scala.Product2" ||
-              qname == "scala.Function" || qname == "scala.Function1" || qname == "scala.Function2" ||
-              qname == "scala.runtime.AbstractFunction0" || qname == "scala.runtime.AbstractFunction1" ||
-              qname == "scala.runtime.AbstractFunction2"
-            )
-          )
-        }
         def packageElem(pack: model.Package): NodeSeq = {
           <xml:group>
             { if (!pack.isRootPackage)
@@ -116,22 +118,64 @@ class Index(universe: Universe, indexModel: IndexModelFactory#IndexModel) extend
                     placeholderSeq ++ createLink(entry, includePlaceholder = false, includeText = true)
                 }
 
-                <li title={ entities.head.qualifiedName }>{
+                <li id={
+                  "template-" + toId(entities.head.qualifiedName)
+                } title={ entities.head.qualifiedName }>{
                   itemContents
                 }</li>
               }
             }</ol>
             <ol class="packages"> {
               for (sp <- pack.packages sortBy (_.name.toLowerCase)) yield
-                <li class="pack" title={ sp.qualifiedName }>{ packageElem(sp) }</li>
+                <li id={
+                  "package-" + toId(sp.qualifiedName)
+                } class="pack" title={ sp.qualifiedName }>{ packageElem(sp) }</li>
             }</ol>
           </xml:group>
         }
         packageElem(universe.rootPackage)
-      }</div></div>
+      }</div></div>{ scriptElement }
     </div>
     </xml:group>
 
+  def toId(str: String) = {
+    val pattern = "[^A-Za-z0-9-]".r
+    pattern.replaceSomeIn(str, (m: scala.util.matching.Regex.Match) => {
+      Some("-" + m.group(0).charAt(0).toInt)
+    })
+  }
+
+  def scriptElement = {
+    val templatesOf = allPackagesWithTemplates
+
+    val elements = templatesOf.keys.map(pack => {
+      List(
+        "{ name: '", pack, "', children: ",
+        templatesOf(pack).map(t => "'" + t + "'").mkString("[", ",", "]"),
+        "}"
+      ).mkString("")
+    })
+
+    <script type="text/javascript">{
+      elements.mkString("Index.PACKAGES = [", ",", "]")
+    }</script>
+  }
+
+  def allPackagesWithTemplates: Map[Package, List[DocTemplateEntity]] = {
+    Map(allPackages.map((key) => {
+      key -> key.templates.filter(t => !t.isPackage && !isExcluded(t))
+    }) : _*)
+  }
+
+  def allPackages: List[Package] = {
+    def f(parent: Package): List[Package] = {
+      parent.packages.flatMap(
+        p => f(p) :+ p
+      )
+    }
+    f(universe.rootPackage).sortBy(_.toString)
+  }
+
   def packageQualifiedName(ety: DocTemplateEntity): String =
     if (ety.inTemplate.isPackage) ety.name else (packageQualifiedName(ety.inTemplate) + "." + ety.name)