mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-19 05:12:48 +00:00
51 lines
777 B
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
|
|
}
|
|
}
|