1 files changed, 77 insertions, 0 deletions
diff --git a/src/data_collection/disks/windows.rs b/src/data_collection/disks/windows.rs
new file mode 100644
index 00000000..ae7bb025
--- /dev/null
+++ b/src/data_collection/disks/windows.rs
@@ -0,0 +1,77 @@
+//! Disk stats via sysinfo.
+
+use itertools::Itertools;
+use sysinfo::{DiskExt, SystemExt};
+
+use super::{keep_disk_entry, DiskHarvest};
+
+use crate::data_collection::{disks::IoCounters, DataCollector};
+
+mod bindings;
+use bindings::*;
+
+/// Returns I/O stats.
+pub(crate) fn io_stats() -> anyhow::Result<Vec<IoCounters>> {
+    let volume_io = all_volume_io()?;
+
+    Ok(volume_io
+        .into_iter()
+        .map_ok(|(performance, volume_name)| {
+            let name = volume_name;
+            let read_bytes = performance.BytesRead as u64;
+            let write_bytes = performance.BytesWritten as u64;
+
+            IoCounters::new(name, read_bytes, write_bytes)
+        })
+        .flatten()
+        .collect::<Vec<_>>())
+}
+
+pub(crate) fn get_disk_usage(collector: &DataCollector) -> anyhow::Result<Vec<DiskHarvest>> {
+    let disks = collector.sys.disks();
+    let disk_filter = &collector.filters.disk_filter;
+    let mount_filter = &collector.filters.mount_filter;
+
+    Ok(disks
+        .iter()
+        .filter_map(|disk| {
+            let name = {
+                let name = disk.name();
+
+                if name.is_empty() {
+                    "No Name".to_string()
+                } else {
+                    name.to_os_string()
+                        .into_string()
+                        .unwrap_or_else(|_| "Name Unavailable".to_string())
+                }
+            };
+
+            let mount_point = disk
+                .mount_point()
+                .as_os_str()
+                .to_os_string()
+                .into_string()
+                .unwrap_or_else(|_| "Mount Unavailable".to_string());
+
+            let volume_name = volume_name_from_mount(&mount_point).ok();
+
+            if keep_disk_entry(&name, &mount_point, disk_filter, mount_filter) {
+                let free_space = disk.available_space();
+                let total_space = disk.total_space();
+                let used_space = total_space - free_space;
+
+                Some(DiskHarvest {
+                    name,
+                    mount_point,
+                    volume_name,
+                    free_space: Some(free_space),
+                    used_space: Some(used_space),
+                    total_space: Some(total_space),
+                })
+            } else {
+                None
+            }
+        })
+        .collect())
+}