fmartingr
/
games-screenshot-manager
mirror of https://github.com/fmartingr/games-screenshot-manager.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Moved Download function to an helper module

This commit is contained in:
Felipe M 2021-01-05 18:22:04 +01:00
parent 0b33e24472
commit fa7016eb61
2 changed files with 35 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/helpers/http.go
View File

@ -1,6 +1,7 @@
package helpers
import (
"io/ioutil"
"net/http"
"net/url"
)
@ -18,3 +19,28 @@ func DoRequest(method string, requestURL string) (*http.Response, error) {
}
return http.DefaultClient.Do(&request)
}
func DownloadURLIntoTempFile(requestURL string) (string, error) {
response, err := DoRequest("GET", requestURL)
if err != nil {
return "", err
}
if response.Body != nil {
defer response.Body.Close()
}
tmpfile, err := ioutil.TempFile("", "games-screenshot-manager")
if err != nil {
return "", err
}
body, err := ioutil.ReadAll(response.Body)
if err != nil {
return "", err
}
tmpfile.Write(body)
return tmpfile.Name(), nil
}

31
pkg/providers/steam/steam.go
View File

@ -88,29 +88,13 @@ func getSteamAppList(c chan SteamAppList) {
c <- steamListResponse.AppList
}
func downloadGameHeaderImage(appId string) string {
response, err := helpers.DoRequest("GET", "https://cdn.cloudflare.steamstatic.com/steam/apps/"+appId+"/header.jpg")
func downloadGameHeaderImage(appId string) (string, error) {
url := "https://cdn.cloudflare.steamstatic.com/steam/apps/" + appId + "/header.jpg"
coverURL, err := helpers.DownloadURLIntoTempFile(url)
if err != nil {
panic(err)
return "", err
}
if response.Body != nil {
defer response.Body.Close()
}
tmpfile, err := ioutil.TempFile("", "games-screenshot-manager-cover")
if err != nil {
panic(err)
}
body, err := ioutil.ReadAll(response.Body)
if err != nil {
panic(err)
}
tmpfile.Write(body)
return tmpfile.Name()
return coverURL, nil
}
func GuessUsers() []string {
@ -188,7 +172,10 @@ func GetGames(downloadCovers bool) []games.Game {
userGame := games.Game{ID: userGameID, Name: steamGame.Name, Provider: providerName, Platform: "PC"}
if downloadCovers {
userGame.Cover = games.Screenshot{Path: downloadGameHeaderImage(userGameID), DestinationName: ".cover"}
coverPath, err := downloadGameHeaderImage(userGameID)
if err == nil {
userGame.Cover = games.Screenshot{Path: coverPath, DestinationName: ".cover"}
}
}
log.Printf("Found Steam game for user %s: %s (%s)", userID, userGame.Name, userGame.ID)