1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-11-19 05:12:48 +00:00
arikawa/httputil/options.go

51 lines
777 B
Go

package httputil
import (
"bytes"
"io"
"io/ioutil"
"net/http"
"github.com/diamondburned/arikawa/json"
)
type RequestOption func(*http.Request) error
func JSONRequest(r *http.Request) error {
r.Header.Set("Content-Type", "application/json")
return nil
}
func WithBody(body io.ReadCloser) func(*http.Request) error {
return func(r *http.Request) error {
r.Body = body
return nil
}
}
func WithJSONBody(
json json.Driver, v interface{}) func(r *http.Request) error {
if v == nil {
return func(*http.Request) error {
return nil
}
}
var buf bytes.Buffer
var err error
go func() {
err = json.EncodeStream(&buf, v)
}()
return func(r *http.Request) error {
if err != nil {
return err
}
r.Body = ioutil.NopCloser(&buf)
return nil
}
}