105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
|
package scw
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
|
||
|
|
"github.com/scaleway/scaleway-sdk-go/internal/auth"
|
||
|
|
"github.com/scaleway/scaleway-sdk-go/internal/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ScalewayRequest contains all the contents related to performing a request on the Scaleway API.
|
||
|
|
type ScalewayRequest struct {
|
||
|
|
Method string
|
||
|
|
Path string
|
||
|
|
Headers http.Header
|
||
|
|
Query url.Values
|
||
|
|
Body io.Reader
|
||
|
|
|
||
|
|
// request options
|
||
|
|
ctx context.Context
|
||
|
|
auth auth.Auth
|
||
|
|
allPages bool
|
||
|
|
}
|
||
|
|
|
||
|
|
// getAllHeaders constructs a http.Header object and aggregates all headers into the object.
|
||
|
|
func (req *ScalewayRequest) getAllHeaders(token auth.Auth, userAgent string, anonymized bool) http.Header {
|
||
|
|
var allHeaders http.Header
|
||
|
|
if anonymized {
|
||
|
|
allHeaders = token.AnonymizedHeaders()
|
||
|
|
} else {
|
||
|
|
allHeaders = token.Headers()
|
||
|
|
}
|
||
|
|
|
||
|
|
allHeaders.Set("User-Agent", userAgent)
|
||
|
|
if req.Body != nil {
|
||
|
|
allHeaders.Set("Content-Type", "application/json")
|
||
|
|
}
|
||
|
|
for key, value := range req.Headers {
|
||
|
|
allHeaders.Del(key)
|
||
|
|
for _, v := range value {
|
||
|
|
allHeaders.Add(key, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return allHeaders
|
||
|
|
}
|
||
|
|
|
||
|
|
// getURL constructs a URL based on the base url and the client.
|
||
|
|
func (req *ScalewayRequest) getURL(baseURL string) (*url.URL, error) {
|
||
|
|
url, err := url.Parse(baseURL + req.Path)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.New("invalid url %s: %s", baseURL+req.Path, err)
|
||
|
|
}
|
||
|
|
url.RawQuery = req.Query.Encode()
|
||
|
|
|
||
|
|
return url, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetBody json marshal the given body and write the json content type
|
||
|
|
// to the request. It also catches when body is a file.
|
||
|
|
func (req *ScalewayRequest) SetBody(body interface{}) error {
|
||
|
|
var contentType string
|
||
|
|
var content io.Reader
|
||
|
|
|
||
|
|
switch b := body.(type) {
|
||
|
|
case *File:
|
||
|
|
contentType = b.ContentType
|
||
|
|
content = b.Content
|
||
|
|
case io.Reader:
|
||
|
|
contentType = "text/plain"
|
||
|
|
content = b
|
||
|
|
default:
|
||
|
|
buf, err := json.Marshal(body)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
contentType = "application/json"
|
||
|
|
content = bytes.NewReader(buf)
|
||
|
|
}
|
||
|
|
|
||
|
|
if req.Headers == nil {
|
||
|
|
req.Headers = http.Header{}
|
||
|
|
}
|
||
|
|
|
||
|
|
req.Headers.Set("Content-Type", contentType)
|
||
|
|
req.Body = content
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (req *ScalewayRequest) apply(opts []RequestOption) {
|
||
|
|
for _, opt := range opts {
|
||
|
|
opt(req)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (req *ScalewayRequest) validate() error {
|
||
|
|
// nothing so far
|
||
|
|
return nil
|
||
|
|
}
|