-
Notifications
You must be signed in to change notification settings - Fork 14
/
client.go
116 lines (101 loc) · 2.19 KB
/
client.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
package go1inch
import (
"bytes"
"context"
"encoding/json"
"errors"
"fmt"
"io/ioutil"
"net/http"
)
const (
inchURL = "https://api.1inch.io/v5.0/"
)
type Network string
const (
Eth Network = "eth"
Bsc Network = "bsc"
Matic Network = "matic"
Optimism Network = "optimism"
Arbitrum Network = "arbitrum"
GnosisChain Network = "gnosis"
Avalanche Network = "avalanche"
Fantom Network = "fantom"
Klaytn Network = "klaytn"
Auror Network = "auror"
)
var (
networks = map[Network]string{
Eth: "1",
Bsc: "56",
Matic: "137",
Optimism: "10",
Arbitrum: "42161",
GnosisChain: "100",
Avalanche: "43114",
Fantom: "250",
Klaytn: "8217",
Auror: "1313161554",
}
)
func NewClient() *Client {
return &Client{
Http: &http.Client{},
}
}
func setQueryParam(endpoint *string, params []map[string]interface{}) {
var first = true
for _, param := range params {
for i := range param {
if first {
*endpoint = fmt.Sprintf("%s?%s=%v", *endpoint, i, param[i])
first = false
} else {
*endpoint = fmt.Sprintf("%s&%s=%v", *endpoint, i, param[i])
}
}
}
}
func (c *Client) doRequest(ctx context.Context, net Network, endpoint, method string, expRes interface{}, reqData interface{}, opts ...map[string]interface{}) (int, error) {
n, ok := networks[net]
if !ok {
return 0, errors.New("invalid network")
}
callURL := fmt.Sprintf("%s%s%s", inchURL, n, endpoint)
var dataReq []byte
var err error
if reqData != nil {
dataReq, err = json.Marshal(reqData)
if err != nil {
return 0, err
}
}
if len(opts) > 0 && len(opts[0]) > 0 {
setQueryParam(&callURL, opts)
}
req, err := http.NewRequestWithContext(ctx, method, callURL, bytes.NewBuffer(dataReq))
if err != nil {
return 0, err
}
resp, err := c.Http.Do(req)
if err != nil {
return 0, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return 0, err
}
switch resp.StatusCode {
case 200:
if expRes != nil {
err = json.Unmarshal(body, expRes)
if err != nil {
return 0, err
}
}
return resp.StatusCode, nil
default:
return resp.StatusCode, fmt.Errorf("%s", body)
}
}