Feat/Flaresolverr support (#12)
* new: feat: add flaresolverr support * chg: feat: add session pool * chg: fix: deadlock error * chg: fix: make it work without flaresolverr
This commit is contained in:
93
requester/requester.go
Normal file
93
requester/requester.go
Normal file
@@ -0,0 +1,93 @@
|
||||
package requester
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"regexp"
|
||||
"time"
|
||||
|
||||
"github.com/felipemarinho97/torrent-indexer/cache"
|
||||
)
|
||||
|
||||
const (
|
||||
shortLivedCacheExpiration = 30 * time.Minute
|
||||
cacheKey = "shortLivedCache"
|
||||
)
|
||||
|
||||
var challangeRegex = regexp.MustCompile(`(?i)(just a moment|cf-chl-bypass|under attack)`)
|
||||
|
||||
type Requster struct {
|
||||
fs *FlareSolverr
|
||||
c *cache.Redis
|
||||
httpClient *http.Client
|
||||
}
|
||||
|
||||
func NewRequester(fs *FlareSolverr, c *cache.Redis) *Requster {
|
||||
return &Requster{fs: fs, httpClient: &http.Client{}, c: c}
|
||||
}
|
||||
|
||||
func (i *Requster) GetDocument(ctx context.Context, url string) (io.ReadCloser, error) {
|
||||
var body io.ReadCloser
|
||||
|
||||
// try request from short-lived cache
|
||||
key := fmt.Sprintf("%s:%s", cacheKey, url)
|
||||
bodyByte, err := i.c.Get(ctx, key)
|
||||
if err == nil {
|
||||
fmt.Printf("returning from short-lived cache: %s\n", url)
|
||||
body = io.NopCloser(bytes.NewReader(bodyByte))
|
||||
return body, nil
|
||||
}
|
||||
|
||||
// try request with plain client
|
||||
resp, err := i.httpClient.Get(url)
|
||||
if err != nil {
|
||||
// try request with flare solverr
|
||||
body, err = i.fs.Get(url)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to do request for url %s: %w", url, err)
|
||||
}
|
||||
} else {
|
||||
defer resp.Body.Close()
|
||||
body = resp.Body
|
||||
}
|
||||
|
||||
bodyByte, err = io.ReadAll(body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to read response body: %w", err)
|
||||
}
|
||||
if hasChallange(bodyByte) {
|
||||
// try request with flare solverr
|
||||
body, err = i.fs.Get(url)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to do request for url %s: %w", url, err)
|
||||
}
|
||||
bodyByte, err = io.ReadAll(body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to read response body: %w", err)
|
||||
}
|
||||
fmt.Printf("request served from flaresolverr: %s\n", url)
|
||||
} else {
|
||||
fmt.Printf("request served from plain client: %s\n", url)
|
||||
}
|
||||
|
||||
// save response to cache if it's not a challange and body is not empty
|
||||
if !hasChallange(bodyByte) && len(bodyByte) > 0 {
|
||||
err = i.c.SetWithExpiration(ctx, key, bodyByte, shortLivedCacheExpiration)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to save response to cache: %v\n", err)
|
||||
}
|
||||
fmt.Printf("saved to cache: %s\n", url)
|
||||
} else {
|
||||
return nil, fmt.Errorf("response is a challange")
|
||||
}
|
||||
|
||||
return io.NopCloser(bytes.NewReader(bodyByte)), nil
|
||||
}
|
||||
|
||||
// hasChallange checks if the body contains a challange by regex matching
|
||||
func hasChallange(body []byte) bool {
|
||||
return challangeRegex.Match(body)
|
||||
}
|
||||
Reference in New Issue
Block a user