- removed ioutils in some spots

- fixed rate limit headers
- added poll data to post
- changed random os to windows for surf client
This commit is contained in:
2026-05-25 16:21:13 -05:00
parent 76c46f0a0d
commit d4a435b52f
2 changed files with 49 additions and 33 deletions
+20 -32
View File
@@ -2,12 +2,10 @@ package reddit
import (
"bytes"
"compress/gzip"
"context"
"encoding/json"
"fmt"
"io"
"io/ioutil"
"net/http"
"net/url"
"reflect"
@@ -36,9 +34,9 @@ const (
headerAccept = "Accept"
headerUserAgent = "User-Agent"
headerRateLimitRemaining = "x-ratelimit-remaining"
headerRateLimitUsed = "x-ratelimit-used"
headerRateLimitReset = "x-ratelimit-reset"
headerRateLimitRemaining = "X-Ratelimit-Remaining"
headerRateLimitUsed = "X-Ratelimit-Used"
headerRateLimitReset = "X-Ratelimit-Reset"
)
var defaultClient, _ = NewReadonlyClient()
@@ -115,13 +113,14 @@ func newClient() *Client {
surf_client := surf.NewClient().
Builder().
Impersonate().
RandomOS().
Windows().
Firefox().
Session().
Build().
Unwrap()
Unwrap().
Std()
client := &Client{client: surf_client.Std(), BaseURL: baseURL, TokenURL: tokenURL}
client := &Client{client: surf_client, BaseURL: baseURL, TokenURL: tokenURL}
client.Account = &AccountService{client: client}
client.Collection = &CollectionService{client: client}
@@ -377,32 +376,21 @@ func (c *Client) Do(ctx context.Context, req *http.Request, v interface{}) (*Res
return response, err
}
body, err := ResponseReader(resp)
if err != nil {
return response, err
}
w, ok := v.(io.Writer)
if ok {
_, err = io.Copy(w, response.Body)
_, err = io.Copy(w, body)
if err != nil {
return response, err
}
} else {
if response.Header.Get("Content-Encoding") == "gzip" {
gzipReader, err := gzip.NewReader(resp.Body)
if err != nil {
return response, fmt.Errorf("Error creating gzip reader: %s", err)
}
defer gzipReader.Close()
b, err := io.ReadAll(gzipReader)
if err != nil {
return response, fmt.Errorf("Error decoding gzip json: %s", err)
}
err = json.Unmarshal(b, v)
if err != nil {
return response, fmt.Errorf("Error unmarshalling json: %s", err)
}
} else {
err = json.NewDecoder(response.Body).Decode(v)
if err != nil {
return response, err
}
err = json.NewDecoder(body).Decode(v)
if err != nil {
return response, err
}
}
@@ -425,7 +413,7 @@ func (c *Client) checkRateLimitBeforeDo(req *http.Request) *RateLimitError {
StatusCode: http.StatusTooManyRequests,
Request: req,
Header: make(http.Header),
Body: ioutil.NopCloser(strings.NewReader("")),
Body: io.NopCloser(strings.NewReader("")),
}
return &RateLimitError{
Rate: rate,
@@ -478,7 +466,7 @@ func CheckResponse(r *http.Response) error {
jsonErrorResponse := &JSONErrorResponse{Response: r}
data, err := ioutil.ReadAll(r.Body)
data, err := io.ReadAll(r.Body)
if err == nil && len(data) > 0 {
json.Unmarshal(data, jsonErrorResponse)
if len(jsonErrorResponse.JSON.Errors) > 0 {
@@ -487,14 +475,14 @@ func CheckResponse(r *http.Response) error {
}
// reset response body
r.Body = ioutil.NopCloser(bytes.NewBuffer(data))
r.Body = io.NopCloser(bytes.NewBuffer(data))
if c := r.StatusCode; c >= 200 && c <= 299 {
return nil
}
errorResponse := &ErrorResponse{Response: r}
data, err = ioutil.ReadAll(r.Body)
data, err = io.ReadAll(r.Body)
if err == nil && len(data) > 0 {
err := json.Unmarshal(data, errorResponse)
if err != nil {