Add expand all on tree cell view

Author: Maxteabag

sqlit/core/keymap.py

@@ -386,6 +386,7 @@ def _build_action_keys(self) -> list[ActionKeyDef]:
             ActionKeyDef("y", "copy_value_view", "value_view"),
             ActionKeyDef("t", "toggle_value_view_mode", "value_view"),
             ActionKeyDef("z", "collapse_all_json_nodes", "value_view"),
+            ActionKeyDef("Z", "expand_all_json_nodes", "value_view"),
         ]
 
 

sqlit/domains/results/state/value_view_active.py

@@ -41,11 +41,13 @@ class ValueViewTreeModeState(State):
 
     def _setup_actions(self) -> None:
         self.allows("collapse_all_json_nodes", key="z", label="Collapse", help="Collapse all nodes")
+        self.allows("expand_all_json_nodes", key="Z", label="Expand", help="Expand all nodes")
         self.allows("toggle_value_view_mode", key="t", label="Syntax View", help="Switch to syntax view")
 
     def get_display_bindings(self, app: InputContext) -> tuple[list[DisplayBinding], list[DisplayBinding]]:
         left: list[DisplayBinding] = [
             DisplayBinding(key="z", label="Collapse", action="collapse_all_json_nodes"),
+            DisplayBinding(key="Z", label="Expand", action="expand_all_json_nodes"),
             DisplayBinding(key="t", label="Syntax View", action="toggle_value_view_mode"),
         ]
 

sqlit/domains/results/ui/mixins/results.py

@@ -307,6 +307,17 @@ def action_collapse_all_json_nodes(self: ResultsMixinHost) -> None:
         except Exception:
             pass
 
+    def action_expand_all_json_nodes(self: ResultsMixinHost) -> None:
+        """Expand all nodes in the JSON tree view."""
+        from sqlit.shared.ui.widgets import InlineValueView
+
+        try:
+            value_view = self.query_one("#value-view", InlineValueView)
+            if value_view.is_visible:
+                value_view.expand_all_nodes()
+        except Exception:
+            pass
+
     def action_copy_cell(self: ResultsMixinHost) -> None:
         """Copy the selected cell to clipboard (or internal clipboard)."""
         table, _columns, _rows, _stacked = self._get_active_results_context()

sqlit/shared/ui/widgets_value_view.py

@@ -86,6 +86,14 @@ def collapse_all_nodes(self) -> None:
             except Exception:
                 pass
 
+    def expand_all_nodes(self) -> None:
+        """Expand all tree nodes."""
+        if self._is_json and self._tree_mode:
+            try:
+                self.query_one("#json-tree", JSONTreeView).action_expand_all()
+            except Exception:
+                pass
+
     def _rebuild(self) -> None:
         """Rebuild the display based on current mode."""
         try: