add rate limits to error wrapper (#39)

models/pinterest_error.go

@@ -12,6 +12,7 @@ import (
 type PinterestError struct {
 	StatusCode int    `json:"status_code"`
 	Message    string `json:"message"`
+	Limit      TypeRatelimit
 }
 
 func (e *PinterestError) Error() string {
@@ -32,6 +33,7 @@ func WrapPinterestError(httpResponse *http.Response, bodyResponse *Response, err
 		return &PinterestError{
 			StatusCode: httpResponse.StatusCode,
 			Message:    bodyResponse.Message,
+			Limit:      GetRatelimit(httpResponse),
 		}
 	}
 

models/rate_limit.go

@@ -0,0 +1,33 @@
+package models
+
+import (
+	"net/http"
+	"strconv"
+	"fmt"
+)
+
+type TypeRatelimit struct {
+	Remaining int
+	Limit     int
+	Refresh   int
+}
+
+func GetLimit(httpResp *http.Response, key string) int {
+	if val, ok := httpResp.Header[key]; ok {
+		rateLimit, err := strconv.Atoi(val[0])
+		if err == nil {
+			fmt.Println(err)
+			return 0
+		}
+		return rateLimit
+	}
+	return 0
+}
+
+func GetRatelimit(httpResp *http.Response) TypeRatelimit {
+	return TypeRatelimit {
+		Remaining: GetLimit(httpResp, "X-Ratelimit-Refresh"),
+		Limit: GetLimit(httpResp, "X-Ratelimit-Limit"),
+		Refresh: GetLimit(httpResp, "X-Ratelimit-Remaining"),
+	}
+}