ratelimit.go

package csfloat

import (
	"net/http"
	"strconv"
	"time"
)

type Ratelimits struct {
	Limit         uint
	Remaining     uint
	Reset         time.Time
	SuggestedWait time.Time
}

// ratelimitsFrom parses the headers to get ratelimiting. It does NOT consume
// the body.
func ratelimitsFrom(response *http.Response) (Ratelimits, error) {
	var ratelimits Ratelimits

	remaining := response.Header.Get("X-Ratelimit-Remaining")
	limit := response.Header.Get("X-Ratelimit-Limit")
	reset := response.Header.Get("X-Ratelimit-Reset")

	now := time.Now()
	remainingInt, err := strconv.ParseUint(remaining, 10, 64)
	if err != nil {
		return ratelimits, err
	}

	limitInt, err := strconv.ParseUint(limit, 10, 64)
	if err != nil {
		return ratelimits, err
	}

	resetInt, err := strconv.Atoi(reset)
	if err != nil {
		return ratelimits, err
	}

	ratelimits.Remaining = uint(remainingInt)
	ratelimits.Limit = uint(limitInt)
	resetTime := time.Unix(int64(resetInt), 0)
	ratelimits.Reset = resetTime
	if remainingInt == 0 {
		ratelimits.SuggestedWait = resetTime
	} else {
		ratelimits.SuggestedWait = now.Add(resetTime.Sub(now) / time.Duration(remainingInt))
	}

	return ratelimits, nil
}