forked from tdeebswihart/aws-signing
-
Notifications
You must be signed in to change notification settings - Fork 4
/
curl.go
71 lines (65 loc) · 1.29 KB
/
curl.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
package config
import (
"bytes"
"io"
"net/http"
"net/url"
"os"
"strings"
)
var (
DefaultCurl = Curl{
Method: "GET",
Headers: http.Header{
"Content-Type": []string{"application/json"},
},
Path: "/",
Aws: DefaultsAws,
}
EnvCurl = Curl{
Aws: EnvAws,
}
)
type Curl struct {
Data string `hcl:"-"`
Method string `hcl:"-"`
Headers http.Header `hcl:"-"`
Path string `hcl:"-"`
Aws Aws `hcl:"aws"`
Debug bool `hcl:"-"`
}
func MergeCurl(cfgs ...Curl) Curl {
rv := Curl{Headers: http.Header{}}
for _, cur := range cfgs {
if cur.Data != "" {
rv.Data = cur.Data
}
if cur.Method != "" {
rv.Method = cur.Method
}
if cur.Headers != nil {
for k, v := range cur.Headers {
rv.Headers[k] = v
}
}
if cur.Path != "" {
rv.Path = cur.Path
}
rv.Aws = MergeAws(rv.Aws, cur.Aws)
rv.Debug = rv.Debug || cur.Debug
}
return rv
}
func (c Curl) RequestUrl() string {
additional := strings.TrimPrefix(strings.TrimPrefix(c.Path, "//"), "/")
requestUrl, _ := url.Parse(strings.TrimSuffix(c.Aws.EndpointUrl().String(), "/") + "/" + additional)
return requestUrl.String()
}
func (c Curl) RequestBody() io.Reader {
if c.Data == "@-" {
return os.Stdin
} else if c.Data == "" {
return nil
}
return bytes.NewBufferString(c.Data)
}