flawz 0.3.0

A Terminal UI for browsing CVEs
Documentation 
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
92
93
94

use clap::Parser;

use crate::{app::AppResult, error::Error, theme::BuiltinTheme};

/// Command line arguments.
#[derive(Debug, Default, Parser)]
#[clap(
    version,
    author = clap::crate_authors!("\n"),
    about,
    rename_all_env = "screaming-snake",
    help_template = "\
{before-help}{name} {version} - {about}
With ♥ by {author-with-newline}
{usage-heading}
  {usage}

{all-args}{after-help}
",
)]
pub struct Args {
    /// A URL where NIST CVE 1.1 feeds can be found.
    #[arg(long, env, default_value = "https://nvd.nist.gov/feeds/json/cve/1.1/")]
    pub url: String,

    /// List of feeds that are going to be synced.
    #[arg(
        short,
        long,
        env,
        num_args(0..),
        default_values_t = ["2002:2024".to_string(), "recent".into(), "modified".into()]
    )]
    pub feeds: Vec<String>,

    /// Path to the SQLite database used to store the synced CVE data.
    #[arg(short, env, long)]
    pub db: Option<String>,

    /// Always fetch feeds.
    #[arg(short = 'u', long)]
    pub force_update: bool,

    /// Do not fetch feeds.
    #[arg(short, long)]
    pub offline: bool,

    /// Start with a search query.
    #[arg(short, env, long)]
    pub query: Option<String>,

    /// Set the theme.
    #[arg(short, long, value_enum, default_value = "dracula")]
    pub theme: BuiltinTheme,
}

impl Args {
    /// Parses and returns the feeds.
    pub fn feeds(&self) -> AppResult<Vec<String>> {
        self.feeds
            .iter()
            .try_fold::<Vec<_>, _, AppResult<_>>(vec![], |mut acc, v| {
                if v.contains(':') {
                    let mut parts = v.split(':');
                    let start = parts
                        .next()
                        .and_then(|v| v.parse::<usize>().ok())
                        .ok_or_else(|| Error::RangeArgsError)?;
                    let end = parts
                        .next()
                        .and_then(|v| v.parse::<usize>().ok())
                        .ok_or_else(|| Error::RangeArgsError)?;
                    acc.extend(
                        (start..=end)
                            .map(|v| v.to_string())
                            .collect::<Vec<String>>(),
                    );
                } else {
                    acc.push(v.clone())
                }
                Ok(acc)
            })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use clap::CommandFactory;
    #[test]
    fn test_args() {
        Args::command().debug_assert();
    }
}