refactor: reformat codebase and apply minor structural cleanup across CLI, guard, and hook modules

Author: fajarhide

src/cli/learn.rs

@@ -91,7 +91,7 @@ pub fn run_learn(args: &[String]) -> Result<()> {
                 .bright_white()
         );
         let report = crate::pipeline::toml_filter::run_inline_tests(
-            &crate::pipeline::toml_filter::load_all_filters(),
+            crate::pipeline::toml_filter::load_all_filters(),
         );
         let total = report.passes + report.failures.len();
 

src/cli/stats.rs

@@ -111,7 +111,10 @@ fn print_help() {
         "--month".cyan()
     );
     println!("  {: <12} Machine-readable JSON output", "--json".cyan());
-    println!("  {: <12} Display breakdown per project path", "--project".cyan());
+    println!(
+        "  {: <12} Display breakdown per project path",
+        "--project".cyan()
+    );
     println!("  {: <12} Show this help message", "--help, -h".cyan());
 
     println!("\n{}", "EXAMPLES:".bold().bright_white());
@@ -582,7 +585,11 @@ fn run_project_stats(args: &[String], store: &Store) -> Result<()> {
     };
 
     let projects = store.get_project_stats(since)?;
-    println!("\n  {} — {} Breakdown", "OMNI Project Analytics".bold().bright_white(), period_label);
+    println!(
+        "\n  {} — {} Breakdown",
+        "OMNI Project Analytics".bold().bright_white(),
+        period_label
+    );
     print_separator();
 
     if projects.is_empty() {
@@ -600,12 +607,19 @@ fn run_project_stats(args: &[String], store: &Store) -> Result<()> {
         let display_path = if path.chars().count() > 30 {
             let mut s: String = path.chars().take(12).collect();
             s.push_str("...");
-            s.extend(path.chars().rev().take(15).collect::<String>().chars().rev());
+            s.extend(
+                path.chars()
+                    .rev()
+                    .take(15)
+                    .collect::<String>()
+                    .chars()
+                    .rev(),
+            );
             s
         } else {
             path
         };
-        
+
         let bar = format_bar(savings);
         let bar_colored = if savings > 80.0 {
             bar.bright_green()

src/guard/config.rs

@@ -1,6 +1,6 @@
+use crate::paths;
 use serde::Deserialize;
 use std::fs;
-use crate::paths;
 
 #[derive(Debug, Deserialize, Default)]
 pub struct OmniConfig {

src/guard/mod.rs

@@ -1,5 +1,5 @@
+pub mod config;
 pub mod env;
 pub mod limits;
 pub mod trust;
 pub mod update;
-pub mod config;

src/hooks/pipe.rs

@@ -186,98 +186,97 @@ fn distill(
         }
     }
 
-    let (output, filter_name, rewind_hash, kept_count, dropped_count, collapse_savings, route) = if let Some(filter) =
-        matched_toml
-    {
-        let out = filter.apply(&input_text);
-        (out, filter.name.clone(), None, 0, 0, None, Route::Keep)
-    } else {
-        let cmd = command_name.unwrap_or("");
-
-        // Pure Command Architecture: Resolve profile
-        let profile = crate::pipeline::registry::resolve_profile(cmd);
-
-        // 1. Initial Scoring
-        let segments = scorer::score_segments(
-            &input_text,
-            profile.segmentation,
-            session.as_ref().and_then(|m| m.lock().ok()).as_deref(),
-        );
-
-        // 2. Collapse
-        let collapse_result = collapse::collapse(&input_text, &profile.collapse);
-        let collapse_savings_data = if collapse_result.original_lines > collapse_result.collapsed_to {
-            Some((collapse_result.original_lines, collapse_result.collapsed_to))
+    let (output, filter_name, rewind_hash, kept_count, dropped_count, collapse_savings, route) =
+        if let Some(filter) = matched_toml {
+            let out = filter.apply(&input_text);
+            (out, filter.name.clone(), None, 0, 0, None, Route::Keep)
         } else {
-            None
-        };
-        let effective_input = collapse_result.collapsed_lines.join("\n");
+            let cmd = command_name.unwrap_or("");
 
-        // 3. Re-score
-        let final_segments = if collapse_result.savings_pct > 0.1 {
-            scorer::score_segments(
-                &effective_input,
+            // Pure Command Architecture: Resolve profile
+            let profile = crate::pipeline::registry::resolve_profile(cmd);
+
+            // 1. Initial Scoring
+            let segments = scorer::score_segments(
+                &input_text,
                 profile.segmentation,
                 session.as_ref().and_then(|m| m.lock().ok()).as_deref(),
-            )
-        } else {
-            segments
-        };
+            );
 
-        // 4. Distill
-        let mut out = crate::distillers::distill_with_command(
-            &final_segments,
-            &effective_input,
-            cmd,
-            session.as_ref().and_then(|m| m.lock().ok()).as_deref(),
-        );
+            // 2. Collapse
+            let collapse_result = collapse::collapse(&input_text, &profile.collapse);
+            let collapse_savings_data =
+                if collapse_result.original_lines > collapse_result.collapsed_to {
+                    Some((collapse_result.original_lines, collapse_result.collapsed_to))
+                } else {
+                    None
+                };
+            let effective_input = collapse_result.collapsed_lines.join("\n");
+
+            // 3. Re-score
+            let final_segments = if collapse_result.savings_pct > 0.1 {
+                scorer::score_segments(
+                    &effective_input,
+                    profile.segmentation,
+                    session.as_ref().and_then(|m| m.lock().ok()).as_deref(),
+                )
+            } else {
+                segments
+            };
+
+            // 4. Distill
+            let mut out = crate::distillers::distill_with_command(
+                &final_segments,
+                &effective_input,
+                cmd,
+                session.as_ref().and_then(|m| m.lock().ok()).as_deref(),
+            );
 
-        // Rewind decision
-        let noise_count = final_segments
-            .iter()
-            .filter(|s| s.final_score() < 0.3)
-            .count();
-        let should_store = noise_count as f32 / final_segments.len().max(1) as f32 > 0.4
-            && final_segments.len() > 20;
-
-        let d_count = noise_count;
-        let k_count = final_segments.len() - d_count;
-
-        let dropped_lines: usize = final_segments
-            .iter()
-            .filter(|s| s.final_score() < 0.3)
-            .map(|s| s.content.lines().count())
-            .sum();
-
-        // Auto-learn trigger
-        if !cmd.is_empty() && input_text.len() > 100 {
-            let poor = final_segments.len() > 5
-                && (d_count as f32 / final_segments.len().max(1) as f32) < 0.3;
-            if poor {
-                crate::session::learn::queue_for_learn(&input_text, cmd);
+            // Rewind decision
+            let noise_count = final_segments
+                .iter()
+                .filter(|s| s.final_score() < 0.3)
+                .count();
+            let should_store = noise_count as f32 / final_segments.len().max(1) as f32 > 0.4
+                && final_segments.len() > 20;
+
+            let d_count = noise_count;
+            let k_count = final_segments.len() - d_count;
+
+            let dropped_lines: usize = final_segments
+                .iter()
+                .filter(|s| s.final_score() < 0.3)
+                .map(|s| s.content.lines().count())
+                .sum();
+
+            // Auto-learn trigger
+            if !cmd.is_empty() && input_text.len() > 100 {
+                let poor = final_segments.len() > 5
+                    && (d_count as f32 / final_segments.len().max(1) as f32) < 0.3;
+                if poor {
+                    crate::session::learn::queue_for_learn(&input_text, cmd);
+                }
             }
-        }
-
-        let mut r_hash = None;
-
-        // Determine Route
-        let ratio = 1.0 - (out.len() as f32 / input_text.len().max(1) as f32);
-        let mut route = if r_hash.is_some() {
-            Route::Rewind
-        } else if ratio >= 0.7 {
-            Route::Keep
-        } else if ratio >= 0.3 {
-            Route::Soft
-        } else {
-            Route::Passthrough
-        };
 
-        if route == Route::Soft {
-            out.push_str("\n[Partial signal - omni learn recommended]\n");
-        }
+            let mut r_hash = None;
+
+            // Determine Route
+            let ratio = 1.0 - (out.len() as f32 / input_text.len().max(1) as f32);
+            let mut route = if r_hash.is_some() {
+                Route::Rewind
+            } else if ratio >= 0.7 {
+                Route::Keep
+            } else if ratio >= 0.3 {
+                Route::Soft
+            } else {
+                Route::Passthrough
+            };
+
+            if route == Route::Soft {
+                out.push_str("\n[Partial signal - omni learn recommended]\n");
+            }
 
-        if should_store {
-            if let Some(s) = store {
+            if should_store && let Some(s) = store {
                 let hash = s.store_rewind(&input_text);
                 if std::io::stdout().is_terminal() {
                     out.push_str(&format!(
@@ -297,25 +296,24 @@ fn distill(
                 r_hash = Some(hash);
                 route = Route::Rewind; // Override if stored in rewind
             }
-        }
 
-        // Safety truncation
-        const MAX_OUTPUT: usize = 50_000;
-        if out.len() > MAX_OUTPUT {
-            out.truncate(MAX_OUTPUT);
-            out.push_str("\n[OMNI: output truncated]\n");
-        }
+            // Safety truncation
+            const MAX_OUTPUT: usize = 50_000;
+            if out.len() > MAX_OUTPUT {
+                out.truncate(MAX_OUTPUT);
+                out.push_str("\n[OMNI: output truncated]\n");
+            }
 
-        (
-            out,
-            cmd.split_whitespace().next().unwrap_or("omni").to_string(),
-            r_hash,
-            k_count,
-            d_count,
-            collapse_savings_data,
-            route,
-        )
-    };
+            (
+                out,
+                cmd.split_whitespace().next().unwrap_or("omni").to_string(),
+                r_hash,
+                k_count,
+                d_count,
+                collapse_savings_data,
+                route,
+            )
+        };
 
     PipelineResult {
         session_id,

src/hooks/post_tool.rs

@@ -220,7 +220,6 @@ pub fn process_payload(
         }
     }
 
-
     if should_store {
         if let Some(ref s) = store {
             let hash = s.store_rewind(&content);