clap/examples/tutorial_builder/04_04_custom.rs

94 lines
3.4 KiB
Rust
Raw Normal View History

// Note: this requires the `cargo` feature
2022-05-23 15:50:11 +00:00
use std::path::PathBuf;
use clap::{arg, command, value_parser, ArgAction, ErrorKind};
fn main() {
// Create application like normal
2022-02-15 14:33:38 +00:00
let mut cmd = command!()
2018-11-14 17:05:06 +00:00
// Add the version arguments
.arg(arg!(--"set-ver" <VER> "set version manually").required(false))
.arg(arg!(--major "auto inc major").action(ArgAction::SetTrue))
.arg(arg!(--minor "auto inc minor").action(ArgAction::SetTrue))
.arg(arg!(--patch "auto inc patch").action(ArgAction::SetTrue))
2018-11-14 17:05:06 +00:00
// Arguments can also be added to a group individually, these two arguments
// are part of the "input" group which is not required
2022-05-23 15:50:11 +00:00
.arg(arg!([INPUT_FILE] "some regular input").value_parser(value_parser!(PathBuf)))
.arg(
arg!(--"spec-in" <SPEC_IN> "some special input argument")
.required(false)
.value_parser(value_parser!(PathBuf)),
)
2018-11-14 17:05:06 +00:00
// Now let's assume we have a -c [config] argument which requires one of
// (but **not** both) the "input" arguments
2022-05-23 15:50:11 +00:00
.arg(
arg!(config: -c <CONFIG>)
.required(false)
.value_parser(value_parser!(PathBuf)),
);
2022-02-14 21:47:20 +00:00
let matches = cmd.get_matches_mut();
// Let's assume the old version 1.2.3
let mut major = 1;
let mut minor = 2;
let mut patch = 3;
// See if --set-ver was used to set the version manually
let version = if let Some(ver) = matches.get_one::<String>("set-ver") {
if *matches.get_one::<bool>("major").expect("defaulted by clap")
|| *matches.get_one::<bool>("minor").expect("defaulted by clap")
|| *matches.get_one::<bool>("patch").expect("defaulted by clap")
{
2022-02-14 21:47:20 +00:00
cmd.error(
ErrorKind::ArgumentConflict,
"Can't do relative and absolute version change",
)
.exit();
}
ver.to_string()
} else {
// Increment the one requested (in a real program, we'd reset the lower numbers)
2018-01-25 04:05:05 +00:00
let (maj, min, pat) = (
*matches.get_one::<bool>("major").expect("defaulted by clap"),
*matches.get_one::<bool>("minor").expect("defaulted by clap"),
*matches.get_one::<bool>("patch").expect("defaulted by clap"),
2018-01-25 04:05:05 +00:00
);
match (maj, min, pat) {
(true, false, false) => major += 1,
(false, true, false) => minor += 1,
(false, false, true) => patch += 1,
_ => {
2022-02-14 21:47:20 +00:00
cmd.error(
ErrorKind::ArgumentConflict,
2022-03-02 05:19:14 +00:00
"Can only modify one version field",
)
.exit();
}
};
format!("{}.{}.{}", major, minor, patch)
};
println!("Version: {}", version);
// Check for usage of -c
if matches.contains_id("config") {
2018-01-25 04:05:05 +00:00
let input = matches
2022-05-23 15:50:11 +00:00
.get_one::<PathBuf>("INPUT_FILE")
.or_else(|| matches.get_one::<PathBuf>("spec-in"))
.unwrap_or_else(|| {
2022-02-14 21:47:20 +00:00
cmd.error(
ErrorKind::MissingRequiredArgument,
"INPUT_FILE or --spec-in is required when using --config",
)
.exit()
2022-05-23 15:50:11 +00:00
})
.display();
2018-01-25 04:05:05 +00:00
println!(
"Doing work using input {} and config {}",
input,
matches.get_one::<PathBuf>("config").unwrap().display()
2018-01-25 04:05:05 +00:00
);
}
}