-
Notifications
You must be signed in to change notification settings - Fork 29
/
engine.go
152 lines (130 loc) · 2.59 KB
/
engine.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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
package render
import (
"bytes"
"encoding/json"
"encoding/xml"
"html/template"
"net/http"
)
// Engine is the generic interface for all responses.
type Engine interface {
Render(http.ResponseWriter, interface{}) error
}
// Head defines the basic ContentType and Status fields.
type Head struct {
ContentType string
Status int
}
// XML built-in renderer.
type XML struct {
Head
Indent bool
Prefix []byte
}
// JSON built-in renderer.
type JSON struct {
Head
Indent bool
Prefix []byte
}
// JSONP built-in renderer.
type JSONP struct {
Head
Indent bool
Callback string
}
// HTML built-in renderer.
type HTML struct {
Head
Name string
Templates *template.Template
}
// Data built-in renderer.
type Data struct {
Head
}
// Write outputs the header content.
func (h Head) Write(w http.ResponseWriter) {
w.Header().Set(ContentType, h.ContentType)
w.WriteHeader(h.Status)
}
// Render a data response.
func (d Data) Render(w http.ResponseWriter, v interface{}) error {
c := w.Header().Get(ContentType)
if c != "" {
d.Head.ContentType = c
}
d.Head.Write(w)
w.Write(v.([]byte))
return nil
}
// Render a JSON response.
func (j JSON) Render(w http.ResponseWriter, v interface{}) error {
var result []byte
var err error
if j.Indent {
result, err = json.MarshalIndent(v, "", " ")
} else {
result, err = json.Marshal(v)
}
if err != nil {
return err
}
// JSON marshaled fine, write out the result.
j.Head.Write(w)
if len(j.Prefix) > 0 {
w.Write(j.Prefix)
}
w.Write(result)
return nil
}
// Render a JSONP response.
func (j JSONP) Render(w http.ResponseWriter, v interface{}) error {
var result []byte
var err error
if j.Indent {
result, err = json.MarshalIndent(v, "", " ")
} else {
result, err = json.Marshal(v)
}
if err != nil {
return err
}
// JSON marshaled fine, write out the result.
j.Head.Write(w)
w.Write([]byte(j.Callback + "("))
w.Write(result)
w.Write([]byte(");"))
return nil
}
// Render an XML response.
func (x XML) Render(w http.ResponseWriter, v interface{}) error {
var result []byte
var err error
if x.Indent {
result, err = xml.MarshalIndent(v, "", " ")
} else {
result, err = xml.Marshal(v)
}
if err != nil {
return err
}
// XML marshaled fine, write out the result.
x.Head.Write(w)
if len(x.Prefix) > 0 {
w.Write(x.Prefix)
}
w.Write(result)
return nil
}
// Render a HTML response.
func (h HTML) Render(w http.ResponseWriter, binding interface{}) error {
out := new(bytes.Buffer)
err := h.Templates.ExecuteTemplate(out, h.Name, binding)
if err != nil {
return err
}
h.Head.Write(w)
w.Write(out.Bytes())
return nil
}