diff options
| -rw-r--r-- | cmd/cetus-nasa/cetus-nasa.go | 26 |
1 files changed, 18 insertions, 8 deletions
diff --git a/cmd/cetus-nasa/cetus-nasa.go b/cmd/cetus-nasa/cetus-nasa.go index 7eb80c2..d90a08b 100644 --- a/cmd/cetus-nasa/cetus-nasa.go +++ b/cmd/cetus-nasa/cetus-nasa.go @@ -75,11 +75,7 @@ func main() { } if !quiet { - fmt.Printf("Title: %s\n\n", apodRes.Title) - fmt.Printf("Copyright: %s\n", apodRes.Copyright) - fmt.Printf("Date: %s\n\n", apodRes.Date) - fmt.Printf("URL: %s\n\n", apodRes.HDURL) - fmt.Printf("Explanation: %s\n", apodRes.Explanation) + printDetails(apodRes) } // if fetchOnly is true then don't set background @@ -87,10 +83,15 @@ func main() { return } picturePath = apodRes.HDURL - err = background.Set(picturePath) - if err != nil { - log.Fatal(err) + + // if media type is an image then set background + if apodRes.MediaType == "image" { + err = background.Set(picturePath) + if err != nil { + log.Fatal(err) + } } + } func parseFlags() { @@ -112,3 +113,12 @@ func parseFlags() { flag.Parse() } + +func printDetails(apodRes nasa.APOD) { + fmt.Printf("Title: %s\n\n", apodRes.Title) + fmt.Printf("Copyright: %s\n", apodRes.Copyright) + fmt.Printf("Date: %s\n\n", apodRes.Date) + fmt.Printf("Media Type: %s\n", apodRes.MediaType) + fmt.Printf("URL: %s\n\n", apodRes.HDURL) + fmt.Printf("Explanation: %s\n", apodRes.Explanation) +} |