NyxErinys/crunchy-cli
1
0
Fork 0
mirror of https://github.com/crunchy-labs/crunchy-cli.git synced 2026-01-21 12:12:00 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add custom useragent for cli request

Browse source
This commit is contained in:
bytedream 2022-04-16 01:07:56 +02:00
parent 3617955bc5
commit 598e460e6c
3 changed files with 47 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

37
cmd/crunchyroll-go/cmd/utils.go
View file

@ -71,7 +71,23 @@ func allLocalesAsStrings() (locales []string) {
return
}
func createOrDefaultClient(proxy string) (*http.Client, error) {
type headerRoundTripper struct {
http.RoundTripper
header map[string]string
}
func (rht headerRoundTripper) RoundTrip(r *http.Request) (*http.Response, error) {
resp, err := rht.RoundTripper.RoundTrip(r)
if err != nil {
return nil, err
}
for k, v := range rht.header {
resp.Header.Set(k, v)
}
return resp, nil
}
func createOrDefaultClient(proxy, useragent string) (*http.Client, error) {
if proxy == "" {
return http.DefaultClient, nil
} else {
@ -80,12 +96,21 @@ func createOrDefaultClient(proxy string) (*http.Client, error) {
if err != nil {
return nil, err
}
var rt http.RoundTripper = &http.Transport{
DisableCompression: true,
Proxy: http.ProxyURL(proxyURL),
}
if useragent != "" {
rt = headerRoundTripper{
RoundTripper: rt,
header: map[string]string{"User-Agent": useragent},
}
}
client := &http.Client{
Transport: &http.Transport{
DisableCompression: true,
Proxy: http.ProxyURL(proxyURL),
},
Timeout: 30 * time.Second,
Transport: rt,
Timeout: 30 * time.Second,
}
return client, nil
}