apollo-backend/internal/reddit/client.go

106 lines
1.9 KiB
Go
Raw Normal View History

2021-05-10 00:51:15 +00:00
package reddit
import (
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"strings"
"time"
)
const (
tokenURL = "https://www.reddit.com/api/v1/access_token"
)
type Client struct {
id string
secret string
}
func NewClient(id, secret string) *Client {
return &Client{id, secret}
}
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) {
tr := &http.Transport{}
client := &http.Client{Transport: tr}
req, err := r.HTTPRequest()
if err != nil {
return nil, err
}
resp, err := client.Do(req)
if err != nil {
return nil, err
}
return ioutil.ReadAll(resp.Body)
}
func (rac *AuthenticatedClient) refreshTokens() (string, string, error) {
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)
fmt.Println(string(body))
return "", "", err
}
func (rac *AuthenticatedClient) MessageInbox() error {
req := NewRequest(
WithMethod("GET"),
WithToken(rac.accessToken),
WithURL("https://oauth.reddit.com/message/inbox.json"),
)
body, _ := rac.request(req)
fmt.Println(string(body))
return nil
}
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
}