feat(api)!: add hook to Sub

Author: KSXGitHub

src/app.rs

@@ -6,6 +6,7 @@ use crate::{
     args::{Args, Quantity, Threads},
     bytes_format::BytesFormat,
     get_size::{GetApparentSize, GetSize},
+    hook,
     json_data::{JsonData, UnitAndTree},
     reporter::{ErrorOnlyReporter, ErrorReport, ProgressAndErrorReporter, ProgressReport},
     runtime_error::RuntimeError,
@@ -201,6 +202,7 @@ impl App {
                     direction: Direction::from_top_down(top_down),
                     bar_alignment: BarAlignment::from_align_right(align_right),
                     size_getter: <$size_getter as GetSizeUtils>::INSTANCE,
+                    hook: hook::DoNothing, // TODO: change this
                     reporter: <$size_getter as CreateReporter<$progress>>::create_reporter(report_error),
                     bytes_format: <$size_getter as GetSizeUtils>::formatter(bytes_format),
                     files,

src/app/sub.rs

@@ -16,11 +16,12 @@ use serde::Serialize;
 use std::{io::stdout, iter::once, num::NonZeroU64, path::PathBuf};
 
 /// The sub program of the main application.
-pub struct Sub<Size, SizeGetter, Report>
+pub struct Sub<Size, SizeGetter, Hook, Report>
 where
     Report: ParallelReporter<Size> + Sync,
     Size: size::Size + Into<u64> + Serialize + Send + Sync,
     SizeGetter: GetSize<Size = Size> + Copy + Sync,
+    Hook: hook::Hook<Size> + Copy + Sync,
     DataTreeReflection<String, Size>: Into<UnitAndTree>,
 {
     /// List of files and/or directories.
@@ -39,6 +40,8 @@ where
     pub max_depth: NonZeroU64,
     /// [Get the size](GetSize) of files/directories.
     pub size_getter: SizeGetter,
+    /// Hook to run after [`Self::size_getter`].
+    pub hook: Hook,
     /// Reports measurement progress.
     pub reporter: Report,
     /// Minimal size proportion required to appear.
@@ -47,11 +50,12 @@ where
     pub no_sort: bool,
 }
 
-impl<Size, SizeGetter, Report> Sub<Size, SizeGetter, Report>
+impl<Size, SizeGetter, Hook, Report> Sub<Size, SizeGetter, Hook, Report>
 where
     Size: size::Size + Into<u64> + Serialize + Send + Sync,
     Report: ParallelReporter<Size> + Sync,
     SizeGetter: GetSize<Size = Size> + Copy + Sync,
+    Hook: hook::Hook<Size> + Copy + Sync,
     DataTreeReflection<String, Size>: Into<UnitAndTree>,
 {
     /// Run the sub program.
@@ -65,6 +69,7 @@ where
             column_width_distribution,
             max_depth,
             size_getter,
+            hook,
             reporter,
             min_ratio,
             no_sort,
@@ -79,7 +84,7 @@ where
                     reporter: &reporter,
                     root,
                     size_getter,
-                    hook: hook::DoNothing, // TODO: change this
+                    hook,
                     max_depth,
                 }
                 .into()
@@ -90,6 +95,7 @@ where
         } else {
             return Sub {
                 files: vec![".".into()],
+                hook,
                 reporter,
                 ..self
             }