79 lines
1.4 KiB
Go
79 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"code.google.com/p/go.net/html"
|
||
|
|
)
|
||
|
|
|
||
|
|
var trans = make(map[string]string)
|
||
|
|
|
||
|
|
func generalNode(n *html.Node) {
|
||
|
|
translate := false
|
||
|
|
if n.Type == html.ElementNode {
|
||
|
|
for _, a := range n.Attr {
|
||
|
|
if a.Key == "translate" {
|
||
|
|
translate = true
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else if n.Type == html.TextNode {
|
||
|
|
v := strings.TrimSpace(n.Data)
|
||
|
|
if len(v) > 1 && !(strings.HasPrefix(v, "{{") && strings.HasSuffix(v, "}}")) {
|
||
|
|
log.Println("Untranslated text node:")
|
||
|
|
log.Print("\t" + v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for c := n.FirstChild; c != nil; c = c.NextSibling {
|
||
|
|
if translate {
|
||
|
|
inTranslate(c)
|
||
|
|
} else {
|
||
|
|
generalNode(c)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func inTranslate(n *html.Node) {
|
||
|
|
if n.Type == html.TextNode {
|
||
|
|
v := strings.TrimSpace(n.Data)
|
||
|
|
if _, ok := trans[v]; !ok {
|
||
|
|
av := strings.Replace(v, "{%", "{{", -1)
|
||
|
|
av = strings.Replace(av, "%}", "}}", -1)
|
||
|
|
trans[v] = av
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
log.Println("translate node with non-text child <")
|
||
|
|
log.Println(n)
|
||
|
|
}
|
||
|
|
if n.FirstChild != nil {
|
||
|
|
log.Println("translate node has children:")
|
||
|
|
log.Println(n.Data)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
fd, err := os.Open(os.Args[1])
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
err = json.NewDecoder(fd).Decode(&trans)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
fd.Close()
|
||
|
|
|
||
|
|
doc, err := html.Parse(os.Stdin)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
generalNode(doc)
|
||
|
|
bs, err := json.MarshalIndent(trans, "", " ")
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
os.Stdout.Write(bs)
|
||
|
|
}
|