implement scan command with save

closes #1

Author: w3irdrobot

.gitignore

@@ -1 +1,2 @@
 target
+.idea

Cargo.lock

@@ -20,8 +20,11 @@ config = { version = "0.14.1", default-features = false, features = [
 ] }
 directories = "5.0.1"
 env_logger = "0.11.5"
+futures = "0.3.31"
 humantime = "2.1.0"
+ipnetwork = "0.20.0"
 log = "0.4.22"
+reqwest = "0.12.9"
 serde = { version = "1.0.215", features = ["derive"] }
 serde_json = "1.0.133"
 serde_with = "3.11.0"

crates/bacli/Cargo.toml

@@ -2,10 +2,12 @@ mod alias;
 mod info;
 mod list;
 mod restart;
+mod scan;
 mod update_settings;
 
 pub use alias::*;
 pub use info::*;
 pub use list::*;
 pub use restart::*;
+pub use scan::*;
 pub use update_settings::*;

crates/bacli/src/commands/mod.rs

@@ -0,0 +1,58 @@
+use std::net::IpAddr;
+use std::time::Duration;
+
+use anyhow::Result;
+use bitaxe_api::prelude::*;
+use comfy_table::Table;
+use futures::future;
+use ipnetwork::IpNetwork;
+use log::debug;
+
+use crate::config::Config;
+use crate::models::ScanArgs;
+
+pub async fn scan(mut config: Config, args: ScanArgs) -> Result<()> {
+    debug!("Scanning network for devices: {:?}", args);
+    let pool = reqwest::Client::builder()
+        .timeout(Duration::from_secs(1))
+        .build()?;
+
+    let network = IpNetwork::with_netmask(args.base, args.mask)?;
+    let devices = future::join_all(network.into_iter().map(|i| check_ip(pool.clone(), i)))
+        .await
+        .into_iter()
+        .filter_map(|res| res.ok());
+
+    let mut table = Table::new();
+    table.set_header(vec!["IP", "Alias", "Board Version", "OS Version"]);
+
+    for (ip, info) in devices {
+        let alias = match config.clone().get_device(&ip.to_string()) {
+            Some(device) => device.alias.clone().unwrap_or("None".to_string()),
+            None => {
+                if args.should_save {
+                    config.upsert_device(&ip, None).await?;
+                }
+                "None".to_string()
+            }
+        };
+
+        table.add_row(vec![
+            ip.to_string(),
+            alias,
+            info.board_version,
+            info.version,
+        ]);
+    }
+
+    println!("{table}");
+
+    Ok(())
+}
+
+async fn check_ip(pool: reqwest::Client, ip: IpAddr) -> Result<(IpAddr, SystemInfo)> {
+    let client = BitaxeClient::new_with_client(pool, &ip);
+    let info = client.system_info().await?;
+
+    Ok((ip, info))
+}

crates/bacli/src/commands/scan.rs

@@ -23,6 +23,7 @@ async fn main() -> anyhow::Result<()> {
         Command::UpdateSettings(args) => update_settings(cfg, args).await?,
         Command::List => list(cfg).await?,
         Command::Alias(args) => alias(cfg, args).await?,
+        Command::Scan(args) => scan(cfg, args).await?,
     }
 
     Ok(())

crates/bacli/src/main.rs

@@ -1,3 +1,5 @@
+use std::net::IpAddr;
+
 use bitaxe_api::models::Settings;
 use clap::{Args, Parser, Subcommand};
 use serde::{Deserialize, Serialize};
@@ -26,6 +28,8 @@ pub enum Command {
     List,
     /// Associate an alias with a base (IP)
     Alias(AliasArgs),
+    /// Scan the local network for devices
+    Scan(ScanArgs),
 }
 
 #[derive(Debug, Clone, Args)]
@@ -55,10 +59,23 @@ pub struct UpdateSetttingsArgs {
 pub struct AliasArgs {
     /// The URL of the device on the local network. This will usually be an IP address.
     pub base: String,
-    /// The alias to reference the IP
+    /// The alias to reference the IP.
     pub alias: String,
 }
 
+#[derive(Debug, Clone, Args)]
+pub struct ScanArgs {
+    /// An IP address in the IP range of the network containing the devices.
+    #[arg(long, default_value = "192.168.1.1")]
+    pub base: IpAddr,
+    /// A mask to apply to the base IP to get the range of available IPs.
+    #[arg(long, default_value = "255.255.255.0")]
+    pub mask: IpAddr,
+    /// Save any new found devices to the config.
+    #[arg(short, long = "save")]
+    pub should_save: bool,
+}
+
 #[serde_as]
 #[skip_serializing_none]
 #[derive(Debug, Clone, Serialize, Deserialize)]