Add Wallhaven support

2 files changed, 171 insertions, 0 deletions

diff --git a/cmd/cetus-wallhaven/cetus-wallhaven.go b/cmd/cetus-wallhaven/cetus-wallhaven.go
new file mode 100644
index 0000000..5314db9
--- /dev/null
+++ b/cmd/cetus-wallhaven/cetus-wallhaven.go
@@ -0,0 +1,151 @@
+package main
+
+import (
+	"encoding/json"
+	"flag"
+	"fmt"
+	"math/rand"
+	"time"
+
+	"framagit.org/andinus/cetus/pkg/background"
+	"framagit.org/andinus/cetus/pkg/cetus"
+	"framagit.org/andinus/cetus/pkg/wallhaven"
+)
+
+type wh struct {
+	Photos []photo `json:"data"`
+	// MetaI  []meta  `json:"meta"`
+}
+
+// type meta struct {
+// 	CurrentPage int64  `json:"current_page"`
+// 	LastPage    int64  `json:"last_page"`
+// 	PerPage     int64  `json:"per_page"`
+// 	Total       int64  `json:"total"`
+// 	Query       string `json:"query"`
+// 	Seed        string `json:"seed"`
+// }
+
+type photo struct {
+	Id       string `json:"id"`
+	Url      string `json:"url"`
+	ShortUrl string `json:"short_url"`
+	// Views      int      `json:"views"`
+	// Favorites  int      `json:"favorites"`
+	Source   string `json:"source"`
+	Purity   string `json:"purity"`
+	Category string `json:"general"`
+	// DimensionX int      `json:"dimension_x"`
+	// DimensionY int      `json:"dimension_y"`
+	Resolution string `json:"resolution"`
+	Ratio      string `json:"ratio"`
+	// FileSize   int      `json:"file_size"`
+	FileType  string   `json:"file_type"`
+	CreatedAt string   `json:"created_at"`
+	Path      string   `json:"path"`
+	Colors    []string `json:"colors"`
+	// Thumbs     []thumb  `json:"thumbs"`
+}
+
+// type thumb struct {
+// 	Small    string `json:"small"`
+// 	Original string `json:"original"`
+// 	Large    string `json:"large"`
+// }
+
+var (
+	t         time.Duration
+	api       string
+	apiKey    string
+	dump      bool
+	quiet     bool
+	random    bool
+	version   bool
+	pathOnly  bool
+	fetchOnly bool
+)
+
+func main() {
+	parseFlags()
+
+	if version {
+		cetus.Version()
+		return
+	}
+	rand.Seed(time.Now().Unix())
+
+	// Convert timeout to seconds
+	t = t * time.Second
+
+	body, err := whBody()
+	if dump {
+		fmt.Println(body)
+		return
+	}
+
+	wh := wh{}
+	err = json.Unmarshal([]byte(body), &wh)
+	cetus.ErrChk("body unmarshal failed", err)
+
+	// if random was set then wh holds list of multiple responses,
+	// choose a random response from the list
+	i := rand.Intn(len(wh.Photos))
+	whPhoto := wh.Photos[i]
+
+	printDetails(whPhoto)
+
+	// if fetchOnly is true then don't set background
+	if fetchOnly {
+		return
+	}
+
+	err = background.Set(whPhoto.Path)
+	cetus.ErrChk("setting background failed", 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")
+	flag.BoolVar(&dump, "dump", false, "Only dump received response")
+	flag.BoolVar(&random, "random", true, "Choose a random image")
+	flag.BoolVar(&pathOnly, "path-only", false, "Print only path of the image")
+
+	flag.StringVar(&api, "api", "https://wallhaven.cc/api/v1/search", "Wallhaven Search API URL")
+	flag.StringVar(&apiKey, "api-key", "", "Wallhaven API Key")
+
+	flag.DurationVar(&t, "timeout", 32*time.Second, "Timeout for http client in seconds")
+	flag.Parse()
+
+}
+
+func printDetails(whPhoto photo) {
+	if quiet {
+		return
+	}
+	if pathOnly {
+		cetus.PrintPath(whPhoto.Path)
+		return
+	}
+	fmt.Printf("Id: %s\n", whPhoto.Id)
+	fmt.Printf("URL: %s\n", whPhoto.Url)
+	fmt.Printf("Short URL: %s\n", whPhoto.ShortUrl)
+	fmt.Printf("Source: %s\n", whPhoto.Source)
+	fmt.Printf("Date: %s\n\n", whPhoto.CreatedAt)
+	fmt.Printf("Resolution: %s\n", whPhoto.Resolution)
+	fmt.Printf("Ratio: %s\n", whPhoto.Ratio)
+	// fmt.Printf("Views: %s\n", whPhoto.Views)
+	// fmt.Printf("Favorites: %s\n", whPhoto.Favorites)
+	fmt.Printf("Category: %s\n", whPhoto.Category)
+}
+
+func whBody() (string, error) {
+	reqInfo := make(map[string]string)
+	reqInfo["api"] = api
+	if random {
+		reqInfo["random"] = "true"
+	}
+
+	body, err := wallhaven.GetWhJson(reqInfo, t)
+	return body, err
+}
diff --git a/pkg/wallhaven/wallhaven.go b/pkg/wallhaven/wallhaven.go
new file mode 100644
index 0000000..25af77c
--- /dev/null
+++ b/pkg/wallhaven/wallhaven.go
@@ -0,0 +1,20 @@
+package wallhaven
+
+import (
+	"time"
+
+	"framagit.org/andinus/cetus/pkg/cetus"
+)
+
+// GetWhJson returns json response received from the api
+func GetWhJson(reqInfo map[string]string, t time.Duration) (string, error) {
+	params := make(map[string]string)
+	params["apikey"] = reqInfo["apiKey"]
+	if reqInfo["random"] == "true" {
+		params["sorting"] = "random"
+
+	}
+
+	body, err := cetus.GetRes(reqInfo["api"], params, t)
+	return string(body), err
+}