Refactor keybindings and improve UX

Keybindings:
- Refactor to use centralized keybinding definitions
- Add leader key menu for discoverable shortcuts
- Add connection picker with fuzzy search

UX improvements:
- Add query execution timing to notifications (e.g. "in 156ms")
- Add JSON/Python literal pretty-printing in value view
- Add zebra striping to results table
- Improve validation feedback in connection forms

Testing:
- Add validation unit tests
- Add UI tests for explorer toggle and fullscreen
- Add keybindings tests
- Add connection picker tests

Author: Maxteabag

sqlit/app.py

@@ -182,7 +182,9 @@ class SSMSTUI(
         Binding("h", "leader_help", show=False),
         Binding("t", "leader_theme", show=False),
         Binding("q", "leader_quit", show=False),
+        Binding("c", "leader_connect", show=False),
         Binding("x", "leader_disconnect", show=False),
+        Binding("z", "leader_cancel", show=False),
         # Regular bindings
         Binding("n", "new_connection", "New", show=False),
         Binding("s", "select_table", "Select", show=False),
@@ -261,6 +263,43 @@ def __init__(self):
         self._table_metadata: dict = {}
         self._columns_loading: set[str] = set()
 
+    @property
+    def object_tree(self) -> Tree:
+        return self.query_one("#object-tree", Tree)
+
+    @property
+    def query_input(self) -> TextArea:
+        return self.query_one("#query-input", TextArea)
+
+    @property
+    def results_table(self) -> DataTable:
+        return self.query_one("#results-table", DataTable)
+
+    @property
+    def sidebar(self):
+        return self.query_one("#sidebar")
+
+    @property
+    def main_panel(self):
+        return self.query_one("#main-panel")
+
+    @property
+    def query_area(self):
+        return self.query_one("#query-area")
+
+    @property
+    def results_area(self):
+        return self.query_one("#results-area")
+
+    @property
+    def status_bar(self) -> Static:
+        return self.query_one("#status-bar", Static)
+
+    @property
+    def autocomplete_dropdown(self):
+        from .widgets import AutocompleteDropdown
+        return self.query_one("#autocomplete-dropdown", AutocompleteDropdown)
+
     def push_screen(self, screen, callback=None, wait_for_dismiss: bool = False):
         """Override push_screen to hide footer when showing modal dialogs."""
         from textual.screen import ModalScreen
@@ -311,21 +350,14 @@ def check_action(self, action: str, parameters: tuple) -> bool | None:
             if action != "leader_key":
                 return False
 
-        try:
-            tree = self.query_one("#object-tree", Tree)
-            query_input = self.query_one("#query-input", TextArea)
-            results_table = self.query_one("#results-table", DataTable)
-        except Exception:
-            return True
-
-        tree_focused = tree.has_focus
-        query_focused = query_input.has_focus
-        results_focused = results_table.has_focus
+        tree_focused = self.object_tree.has_focus
+        query_focused = self.query_input.has_focus
+        results_focused = self.results_table.has_focus
         in_insert_mode = self.vim_mode == VimMode.INSERT
 
-        node = tree.cursor_node
+        node = self.object_tree.cursor_node
         node_type = None
-        is_root = node == tree.root if node else False
+        is_root = node == self.object_tree.root if node else False
         if node and node.data:
             node_type = node.data[0]
 
@@ -428,7 +460,7 @@ def compose(self) -> ComposeResult:
                         yield Static(
                             r"\[r] Results", classes="section-label", id="label-results"
                         )
-                        yield DataTable(id="results-table")
+                        yield DataTable(id="results-table", zebra_stripes=True)
 
             yield Static("Not connected", id="status-bar")
 
@@ -459,11 +491,10 @@ def on_mount(self) -> None:
         self.refresh_tree()
         self._update_footer_bindings()
 
-        tree = self.query_one("#object-tree", Tree)
-        tree.focus()
+        self.object_tree.focus()
         # Move cursor to first node if available
-        if tree.root.children:
-            tree.cursor_line = 0
+        if self.object_tree.root.children:
+            self.object_tree.cursor_line = 0
         self._update_section_labels()
 
         # Check for ODBC drivers

sqlit/ui/mixins/autocomplete.py

@@ -159,46 +159,32 @@ def _show_autocomplete(self, suggestions: list[str], filter_text: str) -> None:
             self._hide_autocomplete()
             return
 
-        dropdown = self.query_one("#autocomplete-dropdown", AutocompleteDropdown)
+        dropdown = self.autocomplete_dropdown
         dropdown.set_items(suggestions, filter_text)
 
-        try:
-            query_input = self.query_one("#query-input", TextArea)
-            cursor_loc = query_input.cursor_location
-            dropdown.styles.offset = (cursor_loc[1] + 2, cursor_loc[0] + 1)
-        except Exception:
-            pass
+        cursor_loc = self.query_input.cursor_location
+        dropdown.styles.offset = (cursor_loc[1] + 2, cursor_loc[0] + 1)
 
         dropdown.show()
         self._autocomplete_visible = True
 
     def _hide_autocomplete(self) -> None:
         """Hide the autocomplete dropdown."""
-        from ...widgets import AutocompleteDropdown
-
-        try:
-            dropdown = self.query_one("#autocomplete-dropdown", AutocompleteDropdown)
-            dropdown.hide()
-            self._autocomplete_visible = False
-        except Exception:
-            pass
+        self.autocomplete_dropdown.hide()
+        self._autocomplete_visible = False
 
     def _apply_autocomplete(self) -> None:
         """Apply the selected autocomplete suggestion."""
-        from ...widgets import AutocompleteDropdown
-
-        dropdown = self.query_one("#autocomplete-dropdown", AutocompleteDropdown)
-        selected = dropdown.get_selected()
+        selected = self.autocomplete_dropdown.get_selected()
 
         if not selected:
             self._hide_autocomplete()
             return
 
         self._autocomplete_just_applied = True
 
-        query_input = self.query_one("#query-input", TextArea)
-        text = query_input.text
-        cursor_loc = query_input.cursor_location
+        text = self.query_input.text
+        cursor_loc = self.query_input.cursor_location
         cursor_pos = self._location_to_offset(text, cursor_loc)
 
         word_start = cursor_pos
@@ -214,11 +200,11 @@ def _apply_autocomplete(self) -> None:
         else:
             new_text = text[:word_start] + selected + text[cursor_pos:]
 
-        query_input.text = new_text
+        self.query_input.text = new_text
 
         new_cursor_pos = word_start + len(selected)
         new_loc = self._offset_to_location(new_text, new_cursor_pos)
-        query_input.cursor_location = new_loc
+        self.query_input.cursor_location = new_loc
 
         self._hide_autocomplete()
 
@@ -280,13 +266,13 @@ def on_text_area_changed(self, event: TextArea.Changed) -> None:
 
     def on_key(self, event) -> None:
         """Handle key events for autocomplete navigation."""
-        from ...widgets import AutocompleteDropdown, VimMode
+        from ...widgets import VimMode
 
         # Handle autocomplete navigation
         if not self._autocomplete_visible:
             return
 
-        dropdown = self.query_one("#autocomplete-dropdown", AutocompleteDropdown)
+        dropdown = self.autocomplete_dropdown
 
         if event.key == "down":
             dropdown.move_selection(1)

sqlit/ui/mixins/connection.py

@@ -115,8 +115,7 @@ def action_disconnect(self) -> None:
         if self.current_connection:
             self._disconnect_silent()
 
-            status = self.query_one("#status-bar", Static)
-            status.update("Disconnected")
+            self.status_bar.update("Disconnected")
 
             self.refresh_tree()
             self.notify("Disconnected")
@@ -130,12 +129,9 @@ def action_new_connection(self) -> None:
 
     def action_edit_connection(self) -> None:
         """Edit the selected connection."""
-        from textual.widgets import Tree
-
         from ..screens import ConnectionScreen
 
-        tree = self.query_one("#object-tree", Tree)
-        node = tree.cursor_node
+        node = self.object_tree.cursor_node
 
         if not node or not node.data:
             return
@@ -183,13 +179,9 @@ def handle_connection_result(self, result: tuple | None) -> None:
 
     def action_delete_connection(self) -> None:
         """Delete the selected connection."""
-        from textual.widgets import Tree
-
         from ..screens import ConfirmScreen
-        from ...config import ConnectionConfig
 
-        tree = self.query_one("#object-tree", Tree)
-        node = tree.cursor_node
+        node = self.object_tree.cursor_node
 
         if not node or not node.data:
             return
@@ -220,10 +212,7 @@ def _do_delete_connection(self, config: "ConnectionConfig") -> None:
 
     def action_connect_selected(self) -> None:
         """Connect to the selected connection."""
-        from textual.widgets import Tree
-
-        tree = self.query_one("#object-tree", Tree)
-        node = tree.cursor_node
+        node = self.object_tree.cursor_node
 
         if not node or not node.data:
             return
@@ -236,3 +225,36 @@ def action_connect_selected(self) -> None:
             if self.current_connection:
                 self._disconnect_silent()
             self.connect_to_server(config)
+
+    def action_show_connection_picker(self) -> None:
+        """Show connection picker dialog."""
+        from ..screens import ConnectionPickerScreen
+
+        if not self.connections:
+            self.notify("No connections configured", severity="warning")
+            return
+
+        self.push_screen(
+            ConnectionPickerScreen(self.connections),
+            self._handle_connection_picker_result,
+        )
+
+    def _handle_connection_picker_result(self, result: str | None) -> None:
+        """Handle connection picker selection."""
+        if result is None:
+            return
+
+        config = next((c for c in self.connections if c.name == result), None)
+        if config:
+            # Select the connection node in the tree
+            for node in self.object_tree.root.children:
+                if node.data and node.data[0] == "connection" and node.data[1].name == result:
+                    self.object_tree.select_node(node)
+                    break
+
+            if self.current_config and self.current_config.name == config.name:
+                self.notify(f"Already connected to {config.name}")
+                return
+            if self.current_connection:
+                self._disconnect_silent()
+            self.connect_to_server(config)