wallhaven_dl/main.go

206 lines
5.1 KiB
Go
Raw Permalink Normal View History

2023-10-17 18:33:29 +00:00
package main
2024-02-25 03:11:14 +00:00
import (
"context"
"fmt"
"log"
"math/rand"
"os"
"os/exec"
"path"
"time"
"github.com/urfave/cli/v3"
"git.asdf.cafe/abs3nt/wallhaven_dl/src/wallhaven"
)
2023-10-17 18:33:29 +00:00
2024-07-20 17:45:00 +00:00
var Version = "dev"
2023-10-17 18:33:29 +00:00
func main() {
2024-02-25 03:11:14 +00:00
app := cli.Command{
EnableShellCompletion: true,
2024-07-20 17:45:00 +00:00
Version: Version,
2024-02-25 03:11:14 +00:00
Name: "wallhaven_dl",
Usage: "Download wallpapers from wallhaven.cc",
Commands: []*cli.Command{
{
Name: "search",
Usage: "Search for wallpapers",
Flags: []cli.Flag{
&cli.StringFlag{
Name: "range",
Aliases: []string{"r"},
Value: "1y",
Validator: func(s string) error {
if s != "1d" && s != "3d" && s != "1w" && s != "1M" && s != "3M" && s != "6M" && s != "1y" {
return fmt.Errorf("range must be '1d', '3d', '1w', '1M', '3M', '6M' or '1y'")
}
return nil
},
Usage: "Time range for top sorting",
},
&cli.StringFlag{
Name: "purity",
Aliases: []string{"p"},
Value: "110",
Validator: func(s string) error {
if len(s) != 3 {
return fmt.Errorf("purity must be 3 characters long")
}
for _, c := range s {
if c != '0' && c != '1' {
return fmt.Errorf("purity must be 0 or 1")
}
}
return nil
},
Usage: "Purity of the wallpapers",
},
&cli.StringFlag{
Name: "categories",
Aliases: []string{"c"},
Value: "010",
Validator: func(s string) error {
if len(s) != 3 {
return fmt.Errorf("categories must be 3 characters long")
}
for _, c := range s {
if c != '0' && c != '1' {
return fmt.Errorf("categories must be 0 or 1")
}
}
return nil
},
Usage: "Categories of the wallpapers",
},
&cli.StringFlag{
Name: "sort",
Aliases: []string{"s"},
Value: "toplist",
Validator: func(s string) error {
if s != "relevance" && s != "random" && s != "date_added" && s != "views" && s != "favorites" &&
s != "toplist" {
return fmt.Errorf(
"sort must be 'relevance', 'random', 'date_added', 'views', 'favorites' or 'toplist'",
)
}
return nil
},
Usage: "Sorting of the wallpapers",
},
&cli.StringFlag{
Name: "order",
Aliases: []string{"o"},
Value: "desc",
Validator: func(s string) error {
if s != "asc" && s != "desc" {
return fmt.Errorf("order must be 'asc' or 'desc'")
}
return nil
},
Usage: "Order of the wallpapers",
},
&cli.IntFlag{
Name: "page",
Aliases: []string{"pg"},
Value: 5,
Usage: "Pages to search",
},
&cli.StringSliceFlag{
Name: "ratios",
Aliases: []string{"rt"},
Value: []string{"16x9", "16x10"},
Usage: "Ratios of the wallpapers",
},
&cli.StringFlag{
Name: "atLeast",
Aliases: []string{"al"},
Value: "2560x1440",
Usage: "Minimum resolution",
},
&cli.StringFlag{
Name: "scriptPath",
Aliases: []string{"sp"},
Value: "",
TakesFile: true,
Usage: "Path to the script to run after downloading",
},
&cli.StringFlag{
Name: "downloadPath",
Aliases: []string{"dp"},
Value: os.Getenv("HOME") + "/Pictures/Wallpapers",
TakesFile: true,
Usage: "Path to download the wallpapers",
},
},
Action: func(ctx context.Context, c *cli.Command) error {
return search(c)
},
},
},
}
if err := app.Run(context.Background(), os.Args); err != nil {
log.Fatal(err)
}
}
func search(c *cli.Command) error {
seed := rand.NewSource(time.Now().UnixNano())
r := rand.New(seed)
s := &wallhaven.Search{
Categories: c.String("categories"),
Purities: c.String("purity"),
Sorting: c.String("sort"),
Order: c.String("order"),
TopRange: c.String("range"),
AtLeast: c.String("atLeast"),
Ratios: c.StringSlice("ratios"),
2024-07-11 22:26:25 +00:00
Page: int64(rand.Intn(int(c.Int("page"))) + 1),
2024-02-25 03:11:14 +00:00
}
query := c.Args().First()
if query != "" {
s.Query = wallhaven.Q{
Tags: []string{query},
}
}
results, err := wallhaven.SearchWallpapers(s)
if err != nil {
return err
}
resultPath, err := getOrDownload(results, r, c.String("downloadPath"))
if err != nil {
return err
}
searchScript := c.String("scriptPath")
if searchScript != "" {
err = runScript(resultPath, searchScript)
if err != nil {
return err
}
}
return nil
}
func getOrDownload(results *wallhaven.SearchResults, r *rand.Rand, downloadPath string) (string, error) {
if len(results.Data) == 0 {
return "", fmt.Errorf("no wallpapers found")
}
result := results.Data[r.Intn(len(results.Data))]
fullPath := path.Join(downloadPath, path.Base(result.Path))
if _, err := os.Stat(fullPath); err != nil {
err = result.Download(path.Join(downloadPath))
if err != nil {
return "", err
}
}
return fullPath, nil
}
func runScript(imgPath, script string) error {
2024-07-21 02:20:20 +00:00
cmd := exec.Command(script, imgPath)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
return cmd.Run()
2023-10-17 18:33:29 +00:00
}