632 lines
18 KiB
Go
632 lines
18 KiB
Go
|
package xpath
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var html *TNode = example()
|
||
|
|
|
||
|
|
func TestCompile(t *testing.T) {
|
||
|
|
var err error
|
||
|
|
_, err = Compile("//a")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("//a should be correct but got error %s", err)
|
||
|
|
}
|
||
|
|
_, err = Compile("//a[id=']/span")
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("//a[id=] should be got correct but is nil")
|
||
|
|
}
|
||
|
|
_, err = Compile("//ul/li/@class")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("//ul/li/@class should be correct but got error %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSelf(t *testing.T) {
|
||
|
|
testXPath(t, html, ".", "html")
|
||
|
|
testXPath(t, html.FirstChild, ".", "head")
|
||
|
|
testXPath(t, html, "self::*", "html")
|
||
|
|
testXPath(t, html.LastChild, "self::body", "body")
|
||
|
|
testXPath2(t, html, "//body/./ul/li/a", 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParent(t *testing.T) {
|
||
|
|
testXPath(t, html.LastChild, "..", "html")
|
||
|
|
testXPath(t, html.LastChild, "parent::*", "html")
|
||
|
|
a := selectNode(html, "//li/a")
|
||
|
|
testXPath(t, a, "parent::*", "li")
|
||
|
|
testXPath(t, html, "//title/parent::head", "head")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAttribute(t *testing.T) {
|
||
|
|
testXPath(t, html, "@lang='en'", "html")
|
||
|
|
testXPath2(t, html, "@lang='zh'", 0)
|
||
|
|
testXPath2(t, html, "//@href", 3)
|
||
|
|
testXPath2(t, html, "//a[@*]", 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRelativePath(t *testing.T) {
|
||
|
|
testXPath(t, html, "head", "head")
|
||
|
|
testXPath(t, html, "/head", "head")
|
||
|
|
testXPath(t, html, "body//li", "li")
|
||
|
|
testXPath(t, html, "/head/title", "title")
|
||
|
|
|
||
|
|
testXPath2(t, html, "/body/ul/li/a", 3)
|
||
|
|
testXPath(t, html, "//title", "title")
|
||
|
|
testXPath(t, html, "//title/..", "head")
|
||
|
|
testXPath(t, html, "//title/../..", "html")
|
||
|
|
testXPath2(t, html, "//a[@href]", 3)
|
||
|
|
testXPath(t, html, "//ul/../footer", "footer")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestChild(t *testing.T) {
|
||
|
|
testXPath(t, html, "/child::head", "head")
|
||
|
|
testXPath(t, html, "/child::head/child::title", "title")
|
||
|
|
testXPath(t, html, "//title/../child::title", "title")
|
||
|
|
testXPath(t, html.Parent, "//child::*", "html")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDescendant(t *testing.T) {
|
||
|
|
testXPath2(t, html, "descendant::*", 15)
|
||
|
|
testXPath2(t, html, "/head/descendant::*", 2)
|
||
|
|
testXPath2(t, html, "//ul/descendant::*", 7) // <li> + <a>
|
||
|
|
testXPath2(t, html, "//ul/descendant::li", 4) // <li>
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAncestor(t *testing.T) {
|
||
|
|
testXPath2(t, html, "/body/footer/ancestor::*", 2) // body>html
|
||
|
|
testXPath2(t, html, "/body/ul/li/a/ancestor::li", 3)
|
||
|
|
testXPath2(t, html, "/body/ul/li/a/ancestor-or-self::li", 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFollowingSibling(t *testing.T) {
|
||
|
|
var list []*TNode
|
||
|
|
list = selectNodes(html, "//li/following-sibling::*")
|
||
|
|
for _, n := range list {
|
||
|
|
if n.Data != "li" {
|
||
|
|
t.Fatalf("expected node is li,but got:%s", n.Data)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
list = selectNodes(html, "//ul/following-sibling::*") // p,footer
|
||
|
|
for _, n := range list {
|
||
|
|
if n.Data != "p" && n.Data != "footer" {
|
||
|
|
t.Fatal("expected node is not one of the following nodes: [p,footer]")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
testXPath(t, html, "//ul/following-sibling::footer", "footer")
|
||
|
|
list = selectNodes(html, "//h1/following::*") // ul>li>a,p,footer
|
||
|
|
if list[0].Data != "ul" {
|
||
|
|
t.Fatal("expected node is not ul")
|
||
|
|
}
|
||
|
|
if list[1].Data != "li" {
|
||
|
|
t.Fatal("expected node is not li")
|
||
|
|
}
|
||
|
|
if list[len(list)-1].Data != "footer" {
|
||
|
|
t.Fatal("expected node is not footer")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPrecedingSibling(t *testing.T) {
|
||
|
|
testXPath(t, html, "/body/footer/preceding-sibling::*", "p")
|
||
|
|
testXPath2(t, html, "/body/footer/preceding-sibling::*", 3) // p,ul,h1
|
||
|
|
list := selectNodes(html, "//h1/preceding::*") // head>title>meta
|
||
|
|
if list[0].Data != "head" {
|
||
|
|
t.Fatal("expected is not head")
|
||
|
|
}
|
||
|
|
if list[1].Data != "title" {
|
||
|
|
t.Fatal("expected is not title")
|
||
|
|
}
|
||
|
|
if list[2].Data != "meta" {
|
||
|
|
t.Fatal("expected is not meta")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStarWide(t *testing.T) {
|
||
|
|
testXPath(t, html, "/head/*", "title")
|
||
|
|
testXPath2(t, html, "//ul/*", 4)
|
||
|
|
testXPath(t, html, "@*", "html")
|
||
|
|
testXPath2(t, html, "/body/h1/*", 0)
|
||
|
|
testXPath2(t, html, `//ul/*/a`, 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNodeTestType(t *testing.T) {
|
||
|
|
testXPath(t, html, "//title/text()", "Hello")
|
||
|
|
testXPath(t, html, "//a[@href='/']/text()", "Home")
|
||
|
|
testXPath2(t, html, "//head/node()", 2)
|
||
|
|
testXPath2(t, html, "//ul/node()", 4)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPosition(t *testing.T) {
|
||
|
|
testXPath3(t, html, "/head[1]", html.FirstChild) // compare to 'head' element
|
||
|
|
ul := selectNode(html, "//ul")
|
||
|
|
testXPath3(t, html, "/head[last()]", html.FirstChild)
|
||
|
|
testXPath3(t, html, "//li[1]", ul.FirstChild)
|
||
|
|
testXPath3(t, html, "//li[4]", ul.LastChild)
|
||
|
|
testXPath3(t, html, "//li[last()]", ul.LastChild)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPredicate(t *testing.T) {
|
||
|
|
testXPath(t, html.Parent, "html[@lang='en']", "html")
|
||
|
|
testXPath(t, html, "//a[@href='/']", "a")
|
||
|
|
testXPath(t, html, "//meta[@name]", "meta")
|
||
|
|
ul := selectNode(html, "//ul")
|
||
|
|
testXPath3(t, html, "//li[position()=4]", ul.LastChild)
|
||
|
|
testXPath3(t, html, "//li[position()=1]", ul.FirstChild)
|
||
|
|
testXPath2(t, html, "//li[position()>0]", 4)
|
||
|
|
testXPath3(t, html, "//a[text()='Home']", selectNode(html, "//a[1]"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOr_And(t *testing.T) {
|
||
|
|
list := selectNodes(html, "//h1|//footer")
|
||
|
|
if len(list) == 0 {
|
||
|
|
t.Fatal("//h1|//footer no any node found")
|
||
|
|
}
|
||
|
|
if list[0].Data != "h1" {
|
||
|
|
t.Fatalf("expected first node of node-set is h1,but got %s", list[0].Data)
|
||
|
|
}
|
||
|
|
if list[1].Data != "footer" {
|
||
|
|
t.Fatalf("expected first node of node-set is footer,but got %s", list[1].Data)
|
||
|
|
}
|
||
|
|
|
||
|
|
list = selectNodes(html, "//a[@id=1 or @id=2]")
|
||
|
|
if list[0] != selectNode(html, "//a[@id=1]") {
|
||
|
|
t.Fatal("node is not equal")
|
||
|
|
}
|
||
|
|
if list[1] != selectNode(html, "//a[@id=2]") {
|
||
|
|
t.Fatal("node is not equal")
|
||
|
|
}
|
||
|
|
list = selectNodes(html, "//a[@id or @href]")
|
||
|
|
if list[0] != selectNode(html, "//a[@id=1]") {
|
||
|
|
t.Fatal("node is not equal")
|
||
|
|
}
|
||
|
|
if list[1] != selectNode(html, "//a[@id=2]") {
|
||
|
|
t.Fatal("node is not equal")
|
||
|
|
}
|
||
|
|
testXPath3(t, html, "//a[@id=1 and @href='/']", selectNode(html, "//a[1]"))
|
||
|
|
testXPath3(t, html, "//a[text()='Home' and @id='1']", selectNode(html, "//a[1]"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFunction(t *testing.T) {
|
||
|
|
testEval(t, html, "boolean(//*[@id])", true)
|
||
|
|
testEval(t, html, "boolean(//*[@x])", false)
|
||
|
|
testEval(t, html, "name(//title)", "title")
|
||
|
|
testXPath2(t, html, "//*[name()='a']", 3)
|
||
|
|
testXPath(t, html, "//*[starts-with(name(),'h1')]", "h1")
|
||
|
|
testXPath(t, html, "//*[ends-with(name(),'itle')]", "title") // Head title
|
||
|
|
testXPath2(t, html, "//*[contains(@href,'a')]", 2)
|
||
|
|
testXPath2(t, html, "//*[starts-with(@href,'/a')]", 2) // a links: `/account`,`/about`
|
||
|
|
testXPath2(t, html, "//*[ends-with(@href,'t')]", 2) // a links: `/account`,`/about`
|
||
|
|
testXPath3(t, html, "//h1[normalize-space(text())='This is a H1']", selectNode(html, "//h1"))
|
||
|
|
testXPath3(t, html, "//title[substring(.,1)='Hello']", selectNode(html, "//title"))
|
||
|
|
testXPath3(t, html, "//title[substring(text(),1,4)='Hell']", selectNode(html, "//title"))
|
||
|
|
testXPath3(t, html, "//title[substring(self::*,1,4)='Hell']", selectNode(html, "//title"))
|
||
|
|
testXPath2(t, html, "//title[substring(child::*,1)]", 0) // Here substring return boolen (false), should it?
|
||
|
|
testXPath2(t, html, "//title[substring(child::*,1) = '']", 1)
|
||
|
|
testXPath3(t, html, "//li[not(a)]", selectNode(html, "//ul/li[4]"))
|
||
|
|
testXPath2(t, html, "//li/a[not(@id='1')]", 2) // //li/a[@id!=1]
|
||
|
|
testXPath2(t, html, "//h1[string-length(normalize-space(' abc ')) = 3]", 1)
|
||
|
|
testXPath2(t, html, "//h1[string-length(normalize-space(self::text())) = 12]", 1)
|
||
|
|
testXPath2(t, html, "//title[string-length(normalize-space(child::*)) = 0]", 1)
|
||
|
|
testXPath2(t, html, "//title[string-length(self::text()) = 5]", 1) // Hello = 5
|
||
|
|
testXPath2(t, html, "//title[string-length(child::*) = 5]", 0)
|
||
|
|
testXPath2(t, html, "//ul[count(li)=4]", 1)
|
||
|
|
testEval(t, html, "true()", true)
|
||
|
|
testEval(t, html, "false()", false)
|
||
|
|
testEval(t, html, "boolean(0)", false)
|
||
|
|
testEval(t, html, "boolean(1)", true)
|
||
|
|
testEval(t, html, "sum(1+2)", float64(3))
|
||
|
|
testEval(t, html, "string(sum(1+2))", "3")
|
||
|
|
testEval(t, html, "sum(1.1+2)", float64(3.1))
|
||
|
|
testEval(t, html, "sum(//a/@id)", float64(6)) // 1+2+3
|
||
|
|
testEval(t, html, `concat("1","2","3")`, "123")
|
||
|
|
testEval(t, html, `concat(" ",//a[@id='1']/@href," ")`, " / ")
|
||
|
|
testEval(t, html, "ceiling(5.2)", float64(6))
|
||
|
|
testEval(t, html, "floor(5.2)", float64(5))
|
||
|
|
testEval(t, html, `substring-before('aa-bb','-')`, "aa")
|
||
|
|
testEval(t, html, `substring-before('aa-bb','a')`, "")
|
||
|
|
testEval(t, html, `substring-before('aa-bb','b')`, "aa-")
|
||
|
|
testEval(t, html, `substring-before('aa-bb','q')`, "")
|
||
|
|
testEval(t, html, `substring-after('aa-bb','-')`, "bb")
|
||
|
|
testEval(t, html, `substring-after('aa-bb','a')`, "a-bb")
|
||
|
|
testEval(t, html, `substring-after('aa-bb','b')`, "b")
|
||
|
|
testEval(t, html, `substring-after('aa-bb','q')`, "")
|
||
|
|
testEval(t, html,
|
||
|
|
`translate('The quick brown fox.', 'abcdefghijklmnopqrstuvwxyz', 'ABCDEFGHIJKLMNOPQRSTUVWXYZ')`,
|
||
|
|
"THE QUICK BROWN FOX.",
|
||
|
|
)
|
||
|
|
testEval(t, html,
|
||
|
|
`translate('The quick brown fox.', 'brown', 'red')`,
|
||
|
|
"The quick red fdx.",
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPanic(t *testing.T) {
|
||
|
|
// starts-with
|
||
|
|
assertPanic(t, func() { testXPath(t, html, "//*[starts-with(0, 0)]", "") })
|
||
|
|
assertPanic(t, func() { testXPath(t, html, "//*[starts-with(name(), 0)]", "") })
|
||
|
|
//ends-with
|
||
|
|
assertPanic(t, func() { testXPath(t, html, "//*[ends-with(0, 0)]", "") })
|
||
|
|
assertPanic(t, func() { testXPath(t, html, "//*[ends-with(name(), 0)]", "") })
|
||
|
|
// contains
|
||
|
|
assertPanic(t, func() { testXPath2(t, html, "//*[contains(0, 0)]", 0) })
|
||
|
|
assertPanic(t, func() { testXPath2(t, html, "//*[contains(@href, 0)]", 0) })
|
||
|
|
// sum
|
||
|
|
assertPanic(t, func() { testXPath3(t, html, "//title[sum('Hello') = 0]", nil) })
|
||
|
|
// substring
|
||
|
|
assertPanic(t, func() { testXPath3(t, html, "//title[substring(.,'')=0]", nil) })
|
||
|
|
assertPanic(t, func() { testXPath3(t, html, "//title[substring(.,4,'')=0]", nil) })
|
||
|
|
assertPanic(t, func() { testXPath3(t, html, "//title[substring(.,4,4)=0]", nil) })
|
||
|
|
//assertPanic(t, func() { testXPath2(t, html, "//title[substring(child::*,0) = '']", 0) }) // Here substring return boolen (false), should it?
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func assertPanic(t *testing.T, f func()) {
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r == nil {
|
||
|
|
t.Errorf("The code did not panic")
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
f()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEvaluate(t *testing.T) {
|
||
|
|
testEval(t, html, "count(//ul/li)", float64(4))
|
||
|
|
testEval(t, html, "//html/@lang", []string{"en"})
|
||
|
|
testEval(t, html, "//title/text()", []string{"Hello"})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOperationOrLogical(t *testing.T) {
|
||
|
|
testXPath3(t, html, "//li[1+1]", selectNode(html, "//li[2]"))
|
||
|
|
testXPath3(t, html, "//li[5 div 2]", selectNode(html, "//li[2]"))
|
||
|
|
testXPath3(t, html, "//li[3 mod 2]", selectNode(html, "//li[1]"))
|
||
|
|
testXPath3(t, html, "//li[3 - 2]", selectNode(html, "//li[1]"))
|
||
|
|
testXPath2(t, html, "//li[position() mod 2 = 0 ]", 2) // //li[2],li[4]
|
||
|
|
testXPath2(t, html, "//a[@id>=1]", 3) // //a[@id>=1] == a[1],a[2],a[3]
|
||
|
|
testXPath2(t, html, "//a[@id<=2]", 2) // //a[@id<=2] == a[1],a[1]
|
||
|
|
testXPath2(t, html, "//a[@id<2]", 1) // //a[@id>=1] == a[1]
|
||
|
|
testXPath2(t, html, "//a[@id!=2]", 2) // //a[@id>=1] == a[1],a[3]
|
||
|
|
testXPath2(t, html, "//a[@id=1 or @id=3]", 2) // //a[@id>=1] == a[1],a[3]
|
||
|
|
testXPath3(t, html, "//a[@id=1 and @href='/']", selectNode(html, "//a[1]"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func testEval(t *testing.T, root *TNode, expr string, expected interface{}) {
|
||
|
|
v := MustCompile(expr).Evaluate(createNavigator(root))
|
||
|
|
if it, ok := v.(*NodeIterator); ok {
|
||
|
|
exp, ok := expected.([]string)
|
||
|
|
if !ok {
|
||
|
|
t.Fatalf("expected value, got: %#v", v)
|
||
|
|
}
|
||
|
|
got := iterateNavs(it)
|
||
|
|
if len(exp) != len(got) {
|
||
|
|
t.Fatalf("expected: %#v, got: %#v", exp, got)
|
||
|
|
}
|
||
|
|
for i, n1 := range exp {
|
||
|
|
n2 := got[i]
|
||
|
|
if n1 != n2.Value() {
|
||
|
|
t.Fatalf("expected: %#v, got: %#v", n1, n2)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if v != expected {
|
||
|
|
t.Fatalf("expected: %#v, got: %#v", expected, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testXPath(t *testing.T, root *TNode, expr string, expected string) {
|
||
|
|
node := selectNode(root, expr)
|
||
|
|
if node == nil {
|
||
|
|
t.Fatalf("`%s` returns node is nil", expr)
|
||
|
|
}
|
||
|
|
if node.Data != expected {
|
||
|
|
t.Fatalf("`%s` expected node is %s,but got %s", expr, expected, node.Data)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testXPath2(t *testing.T, root *TNode, expr string, expected int) {
|
||
|
|
list := selectNodes(root, expr)
|
||
|
|
if len(list) != expected {
|
||
|
|
t.Fatalf("`%s` expected node numbers is %d,but got %d", expr, expected, len(list))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testXPath3(t *testing.T, root *TNode, expr string, expected *TNode) {
|
||
|
|
node := selectNode(root, expr)
|
||
|
|
if node == nil {
|
||
|
|
t.Fatalf("`%s` returns node is nil", expr)
|
||
|
|
}
|
||
|
|
if node != expected {
|
||
|
|
t.Fatalf("`%s` %s != %s", expr, node.Value(), expected.Value())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func iterateNavs(t *NodeIterator) []*TNodeNavigator {
|
||
|
|
var nodes []*TNodeNavigator
|
||
|
|
for t.MoveNext() {
|
||
|
|
node := t.Current().(*TNodeNavigator)
|
||
|
|
nodes = append(nodes, node)
|
||
|
|
}
|
||
|
|
return nodes
|
||
|
|
}
|
||
|
|
|
||
|
|
func iterateNodes(t *NodeIterator) []*TNode {
|
||
|
|
var nodes []*TNode
|
||
|
|
for t.MoveNext() {
|
||
|
|
node := (t.Current().(*TNodeNavigator)).curr
|
||
|
|
nodes = append(nodes, node)
|
||
|
|
}
|
||
|
|
return nodes
|
||
|
|
}
|
||
|
|
|
||
|
|
func selectNode(root *TNode, expr string) (n *TNode) {
|
||
|
|
t := Select(createNavigator(root), expr)
|
||
|
|
if t.MoveNext() {
|
||
|
|
n = (t.Current().(*TNodeNavigator)).curr
|
||
|
|
}
|
||
|
|
return n
|
||
|
|
}
|
||
|
|
|
||
|
|
func selectNodes(root *TNode, expr string) []*TNode {
|
||
|
|
t := Select(createNavigator(root), expr)
|
||
|
|
return iterateNodes(t)
|
||
|
|
}
|
||
|
|
|
||
|
|
func createNavigator(n *TNode) *TNodeNavigator {
|
||
|
|
return &TNodeNavigator{curr: n, root: n, attr: -1}
|
||
|
|
}
|
||
|
|
|
||
|
|
type Attribute struct {
|
||
|
|
Key, Value string
|
||
|
|
}
|
||
|
|
|
||
|
|
type TNode struct {
|
||
|
|
Parent, FirstChild, LastChild, PrevSibling, NextSibling *TNode
|
||
|
|
|
||
|
|
Type NodeType
|
||
|
|
Data string
|
||
|
|
Attr []Attribute
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNode) Value() string {
|
||
|
|
if n.Type == TextNode {
|
||
|
|
return n.Data
|
||
|
|
}
|
||
|
|
|
||
|
|
var buff bytes.Buffer
|
||
|
|
var output func(*TNode)
|
||
|
|
output = func(node *TNode) {
|
||
|
|
if node.Type == TextNode {
|
||
|
|
buff.WriteString(node.Data)
|
||
|
|
}
|
||
|
|
for child := node.FirstChild; child != nil; child = child.NextSibling {
|
||
|
|
output(child)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
output(n)
|
||
|
|
return buff.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// TNodeNavigator is for navigating TNode.
|
||
|
|
type TNodeNavigator struct {
|
||
|
|
curr, root *TNode
|
||
|
|
attr int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) NodeType() NodeType {
|
||
|
|
if n.curr.Type == ElementNode && n.attr != -1 {
|
||
|
|
return AttributeNode
|
||
|
|
}
|
||
|
|
return n.curr.Type
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) LocalName() string {
|
||
|
|
if n.attr != -1 {
|
||
|
|
return n.curr.Attr[n.attr].Key
|
||
|
|
}
|
||
|
|
return n.curr.Data
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) Prefix() string {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) Value() string {
|
||
|
|
switch n.curr.Type {
|
||
|
|
case CommentNode:
|
||
|
|
return n.curr.Data
|
||
|
|
case ElementNode:
|
||
|
|
if n.attr != -1 {
|
||
|
|
return n.curr.Attr[n.attr].Value
|
||
|
|
}
|
||
|
|
var buf bytes.Buffer
|
||
|
|
node := n.curr.FirstChild
|
||
|
|
for node != nil {
|
||
|
|
if node.Type == TextNode {
|
||
|
|
buf.WriteString(strings.TrimSpace(node.Data))
|
||
|
|
}
|
||
|
|
node = node.NextSibling
|
||
|
|
}
|
||
|
|
return buf.String()
|
||
|
|
case TextNode:
|
||
|
|
return n.curr.Data
|
||
|
|
}
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) Copy() NodeNavigator {
|
||
|
|
n2 := *n
|
||
|
|
return &n2
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToRoot() {
|
||
|
|
n.curr = n.root
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToParent() bool {
|
||
|
|
if node := n.curr.Parent; node != nil {
|
||
|
|
n.curr = node
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToNextAttribute() bool {
|
||
|
|
if n.attr >= len(n.curr.Attr)-1 {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
n.attr++
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToChild() bool {
|
||
|
|
if node := n.curr.FirstChild; node != nil {
|
||
|
|
n.curr = node
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToFirst() bool {
|
||
|
|
if n.curr.PrevSibling == nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
for {
|
||
|
|
node := n.curr.PrevSibling
|
||
|
|
if node == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
n.curr = node
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) String() string {
|
||
|
|
return n.Value()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToNext() bool {
|
||
|
|
if node := n.curr.NextSibling; node != nil {
|
||
|
|
n.curr = node
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveToPrevious() bool {
|
||
|
|
if node := n.curr.PrevSibling; node != nil {
|
||
|
|
n.curr = node
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNodeNavigator) MoveTo(other NodeNavigator) bool {
|
||
|
|
node, ok := other.(*TNodeNavigator)
|
||
|
|
if !ok || node.root != n.root {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
n.curr = node.curr
|
||
|
|
n.attr = node.attr
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func createNode(data string, typ NodeType) *TNode {
|
||
|
|
return &TNode{Data: data, Type: typ, Attr: make([]Attribute, 0)}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNode) createChildNode(data string, typ NodeType) *TNode {
|
||
|
|
m := createNode(data, typ)
|
||
|
|
m.Parent = n
|
||
|
|
if n.FirstChild == nil {
|
||
|
|
n.FirstChild = m
|
||
|
|
} else {
|
||
|
|
n.LastChild.NextSibling = m
|
||
|
|
m.PrevSibling = n.LastChild
|
||
|
|
}
|
||
|
|
n.LastChild = m
|
||
|
|
return m
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNode) appendNode(data string, typ NodeType) *TNode {
|
||
|
|
m := createNode(data, typ)
|
||
|
|
m.Parent = n.Parent
|
||
|
|
n.NextSibling = m
|
||
|
|
m.PrevSibling = n
|
||
|
|
if n.Parent != nil {
|
||
|
|
n.Parent.LastChild = m
|
||
|
|
}
|
||
|
|
return m
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *TNode) addAttribute(k, v string) {
|
||
|
|
n.Attr = append(n.Attr, Attribute{k, v})
|
||
|
|
}
|
||
|
|
|
||
|
|
func example() *TNode {
|
||
|
|
/*
|
||
|
|
<html lang="en">
|
||
|
|
<head>
|
||
|
|
<title>Hello</title>
|
||
|
|
<meta name="language" content="en"/>
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<h1> This is a H1 </h1>
|
||
|
|
<ul>
|
||
|
|
<li><a id="1" href="/">Home</a></li>
|
||
|
|
<li><a id="2" href="/about">about</a></li>
|
||
|
|
<li><a id="3" href="/account">login</a></li>
|
||
|
|
<li></li>
|
||
|
|
</ul>
|
||
|
|
<p>
|
||
|
|
Hello,This is an example for gxpath.
|
||
|
|
</p>
|
||
|
|
<footer>footer script</footer>
|
||
|
|
</body>
|
||
|
|
</html>
|
||
|
|
*/
|
||
|
|
doc := createNode("", RootNode)
|
||
|
|
xhtml := doc.createChildNode("html", ElementNode)
|
||
|
|
xhtml.addAttribute("lang", "en")
|
||
|
|
|
||
|
|
// The HTML head section.
|
||
|
|
head := xhtml.createChildNode("head", ElementNode)
|
||
|
|
n := head.createChildNode("title", ElementNode)
|
||
|
|
n = n.createChildNode("Hello", TextNode)
|
||
|
|
n = head.createChildNode("meta", ElementNode)
|
||
|
|
n.addAttribute("name", "language")
|
||
|
|
n.addAttribute("content", "en")
|
||
|
|
// The HTML body section.
|
||
|
|
body := xhtml.createChildNode("body", ElementNode)
|
||
|
|
n = body.createChildNode("h1", ElementNode)
|
||
|
|
n = n.createChildNode(" This is a H1 ", TextNode)
|
||
|
|
ul := body.createChildNode("ul", ElementNode)
|
||
|
|
n = ul.createChildNode("li", ElementNode)
|
||
|
|
n = n.createChildNode("a", ElementNode)
|
||
|
|
n.addAttribute("id", "1")
|
||
|
|
n.addAttribute("href", "/")
|
||
|
|
n = n.createChildNode("Home", TextNode)
|
||
|
|
n = ul.createChildNode("li", ElementNode)
|
||
|
|
n = n.createChildNode("a", ElementNode)
|
||
|
|
n.addAttribute("id", "2")
|
||
|
|
n.addAttribute("href", "/about")
|
||
|
|
n = n.createChildNode("about", TextNode)
|
||
|
|
n = ul.createChildNode("li", ElementNode)
|
||
|
|
n = n.createChildNode("a", ElementNode)
|
||
|
|
n.addAttribute("id", "3")
|
||
|
|
n.addAttribute("href", "/account")
|
||
|
|
n = n.createChildNode("login", TextNode)
|
||
|
|
n = ul.createChildNode("li", ElementNode)
|
||
|
|
|
||
|
|
n = body.createChildNode("p", ElementNode)
|
||
|
|
n = n.createChildNode("Hello,This is an example for gxpath.", TextNode)
|
||
|
|
|
||
|
|
n = body.createChildNode("footer", ElementNode)
|
||
|
|
n = n.createChildNode("footer script", TextNode)
|
||
|
|
|
||
|
|
return xhtml
|
||
|
|
}
|