src/compiler/scala/tools/nsc/interpreter/XMLCompletion.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

/* NSC -- new Scala compiler
 * Copyright 2005-2010 LAMP/EPFL
 * @author Paul Phillips
 */

package scala.tools.nsc
package interpreter

import xml.{ XML, Group, Node, NodeSeq }
import XMLCompletion._
import scala.collection.{ mutable, immutable }

class XMLCompletion(root: Node) extends CompletionAware {
  private val nodeCache = new mutable.HashMap[String, Node]
  private def getNode(s: String): Option[Node] = {
    completions // make sure cache is populated
    nodeCache get s
  }

  lazy val completions: List[String] = {
    def children = root.child.toList
    def uniqueTags = children groupBy (_.label) filter (_._2.size == 1) map (_._1)
    val uniqs = uniqueTags.toList

    children.foldLeft(List[String]())((res, node) => {
      val name = node.label
      def count = res filter (_ startsWith (name + "[")) size  // ]
      val suffix = if (uniqs contains name) "" else "[%d]" format (count + 1)
      val s = name + suffix

      nodeCache(s) = node

      s :: res
    }).sorted
  }
  def completions(verbosity: Int) = completions

  override def execute(id: String)  = getNode(id)
  override def follow(id: String)   = getNode(id) map (x => new XMLCompletion(x))
}

object XMLCompletion {
  def apply(x: Node) = new XMLCompletion(x)
}