41 lines
1.1 KiB
Rust
41 lines
1.1 KiB
Rust
|
use clap::{Arg, App};
|
||
|
|
use glob::Pattern as GlobPattern;
|
||
|
|
|
||
|
|
pub fn filter_nodes(nodes: &Vec<String>, filter: &str) -> Vec<String> {
|
||
|
|
let filters: Vec<GlobPattern> = filter.split(",").map(|pattern| GlobPattern::new(pattern).unwrap()).collect();
|
||
|
|
|
||
|
|
if filters.len() > 0 {
|
||
|
|
nodes.iter().filter(|name| {
|
||
|
|
for filter in filters.iter() {
|
||
|
|
if filter.matches(name) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
false
|
||
|
|
}).cloned().collect()
|
||
|
|
} else {
|
||
|
|
nodes.to_owned()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn register_common_args<'a, 'b>(command: App<'a, 'b>) -> App<'a, 'b> {
|
||
|
|
command
|
||
|
|
.arg(Arg::with_name("config")
|
||
|
|
.short("f")
|
||
|
|
.long("config")
|
||
|
|
.help("Path to a Hive expression")
|
||
|
|
.default_value("hive.nix")
|
||
|
|
.required(true))
|
||
|
|
.arg(Arg::with_name("on")
|
||
|
|
.long("on")
|
||
|
|
.help("Select a list of machines")
|
||
|
|
.long_help(r#"The list is comma-separated and globs are supported.
|
||
|
|
Valid examples:
|
||
|
|
|
||
|
|
- host1,host2,host3
|
||
|
|
- edge-*
|
||
|
|
- edge-*,core-*"#)
|
||
|
|
.takes_value(true))
|
||
|
|
}
|