tmp: save it for later

Author: KSXGitHub

Cargo.lock

@@ -67,6 +67,9 @@ serde = { version = "^1.0.219", optional = true }
 serde_json = { version = "^1.0.140", optional = true }
 sysinfo = "^0.35.2"
 
+[target.'cfg(unix)'.dependencies]
+dashmap = "^6.1.0"
+
 [dev-dependencies]
 build-fs-tree = "^0.7.1"
 command-extra = "^1.0.0"

Cargo.toml

@@ -0,0 +1,48 @@
+use std::{fs::Metadata, path::Path};
+
+#[cfg(unix)]
+use dashmap::DashMap;
+#[cfg(unix)]
+use std::{os::unix::fs::MetadataExt, path::PathBuf};
+
+/// Hook to run with a [`Path`] and its corresponding [`Metadata`].
+pub trait Hook {
+    fn run_hook(&self, path: &Path, metadata: &Metadata);
+}
+
+/// A [hook](Hook) that does nothing.
+#[derive(Debug, Clone, Copy)]
+pub struct DoNothing;
+impl Hook for DoNothing {
+    fn run_hook(&self, _: &Path, _: &Metadata) {}
+}
+
+/// A [hook](Hook) that record files with more than 1 links.
+#[cfg(unix)] // Windows does have `MetadataExt::number_of_links` but it requires Nightly
+#[derive(Debug, Clone, Copy)]
+pub struct RecordHardLink<'a> {
+    /// Map an inode number to its detected paths.
+    storage: &'a DashMap<u64, Vec<PathBuf>>, // TODO: benchmark against Mutex<HashMap<u64, Vec<PathBuf>>>
+}
+
+#[cfg(unix)]
+impl<'a> RecordHardLink<'a> {
+    /// Create a [hook](Hook) to record files with more than 1 links.
+    pub fn new(storage: &'a DashMap<u64, Vec<PathBuf>>) -> Self {
+        RecordHardLink { storage }
+    }
+}
+
+#[cfg(unix)]
+impl<'a> Hook for RecordHardLink<'a> {
+    fn run_hook(&self, path: &Path, metadata: &Metadata) {
+        if metadata.is_dir() || metadata.nlink() <= 1 {
+            return;
+        }
+
+        self.storage
+            .entry(metadata.ino())
+            .or_default()
+            .push(path.to_path_buf());
+    }
+}

src/hook.rs

@@ -33,6 +33,7 @@ pub mod bytes_format;
 pub mod data_tree;
 pub mod fs_tree_builder;
 pub mod get_size;
+pub mod hook;
 pub mod json_data;
 pub mod os_string_display;
 pub mod reporter;