path: root/cmd/cetus-nasa/cetus-nasa.go

package main

import (
	"flag"
	"fmt"
	"log"
	"time"

	"framagit.org/andinus/cetus/pkg/background"
	"framagit.org/andinus/cetus/pkg/cetus"
	"framagit.org/andinus/cetus/pkg/nasa"
)

var (
	quiet     bool
	version   bool
	fetchOnly bool
	pathOnly  bool

	api         string
	apiKey      string
	date        string
	random      bool
	dateHelp    string
	dateDefault string
	timeout     time.Duration

	err     error
	apodRes nasa.APOD
)

func main() {
	parseFlags()

	if version {
		cetus.Version()
		return
	}

	// Convert timeout to seconds
	timeout = timeout * time.Second

	if random {
		date = nasa.RandDate()
	}

	// get response from api
	apodRes, err = getAPODRes()
	if err != nil {
		if len(apodRes.Msg) != 0 {
			log.Println("Message: ", apodRes.Msg)
		}
		log.Fatal(err)
	}

	printDetails(apodRes)

	// if fetchOnly is true then don't set background
	if fetchOnly {
		return
	}

	// if media type is an image then set background
	if apodRes.MediaType == "image" {
		err = background.Set(apodRes.HDURL)
		if err != nil {
			log.Fatal(err)
		}
	}

}

func parseFlags() {
	flag.BoolVar(&quiet, "quiet", false, "No output")
	flag.BoolVar(&version, "version", false, "Cetus version")
	flag.BoolVar(&fetchOnly, "fetch-only", false, "Don't set background, only fetch info")

	dateHelp = fmt.Sprintf("Choose a random date between 1995-06-16 & %s",
		time.Now().UTC().Format("2006-01-02"))
	flag.BoolVar(&random, "random", false, dateHelp)
	flag.BoolVar(&pathOnly, "path-only", false, "Print only path of the image")

	flag.StringVar(&api, "api", "https://api.nasa.gov/planetary/apod", "APOD API URL")
	flag.StringVar(&apiKey, "api-key", "DEMO_KEY", "api.nasa.gov key for expanded usage")

	dateDefault = time.Now().UTC().Format("2006-01-02")
	flag.StringVar(&date, "date", dateDefault, "Date of the APOD image to retrieve")

	flag.DurationVar(&timeout, "timeout", 32*time.Second, "Timeout for http client in seconds")
	flag.Parse()

}

func printDetails(apodRes nasa.APOD) {
	if quiet {
		return
	}
	if pathOnly {
		cetus.PrintPath(apodRes.HDURL)
		return
	}
	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)
	if apodRes.MediaType == "image" {
		fmt.Printf("URL: %s\n\n", apodRes.HDURL)
	} else {
		fmt.Printf("URL: %s\n\n", apodRes.URL)
	}
	fmt.Printf("Explanation: %s\n", apodRes.Explanation)
}

func getAPODRes() (nasa.APOD, error) {
	var apodInfo map[string]string
	apodInfo = make(map[string]string)
	apodInfo["api"] = api
	apodInfo["apiKey"] = apiKey
	apodInfo["date"] = date

	apodRes, err = nasa.APODPath(apodInfo, timeout)

	return apodRes, err
}