apollo-backend/internal/reddit/client.go

184 lines
3.8 KiB
Go
Raw Normal View History

2021-05-10 00:51:15 +00:00
package reddit
import (
"encoding/json"
2021-07-12 18:36:08 +00:00
"fmt"
2021-05-10 00:51:15 +00:00
"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-10 18:51:42 +00:00
func SplitID(id string) (string, string) {
if parts := strings.Split(id, "_"); len(parts) == 2 {
return parts[0], parts[1]
}
return "", ""
}
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) {
2021-07-08 02:44:46 +00:00
if info.Reused {
2021-07-08 02:54:26 +00:00
statsd.Incr("reddit.api.connections.reused", []string{}, 0.1)
2021-07-08 02:44:46 +00:00
if info.WasIdle {
idleTime := float64(int64(info.IdleTime) / int64(time.Millisecond))
statsd.Histogram("reddit.api.connections.idle_time", idleTime, []string{}, 0.1)
}
2021-07-08 02:19:02 +00:00
} else {
2021-07-08 02:54:26 +00:00
statsd.Incr("reddit.api.connections.created", []string{}, 0.1)
2021-07-08 02:19:02 +00:00
}
},
}
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))
2021-07-08 23:26:15 +00:00
start := time.Now()
resp, err := rac.client.Do(req)
2021-07-08 23:26:15 +00:00
rac.statsd.Incr("reddit.api.calls", r.tags, 0.1)
rac.statsd.Histogram("reddit.api.latency", float64(time.Now().Sub(start).Milliseconds()), r.tags, 0.1)
2021-05-10 00:51:15 +00:00
if err != nil {
2021-07-08 23:26:15 +00:00
rac.statsd.Incr("reddit.api.errors", r.tags, 0.1)
2021-05-10 00:51:15 +00:00
return nil, err
}
defer resp.Body.Close()
2021-05-10 00:51:15 +00:00
2021-07-12 18:36:08 +00:00
bb, err := ioutil.ReadAll(resp.Body)
if err != nil {
rac.statsd.Incr("reddit.api.errors", r.tags, 0.1)
return nil, err
}
if resp.StatusCode != 200 {
rac.statsd.Incr("reddit.api.errors", r.tags, 0.1)
// Try to parse a json error. Otherwise we generate a generic one
rerr := &Error{}
if jerr := json.Unmarshal(bb, rerr); jerr != nil {
return nil, fmt.Errorf("error from reddit: %d", resp.StatusCode)
}
return nil, rerr
}
return bb, nil
2021-05-10 00:51:15 +00:00
}
2021-06-24 02:19:43 +00:00
func (rac *AuthenticatedClient) RefreshTokens() (*RefreshTokenResponse, error) {
2021-05-10 00:51:15 +00:00
req := NewRequest(
2021-07-08 23:26:15 +00:00
WithTags([]string{"url:/api/v1/access_token"}),
2021-05-10 00:51:15 +00:00
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(
2021-07-08 23:55:14 +00:00
WithTags([]string{"url:/api/v1/message/inbox"}),
2021-05-10 00:51:15 +00:00
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(
2021-07-08 23:26:15 +00:00
WithTags([]string{"url:/api/v1/me"}),
2021-05-10 00:51:15 +00:00
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
}