feat: pass extra_options to database drivers

- Add extra_options pass-through to all adapters, allowing users to pass
  custom driver parameters via connections.json or CLI URLs
- Add Snowflake authentication dropdown with support for:
  - Username & Password (default)
  - SSO (Browser)
  - Key Pair (JWT)
  - OAuth Token
- Add conditional fields for private key file and password when JWT is selected

Author: Maxteabag

sqlit/domains/connections/providers/clickhouse/adapter.py

@@ -141,6 +141,7 @@ def connect(self, config: ConnectionConfig) -> Any:
             if tls_mode != TLS_MODE_DEFAULT:
                 connect_args["verify"] = tls_mode != TLS_MODE_REQUIRE
 
+        connect_args.update(config.extra_options)
         client = clickhouse_connect.get_client(**connect_args)
         return client
 

sqlit/domains/connections/providers/cockroachdb/adapter.py

@@ -87,6 +87,7 @@ def connect(self, config: ConnectionConfig) -> Any:
             if tls_key_password:
                 connect_args["sslpassword"] = tls_key_password
 
+        connect_args.update(config.extra_options)
         conn = psycopg2.connect(**connect_args)
         # Enable autocommit to avoid transaction issues
         conn.autocommit = True

sqlit/domains/connections/providers/db2/adapter.py

@@ -81,7 +81,9 @@ def connect(self, config: ConnectionConfig) -> Any:
             f"UID={endpoint.username};"
             f"PWD={endpoint.password};"
         )
-        return ibm_db_dbi.connect(conn_str, "", "")
+        connect_args: dict[str, Any] = {}
+        connect_args.update(config.extra_options)
+        return ibm_db_dbi.connect(conn_str, "", "", **connect_args)
 
     def get_databases(self, conn: Any) -> list[str]:
         return []

sqlit/domains/connections/providers/duckdb/adapter.py

@@ -83,7 +83,9 @@ def connect(self, config: ConnectionConfig) -> Any:
             raise ValueError("DuckDB connections require a file endpoint.")
         file_path = resolve_file_path(str(file_endpoint.path))
         duckdb_any: Any = duckdb
-        return duckdb_any.connect(str(file_path))
+        connect_args: dict[str, Any] = {}
+        connect_args.update(config.extra_options)
+        return duckdb_any.connect(str(file_path), **connect_args)
 
     def get_databases(self, conn: Any) -> list[str]:
         """DuckDB doesn't support multiple databases - return empty list."""

sqlit/domains/connections/providers/firebird/adapter.py

@@ -72,13 +72,15 @@ def connect(self, config: "ConnectionConfig") -> Any:
         endpoint = config.tcp_endpoint
         if endpoint is None:
             raise ValueError("Firebird connections require a TCP-style endpoint.")
-        conn = firebirdsql.connect(
-            host=endpoint.host or "localhost",
-            port=int(endpoint.port) if endpoint.port else 3050,
-            database=endpoint.database or "security.db",
-            user=endpoint.username,
-            password=endpoint.password,
-        )
+        connect_args: dict[str, Any] = {
+            "host": endpoint.host or "localhost",
+            "port": int(endpoint.port) if endpoint.port else 3050,
+            "database": endpoint.database or "security.db",
+            "user": endpoint.username,
+            "password": endpoint.password,
+        }
+        connect_args.update(config.extra_options)
+        conn = firebirdsql.connect(**connect_args)
         return conn
 
     def get_databases(self, conn: Any) -> list[str]:

sqlit/domains/connections/providers/flight/adapter.py

@@ -130,6 +130,7 @@ def connect(self, config: ConnectionConfig) -> Any:
         if use_tls and config.get_option("flight_skip_verify", "false") == "true":
             db_kwargs["adbc.flight.sql.client_option.tls_skip_verify"] = "true"
 
+        db_kwargs.update(config.extra_options)
         conn = flight_sql.connect(uri, db_kwargs=db_kwargs)
 
         # Store the catalog/database for later use

sqlit/domains/connections/providers/hana/adapter.py

@@ -82,6 +82,7 @@ def connect(self, config: ConnectionConfig) -> Any:
         if schema:
             connect_args["currentSchema"] = schema
 
+        connect_args.update(config.extra_options)
         return hdbcli.connect(**connect_args)
 
     def get_databases(self, conn: Any) -> list[str]:

sqlit/domains/connections/providers/mariadb/adapter.py

@@ -100,6 +100,7 @@ def connect(self, config: ConnectionConfig) -> Any:
                 connect_args["ssl_verify_cert"] = tls_mode_verifies_cert(tls_mode)
                 connect_args["ssl_verify_identity"] = tls_mode_verifies_hostname(tls_mode)
 
+        connect_args.update(config.extra_options)
         conn = mariadb_any.connect(**connect_args)
         self._supports_sequences = self._detect_sequences_support(conn)
         return conn

sqlit/domains/connections/providers/mssql/adapter.py

@@ -180,6 +180,9 @@ def connect(self, config: ConnectionConfig) -> Any:
         )
 
         conn_str = self._build_connection_string(config)
+        # Append extra_options to connection string
+        for key, value in config.extra_options.items():
+            conn_str += f"{key}={value};"
         return mssql_python.connect(conn_str)
 
     def get_databases(self, conn: Any) -> list[str]:

sqlit/domains/connections/providers/mysql/adapter.py

@@ -110,4 +110,5 @@ def connect(self, config: ConnectionConfig) -> Any:
             ssl_params["check_hostname"] = tls_mode_verifies_hostname(tls_mode)
             connect_args["ssl"] = ssl_params
 
+        connect_args.update(config.extra_options)
         return pymysql.connect(**connect_args)