From d7a40e6c6c5658b7871db6dc75752d85ed1458eb Mon Sep 17 00:00:00 2001
From: nardor <nard0r@protonmail.com>
Date: Mon, 5 Aug 2024 12:00:26 +0200
Subject: [PATCH] feat(diff): add nodetype, metadata, symlink statistics types

---
 src/commands/diff.rs | 63 +++++++++++++++++++++++++++++++++++++-------
 1 file changed, 54 insertions(+), 9 deletions(-)

diff --git a/src/commands/diff.rs b/src/commands/diff.rs
index f36905891..57d8e857d 100644
--- a/src/commands/diff.rs
+++ b/src/commands/diff.rs
@@ -197,6 +197,11 @@ struct DiffStatistics {
     directories_removed: usize,
     others_added: usize,
     others_removed: usize,
+    node_type_changed: usize,
+    metadata_changed: usize,
+    symlink_added: usize,
+    symlink_removed: usize,
+    symlink_changed: usize,
 }
 
 impl DiffStatistics {
@@ -204,6 +209,7 @@ impl DiffStatistics {
         match node_type {
             NodeType::File => self.files_removed += 1,
             NodeType::Dir => self.directories_removed += 1,
+            NodeType::Symlink { .. } => self.symlink_removed += 1,
             _ => self.others_removed += 1,
         }
     }
@@ -212,6 +218,7 @@ impl DiffStatistics {
         match node_type {
             NodeType::File => self.files_added += 1,
             NodeType::Dir => self.directories_added += 1,
+            NodeType::Symlink { .. } => self.symlink_added += 1,
             _ => self.others_added += 1,
         }
     }
@@ -219,22 +226,60 @@ impl DiffStatistics {
     fn changed_file(&mut self) {
         self.files_changed += 1;
     }
+
+    fn changed_node_type(&mut self) {
+        self.node_type_changed += 1;
+    }
+
+    fn changed_metadata(&mut self) {
+        self.metadata_changed += 1;
+    }
+
+    fn changed_symlink(&mut self) {
+        self.symlink_changed += 1;
+    }
 }
 
 impl Display for DiffStatistics {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         f.write_fmt(format_args!(
-            "Files:\t{} new,\t{} removed, \t{} changed\n",
+            "Files   :\t{} new,\t{} removed, \t{} changed\n",
             self.files_added, self.files_removed, self.files_changed
         ))?;
+        // symlink
+        if self.symlink_added != 0 || self.symlink_removed != 0 || self.symlink_changed != 0 {
+            f.write_fmt(format_args!(
+                "Symlinks:\t{} new,\t{} removed, \t{} changed\n",
+                self.symlink_added, self.symlink_removed, self.symlink_changed
+            ))?;
+        }
         f.write_fmt(format_args!(
-            "Dirs:\t{} new,\t{} removed\n",
+            "Dirs    :\t{} new,\t{} removed\n",
             self.directories_added, self.directories_removed
         ))?;
-        f.write_fmt(format_args!(
-            "Others:\t{} new,\t{} removed",
-            self.others_added, self.others_removed
-        ))
+        if self.others_added != 0 || self.others_removed != 0 {
+            f.write_fmt(format_args!(
+                "Others  :\t{} new,\t{} removed\n",
+                self.others_added, self.others_removed
+            ))?;
+        }
+
+        // node type
+        if self.node_type_changed != 0 {
+            f.write_fmt(format_args!(
+                "NodeType:\t{} changed\n",
+                self.node_type_changed
+            ))?;
+        }
+
+        // metadata
+        if self.metadata_changed != 0 {
+            f.write_fmt(format_args!(
+                "Metadata:\t{} changed\n",
+                self.metadata_changed
+            ))?;
+        }
+        Ok(())
     }
 }
 
@@ -294,7 +339,7 @@ fn diff(
                     tpe if tpe != &node2.node_type => {
                         // type was changed
                         println!("T    {path:?}");
-                        diff_statistics.changed_file();
+                        diff_statistics.changed_node_type();
                     }
                     NodeType::File if !no_content && !file_identical(path, node1, node2)? => {
                         println!("M    {path:?}");
@@ -302,12 +347,12 @@ fn diff(
                     }
                     NodeType::File if metadata && node1.meta != node2.meta => {
                         println!("U    {path:?}");
-                        diff_statistics.changed_file();
+                        diff_statistics.changed_metadata();
                     }
                     NodeType::Symlink { .. } => {
                         if node1.node_type.to_link() != node1.node_type.to_link() {
                             println!("U    {path:?}");
-                            diff_statistics.changed_file();
+                            diff_statistics.changed_symlink();
                         }
                     }
                     _ => {} // no difference to show