ripjson 0.9.11

A fast and lean way to grep in JSON files.
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

use getopts::Options;
use std::env;
use std::io::IsTerminal;

fn print_usage(program: &str, opts: Options) {
    let brief = format!("Usage: {} <regex> <files> [options]", program);
    print!("{}", opts.usage(&brief));
}

fn print_version() {
    const VERSION: Option<&'static str> = option_env!("CARGO_PKG_VERSION");
    println!("Ripjson v{}", VERSION.unwrap_or("unknown"));
}

fn add_options(opts: &mut Options) {
    opts.optflag("i", "ignore-case", "Search case insensitively.");
    opts.optflag("s", "sensitive-case", "Search case sensitively [default].");
    opts.optflag("h", "help", "Print this help menu.");
    opts.optflag("v", "version", "Print version.");
    opts.optopt(
        "",
        "color",
        "Color output.\nWHEN can be never, always, or auto [default].",
        "<WHEN>",
    );
}

pub fn parse_options() -> (String, Vec<String>, bool) {
    let args: Vec<String> = env::args().collect();
    let program = args[0].clone();
    let mut opts = Options::new();
    add_options(&mut opts);

    let matches = match opts.parse(&args[1..]) {
        Ok(m) => m,
        Err(_f) => {
            print_usage(&program, opts);
            std::process::exit(1)
        }
    };

    if matches.opt_present("h") {
        print_usage(&program, opts);
        std::process::exit(0);
    }

    if matches.opt_present("v") {
        print_version();
        std::process::exit(0);
    }

    let color_output = if matches.opt_present("color") {
        match matches.opt_str("color").unwrap().as_ref() {
            "never" => false,
            "always" => true,
            "auto" => std::io::stdout().is_terminal(),
            _ => {
                println!(
                    "Error: unknown color mode '{}' specified.\n",
                    matches.opt_str("color").unwrap()
                );
                print_usage(&program, opts);
                std::process::exit(1)
            }
        }
    } else {
        std::io::stdout().is_terminal()
    };

    let ignore_case = matches.opt_present("i") && !matches.opt_present("s");

    let regex = format!(
        "{}({})",
        if ignore_case { "(?i)" } else { "" },
        if !matches.free.is_empty() {
            matches.free[0].clone()
        } else {
            println!("Error: no regex specified.\n");
            print_usage(&program, opts);
            std::process::exit(1)
        }
    );

    let files = if matches.free.len() > 1 {
        matches.free[1..].to_vec()
    } else {
        // Read from stdin if no file specified
        vec![String::from("-")]
    };

    (regex, files, color_output)
}