f

ms-jpq · Dec 9, 2024 · 30cbcf8 · 30cbcf8
1 parent d7953de
commit 30cbcf8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/chadtree/transitions/redraw.py b/chadtree/transitions/redraw.py
@@ -88,7 +88,7 @@ def _update(
 
 async def redraw(state: State, focus: Optional[PurePath]) -> Derived:
     derived = await _derived(state)
-    focus_row = derived.path_row_lookup.get(focus) if focus else None
+    focus_row = derived.path_row_lookup.get(focus) if state.follow and focus else None
     buf_name = _buf_name(state.root.path)
 
     ns = await Nvim.create_namespace(_NS)

diff --git a/chadtree/transitions/shared/current.py b/chadtree/transitions/shared/current.py
@@ -29,7 +29,7 @@ async def new_current_file(state: State, current: PurePath) -> Stage:
     else:
         new_state = await forward(state, current=current)
 
-    focus = current if state.follow else None
+    focus = new_state.current
     return Stage(new_state, focus=focus)
 
 

diff --git a/chadtree/transitions/shared/refresh.py b/chadtree/transitions/shared/refresh.py
@@ -62,7 +62,7 @@ async def refresh(state: State) -> Stage:
         current_ancestors if state.follow else frozenset()
     )
     new_index = index if new_current else index | parent_paths
-    focus = current if state.follow else None
+    focus = current
 
     new_state = await forward(
         state,