apollo-backend/internal/reddit/client.go

147 lines
2.7 KiB
Go
Raw Normal View History

2021-05-10 00:51:15 +00:00
package reddit
import (
"encoding/json"
"io/ioutil"
"net/http"
2021-07-08 02:19:02 +00:00
"net/http/httptrace"
2021-05-10 00:51:15 +00:00
"strings"
"time"
2021-07-08 02:19:02 +00:00
"github.com/DataDog/datadog-go/statsd"
"github.com/valyala/fastjson"
2021-05-10 00:51:15 +00:00
)
const (
tokenURL = "https://www.reddit.com/api/v1/access_token"
)
type Client struct {
id string
secret string
client *http.Client
2021-07-08 02:19:02 +00:00
tracer *httptrace.ClientTrace
parser *fastjson.Parser
2021-07-08 02:19:02 +00:00
statsd *statsd.Client
2021-05-10 00:51:15 +00:00
}
2021-07-08 02:19:02 +00:00
func NewClient(id, secret string, statsd *statsd.Client) *Client {
tracer := &httptrace.ClientTrace{
GotConn: func(info httptrace.GotConnInfo) {
if info.Reused {
statsd.Incr("reddit.api.connections.reused", []string{}, 0.1)
} else {
statsd.Incr("reddit.api.connections.created", []string{}, 0.1)
}
},
}
2021-07-08 02:25:57 +00:00
client := &http.Client{}
parser := &fastjson.Parser{}
return &Client{
id,
secret,
client,
2021-07-08 02:19:02 +00:00
tracer,
parser,
2021-07-08 02:19:02 +00:00
statsd,
}
2021-05-10 00:51:15 +00:00
}
type AuthenticatedClient struct {
*Client
refreshToken string
accessToken string
expiry *time.Time
}
func (rc *Client) NewAuthenticatedClient(refreshToken, accessToken string) *AuthenticatedClient {
return &AuthenticatedClient{rc, refreshToken, accessToken, nil}
}
func (rac *AuthenticatedClient) request(r *Request) ([]byte, error) {
req, err := r.HTTPRequest()
if err != nil {
return nil, err
}
2021-07-08 02:19:02 +00:00
req = req.WithContext(httptrace.WithClientTrace(req.Context(), rac.tracer))
resp, err := rac.client.Do(req)
2021-05-10 00:51:15 +00:00
if err != nil {
return nil, err
}
defer resp.Body.Close()
2021-05-10 00:51:15 +00:00
return ioutil.ReadAll(resp.Body)
}
2021-06-24 02:19:43 +00:00
func (rac *AuthenticatedClient) RefreshTokens() (*RefreshTokenResponse, error) {
2021-05-10 00:51:15 +00:00
req := NewRequest(
WithMethod("POST"),
WithURL(tokenURL),
WithBody("grant_type", "refresh_token"),
WithBody("refresh_token", rac.refreshToken),
WithBasicAuth(rac.id, rac.secret),
)
body, err := rac.request(req)
2021-06-24 02:19:43 +00:00
if err != nil {
return nil, err
}
rtr := &RefreshTokenResponse{}
json.Unmarshal([]byte(body), rtr)
return rtr, nil
2021-05-10 00:51:15 +00:00
}
2021-06-24 02:19:43 +00:00
func (rac *AuthenticatedClient) MessageInbox(from string) (*MessageListingResponse, error) {
2021-05-10 00:51:15 +00:00
req := NewRequest(
WithMethod("GET"),
WithToken(rac.accessToken),
WithURL("https://oauth.reddit.com/message/inbox.json"),
2021-06-24 02:19:43 +00:00
WithQuery("before", from),
2021-05-10 00:51:15 +00:00
)
2021-06-24 02:19:43 +00:00
body, err := rac.request(req)
if err != nil {
return nil, err
}
mlr := &MessageListingResponse{}
json.Unmarshal([]byte(body), mlr)
return mlr, nil
2021-05-10 00:51:15 +00:00
}
type MeResponse struct {
Name string
}
func (mr *MeResponse) NormalizedUsername() string {
return strings.ToLower(mr.Name)
}
func (rac *AuthenticatedClient) Me() (*MeResponse, error) {
req := NewRequest(
WithMethod("GET"),
WithToken(rac.accessToken),
WithURL("https://oauth.reddit.com/api/v1/me"),
)
body, err := rac.request(req)
if err != nil {
return nil, err
}
mr := &MeResponse{}
err = json.Unmarshal(body, mr)
return mr, err
}