Cloud discovery (#50)

* Azure discovery improvements and auto firewall exception adding

* Azure firewall detection on existing connections

* Improve package installation flow

* Improve driver failed to load proccess

* Fix default region

* Improve azure discovery

* Ux improvements

* Cloud discovery refacotirng

* Add redshift and bigquery adapters. Add AWS and GCP discovery

* Connection picker polish

* Mock

* Fix MySQL dependency: use PyMySQL instead of mysql-connector-python

* Fix test failures: Windows auth check, Docker tab navigation, password visibility tests

---------

Co-authored-by: Peter Adams <18162810+Maxteabag@users.noreply.github.com>

Author: Maxteabag

pyproject.toml

@@ -39,7 +39,7 @@ dynamic = ["version"]
 all = [
     "psycopg2-binary>=2.9.0",
     "mssql-python>=1.1.0",
-    "mysql-connector-python>=9.1.0",  # min avoids known CVEs
+    "PyMySQL>=1.1.0",
     "mariadb>=1.1.0",
     "oracledb>=2.0.0",
     "duckdb>=1.1.0",  # min avoids known CVEs
@@ -55,7 +55,7 @@ all = [
 postgres = ["psycopg2-binary>=2.9.0"]
 cockroachdb = ["psycopg2-binary>=2.9.0"]
 mssql = ["mssql-python>=1.1.0"]
-mysql = ["mysql-connector-python>=9.1.0"]  # min avoids known CVEs
+mysql = ["PyMySQL>=1.1.0"]
 mariadb = ["mariadb>=1.1.0"]
 oracle = ["oracledb>=2.0.0"]
 duckdb = ["duckdb>=1.1.0"]  # min avoids known CVEs

sqlit/cli.py

@@ -151,6 +151,11 @@ def main() -> int:
         metavar="COUNT",
         help="Maximum rows to fetch and render (default: 10000). Use for performance testing.",
     )
+    parser.add_argument(
+        "--mock-cloud",
+        action="store_true",
+        help="Use mock cloud provider data (Azure, AWS, GCP) for demos/screenshots.",
+    )
     parser.add_argument(
         "--profile-startup",
         action="store_true",
@@ -255,6 +260,11 @@ def main() -> int:
         help="Maximum rows to fetch (default: 1000, use 0 for unlimited)",
     )
 
+    # Docker discovery command
+    docker_parser = subparsers.add_parser("docker", help="Docker container discovery")
+    docker_subparsers = docker_parser.add_subparsers(dest="docker_command", help="Docker commands")
+    docker_subparsers.add_parser("list", help="List detected database containers")
+
     startup_mark = time.perf_counter()
     args = parser.parse_args(filtered_argv[1:])  # Skip program name
     if args.settings:
@@ -281,6 +291,10 @@ def main() -> int:
         os.environ["SQLIT_MAX_ROWS"] = str(args.max_rows)
     else:
         os.environ.pop("SQLIT_MAX_ROWS", None)
+    if args.mock_cloud:
+        os.environ["SQLIT_MOCK_CLOUD"] = "1"
+    else:
+        os.environ.pop("SQLIT_MOCK_CLOUD", None)
     if args.profile_startup:
         os.environ["SQLIT_PROFILE_STARTUP"] = "1"
     else:
@@ -388,6 +402,15 @@ def main() -> int:
     if args.command == "query":
         return cmd_query(args)
 
+    if args.command == "docker":
+        from .commands import cmd_docker_list
+
+        if args.docker_command == "list":
+            return cmd_docker_list(args)
+        else:
+            docker_parser.print_help()
+            return 1
+
     parser.print_help()
     return 1
 

sqlit/commands.py

@@ -345,6 +345,54 @@ def _output_table(columns: list[str], rows: list[tuple], truncated: bool) -> Non
         print(f"\n({len(rows)} row(s) returned)")
 
 
+def cmd_docker_list(args: Any) -> int:
+    """List detected Docker database containers."""
+    from .services.docker_detector import (
+        ContainerStatus,
+        DockerStatus,
+        detect_database_containers,
+    )
+
+    status, containers = detect_database_containers()
+
+    if status == DockerStatus.NOT_INSTALLED:
+        print("Error: Docker Python library not installed.")
+        print("Install it with: pip install docker")
+        return 1
+    elif status == DockerStatus.NOT_RUNNING:
+        print("Error: Docker is not running.")
+        return 1
+    elif status == DockerStatus.NOT_ACCESSIBLE:
+        print("Error: Docker is not accessible (permission denied).")
+        print("Try adding your user to the docker group or running with sudo.")
+        return 1
+
+    if not containers:
+        print("No database containers found.")
+        return 0
+
+    running = [c for c in containers if c.status == ContainerStatus.RUNNING]
+    exited = [c for c in containers if c.status == ContainerStatus.EXITED]
+
+    print(f"{'Container':<25} {'Type':<12} {'Port':<8} {'Database':<15} {'Status':<10}")
+    print("-" * 75)
+
+    for c in running:
+        port_str = str(c.port) if c.port else "-"
+        db_str = c.database[:13] + ".." if c.database and len(c.database) > 15 else (c.database or "-")
+        name_str = c.container_name[:23] + ".." if len(c.container_name) > 25 else c.container_name
+        print(f"{name_str:<25} {c.db_type:<12} {port_str:<8} {db_str:<15} {'running':<10}")
+
+    for c in exited:
+        port_str = "-"
+        db_str = c.database[:13] + ".." if c.database and len(c.database) > 15 else (c.database or "-")
+        name_str = c.container_name[:23] + ".." if len(c.container_name) > 25 else c.container_name
+        print(f"{name_str:<25} {c.db_type:<12} {port_str:<8} {db_str:<15} {'exited':<10}")
+
+    print(f"\nFound {len(running)} running, {len(exited)} exited database container(s).")
+    return 0
+
+
 def cmd_query(
     args: Any,
     *,

sqlit/config.py

@@ -133,6 +133,7 @@ class AuthType(Enum):
     AD_PASSWORD = "ad_password"
     AD_INTERACTIVE = "ad_interactive"
     AD_INTEGRATED = "ad_integrated"
+    AD_DEFAULT = "ad_default"  # Uses Azure CLI / environment credentials
 
 
 AUTH_TYPE_LABELS = {
@@ -141,6 +142,7 @@ class AuthType(Enum):
     AuthType.AD_PASSWORD: "Microsoft Entra Password",
     AuthType.AD_INTERACTIVE: "Microsoft Entra MFA",
     AuthType.AD_INTEGRATED: "Microsoft Entra Integrated",
+    AuthType.AD_DEFAULT: "Microsoft Entra Default (CLI)",
 }
 
 
@@ -229,6 +231,7 @@ def get_source_emoji(self) -> str:
 # Source emoji mapping
 SOURCE_EMOJIS: dict[str, str] = {
     "docker": "🐳 ",
+    "azure": "",
 }
 
 

sqlit/db/adapters/__init__.py

@@ -19,6 +19,8 @@
     "DatabaseAdapter",
     "TableInfo",
     # Adapter classes (lazy via __getattr__)
+    "AthenaAdapter",
+    "BigQueryAdapter",
     "ClickHouseAdapter",
     "CockroachDBAdapter",
     "DuckDBAdapter",
@@ -27,17 +29,19 @@
     "MySQLAdapter",
     "OracleAdapter",
     "PostgreSQLAdapter",
+    "RedshiftAdapter",
     "SQLiteAdapter",
     "SQLServerAdapter",
     "SupabaseAdapter",
     "TursoAdapter",
-    "AthenaAdapter",
     # Factory helpers
     "get_adapter",
     "get_supported_adapter_db_types",
 ]
 
 if TYPE_CHECKING:
+    from .athena import AthenaAdapter
+    from .bigquery import BigQueryAdapter
     from .clickhouse import ClickHouseAdapter
     from .cockroachdb import CockroachDBAdapter
     from .duckdb import DuckDBAdapter
@@ -47,10 +51,10 @@
     from .mysql import MySQLAdapter
     from .oracle import OracleAdapter
     from .postgresql import PostgreSQLAdapter
+    from .redshift import RedshiftAdapter
     from .sqlite import SQLiteAdapter
     from .supabase import SupabaseAdapter
     from .turso import TursoAdapter
-    from .athena import AthenaAdapter
 
 
 def get_adapter(db_type: str) -> DatabaseAdapter: