forked from dskinner/damsel
/
main.go
65 lines (55 loc) · 1.21 KB
/
main.go
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
package main
import (
"encoding/json"
"flag"
"fmt"
"dasa.cc/damsel/dmsl"
"dasa.cc/damsel/dmsl/parse"
"log"
"os"
"runtime/pprof"
)
var filename = flag.String("f", "", "file to parse")
var debug = flag.Bool("d", false, "print parser debug info")
var pprint = flag.Bool("pprint", false, "pretty print output")
var cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file")
var data = flag.String("data", "", "json string to decode as data for template")
func main() {
flag.Parse()
if *cpuprofile != "" {
f, err := os.Create(*cpuprofile)
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
parse.DocParse(dmsl.Open("tests/bigtable2.dmsl", ""))
return
}
if *filename == "" {
flag.Usage()
return
}
if *pprint {
parse.Pprint = true
}
if *debug {
r, err := parse.DocParse(dmsl.Open(*filename, ""))
if err != nil {
fmt.Println(err)
}
fmt.Println(r)
} else {
t, err := dmsl.ParseFile(*filename)
var d interface{}
if err = json.Unmarshal([]byte(*data), &d); err != nil {
fmt.Println(err.Error())
}
result, err := t.Execute(d.([]interface{}))
if err != nil {
fmt.Println(err)
} else {
fmt.Println(result)
}
}
}