141 lines
3.8 KiB
Go
141 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"tildegit.org/andinus/cetus/background"
|
|
"tildegit.org/andinus/cetus/bpod"
|
|
"tildegit.org/andinus/cetus/cache"
|
|
"tildegit.org/andinus/cetus/notification"
|
|
)
|
|
|
|
func execBPOD() {
|
|
bpodApi := getEnv("BPOD_API", "https://www.bing.com/HPImageArchive.aspx")
|
|
|
|
// reqInfo holds all the parameters that needs to be sent with
|
|
// the request. GetJson() will pack apiKey & date in params
|
|
// map before sending it to another function. Adding params
|
|
// here will not change the behaviour of the function, changes
|
|
// have to be made in GetJson() too.
|
|
reqInfo = make(map[string]string)
|
|
reqInfo["api"] = bpodApi
|
|
|
|
if random {
|
|
reqInfo["random"] = "true"
|
|
}
|
|
|
|
cacheDir := fmt.Sprintf("%s/%s", cache.GetDir(), "bpod")
|
|
os.MkdirAll(cacheDir, os.ModePerm)
|
|
|
|
body, err = bpod.GetJson(reqInfo)
|
|
if err != nil {
|
|
err = fmt.Errorf("%s\n%s",
|
|
"bpod.go: failed to get json response from api",
|
|
err.Error())
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if dump {
|
|
fmt.Println(body)
|
|
}
|
|
|
|
res, err := bpod.UnmarshalJson(body)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Correct format
|
|
res.URL = fmt.Sprintf("%s%s", "https://www.bing.com", res.URL)
|
|
dt, err := time.Parse("20060102", res.StartDate)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
res.StartDate = dt.Format("2006-01-02")
|
|
|
|
// Save response in cache after marshalling it again, we do
|
|
// this instead of saving the response so as to not break the
|
|
// format in which cache is saved. If random flag was passed
|
|
// then the response will contain all 7 values so we have to
|
|
// marshal it but why not save non-random directly? Because
|
|
// that means the format in which both are saved will be
|
|
// different. One will be the raw response whereas other will
|
|
// be marshalled response. We're currently not using this body
|
|
// cache but this is just to save information.
|
|
file := fmt.Sprintf("%s/%s.json", cacheDir, res.StartDate)
|
|
body, err = bpod.MarshalJson(res)
|
|
if err != nil {
|
|
// We should warn the user if this returns an error
|
|
// but the program shouldn't exit.
|
|
log.Println("bpod.go: failed to marshal res to body, not saving cache")
|
|
} else {
|
|
err = ioutil.WriteFile(file, []byte(body), 0644)
|
|
// Not being able to write to the cache file is a
|
|
// small error and the program shouldn't exit but
|
|
// should continue after printing the log so that the
|
|
// user can investigate it later.
|
|
if err != nil {
|
|
err = fmt.Errorf("%s%s\n%s",
|
|
"bpod.go: failed to write body to file: ", file,
|
|
err.Error())
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
// Send a desktop notification if notify flag was passed.
|
|
if notify {
|
|
n := notification.Notif{}
|
|
n.Title = res.Title
|
|
n.Message = fmt.Sprintf("%s\n\n%s",
|
|
res.StartDate,
|
|
res.Copyright)
|
|
|
|
err = n.Notify()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
if print {
|
|
fmt.Printf("Title: %s\n\n", res.Title)
|
|
fmt.Printf("Copyright: %s\n", res.Copyright)
|
|
fmt.Printf("Copyright Link: %s\n", res.CopyrightLink)
|
|
fmt.Printf("Date: %s\n\n", res.StartDate)
|
|
fmt.Printf("URL: %s\n", res.URL)
|
|
}
|
|
|
|
// Proceed only if the command was set because if it was fetch
|
|
// then it's already finished & should exit now.
|
|
if os.Args[1] == "fetch" {
|
|
os.Exit(0)
|
|
}
|
|
|
|
// Try to set background only if the media type is an image.
|
|
// First it downloads the image to the cache directory and
|
|
// then tries to set it with feh. If the download fails then
|
|
// it exits with a non-zero exit code.
|
|
imgFile := fmt.Sprintf("%s/%s", cacheDir, res.Title)
|
|
|
|
// Check if the file is available locally, if it is then don't
|
|
// download it again and set it from disk
|
|
if _, err := os.Stat(imgFile); os.IsNotExist(err) {
|
|
err = background.Download(imgFile, res.URL)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
} else {
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
err = background.SetFromFile(imgFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|