diff options
author | Robin Krahl <robin.krahl@ireas.org> | 2018-12-18 00:39:15 +0100 |
---|---|---|
committer | Daniel Mueller <deso@posteo.net> | 2018-12-21 18:41:19 -0800 |
commit | d137415a69007a90569ebbf38a92424fba60b997 (patch) | |
tree | 08bdb1fc4f06bca29d36fdfc9e04c8c5c0086119 /argparse/examples/subcommands.rs | |
parent | 170e1df9d96f628af8cf8107f29227b90ae9350b (diff) | |
download | nitrocli-d137415a69007a90569ebbf38a92424fba60b997.tar.gz nitrocli-d137415a69007a90569ebbf38a92424fba60b997.tar.bz2 |
Add argparse 0.2.2 as a dependency
This patch adds the crate rust-argparse [0] in version 0.2.2 as a
dependency, as discussed in issue #4.
[0] https://github.com/tailhook/rust-argparse
Import subrepo argparse/:argparse at 0de60a5e6d9ee1a3570d6089afd3ccd6ed7480c5
Diffstat (limited to 'argparse/examples/subcommands.rs')
-rw-r--r-- | argparse/examples/subcommands.rs | 88 |
1 files changed, 88 insertions, 0 deletions
diff --git a/argparse/examples/subcommands.rs b/argparse/examples/subcommands.rs new file mode 100644 index 0000000..8fb061a --- /dev/null +++ b/argparse/examples/subcommands.rs @@ -0,0 +1,88 @@ +use std::str::FromStr; +use std::io::{stdout, stderr}; +extern crate argparse; + +use argparse::{ArgumentParser, StoreTrue, Store, List}; + +#[allow(non_camel_case_types)] +#[derive(Debug)] +enum Command { + play, + record, +} + +impl FromStr for Command { + type Err = (); + fn from_str(src: &str) -> Result<Command, ()> { + return match src { + "play" => Ok(Command::play), + "record" => Ok(Command::record), + _ => Err(()), + }; + } +} + + + +fn play_command(verbose: bool, args: Vec<String>) { + let mut output = "".to_string(); + { + let mut ap = ArgumentParser::new(); + ap.set_description("Plays a sound"); + ap.refer(&mut output) + .add_option(&["--output"], Store, + r#"Output sink to play to"#); + match ap.parse(args, &mut stdout(), &mut stderr()) { + Ok(()) => {} + Err(x) => { + std::process::exit(x); + } + } + } + println!("Verbosity: {}, Output: {}", verbose, output); +} + +fn record_command(verbose: bool, args: Vec<String>) { + let mut input = "".to_string(); + { + let mut ap = ArgumentParser::new(); + ap.set_description("Records a sound"); + ap.refer(&mut input) + .add_option(&["--input"], Store, + r#"Output source to record from"#); + match ap.parse(args, &mut stdout(), &mut stderr()) { + Ok(()) => {} + Err(x) => { + std::process::exit(x); + } + } + } + println!("Verbosity: {}, Input: {}", verbose, input); +} + +fn main() { + let mut verbose = false; + let mut subcommand = Command::play; + let mut args = vec!(); + { + let mut ap = ArgumentParser::new(); + ap.set_description("Plays or records sound"); + ap.refer(&mut verbose) + .add_option(&["-v", "--verbose"], StoreTrue, + "Be verbose"); + ap.refer(&mut subcommand).required() + .add_argument("command", Store, + r#"Command to run (either "play" or "record")"#); + ap.refer(&mut args) + .add_argument("arguments", List, + r#"Arguments for command"#); + ap.stop_on_first_argument(true); + ap.parse_args_or_exit(); + } + + args.insert(0, format!("subcommand {:?}", subcommand)); + match subcommand { + Command::play => play_command(verbose, args), + Command::record => record_command(verbose, args), + } +} |