-
Notifications
You must be signed in to change notification settings - Fork 0
/
api_client.go
93 lines (78 loc) · 2 KB
/
api_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
package cli
import (
"fmt"
"net/http"
"github.com/go-resty/resty/v2"
)
var MockHttpClientFunc func(client *http.Client)
func NewApiClient(config Config) (*resty.Client, error) {
err := config.RequireServerConfig()
if err != nil {
return nil, err
}
client := resty.New()
client.SetHostURL(config.ServerBaseURL + "/v1")
client.SetDebug(config.Debug)
if MockHttpClientFunc != nil {
MockHttpClientFunc(client.GetClient())
}
return client, nil
}
func NewApiRequest(config Config, state State) (*resty.Request, error) {
err := state.RequireAuthToken()
if err != nil {
return nil, err
}
r, err := NewApiRequestWithoutAuth(config)
if err != nil {
return nil, err
}
if len(config.BasicAuthUser) > 0 || len(config.BasicAuthPassword) > 0 {
r.SetBasicAuth(config.BasicAuthUser, config.BasicAuthPassword)
}
r.SetAuthToken(state.AuthToken)
return r, nil
}
func NewApiRequestWithoutAuth(config Config) (*resty.Request, error) {
client, err := NewApiClient(config)
if err != nil {
return nil, err
}
r := client.R()
r.SetError(&map[string]interface{}{})
r.SetHeader("User-Agent", "sqedule-cli/"+VersionString)
r.SetHeader("Accept", "application/json")
return r, nil
}
func GetApiErrorMessage(resp *resty.Response) string {
e := resp.Error()
if e == nil {
return fmt.Sprintf("unknown error (HTTP %s)", resp.Status())
}
object, ok := e.(*map[string]interface{})
if !ok {
return fmt.Sprintf("HTTP %s: %#v", resp.Status(), object)
}
if object == nil || *object == nil {
return fmt.Sprintf("unknown error (HTTP %s)", resp.Status())
}
if result, ok := (*object)["message"]; ok {
if str, ok := result.(string); ok {
return str
} else {
return fmt.Sprintf("%#v", result)
}
}
if result, ok := (*object)["error"]; ok {
if str, ok := result.(string); ok {
return str
} else {
return fmt.Sprintf("%#v", result)
}
}
if len(*object) == 0 {
return fmt.Sprintf("unknown error (HTTP %s)", resp.Status())
} else {
return fmt.Sprintf("HTTP %s: %#v", resp.Status(), object)
}
}