Mirrors/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2025-12-18 04:44:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add User Agent to image download reqs (#1222)

Browse Source
This commit is contained in:
bnkai
2021-03-23 23:12:11 +02:00
committed by GitHub
parent 73a8bad1bc
commit 68d4a4fe42
4 changed files with 55 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/scraper/image.go
View File

@@ -2,6 +2,7 @@ package scraper
import (
"crypto/tls"
"fmt"
"io/ioutil"
"net/http"
"strings"
@@ -113,6 +114,10 @@ func getImage(url string, globalConfig GlobalConfig) (*string, error) {
return nil, err
}
if resp.StatusCode >= 400 {
return nil, fmt.Errorf("http error %d", resp.StatusCode)
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)

4
pkg/scraper/url.go
View File

@@ -79,6 +79,10 @@ func loadURL(url string, scraperConfig config, globalConfig GlobalConfig) (io.Re
if err != nil {
return nil, err
}
if resp.StatusCode >= 400 {
return nil, fmt.Errorf("http error %d", resp.StatusCode)
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)