slap-cli 1.0.0

Painless shell argument parsing.
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

use {
    crate::Shell,
    clap::{App, AppSettings, Arg},
    std::str,
};

#[derive(Clone)]
pub struct AppWrapper<'a, 'b>
where
    'a: 'b,
{
    pub app: App<'a, 'b>,
    pub help_msg: String,
    pub version_msg: String,
}

impl<'a, 'b> AppWrapper<'a, 'b>
where
    'a: 'b,
{
    pub fn new(
        app: App<'a, 'b>,
        modify_app: impl FnOnce(App<'a, 'b>) -> App<'a, 'b>,
    ) -> anyhow::Result<Self> {
        let app = app
            .settings(&[
                AppSettings::DisableHelpFlags,
                AppSettings::DisableVersion,
                AppSettings::DisableHelpSubcommand,
            ])
            .arg(
                Arg::with_name("help")
                    .short("h")
                    .long("help")
                    .help("Prints help information"),
            )
            .arg(
                Arg::with_name("version")
                    .short("V")
                    .long("version")
                    .help("Prints version information"),
            );
        let app = modify_app(app);

        let mut help_msg = Vec::new();
        app.write_help(&mut help_msg)?;
        let help_msg = str::from_utf8(&help_msg)?;
        let mut version_msg = Vec::new();
        app.write_long_version(&mut version_msg)?;
        let version_msg = str::from_utf8(&version_msg)?;

        Ok(Self {
            app,
            help_msg: help_msg.into(),
            version_msg: version_msg.into(),
        })
    }

    // FIXME: Fix ZSH not generating the code for completion.
    pub fn completions_script(&mut self, bin_name: &str, shell: &Shell) -> anyhow::Result<String> {
        let mut completions_script = Vec::new();
        self.app
            .gen_completions_to(bin_name, shell.clone().into(), &mut completions_script);
        Ok(str::from_utf8(&completions_script)?.trim_end().into())
    }
}