Check media type

Today's media type was video & that broke the program, currently this
will check the media type & set background only if it's an image. It
exits if media type is not an image, this behaviour might change
later.

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)
+}