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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package curl
import (
"io"
"net/url"
"os/exec"
"runtime"
"github.com/alessio/shellescape"
"github.com/vespa-engine/vespa/client/go/internal/util"
)
type header struct {
key string
value string
}
type Command struct {
Path string
Method string
PrivateKey string
CaCertificate string
Certificate string
bodyFile string
bodyInput io.Reader
url *url.URL
headers []header
rawArgs []string
}
func (c *Command) GetUrlPrefix() string {
return c.url.Scheme + "://" + c.url.Host
}
func (c *Command) WithBodyFile(fn string) {
if c.bodyInput != nil {
util.JustExitMsg("cannot use both WithBodyFile and WithBodyInput")
}
c.bodyFile = fn
}
func (c *Command) WithBodyInput(r io.Reader) {
if c.bodyFile != "" {
util.JustExitMsg("cannot use both WithBodyFile and WithBodyInput")
}
c.bodyInput = r
}
func (c *Command) Args() []string {
var args []string
if c.PrivateKey != "" {
args = append(args, "--key", c.PrivateKey)
}
if c.Certificate != "" {
args = append(args, "--cert", c.Certificate)
}
if c.CaCertificate != "" {
args = append(args, "--cacert", c.CaCertificate)
}
if c.Method != "" {
args = append(args, "-X", c.Method)
}
for _, header := range c.headers {
args = append(args, "-H", header.key+": "+header.value)
}
if c.bodyFile != "" {
args = append(args, "--data-binary", "@"+c.bodyFile)
} else if c.bodyInput != nil {
args = append(args, "--data-binary", "@-")
}
args = append(args, c.rawArgs...)
args = append(args, c.url.String())
return args
}
func (c *Command) String() string {
args := []string{c.Path}
args = append(args, c.Args()...)
return shellescape.QuoteCommand(args)
}
func (c *Command) Header(key, value string) {
c.headers = append(c.headers, header{key: key, value: value})
}
func (c *Command) Param(key, value string) {
query := c.url.Query()
query.Set(key, value)
c.url.RawQuery = query.Encode()
}
func (c *Command) Run(stdout, stderr io.Writer) error {
cmd := exec.Command(c.Path, c.Args()...)
cmd.Stdout = stdout
cmd.Stderr = stderr
cmd.Stdin = c.bodyInput
if err := cmd.Start(); err != nil {
return err
}
return cmd.Wait()
}
func Post(url string) (*Command, error) { return curl("POST", url) }
func Get(url string) (*Command, error) { return curl("", url) }
func RawArgs(url string, args ...string) (*Command, error) {
c, err := curl("", url)
if err != nil {
return nil, err
}
c.rawArgs = args
return c, nil
}
func curl(method, rawurl string) (*Command, error) {
path := "curl"
if runtime.GOOS == "windows" {
path = "curl.exe"
}
realURL, err := url.Parse(rawurl)
if err != nil {
return nil, err
}
return &Command{
Path: path,
Method: method,
url: realURL,
}, nil
}
|