Initial commit

src/util.rs

@@ -0,0 +1,40 @@
+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))
+}