From a5ff830a5b2338581e9326bed68d1c219d8c6157 Mon Sep 17 00:00:00 2001
From: Andinus <andinus@inventati.org>
Date: Sun, 15 Mar 2020 12:54:41 +0530
Subject: 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.
---
 cmd/cetus-nasa/cetus-nasa.go | 26 ++++++++++++++++++--------
 1 file changed, 18 insertions(+), 8 deletions(-)

(limited to 'cmd/cetus-nasa/cetus-nasa.go')

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)
+}
-- 
cgit 1.4.1-2-gfad0