forked from liveforeverx/chello
/
jsonrpc2_request.go
95 lines (85 loc) · 2.05 KB
/
jsonrpc2_request.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
package main
import (
"bytes"
"encoding/json"
"fmt"
"strconv"
"strings"
"time"
)
type jsonRPCRequest struct {
JsonRPC string `json:"jsonrpc"`
Id string `json:"id"`
Method string `json:"method"`
Params interface{} `json:"params,omitempty"`
}
func NewJSONRPCRequest(id string) *jsonRPCRequest {
var req = new(jsonRPCRequest)
req.JsonRPC = "2.0"
req.Id = id
if id == "" {
req.Id = strconv.FormatInt(time.Now().UnixNano(), 10)
}
return req
}
func (r *jsonRPCRequest) setMethodAndParams(args []string) error {
switch {
case len(args) == 0:
return fmt.Errorf("missing 'method' (first argument)")
case len(args) == 1:
r.Method = args[0]
case len(args) > 1:
r.Method = args[0]
if strings.ContainsRune(args[1], '=') {
// if first argument conatins a '=' i assume
// a list of key-value pairs, forming a map
// or an 'object' in javascript terms
r.Params = kvPairsToMap(args[1:])
} else if len(args[1]) > 1 && strings.TrimLeft(args[1], " \n\t")[0] == '{' {
raw := json.RawMessage(args[1])
if ok, err := isValidJSON(args[1]); !ok {
return err
}
r.Params = &raw
} else {
r.Params = args[1:]
}
}
return nil
}
func (r *jsonRPCRequest) toJSON() (string, error) {
var buf = bytes.NewBuffer(nil)
var encoder = json.NewEncoder(buf)
if err := encoder.Encode(r); err != nil {
return "", err
}
return buf.String(), nil
}
func kvPairsToMap(pairs []string) map[string]interface{} {
var m = make(map[string]interface{})
for _, elem := range pairs {
var (
parts = strings.SplitN(elem, "=", 2)
key = parts[0]
val interface{}
)
if len(parts) == 2 {
// if the value can be parsed as a float, we assume
// a json.Number
if _, err := strconv.ParseFloat(parts[1], 64); err == nil {
val = json.Number(parts[1])
} else {
val = parts[1]
}
}
m[key] = val
}
return m
}
func isValidJSON(s string) (bool, error) {
// is it valid json? try to decode it
var obj interface{}
var decoder = json.NewDecoder(strings.NewReader(s))
err := decoder.Decode(&obj)
return err == nil, err
}