diff options
Diffstat (limited to 'main.go')
| -rw-r--r-- | main.go | 154 |
1 files changed, 103 insertions, 51 deletions
diff --git a/main.go b/main.go index 3f52fb8..95e91fa 100644 --- a/main.go +++ b/main.go @@ -20,84 +20,105 @@ import ( "fmt" "io/ioutil" "log" + "math/rand" "net/http" "os/exec" + "strconv" "time" ) var ( - err error timeout time.Duration + + apodAPI string + apodAPIKey string + bpodAPI string + bpodNum int + unsplashAPI string ) func main() { - var ( - imgPath string + rand.Seed(time.Now().Unix()) - apod bool - apodAPI string - apodAPIKey string + var ( + err error - bpod bool - bpodAPI string + imgPath string + mode string + src string + srcArr []string = []string{ + "apod", + "bpod", + "unsplash", + } ) // Parse flags passed to program - flag.StringVar(&imgPath, "img-path", "", "Image to set as wallpaper") + flag.StringVar(&src, "src", "random", "Source for the image") + flag.StringVar(&mode, "mode", "random", "Daily, Weekly or Random wallpaper") - flag.BoolVar(&apod, "apod", false, "Set Astronomy Picture of the Day as wallpaper") flag.StringVar(&apodAPI, "apod-api", "https://api.nasa.gov/planetary/apod", "APOD API URL") flag.StringVar(&apodAPIKey, "apod-api-key", "DEMO_KEY", "APOD API Key") - - flag.BoolVar(&bpod, "bpod", false, "Set Bing Photo of the Day as wallpaper") flag.StringVar(&bpodAPI, "bpod-api", "https://www.bing.com/HPImageArchive.aspx", "BPOD API URL") - + flag.IntVar(&bpodNum, "bpod-num", 16, "BPOD Number of images to fetch") flag.DurationVar(&timeout, "timeout", 16, "Timeout for http client") - flag.Parse() - if len(imgPath) > 0 { - err = setWall(imgPath) - errChk(err) - return + if src == "random" { + src = srcArr[rand.Intn(len(srcArr))] } - if apod { - apodI := make(map[string]string) - apodI["api"] = apodAPI - apodI["apiKey"] = apodAPIKey - - err = setWallFromAPOD(apodI) - errChk(err) - return + // Check if the source is known + if !contains(srcArr, src) { + log.Fatal("Error: Unknown Source") } - if bpod { - bpodI := make(map[string]string) - bpodI["api"] = bpodAPI + imgPath, err = parseSrcAndGetPath(src, mode) + errChk(err) + + err = setWall(imgPath) + errChk(err) +} - err = setWallFromBPOD(bpodI) - errChk(err) - return +func contains(arr []string, str string) bool { + for _, i := range arr { + if i == str { + return true + } } + return false } -// Calls feh to set the wallpaper -func setWall(imgPath string) error { - feh, err := exec.LookPath("feh") - if err != nil { - fmt.Println("Error: feh is not in $PATH") - return err +// Gets image path from src +func parseSrcAndGetPath(src string, mode string) (string, error) { + var err error + var imgPath string + + switch src { + case "apod": + fmt.Println("Astronomy Picture of the Day") + imgPath, err = getPathAPOD(mode) + case "bpod": + fmt.Println("Bing Photo of the Day") + imgPath, err = getPathBPOD(mode) + case "unsplash": + fmt.Println("Unsplash Source") + imgPath, err = getPathUnsplash(mode) } - fmt.Printf("Path to set as Wallpaper: %s\n", imgPath) + return imgPath, err +} - err = exec.Command(feh, "--bg-fill", imgPath).Run() - return err +func getPathAPOD(mode string) (string, error) { + var err error + var imgPath string + return imgPath, err } -// Get url of Bing Photo of the Day & pass it to setWall() -func setWallFromBPOD(bpodI map[string]string) error { +func getPathBPOD(mode string) (string, error) { + var err error + var imgPath string + type Images struct { StartDate string `json:"startdate"` FullStartDate string `json:"fullstartdate"` @@ -116,34 +137,65 @@ func setWallFromBPOD(bpodI map[string]string) error { bpodNow := bpodRes{} - req, err := http.NewRequest(http.MethodGet, bpodI["api"], nil) + req, err := http.NewRequest(http.MethodGet, bpodAPI, nil) if err != nil { - return err + return "", err } q := req.URL.Query() q.Add("format", "js") - q.Add("n", "1") + + switch mode { + case "daily": + q.Add("n", "1") + case "random": + // Fetches 16 images (only info) & chooses a random image + q.Add("n", strconv.Itoa(bpodNum)) + default: + return "", fmt.Errorf("Error: Unknown Mode") + } + req.URL.RawQuery = q.Encode() res, err := getRes(req) if err != nil { - fmt.Printf("Error: GET %s\n", bpodI["api"]) - return err + return "", err } defer res.Body.Close() apiBody, err := ioutil.ReadAll(res.Body) if err != nil { - return err + return "", err } err = json.Unmarshal([]byte(apiBody), &bpodNow) if err != nil { + return "", err + } + + // Choose a random image + var i int = rand.Intn(len(bpodNow.Image)) + imgPath = fmt.Sprintf("%s%s", "https://www.bing.com", bpodNow.Image[i].URL) + + return imgPath, err +} + +func getPathUnsplash(mode string) (string, error) { + var err error + var imgPath string + return imgPath, err +} + +// Calls feh to set the wallpaper +func setWall(imgPath string) error { + feh, err := exec.LookPath("feh") + if err != nil { + fmt.Println("Error: feh is not in $PATH") return err } - // Set Bing Photo of the Day as wallpaper - err = setWall(fmt.Sprintf("%s%s", "https://www.bing.com", bpodNow.Image[0].URL)) + fmt.Printf("Path to set as Wallpaper: %s\n", imgPath) + + err = exec.Command(feh, "--bg-fill", imgPath).Run() return err } |