🐛 Enforce max_depth limit in flat and tree renderers

Add early return in recursive rendering functions to skip nodes beyond max_depth, ensuring the parameter is respected and preventing excessive output or recursion in deep directory structures.

Author: elifarley

src/cedarmapper/render/flat.py

@@ -61,6 +61,10 @@ def render_flat(
     rows: list[_DisplayRow] = []
 
     def _render_node(node: DirInfo | FileInfo, depth: int, parent_path: str) -> None:
+        # Only include this node if it's within max_depth
+        if depth > max_depth:
+            return
+
         display_path = (
             f"{parent_path}{node.path.name}"
             if isinstance(node, FileInfo)

src/cedarmapper/render/tree.py

@@ -62,6 +62,10 @@ def _render_node(
         is_last_stack: list[bool],  # stack indicating whether current ancestor is last
         parent_display_path: str,
     ) -> None:
+        # Only include this node if it's within max_depth
+        if depth > max_depth:
+            return
+
         # Build prefix either with numbered indent or with branch characters
         if numbered_indent:
             prefix = f"{depth:>3} "

tests/test_max_depth_repro.py

@@ -0,0 +1,55 @@
+import tempfile
+from pathlib import Path
+from cedarmapper.cli.main import run_cli
+
+def test_max_depth_filtering():
+    """
+    Verify strict max-depth filtering behavior.
+    Structure:
+      root/
+        d1.txt          (depth 1)
+        sub1/           (depth 1)
+          d2.txt        (depth 2)
+          sub2/         (depth 2)
+            d3.txt      (depth 3)
+    """
+    with tempfile.TemporaryDirectory() as tmpdir:
+        root = Path(tmpdir)
+
+        # Create structure
+        (root / "d1.txt").write_text("depth 1 file")
+
+        sub1 = root / "sub1"
+        sub1.mkdir()
+        (sub1 / "d2.txt").write_text("depth 2 file")
+
+        sub2 = sub1 / "sub2"
+        sub2.mkdir()
+        (sub2 / "d3.txt").write_text("depth 3 file")
+
+        # Case 0: max-depth 0 (root only) - usually just shows the root itself or immediate children depending on definition
+        # But commonly max-depth 0 means "just the argument itself".
+        # However, many tools treat max-depth 1 as "argument + immediate children".
+        # Let's test standard assumptions:
+
+        # Test max-depth 1: Should show d1.txt and sub1/, but NOT d2.txt or sub2
+        code, out = run_cli([str(root), "--max-depth", "1", "--tree"])
+        assert code == 0
+        assert "d1.txt" in out, "depth 1 file should be visible"
+        assert "sub1/" in out, "depth 1 dir should be visible"
+        assert "d2.txt" not in out, "depth 2 file should NOT be visible with max-depth 1"
+        assert "sub2/" not in out, "depth 2 dir should NOT be visible with max-depth 1"
+        assert "d3.txt" not in out, "depth 3 file should NOT be visible with max-depth 1"
+
+        # Test max-depth 2: Should show d2.txt and sub2/, but NOT d3.txt
+        code, out = run_cli([str(root), "--max-depth", "2", "--tree"])
+        assert code == 0
+        assert "d1.txt" in out
+        assert "d2.txt" in out, "depth 2 file should be visible with max-depth 2"
+        assert "d3.txt" not in out, "depth 3 file should NOT be visible with max-depth 2"
+
+        # Test flat mode as well
+        code, out = run_cli([str(root), "--max-depth", "1"])
+        assert code == 0
+        assert "d1.txt" in out
+        assert "d2.txt" not in out