package flaresolverr

import (
	"net/http"
	"time"

	"resty.dev/v3"
)

func NewRestyClient() *resty.Client {
	restyClient := resty.New()
	restyClient.SetRetryCount(3).SetRetryWaitTime(3 * time.Second).AddRetryConditions(
		func(r *resty.Response, err error) bool {
			return err != nil || r.StatusCode() == http.StatusTooManyRequests
		},
	)
	restyClient.SetLogger(disableLogger{}).SetHeader("Accept-Charset", "utf-8")
	return restyClient
}

type disableLogger struct{}

func (d disableLogger) Errorf(format string, v ...interface{}) {}
func (d disableLogger) Warnf(format string, v ...interface{})  {}
func (d disableLogger) Debugf(format string, v ...interface{}) {}