-
Notifications
You must be signed in to change notification settings - Fork 39
/
utils.go
111 lines (101 loc) · 2.19 KB
/
utils.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
// (c) 2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package api
import (
"encoding/json"
"fmt"
"io"
"net/http"
"net/url"
"strings"
"github.com/gocraft/web"
)
// ErrorResponse represents an API error to return to the caller
type ErrorResponse struct {
Code int `json:"code"`
Message string `json:"message"`
}
// WriteJSON writes the given bytes to the http response as JSON
func WriteJSON(w http.ResponseWriter, msg []byte) {
w.WriteHeader(200)
_, _ = w.Write(msg)
}
// WriteErr writes the given error message to the http response
func WriteErr(w http.ResponseWriter, code int, msg string) {
errBytes, err := json.Marshal(&ErrorResponse{
Code: code,
Message: msg,
})
if err != nil {
w.WriteHeader(500)
fmt.Fprint(w, `{"code": 500, "message": "failed to generate correct error message"}`)
}
w.WriteHeader(code)
_, _ = w.Write(errBytes)
}
func ParseGet(r *web.Request, n int64) (url.Values, error) {
if r == nil {
return r.URL.Query(), nil
}
pf := r.Body
if pf == nil {
return r.URL.Query(), nil
}
buf := new(strings.Builder)
_, err := io.CopyN(buf, pf, n)
switch err {
case io.EOF:
case nil:
default:
return r.URL.Query(), err
}
if buf.Len() > 0 {
getq, err := url.ParseQuery(buf.String())
switch err {
case nil:
return mergeValues(getq, r.URL.Query()), nil
default:
return r.URL.Query(), err
}
}
return r.URL.Query(), nil
}
func ParseGetJSON(r *web.Request, n int64) (url.Values, error) {
if r == nil {
return r.URL.Query(), nil
}
pf := r.Body
if pf == nil {
return r.URL.Query(), nil
}
buf := new(strings.Builder)
_, err := io.CopyN(buf, pf, n)
switch err {
case io.EOF:
case nil:
default:
return r.URL.Query(), err
}
if buf.Len() > 0 {
s := buf.String()
qdata := make(map[string][]string)
err := json.Unmarshal([]byte(s), &qdata)
switch err {
case nil:
return mergeValues(qdata, r.URL.Query()), nil
default:
return r.URL.Query(), err
}
}
return r.URL.Query(), nil
}
func mergeValues(a url.Values, b url.Values) url.Values {
for k, v := range b {
if _, present := a[k]; present {
a[k] = append(a[k], v...)
} else {
a[k] = append([]string{}, v...)
}
}
return a
}