/
serializer.go
121 lines (104 loc) · 2.08 KB
/
serializer.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
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
package gocalc
import (
"fmt"
"io"
)
type serializer struct {
buffer buffer
indent int
ignore bool
}
func newSerializer() *serializer {
return &serializer{}
}
func (s *serializer) serialize(w io.Writer) {
w.Write(s.buffer)
}
type buffer []byte
func (b *buffer) Write(p []byte) (n int, err error) {
*b = append(*b, p...)
return len(p), nil
}
func (s *serializer) printf(format string, args ...interface{}) {
s.printIndent()
fmt.Fprintf(&s.buffer, format, args...)
}
func (s *serializer) println(str string) {
s.printIndent()
fmt.Fprintln(&s.buffer, str)
}
func (s *serializer) visitBinaryExpr(b *binaryExpr) {
s.println("*binaryExpr {")
s.indent++
s.printf("lhs: ")
s.ignore = true
b.left.accept(s)
s.printf("op: %s\n", b.op.val)
s.printf("rhs: ")
s.ignore = true
b.right.accept(s)
s.indent--
s.println("}")
}
func (s *serializer) visitFuncExpr(f *funcExpr) {
s.println("*funcExpr {")
s.indent++
s.printf("func: %s\n", f.function)
s.printf("args (len: %d) {\n", len(f.args))
s.indent++
for _, arg := range f.args {
arg.accept(s)
}
s.indent--
s.println("}")
s.indent--
s.println("}")
}
func (s *serializer) visitUnaryExpr(u *unaryExpr) {
s.println("*unaryExpr {")
s.indent++
s.printf("op: %v\n", u.op.val)
s.printf("expr: ")
s.ignore = true
u.expr.accept(s)
s.indent--
s.println("}")
}
func (s *serializer) visitBoolExpr(b *boolExpr) {
s.println("*boolExpr {")
s.indent++
s.printf("val: %t\n", b.val)
s.indent--
s.println("}")
}
func (s *serializer) visitFloatExpr(f *floatExpr) {
s.println("*floatExpr {")
s.indent++
s.printf("val: %f\n", f.val)
s.indent--
s.println("}")
}
func (s *serializer) visitIntExpr(i *intExpr) {
s.println("*intExpr {")
s.indent++
s.printf("val: %d\n", i.val)
s.indent--
s.println("}")
}
func (s *serializer) visitParamExpr(e *paramExpr) {
s.println("*identifier {")
s.indent++
s.printf("val: \"%s\"\n", e.identifier)
s.indent--
s.println("}")
}
var indent = []byte(". ")
func (s *serializer) printIndent() {
if s.ignore {
s.ignore = false
return
}
for i := 0; i < s.indent; i++ {
s.buffer.Write(indent)
}
}