// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors. // SPDX-License-Identifier: GPL-2.0-or-later package btn import ( "context" "io" "log" "net/http" "time" "github.com/autobrr/autobrr/internal/domain" "github.com/autobrr/autobrr/pkg/jsonrpc" "github.com/autobrr/autobrr/pkg/sharedhttp" "golang.org/x/time/rate" ) const DefaultURL = "https://api.broadcasthe.net/" type ApiClient interface { GetTorrentByID(ctx context.Context, torrentID string, freeleechToken bool) (*domain.TorrentBasic, error) TestAPI(ctx context.Context) (bool, error) } type OptFunc func(*Client) func WithUrl(url string) OptFunc { return func(c *Client) { c.url = url } } type Client struct { rpcClient jsonrpc.Client rateLimiter *rate.Limiter APIKey string url string Log *log.Logger } func NewClient(apiKey string, opts ...OptFunc) ApiClient { c := &Client{ url: DefaultURL, rateLimiter: rate.NewLimiter(rate.Every(150*time.Hour), 1), // 150 rpcRequest every 1 hour APIKey: apiKey, } for _, opt := range opts { opt(c) } c.rpcClient = jsonrpc.NewClientWithOpts(c.url, &jsonrpc.ClientOpts{ Headers: map[string]string{ "User-Agent": "autobrr", }, HTTPClient: &http.Client{ Timeout: time.Second * 60, Transport: sharedhttp.Transport, }, }) if c.Log == nil { c.Log = log.New(io.Discard, "", log.LstdFlags) } return c }