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 }) }