Merge pull request #114 from Maxteabag/cli-existing-connectionf

Cli existing connection

Author: Maxteabag

README.md

@@ -136,6 +136,9 @@ sqlit --mock=sqlite-demo
 ### CLI
 
 ```bash
+sqlit -c "MyConnection"
+sqlit --connection "MyConnection"
+
 # Run a query
 sqlit query -c "MyConnection" -q "SELECT * FROM Users"
 

sqlit/cli.py

@@ -392,6 +392,12 @@ def main() -> int:
         action="store_true",
         help="Show idle scheduler status in the status bar.",
     )
+    parser.add_argument(
+        "-c",
+        "--connection",
+        metavar="NAME",
+        help="Connect to a saved connection by name (opens TUI with only this connection)",
+    )
 
     subparsers = parser.add_subparsers(dest="command", help="Available commands")
 
@@ -512,21 +518,38 @@ def main() -> int:
                 return 1
             services.apply_mock_profile(mock_profile)
 
-        temp_config = None
+        startup_config = None
+        exclusive_connection = False
         try:
-            # Check for connection URL first (extracted before argparse)
-            if connection_url:
-                temp_config = parse_connection_url(
+            # Check for saved connection by name first
+            if args.connection:
+                saved_connections = services.connection_store.load_all(load_credentials=False)
+                matching = [c for c in saved_connections if c.name == args.connection]
+                if not matching:
+                    print(f"Error: Connection '{args.connection}' not found")
+                    print("Available connections:")
+                    for conn in saved_connections:
+                        print(f"  - {conn.name}")
+                    return 1
+                startup_config = matching[0]
+                exclusive_connection = True
+            # Check for connection URL (extracted before argparse)
+            elif connection_url:
+                startup_config = parse_connection_url(
                     connection_url,
                     name=getattr(args, "name", None),
                 )
             else:
-                temp_config = _build_temp_connection(args)
+                startup_config = _build_temp_connection(args)
         except ValueError as exc:
             print(f"Error: {exc}")
             return 1
 
-        app = SSMSTUI(services=services, startup_connection=temp_config)
+        app = SSMSTUI(
+            services=services,
+            startup_connection=startup_config,
+            exclusive_connection=exclusive_connection,
+        )
         app.run()
         if getattr(app, "_restart_requested", False):
             argv = getattr(app, "_restart_argv", None) or app._compute_restart_argv()

sqlit/domains/explorer/ui/tree/builder.py

@@ -75,6 +75,7 @@ def _add_connection_node(
     is_connected: bool,
     is_connecting: bool,
     spinner: str | None,
+    skip_folder: bool = False,
 ) -> Any:
     if is_connected:
         label = host._format_connection_label(config, "connected")
@@ -83,8 +84,11 @@ def _add_connection_node(
     else:
         label = host._format_connection_label(config, "idle")
 
-    folder_parts = _split_folder_path(getattr(config, "folder_path", ""))
-    parent = _ensure_connection_folder_path(host, folder_parts)
+    if skip_folder:
+        parent = host.object_tree.root
+    else:
+        folder_parts = _split_folder_path(getattr(config, "folder_path", ""))
+        parent = _ensure_connection_folder_path(host, folder_parts)
 
     node = parent.add(label)
     node.data = ConnectionNode(config=config)
@@ -206,28 +210,35 @@ def refresh_tree(host: TreeMixinHost) -> None:
     connecting_name = connecting_config.name if connecting_config else None
     connecting_spinner = host._connect_spinner_frame() if connecting_config else None
 
-    # Check for active direct connection or pending startup connection
+    # Check for active direct connection, exclusive connection, or pending startup connection
     direct_config = getattr(host, "_direct_connection_config", None)
     startup_config = getattr(host, "_startup_connect_config", None)
+    exclusive_connection = getattr(host, "_exclusive_connection", False)
     direct_active = (
         direct_config is not None
         and host.current_config is not None
         and direct_config.name == host.current_config.name
     )
-    # Also hide saved connections when startup connection is pending (before it's connected)
+    # Hide saved connections when startup connection is pending (before it's connected)
     startup_pending = startup_config is not None and not any(
         c.name == startup_config.name for c in host.connections
     )
+    # Exclusive mode: only show the specified connection (even if it's saved)
+    exclusive_active = exclusive_connection and startup_config is not None
     if direct_active and host.current_config is not None:
         connections = [host.current_config]
+    elif exclusive_active:
+        connections = [startup_config]
     elif startup_pending:
         connections = [startup_config]
     else:
         connections = list(host.connections)
     if connecting_config and not any(c.name == connecting_config.name for c in connections):
         connections = connections + [connecting_config]
     connections = _sort_connections_for_display(connections)
-    _build_connection_folders(host, connections)
+    # Skip folder structure in exclusive mode
+    if not exclusive_active:
+        _build_connection_folders(host, connections)
 
     for conn in connections:
         is_connected = host.current_config is not None and conn.name == host.current_config.name
@@ -238,6 +249,7 @@ def refresh_tree(host: TreeMixinHost) -> None:
             is_connected=is_connected,
             is_connecting=is_connecting,
             spinner=connecting_spinner,
+            skip_folder=exclusive_active,
         )
 
     restore_subtree_expansion(host, host.object_tree.root)
@@ -263,28 +275,35 @@ def refresh_tree_chunked(
     connecting_name = connecting_config.name if connecting_config else None
     connecting_spinner = host._connect_spinner_frame() if connecting_config else None
 
-    # Check for active direct connection or pending startup connection
+    # Check for active direct connection, exclusive connection, or pending startup connection
     direct_config = getattr(host, "_direct_connection_config", None)
     startup_config = getattr(host, "_startup_connect_config", None)
+    exclusive_connection = getattr(host, "_exclusive_connection", False)
     direct_active = (
         direct_config is not None
         and host.current_config is not None
         and direct_config.name == host.current_config.name
     )
-    # Also hide saved connections when startup connection is pending (before it's connected)
+    # Hide saved connections when startup connection is pending (before it's connected)
     startup_pending = startup_config is not None and not any(
         c.name == startup_config.name for c in host.connections
     )
+    # Exclusive mode: only show the specified connection (even if it's saved)
+    exclusive_active = exclusive_connection and startup_config is not None
     if direct_active and host.current_config is not None:
         connections = [host.current_config]
+    elif exclusive_active:
+        connections = [startup_config]
     elif startup_pending:
         connections = [startup_config]
     else:
         connections = list(host.connections)
     if connecting_config and not any(c.name == connecting_config.name for c in connections):
         connections = connections + [connecting_config]
     connections = _sort_connections_for_display(connections)
-    _build_connection_folders(host, connections)
+    # Skip folder structure in exclusive mode
+    if not exclusive_active:
+        _build_connection_folders(host, connections)
 
     def schedule_populate() -> None:
         if getattr(host, "_tree_refresh_token", None) is not token:
@@ -343,6 +362,7 @@ def populate_once() -> None:
                 is_connected=is_connected,
                 is_connecting=is_connecting,
                 spinner=connecting_spinner,
+                skip_folder=exclusive_active,
             )
 
         def finish_sync() -> None:
@@ -369,6 +389,7 @@ def add_batch() -> None:
                 is_connected=is_connected,
                 is_connecting=is_connecting,
                 spinner=connecting_spinner,
+                skip_folder=exclusive_active,
             )
         idx = end
         if idx < len(connections):

sqlit/domains/shell/app/main.py

@@ -98,13 +98,15 @@ def __init__(
         services: AppServices | None = None,
         runtime: RuntimeConfig | None = None,
         startup_connection: ConnectionConfig | None = None,
+        exclusive_connection: bool = False,
     ):
         super().__init__()
         self.services = services or build_app_services(runtime or RuntimeConfig.from_env())
         from sqlit.core.connection_manager import ConnectionManager
 
         self._connection_manager = ConnectionManager(self.services)
         self._startup_connection = startup_connection
+        self._exclusive_connection = exclusive_connection
         self._startup_connect_config: ConnectionConfig | None = None
         self._debug_mode = self.services.runtime.debug_mode
         self._debug_idle_scheduler = self.services.runtime.debug_idle_scheduler