package httputil import ( "io" "net/http" "net/url" "github.com/diamondburned/arikawa/v2/utils/httputil/httpdriver" "github.com/diamondburned/arikawa/v2/utils/json" ) type RequestOption func(httpdriver.Request) error type ResponseFunc func(httpdriver.Request, httpdriver.Response) error func PrependOptions(opts []RequestOption, prepend ...RequestOption) []RequestOption { if len(opts) == 0 { return prepend } return append(prepend, opts...) } func JSONRequest(r httpdriver.Request) error { r.AddHeader(http.Header{ "Content-Type": {"application/json"}, }) return nil } func MultipartRequest(r httpdriver.Request) error { r.AddHeader(http.Header{ "Content-Type": {"multipart/form-data"}, }) return nil } func WithHeaders(headers http.Header) RequestOption { return func(r httpdriver.Request) error { r.AddHeader(headers) return nil } } func WithContentType(ctype string) RequestOption { return func(r httpdriver.Request) error { r.AddHeader(http.Header{ "Content-Type": {ctype}, }) return nil } } func WithSchema(schema SchemaEncoder, v interface{}) RequestOption { return func(r httpdriver.Request) error { var params url.Values if p, ok := v.(url.Values); ok { params = p } else { p, err := schema.Encode(v) if err != nil { return err } params = p } r.AddQuery(params) return nil } } func WithBody(body io.ReadCloser) RequestOption { return func(r httpdriver.Request) error { r.WithBody(body) return nil } } // WithJSONBody inserts a JSON body into the request. This ignores JSON errors. func WithJSONBody(v interface{}) RequestOption { if v == nil { return func(httpdriver.Request) error { return nil } } var rp, wp = io.Pipe() go func() { wp.CloseWithError(json.EncodeStream(wp, v)) }() return func(r httpdriver.Request) error { r.AddHeader(http.Header{ "Content-Type": {"application/json"}, }) r.WithBody(rp) return nil } }