59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/diamondburned/arikawa/api/rate"
|
|
"github.com/diamondburned/arikawa/httputil"
|
|
)
|
|
|
|
const (
|
|
BaseEndpoint = "https://discordapp.com/api"
|
|
APIVersion = "6"
|
|
|
|
Endpoint = BaseEndpoint + "/v" + APIVersion + "/"
|
|
EndpointGateway = Endpoint + "gateway"
|
|
EndpointGatewayBot = EndpointGateway + "/bot"
|
|
EndpointWebhooks = Endpoint + "webhooks/"
|
|
)
|
|
|
|
var UserAgent = "DiscordBot (https://github.com/diamondburned/arikawa, v0.0.1)"
|
|
|
|
type Client struct {
|
|
httputil.Client
|
|
Limiter *rate.Limiter
|
|
|
|
Token string
|
|
}
|
|
|
|
func NewClient(token string) *Client {
|
|
cli := &Client{
|
|
Client: httputil.NewClient(),
|
|
Limiter: rate.NewLimiter(),
|
|
Token: token,
|
|
}
|
|
|
|
tw := httputil.NewTransportWrapper()
|
|
tw.Pre = func(r *http.Request) error {
|
|
if r.Header.Get("Authorization") == "" {
|
|
r.Header.Set("Authorization", cli.Token)
|
|
}
|
|
|
|
if r.UserAgent() == "" {
|
|
r.Header.Set("User-Agent", UserAgent)
|
|
}
|
|
|
|
r.Header.Set("X-RateLimit-Precision", "millisecond")
|
|
|
|
// Rate limit stuff
|
|
return cli.Limiter.Acquire(r.Context(), r.URL.Path)
|
|
}
|
|
tw.Post = func(r *http.Response) error {
|
|
return cli.Limiter.Release(r.Request.URL.Path, r.Header)
|
|
}
|
|
|
|
cli.Client.Transport = tw
|
|
|
|
return cli
|
|
}
|