arikawa/httputil/options.go

package httputil

import (
	"bytes"
	"io"
	"io/ioutil"
	"net/http"

	"git.sr.ht/~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
	}
}