mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-09-19 18:12:55 +00:00
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package twitch
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const subInfoCacheTimeout = 300 * time.Second
|
|
|
|
type (
|
|
subInfo struct {
|
|
Total int64 `json:"total"`
|
|
Points int64 `json:"points"`
|
|
}
|
|
)
|
|
|
|
// GetBroadcasterSubscriptionCount gets a list of users that subscribe to the specified broadcaster.
|
|
func (c *Client) GetBroadcasterSubscriptionCount(ctx context.Context, broadcasterName string) (subCount, subPoints int64, err error) {
|
|
cacheKey := []string{"broadcasterSubscriptionCountByChannel", broadcasterName}
|
|
if d := c.apiCache.Get(cacheKey); d != nil {
|
|
data := d.(subInfo)
|
|
return data.Total, data.Points, nil
|
|
}
|
|
|
|
broadcaster, err := c.GetIDForUsername(ctx, broadcasterName)
|
|
if err != nil {
|
|
return 0, 0, errors.Wrap(err, "getting ID for broadcaster name")
|
|
}
|
|
|
|
var data subInfo
|
|
|
|
if err = c.Request(ctx, ClientRequestOpts{
|
|
AuthType: AuthTypeBearerToken,
|
|
Method: http.MethodGet,
|
|
OKStatus: http.StatusOK,
|
|
Out: &data,
|
|
URL: fmt.Sprintf("https://api.twitch.tv/helix/subscriptions?broadcaster_id=%s", broadcaster),
|
|
}); err != nil {
|
|
return 0, 0, errors.Wrap(err, "executing request")
|
|
}
|
|
|
|
// Lets not annoy the API but only ask every 5m
|
|
c.apiCache.Set(cacheKey, subInfoCacheTimeout, data)
|
|
|
|
return data.Total, data.Points, nil
|
|
}
|