mirror of
https://github.com/christianselig/apollo-backend
synced 2024-11-13 07:27:43 +00:00
136 lines
2.5 KiB
Go
136 lines
2.5 KiB
Go
package reddit
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
const userAgent = "server:apollo-backend:v1.0 (by /u/iamthatis) contact me@christianselig.com"
|
|
|
|
type Request struct {
|
|
body url.Values
|
|
query url.Values
|
|
method string
|
|
token string
|
|
url string
|
|
auth string
|
|
tags []string
|
|
emptyResponseBytes int
|
|
retry bool
|
|
client *http.Client
|
|
}
|
|
|
|
type RequestOption func(*Request)
|
|
|
|
func NewRequest(opts ...RequestOption) *Request {
|
|
req := &Request{
|
|
body: url.Values{},
|
|
query: url.Values{},
|
|
method: "GET",
|
|
url: "",
|
|
|
|
token: "",
|
|
auth: "",
|
|
|
|
tags: nil,
|
|
|
|
emptyResponseBytes: 0,
|
|
retry: true,
|
|
client: nil,
|
|
}
|
|
|
|
req.query.Set("raw_json", "1")
|
|
|
|
for _, opt := range opts {
|
|
opt(req)
|
|
}
|
|
|
|
return req
|
|
}
|
|
|
|
func (r *Request) HTTPRequest(ctx context.Context) (*http.Request, error) {
|
|
req, err := http.NewRequestWithContext(ctx, r.method, r.url, strings.NewReader(r.body.Encode()))
|
|
req.URL.RawQuery = r.query.Encode()
|
|
|
|
req.Header.Add("Accept", "application/json")
|
|
req.Header.Add("User-Agent", userAgent)
|
|
|
|
if r.token != "" {
|
|
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", r.token))
|
|
}
|
|
|
|
if r.auth != "" {
|
|
req.Header.Add("Authorization", fmt.Sprintf("Basic %s", r.auth))
|
|
}
|
|
|
|
return req, err
|
|
}
|
|
|
|
func WithTags(tags []string) RequestOption {
|
|
return func(req *Request) {
|
|
req.tags = tags
|
|
}
|
|
}
|
|
|
|
func WithMethod(method string) RequestOption {
|
|
return func(req *Request) {
|
|
req.method = method
|
|
}
|
|
}
|
|
|
|
func WithURL(url string) RequestOption {
|
|
return func(req *Request) {
|
|
req.url = url
|
|
}
|
|
}
|
|
|
|
func WithBasicAuth(user, password string) RequestOption {
|
|
return func(req *Request) {
|
|
encoded := base64.StdEncoding.EncodeToString([]byte(user + ":" + password))
|
|
req.auth = encoded
|
|
}
|
|
}
|
|
|
|
func WithToken(token string) RequestOption {
|
|
return func(req *Request) {
|
|
req.token = token
|
|
}
|
|
}
|
|
|
|
func WithBody(key, val string) RequestOption {
|
|
return func(req *Request) {
|
|
req.body.Set(key, val)
|
|
}
|
|
}
|
|
|
|
func WithQuery(key, val string) RequestOption {
|
|
if val == "" {
|
|
return func(req *Request) {}
|
|
}
|
|
|
|
return func(req *Request) {
|
|
req.query.Set(key, val)
|
|
}
|
|
}
|
|
|
|
func WithEmptyResponseBytes(bytes int) RequestOption {
|
|
return func(req *Request) {
|
|
req.emptyResponseBytes = bytes
|
|
}
|
|
}
|
|
|
|
func WithRetry(retry bool) RequestOption {
|
|
return func(req *Request) {
|
|
req.retry = retry
|
|
}
|
|
}
|
|
|
|
func WithClient(client *http.Client) RequestOption {
|
|
return func(req *Request) {
|
|
req.client = client
|
|
}
|
|
}
|