abs3nt/wallhaven_dl
1
0
Fork 0
Code Issues 1 Pull Requests 1 Actions Packages Projects Releases 4 Wiki Activity

refac

Browse Source
This commit is contained in:
abs3nt 2023-10-22 12:52:07 -07:00
parent 7f7c77ae9a
commit d71eb6a355
Signed by: abs3nt
GPG Key ID: A7BD96A8BAB04C09
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.go
View File

@ -19,7 +19,8 @@ var cli struct {
Query string `arg:"" name:"query" help:"what to search for." type:"string"`
} `cmd:"" help:"search for wallpaper"`
Top struct {
Purity string `arg:"" name:"purity" optional:"" help:"purity of results"`
Purity string `name:"purity" optional:"" help:"purity of results"`
Range string `name:"range" optional:"" help:"range for toplist"`
} `cmd:"" help:"random toplist wallpaper"`
Img struct {
Path string `arg:"" name:"path" help:"path to image or directory." type:"path"`
@ -34,8 +35,8 @@ func main() {
if err != nil {
panic(err)
}
case "top", "top <purity>":
err := setTop(cli.Top.Purity)
case "top":
err := setTop(cli.Top.Purity, cli.Top.Range)
if err != nil {
panic(err)
}
@ -97,7 +98,7 @@ func searchAndSet(query string) error {
return nil
}
func setTop(purity string) error {
func setTop(purity, topRange string) error {
seed := rand.NewSource(time.Now().UnixNano())
r := rand.New(seed)
s := &wallhaven.Search{
@ -105,7 +106,7 @@ func setTop(purity string) error {
Purities: "110",
Sorting: wallhaven.Toplist,
Order: wallhaven.Desc,
TopRange: "1m",
TopRange: "6m",
AtLeast: wallhaven.Resolution{Width: 2560, Height: 1400},
Ratios: []wallhaven.Ratio{
{Horizontal: 16, Vertical: 9},
@ -116,6 +117,9 @@ func setTop(purity string) error {
if purity != "" {
s.Purities = purity
}
if topRange != "" {
s.TopRange = topRange
}
results, err := wallhaven.SearchWallpapers(s)
if err != nil {
return err