72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
|
package rancher
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
rootUrl url.URL
|
||
|
|
client http.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(rootUrl url.URL) Client {
|
||
|
|
return Client{rootUrl, http.Client{}}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) Get(path string, result interface{}) error {
|
||
|
|
return c.do("GET", path, nil, result)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) Put(path string, data, result interface{}) error {
|
||
|
|
var body *bytes.Buffer
|
||
|
|
if data != nil {
|
||
|
|
b, err := json.Marshal(data)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("error marshaling body: {}", err)
|
||
|
|
}
|
||
|
|
body = bytes.NewBuffer(b)
|
||
|
|
}
|
||
|
|
return c.do("PUT", path, body, result)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) do(method, path string, body io.Reader, result interface{}) error {
|
||
|
|
url := c.rootUrl
|
||
|
|
url.Path += path
|
||
|
|
req, err := http.NewRequest(method, url.String(), body)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
user := c.rootUrl.User
|
||
|
|
if user == nil {
|
||
|
|
return fmt.Errorf("user and password must be set in url")
|
||
|
|
}
|
||
|
|
password, isSet := user.Password()
|
||
|
|
if !isSet {
|
||
|
|
return fmt.Errorf("password must be set in url")
|
||
|
|
}
|
||
|
|
req.SetBasicAuth(user.Username(), password)
|
||
|
|
resp, err := c.client.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
respBody, err := ioutil.ReadAll(resp.Body)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if resp.StatusCode != 200 {
|
||
|
|
return fmt.Errorf("Expected to API to return 200, got %s: %s",
|
||
|
|
resp.Status,
|
||
|
|
respBody)
|
||
|
|
}
|
||
|
|
if result != nil {
|
||
|
|
return json.Unmarshal(respBody, result)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|