git.delta.rocks / jrsonnet / refs/commits / a2ea38b80ded

--- /dev/null
+++ b/cmds/fleet/src/cmds/info.rs
@@ -0,0 +1,81 @@
+use std::collections::BTreeSet;
+
+use crate::host::Config;
+use anyhow::{ensure, Result};
+use structopt::StructOpt;
+
+#[derive(StructOpt)]
+pub struct Info {
+	#[structopt(long)]
+	json: bool,
+	#[structopt(subcommand)]
+	cmd: InfoCmd,
+}
+
+#[derive(StructOpt)]
+pub enum InfoCmd {
+	/// List hosts
+	ListHosts {
+		#[structopt(long)]
+		tagged: Vec<String>,
+	},
+	/// List ips
+	HostIps {
+		host: String,
+		#[structopt(long)]
+		external: bool,
+		#[structopt(long)]
+		internal: bool,
+	},
+}
+
+impl Info {
+	pub fn run(self, config: &Config) -> Result<()> {
+		let mut data = Vec::new();
+		match self.cmd {
+			InfoCmd::ListHosts { ref tagged } => {
+				'host: for host in config.list_hosts()? {
+					if !tagged.is_empty() {
+						let tags: Vec<String> = config.config_attr(&host, "tags")?;
+						for tag in tagged {
+							if !tags.contains(&tag) {
+								continue 'host;
+							}
+						}
+					}
+					data.push(host);
+				}
+			}
+			InfoCmd::HostIps {
+				host,
+				external,
+				internal,
+			} => {
+				ensure!(
+					external || internal,
+					"at leas one of --external or --internal must be set"
+				);
+				let mut out = <BTreeSet<String>>::new();
+				if external {
+					out.extend(config.config_attr::<Vec<String>>(&host, "network.externalIps")?);
+				}
+				if internal {
+					out.extend(config.config_attr::<Vec<String>>(&host, "network.internalIps")?);
+				}
+				for ip in out {
+					data.push(ip);
+				}
+			}
+		}
+
+		if self.json {
+			let v = serde_json::to_string_pretty(&data)?;
+			print!("{}", v);
+		} else {
+			for v in data {
+				println!("{}", v);
+			}
+		}
+		Ok(())
+	}
+}