summaryrefslogtreecommitdiff
path: root/argparse/examples/subcommands.rs
diff options
context:
space:
mode:
authorRobin Krahl <robin.krahl@ireas.org>2018-12-18 00:39:15 +0100
committerDaniel Mueller <deso@posteo.net>2018-12-21 18:41:19 -0800
commitd137415a69007a90569ebbf38a92424fba60b997 (patch)
tree08bdb1fc4f06bca29d36fdfc9e04c8c5c0086119 /argparse/examples/subcommands.rs
parent170e1df9d96f628af8cf8107f29227b90ae9350b (diff)
downloadnitrocli-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.rs88
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),
+ }
+}