Adding wifi speed feature for linux

src/connectivity/mod.rs

@@ -1,11 +1,12 @@
+use std::{fmt, io};
+use crate::platforms::WifiError;
+
 #[cfg(target_os = "windows")]
 mod handlers;
 mod providers;
 #[cfg(target_os = "windows")]
 mod stubs;
 
-use crate::platforms::WifiError;
-use std::{fmt, io};
 
 /// Wireless network connectivity functionality.
 pub trait Connectivity: fmt::Debug {
@@ -14,6 +15,10 @@ pub trait Connectivity: fmt::Debug {
 
     /// Disconnects from a wireless network currently connected to.
     fn disconnect(&self) -> Result<bool, WifiConnectionError>;
+
+    // Determines speed when connected to a network
+    // Adding unimplemented block
+    fn speed(&self) -> Result<String, WifiConnectionError> { unimplemented!(); }
 }
 
 /// Error that occurs when attempting to connect to a wireless network.

src/connectivity/providers/linux.rs

@@ -1,6 +1,8 @@
 use crate::connectivity::{Connectivity, WifiConnectionError};
 use crate::platforms::{Connection, WiFi, WifiError, WifiInterface};
-use std::process::Command;
+use std::process::{Command, Stdio};
+use std::str;
+use std::format;
 
 /// Wireless network connectivity functionality.
 impl Connectivity for WiFi {
@@ -51,4 +53,26 @@ impl Connectivity for WiFi {
             .as_ref()
             .contains("disconnect"))
     }
+
+    fn speed(&self) -> Result<String, WifiConnectionError> {
+        let nmcli_speed = Command::new("nmcli")
+            .arg("dev")
+            .arg("wifi")
+            .arg("list")
+            .stdout(Stdio::piped())
+            .spawn()
+            .unwrap();
+
+        let piped_awk = Command::new("awk")
+            .arg(r"/\*/{if (NR!=1) {print $6}}")
+            .stdin(Stdio::from(nmcli_speed.stdout.unwrap())) // Pipe through.
+            .stdout(Stdio::piped())
+            .spawn()
+            .unwrap();
+
+        let result = piped_awk.wait_with_output().unwrap();
+        let result = str::from_utf8(&result.stdout).unwrap().to_string();
+        let formatted_result = format!("{} Mbit/s", result.trim());
+        Ok(formatted_result)
+    }
 }

src/main.rs

@@ -2,9 +2,12 @@ mod connectivity;
 mod hotspot;
 mod platforms;
 
+use std::env::args;
 use crate::connectivity::{Connectivity, WifiConnectionError};
 use crate::platforms::{Config, WiFi};
 
+use std::process::{Command, Stdio};
+
 fn main() -> Result<(), WifiConnectionError> {
     let config = Some(Config {
         interface: Some("wlo1"),
@@ -24,5 +27,8 @@ fn main() -> Result<(), WifiConnectionError> {
         Err(err) => println!("The following error occurred: {:?}", err),
     }
 
+    let speed = wifi.speed().unwrap();
+    println!("Speed is: {}", speed);
+
     Ok(())
 }