idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 08:49:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: download client and ipt download url

Browse source
This commit is contained in:
Ludvig Lundgren 2021-08-30 20:54:14 +02:00
parent 9aaf79e7bb
commit 2cba7a90e0
3 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
internal/client/http.go
View file

@ -45,6 +45,7 @@ func (c *HttpClient) DownloadFile(url string, opts map[string]string) (*Download
// Create the file
out, err := os.Create(tmpFileName)
if err != nil {
log.Error().Stack().Err(err).Msgf("error creating temp file: %v", tmpFileName)
return nil, err
}
@ -53,20 +54,24 @@ func (c *HttpClient) DownloadFile(url string, opts map[string]string) (*Download
// Get the data
resp, err := http.Get(url)
if err != nil {
// TODO better error message
log.Error().Stack().Err(err).Msgf("error downloading file %v from %v", tmpFileName, url)
return nil, err
}
defer resp.Body.Close()
// retry logic
log.Trace().Msgf("downloaded file response: %v - status: %v", resp.Status, resp.StatusCode)
if resp.StatusCode != 200 {
log.Error().Stack().Err(err).Msgf("error downloading file: %v - bad status: %d", tmpFileName, resp.StatusCode)
return nil, err
}
// Write the body to file
_, err = io.Copy(out, resp.Body)
if err != nil {
log.Error().Stack().Err(err).Msgf("error writing downloaded file: %v", tmpFileName)
return nil, err
}
@ -77,5 +82,7 @@ func (c *HttpClient) DownloadFile(url string, opts map[string]string) (*Download
FileName: tmpFileName,
}
log.Trace().Msgf("successfully downloaded file: %v", tmpFileName)
return &res, nil
}