wallheaven 0.2.2

Random image fetcher for wallhaven.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use clap::Parser;

/// Wallpaper fetcher for wallhaven.cc
#[derive(Parser, Debug)]
#[command(version, about)]
pub struct Flags {
    /// Config file to use
    #[arg(long)]
    pub config: Option<String>,

    /// Username for wallhaven.cc
    #[arg(long)]
    pub username: Option<String>,

    /// API key from wallhaven.cc
    #[arg(long)]
    pub apikey: Option<String>,

    /// Specify the collection name to fetch wallpaper from (requires username and apikey)
    #[arg(short, long, conflicts_with_all = ["random", "toplist", "hot", "id"])]
    pub collection: Option<String>,

    /// Query to search for random wallpaper
    #[arg(short, long, conflicts_with_all = ["collection", "toplist", "hot", "id"])]
    pub random: Option<String>,

    /// Select random image from toplist
    #[arg(short, long, conflicts_with_all = ["collection", "hot", "id", "random"])]
    pub toplist: bool,

    /// Time range to search toplist (1d|3d|1w|1M|3M|6M|1y)
    #[arg(long)]
    pub range: Option<String>,

    /// Select random image from hot
    #[arg(long)]
    pub hot: bool,

    /// Set wallpaper by wallhaven id
    #[arg(short, long, conflicts_with_all = ["collection", "toplist", "hot", "random"])]
    pub id: Option<String>,

    /// Set purity filter as bits (sfw|sketchy|nsfw) eg. 110 = sfw & sketchy
    #[arg(short, long, value_parser = check3bit)]
    pub purity: Option<String>,

    /// Set category filter as bits (general|anime|people) eg. 101 = general & people
    #[arg(long, alias = "cat", value_parser = check3bit)]
    pub categories: Option<String>,

    /// Pages to search for random image (higher = slower) max: 10
    #[arg(long, value_parser = clap::value_parser!(i64).range(1..11))]
    pub pages: Option<i64>,

    /// Time in seconds to use cache file before requery of wallhaven.cc
    #[arg(short, long)]
    pub expiry: Option<i64>,

    /// External script to pass selected file to (set wallpaper etc.)
    #[arg(short, long)]
    pub script: Option<String>,

    /// Returns the path to the current set wallpaper
    #[arg(short, long = "file")]
    pub file: bool,

    /// Returns the link to the current set wallpaper on wallhaven.cc
    #[arg(short, long = "url")]
    pub url: bool,

    /// Return the last search query made with random
    #[arg(short, long)]
    pub last: bool,

    /// Deletes wallpaper cache folder
    #[arg(short, long)]
    pub delete: bool,
}

pub fn cli_args() -> Flags {
    let flags = Flags::parse();
    return flags;
}

pub fn check3bit(bits: &str) -> Result<String, String> {
    if bits.len() == 3 && bits.chars().all(|c| c == '0' || c == '1') {
        return Ok(bits.to_string());
    } else {
        return Err(format!("should be bits e.g. 110"));
    }
}