1 files changed, 51 insertions, 0 deletions
diff --git a/src/data_collection/network/sysinfo.rs b/src/data_collection/network/sysinfo.rs
new file mode 100644
index 00000000..a5b85d9a
--- /dev/null
+++ b/src/data_collection/network/sysinfo.rs
@@ -0,0 +1,51 @@
+//! Gets network data via sysinfo.
+
+use std::time::Instant;
+
+use super::NetworkHarvest;
+use crate::app::filter::Filter;
+
+// TODO: Eventually make it so that this thing also takes individual usage into account, so we can show per-interface!
+pub fn get_network_data(
+    sys: &sysinfo::System, prev_net_access_time: Instant, prev_net_rx: &mut u64,
+    prev_net_tx: &mut u64, curr_time: Instant, filter: &Option<Filter>,
+) -> NetworkHarvest {
+    use sysinfo::{NetworkExt, SystemExt};
+
+    let mut total_rx: u64 = 0;
+    let mut total_tx: u64 = 0;
+
+    let networks = sys.networks();
+    for (name, network) in networks {
+        let to_keep = if let Some(filter) = filter {
+            filter.keep_entry(name)
+        } else {
+            true
+        };
+
+        if to_keep {
+            total_rx += network.total_received() * 8;
+            total_tx += network.total_transmitted() * 8;
+        }
+    }
+
+    let elapsed_time = curr_time.duration_since(prev_net_access_time).as_secs_f64();
+
+    let (rx, tx) = if elapsed_time == 0.0 {
+        (0, 0)
+    } else {
+        (
+            ((total_rx.saturating_sub(*prev_net_rx)) as f64 / elapsed_time) as u64,
+            ((total_tx.saturating_sub(*prev_net_tx)) as f64 / elapsed_time) as u64,
+        )
+    };
+
+    *prev_net_rx = total_rx;
+    *prev_net_tx = total_tx;
+    NetworkHarvest {
+        rx,
+        tx,
+        total_rx,
+        total_tx,
+    }
+}