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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
/* __ *\
** ________ ___ / / ___ Scala API **
** / __/ __// _ | / / / _ | (c) 2003-2009, LAMP/EPFL **
** __\ \/ /__/ __ |/ /__/ __ | http://scala-lang.org/ **
** /____/\___/_/ |_/____/_/ | | **
** |/ **
\* */
// $Id$
package scala.xml
import collection.immutable
import collection.SequenceLike
import collection.mutable.{Builder, ListBuffer}
import collection.generic.BuilderFactory
/** This object ...
*
* @author Burak Emir
* @version 1.0
*/
object NodeSeq {
final val Empty = fromSeq(Nil)
def fromSeq(s: Seq[Node]): NodeSeq = new NodeSeq {
def theSeq = s
}
type Coll = NodeSeq
implicit def builderFactory: BuilderFactory[Node, NodeSeq, Coll] = new BuilderFactory[Node, NodeSeq, Coll] { def apply(from: Coll) = newBuilder }
def newBuilder: Builder[Node, NodeSeq] = new ListBuffer[Node] mapResult fromSeq
implicit def seqToNodeSeq(s: Seq[Node]): NodeSeq = fromSeq(s)
}
/** This class implements a wrapper around <code>Seq[Node]</code> that
* adds XPath and comprehension methods.
*
* @author Burak Emir
* @version 1.0
*/
abstract class NodeSeq extends immutable.Sequence[Node] with SequenceLike[Node, NodeSeq] {
import NodeSeq.seqToNodeSeq // import view magic for NodeSeq wrappers
/** Creates a list buffer as builder for this class */
override protected[this] def newBuilder = NodeSeq.newBuilder
def theSeq: Seq[Node]
def length = theSeq.length
override def iterator = theSeq.iterator
def apply(i: Int): Node = theSeq(i)
def apply(f: Node => Boolean): NodeSeq = filter(f)
/** structural equality */
override def equals(x: Any): Boolean = x match {
case z:Node => (length == 1) && z == apply(0)
case z:Seq[_] => sameElements(z)
case z:String => text == z
case _ => false
}
/** Projection function. Similar to XPath, use <code>this \ "foo"</code>
* to get a list of all elements of this sequence that are labelled with
* <code>"foo"</code>. Use <code>\ "_"</code> as a wildcard. Use
* <code>ns \ "@foo"</code> to get the unprefixed attribute "foo".
* Use <code>ns \ "@{uri}foo"</code> to get the prefixed attribute
* "pre:foo" whose prefix "pre" is resolved to the namespace "uri".
* For attribute projections, the resulting NodeSeq attribute values are
* wrapped in a Group.
* There is no support for searching a prefixed attribute by
* its literal prefix.
* The document order is preserved.
*
* @param that ...
* @return ...
*/
def \(that: String): NodeSeq = {
def atResult = {
def fail = throw new IllegalArgumentException(that)
lazy val y = this(0)
val attr =
if (that.length == 1) fail
else if (that(1) == '{') {
val i = that indexOf '}'
if (i == -1) fail
val (uri, key) = (that.substring(2,i), that.substring(i+1, that.length()))
if (uri == "" || key == "") fail
else y.attribute(uri, key)
}
else y.attribute(that.substring(1))
attr match {
case Some(x) => Group(x)
case _ => NodeSeq.Empty
}
}
def makeSeq(cond: (Node) => Boolean) =
NodeSeq fromSeq (this flatMap (_.child) filter cond)
that match {
case "_" => makeSeq(!_.isAtom)
case _ if (that(0) == '@' && this.length == 1) => atResult
case _ => makeSeq(_.label == that)
}
}
/** projection function. Similar to XPath, use <code>this \\ 'foo</code>
* to get a list of all elements of this sequence that are labelled with
* <code>"foo"</code>. Use <code>\\ "_"</code> as a wildcard. Use
* <code>ns \\ "@foo"</code> to get the unprefixed attribute "foo".
* Use <code>ns \\ "@{uri}foo"</code> to get each prefixed attribute
* "pre:foo" whose prefix "pre" is resolved to the namespace "uri".
* For attribute projections, the resulting NodeSeq attribute values are
* wrapped in a Group.
* There is no support for searching a prefixed attribute by
* its literal prefix.
* The document order is preserved.
*
* @param that ...
* @return ...
*/
def \\ (that: String): NodeSeq = {
def filt(cond: (Node) => Boolean) = this flatMap (_.descendant_or_self) filter cond
that match {
case "_" => filt(!_.isAtom)
case _ if that(0) == '@' => filt(!_.isAtom) flatMap (_ \ that)
case _ => filt(x => !x.isAtom && x.label == that)
}
}
override def toString(): String = theSeq.mkString
def text: String = this map (_.text) mkString
}
|