dotcloud/docker

package request

import (
    "bytes"
    "encoding/json"
    "io"
    "net/http"
    "strings"
)

// Options defines request options, like request modifiers and which host to target
type Options struct {
    host             string
    requestModifiers []func(*http.Request) error
}

// Host creates a modifier that sets the specified host as the request URL host
func Host(host string) func(*Options) {
    return func(o *Options) {
        o.host = host
    }
}

// With adds a request modifier to the options
func With(f func(*http.Request) error) func(*Options) {
    return func(o *Options) {
        o.requestModifiers = append(o.requestModifiers, f)
    }
}

// Method creates a modifier that sets the specified string as the request method
func Method(method string) func(*Options) {
    return With(func(req *http.Request) error {
        req.Method = method
        return nil
    })
}

// RawString sets the specified string as body for the request
func RawString(content string) func(*Options) {
    return RawContent(io.NopCloser(strings.NewReader(content)))
}

// RawContent sets the specified reader as body for the request
func RawContent(reader io.ReadCloser) func(*Options) {
    return With(func(req *http.Request) error {
        req.Body = reader
        return nil
    })
}

// ContentType sets the specified Content-Type request header
func ContentType(contentType string) func(*Options) {
    return With(func(req *http.Request) error {
        req.Header.Set("Content-Type", contentType)
        return nil
    })
}

// JSON sets the Content-Type request header to json
func JSON(o *Options) {
    ContentType("application/json")(o)
}

// JSONBody creates a modifier that encodes the specified data to a JSON string and set it as request body. It also sets
// the Content-Type header of the request.
func JSONBody(data interface{}) func(*Options) {
    return With(func(req *http.Request) error {
        jsonData := bytes.NewBuffer(nil)
        if err := json.NewEncoder(jsonData).Encode(data); err != nil {
            return err
        }
        req.Body = io.NopCloser(jsonData)
        req.Header.Set("Content-Type", "application/json")
        return nil
    })
}