Implement dirs-only option

src/render/context/mod.rs

@@ -113,6 +113,10 @@ pub struct Context {
     #[arg(long)]
     /// Print completions for a given shell to stdout
     pub completions: Option<clap_complete::Shell>,
+
+    #[arg(short = 'D', long)]
+    /// Only show directories
+    pub dirs_only: bool,
 }
 
 impl Context {

src/render/tree/mod.rs

@@ -89,6 +89,9 @@ impl Tree {
                 let mut root_id = None;
 
                 while let Ok(TraversalState::Ongoing(node)) = rx.recv() {
+                    if ctx.dirs_only && !node.is_dir() {
+                        continue
+                    }
                     if node.is_dir() {
                         let node_path = node.path();
 

tests/dirs_only.rs

@@ -0,0 +1,24 @@
+use indoc::indoc;
+
+mod utils;
+
+#[test]
+fn level() {
+    assert_eq!(
+        utils::run_cmd(&[
+            "--sort",
+            "name",
+            "--no-config",
+            "--dirs-only",
+            "tests/data"
+        ]),
+        indoc!(
+            "
+            data 
+            ├─ dream_cycle 
+            ├─ lipsum 
+            └─ the_yellow_king"
+        ),
+        "Failed to print dirs-only."
+    )
+}