mirror of
https://github.com/idanoo/autobrr
synced 2025-07-24 09:19:13 +00:00
306 lines
8.1 KiB
Go
306 lines
8.1 KiB
Go
// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
package ggn
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
"github.com/autobrr/autobrr/pkg/errors"
|
|
"github.com/autobrr/autobrr/pkg/sharedhttp"
|
|
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
const DefaultURL = "https://gazellegames.net/api.php"
|
|
|
|
var ErrUnauthorized = errors.New("unauthorized: bad credentials")
|
|
var ErrForbidden = errors.New("forbidden")
|
|
var ErrTooManyRequests = errors.New("too many requests: rate-limit reached")
|
|
|
|
type ApiClient interface {
|
|
GetTorrentByID(ctx context.Context, torrentID string, freelechToken bool) (*domain.TorrentBasic, error)
|
|
TestAPI(ctx context.Context) (bool, error)
|
|
}
|
|
|
|
type Client struct {
|
|
url string
|
|
client *http.Client
|
|
rateLimiter *rate.Limiter
|
|
APIKey string
|
|
}
|
|
|
|
type OptFunc func(*Client)
|
|
|
|
func WithUrl(url string) OptFunc {
|
|
return func(c *Client) {
|
|
c.url = url
|
|
}
|
|
}
|
|
|
|
func NewClient(apiKey string, opts ...OptFunc) ApiClient {
|
|
c := &Client{
|
|
url: DefaultURL,
|
|
client: &http.Client{
|
|
Timeout: time.Second * 30,
|
|
Transport: sharedhttp.Transport,
|
|
},
|
|
rateLimiter: rate.NewLimiter(rate.Every(5*time.Second), 1), // 5 request every 10 seconds
|
|
APIKey: apiKey,
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
opt(c)
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
type Group struct {
|
|
BbWikiBody string `json:"bbWikiBody"`
|
|
WikiBody string `json:"wikiBody"`
|
|
WikiImage string `json:"wikiImage"`
|
|
Id int `json:"id"`
|
|
Name string `json:"name"`
|
|
Aliases []string `json:"aliases"`
|
|
Year int `json:"year"`
|
|
CategoryId int `json:"categoryId"`
|
|
CategoryName string `json:"categoryName"`
|
|
MasterGroup int `json:"masterGroup"`
|
|
Time string `json:"time"`
|
|
Tags []string `json:"tags"`
|
|
Platform string `json:"platform"`
|
|
}
|
|
|
|
type GameInfo struct {
|
|
Screenshots []string `json:"screenshots"`
|
|
Trailer string `json:"trailer"`
|
|
Rating string `json:"rating"`
|
|
MetaRating struct {
|
|
Score string `json:"score"`
|
|
Percent string `json:"percent"`
|
|
Link string `json:"link"`
|
|
} `json:"metaRating"`
|
|
IgnRating struct {
|
|
Score string `json:"score"`
|
|
Percent string `json:"percent"`
|
|
Link string `json:"link"`
|
|
} `json:"ignRating"`
|
|
GamespotRating struct {
|
|
Score string `json:"score"`
|
|
Percent string `json:"percent"`
|
|
Link string `json:"link"`
|
|
} `json:"gamespotRating"`
|
|
Weblinks struct {
|
|
GamesWebsite string `json:"GamesWebsite"`
|
|
Wikipedia string `json:"Wikipedia"`
|
|
Giantbomb string `json:"Giantbomb"`
|
|
GameFAQs string `json:"GameFAQs"`
|
|
PCGamingWiki string `json:"PCGamingWiki"`
|
|
Steam string `json:"Steam"`
|
|
Amazon string `json:"Amazon"`
|
|
GOG string `json:"GOG"`
|
|
HowLongToBeat string `json:"HowLongToBeat"`
|
|
} `json:"weblinks"`
|
|
//`json:"gameInfo"`
|
|
}
|
|
|
|
type Torrent struct {
|
|
Id int `json:"id"`
|
|
InfoHash string `json:"infoHash"`
|
|
Type string `json:"type"`
|
|
Link string `json:"link"`
|
|
Format string `json:"format"`
|
|
Encoding string `json:"encoding"`
|
|
Region string `json:"region"`
|
|
Language string `json:"language"`
|
|
Remastered bool `json:"remastered"`
|
|
RemasterYear int `json:"remasterYear"`
|
|
RemasterTitle string `json:"remasterTitle"`
|
|
Scene bool `json:"scene"`
|
|
HasCue bool `json:"hasCue"`
|
|
ReleaseTitle string `json:"releaseTitle"`
|
|
ReleaseType string `json:"releaseType"`
|
|
GameDOXType string `json:"gameDOXType"`
|
|
GameDOXVersion string `json:"gameDOXVersion"`
|
|
FileCount int `json:"fileCount"`
|
|
Size uint64 `json:"size"`
|
|
Seeders int `json:"seeders"`
|
|
Leechers int `json:"leechers"`
|
|
Snatched int `json:"snatched"`
|
|
FreeTorrent bool `json:"freeTorrent"`
|
|
NeutralTorrent bool `json:"neutralTorrent"`
|
|
Reported bool `json:"reported"`
|
|
Time string `json:"time"`
|
|
BbDescription string `json:"bbDescription"`
|
|
Description string `json:"description"`
|
|
FileList []struct {
|
|
Ext string `json:"ext"`
|
|
Size string `json:"size"`
|
|
Name string `json:"name"`
|
|
} `json:"fileList"`
|
|
FilePath string `json:"filePath"`
|
|
UserId int `json:"userId"`
|
|
Username string `json:"username"`
|
|
}
|
|
|
|
type TorrentResponse struct {
|
|
Group Group `json:"group"`
|
|
Torrent Torrent `json:"torrent"`
|
|
}
|
|
|
|
type Response struct {
|
|
Status string `json:"status"`
|
|
Response TorrentResponse `json:"response,omitempty"`
|
|
Error string `json:"error,omitempty"`
|
|
}
|
|
|
|
type GetIndexResponse struct {
|
|
Status string `json:"status"`
|
|
Response struct {
|
|
ApiVersion string `json:"api_version"`
|
|
} `json:"response"`
|
|
}
|
|
|
|
func (c *Client) Do(req *http.Request) (*http.Response, error) {
|
|
ctx := context.Background()
|
|
err := c.rateLimiter.Wait(ctx) // This is a blocking call. Honors the rate limit
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error waiting for ratelimiter")
|
|
}
|
|
resp, err := c.client.Do(req)
|
|
if err != nil {
|
|
return resp, errors.Wrap(err, "error making request")
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (c *Client) get(ctx context.Context, url string) (*http.Response, error) {
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, http.NoBody)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "ggn client request error : %s", url)
|
|
}
|
|
|
|
req.Header.Add("X-API-Key", c.APIKey)
|
|
req.Header.Set("User-Agent", "autobrr")
|
|
|
|
res, err := c.Do(req)
|
|
if err != nil {
|
|
return res, errors.Wrap(err, "ggn client request error : %s", url)
|
|
}
|
|
|
|
if res.StatusCode == http.StatusUnauthorized {
|
|
return res, ErrUnauthorized
|
|
} else if res.StatusCode == http.StatusForbidden {
|
|
return res, ErrForbidden
|
|
} else if res.StatusCode == http.StatusTooManyRequests {
|
|
return res, ErrTooManyRequests
|
|
}
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func (c *Client) getJSON(ctx context.Context, params url.Values, data any) error {
|
|
reqUrl := fmt.Sprintf("%s?%s", c.url, params.Encode())
|
|
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, reqUrl, http.NoBody)
|
|
if err != nil {
|
|
return errors.Wrap(err, "ggn client request error : %s", reqUrl)
|
|
}
|
|
|
|
req.Header.Add("X-API-Key", c.APIKey)
|
|
req.Header.Set("User-Agent", "autobrr")
|
|
|
|
res, err := c.Do(req)
|
|
if err != nil {
|
|
return errors.Wrap(err, "ggn client request error : %s", reqUrl)
|
|
}
|
|
|
|
defer res.Body.Close()
|
|
|
|
if res.StatusCode == http.StatusUnauthorized {
|
|
return ErrUnauthorized
|
|
} else if res.StatusCode == http.StatusForbidden {
|
|
return ErrForbidden
|
|
} else if res.StatusCode == http.StatusTooManyRequests {
|
|
return ErrTooManyRequests
|
|
}
|
|
|
|
reader := bufio.NewReader(res.Body)
|
|
|
|
if err := json.NewDecoder(reader).Decode(&data); err != nil {
|
|
return errors.Wrap(err, "error unmarshal body")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) GetTorrentByID(ctx context.Context, torrentID string, freeleechToken bool) (*domain.TorrentBasic, error) {
|
|
if torrentID == "" {
|
|
return nil, errors.New("ggn client: must have torrentID")
|
|
}
|
|
|
|
var response *Response
|
|
|
|
params := url.Values{}
|
|
params.Add("request", "torrent")
|
|
params.Add("id", torrentID)
|
|
if freeleechToken {
|
|
params.Add("usetoken", "1")
|
|
}
|
|
|
|
err := c.getJSON(ctx, params, &response)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error getting data")
|
|
}
|
|
|
|
if response.Status != "success" {
|
|
return nil, errors.New("bad status: %s", response.Status)
|
|
}
|
|
|
|
t := &domain.TorrentBasic{
|
|
Id: strconv.Itoa(response.Response.Torrent.Id),
|
|
InfoHash: response.Response.Torrent.InfoHash,
|
|
Size: strconv.FormatUint(response.Response.Torrent.Size, 10),
|
|
}
|
|
|
|
return t, nil
|
|
}
|
|
|
|
// TestAPI try api access against index
|
|
func (c *Client) TestAPI(ctx context.Context) (bool, error) {
|
|
resp, err := c.GetIndex(ctx)
|
|
if err != nil {
|
|
return false, errors.Wrap(err, "test api error")
|
|
}
|
|
|
|
if resp == nil {
|
|
return false, nil
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
func (c *Client) GetIndex(ctx context.Context) (*GetIndexResponse, error) {
|
|
var response *GetIndexResponse
|
|
|
|
params := url.Values{}
|
|
err := c.getJSON(ctx, params, &response)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error getting data")
|
|
}
|
|
|
|
if response.Status != "success" {
|
|
return nil, errors.New("bad status: %s", response.Status)
|
|
}
|
|
|
|
return response, nil
|
|
}
|