jl-db-comp 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

jl_db_comp/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.1.10'
+__version__ = VERSION = '0.1.11'

jl_db_comp/routes.py

@@ -1,4 +1,8 @@
+import hashlib
 import json
+import threading
+import time
+from contextlib import contextmanager
 from urllib.parse import unquote
 
 from jupyter_server.base.handlers import APIHandler
@@ -13,11 +17,133 @@ from .connections import (
 
 try:
     import psycopg2
+    import psycopg2.pool
+    from psycopg2 import sql as pgsql
     PSYCOPG2_AVAILABLE = True
 except ImportError:
     PSYCOPG2_AVAILABLE = False
 
 
+# ---------------------------------------------------------------------------
+# Connection pool
+# ---------------------------------------------------------------------------
+
+_pools: dict = {}
+_pool_lock = threading.Lock()
+
+
+@contextmanager
+def _pooled_connection(db_url: str):
+    """Borrow a connection from a per-URL pool, return it on exit.
+
+    Creates the pool lazily on first use (minconn=1, maxconn=5).
+    Sets ``autocommit=True`` since all queries are read-only.
+    On ``OperationalError`` the connection is discarded instead of returned.
+    """
+    conn = None
+    discard = False
+    try:
+        with _pool_lock:
+            if db_url not in _pools:
+                _pools[db_url] = psycopg2.pool.ThreadedConnectionPool(
+                    minconn=1, maxconn=5, dsn=db_url,
+                )
+        conn = _pools[db_url].getconn()
+        conn.autocommit = True
+        yield conn
+    except psycopg2.OperationalError:
+        discard = True
+        raise
+    finally:
+        if conn is not None:
+            try:
+                with _pool_lock:
+                    pool = _pools.get(db_url)
+                    if pool:
+                        pool.putconn(conn, close=discard)
+            except Exception:
+                pass
+
+
+# ---------------------------------------------------------------------------
+# Metadata cache (TTL = 120 s)
+# ---------------------------------------------------------------------------
+
+class _MetadataCache:
+    """Thread-safe TTL cache for database metadata."""
+
+    def __init__(self, ttl_seconds: float = 120.0):
+        self._ttl = ttl_seconds
+        self._data: dict = {}
+        self._lock = threading.Lock()
+
+    def get(self, key: str):
+        """Return cached value, or ``None`` if missing / expired."""
+        with self._lock:
+            entry = self._data.get(key)
+            if entry is None:
+                return None
+            value, expiry = entry
+            if time.monotonic() > expiry:
+                del self._data[key]
+                return None
+            return value
+
+    def put(self, key: str, value):
+        """Store *value* with TTL starting now."""
+        with self._lock:
+            self._data[key] = (value, time.monotonic() + self._ttl)
+
+    def clear(self):
+        """Drop every entry."""
+        with self._lock:
+            self._data.clear()
+
+
+_cache = _MetadataCache(ttl_seconds=120)
+
+
+# ---------------------------------------------------------------------------
+# Helpers
+# ---------------------------------------------------------------------------
+
+def _url_id(db_url: str) -> str:
+    """Short hash of a connection URL for use in cache keys."""
+    return hashlib.sha256(db_url.encode()).hexdigest()[:12]
+
+
+def _filter_by_prefix(items: list, prefix: str) -> list:
+    """Return items whose ``name`` starts with *prefix* (case-insensitive)."""
+    if not prefix:
+        return items
+    lp = prefix.lower()
+    return [item for item in items if item["name"].lower().startswith(lp)]
+
+
+def _jsonb_path_expr(column: str, path: list = None):
+    """Build a safe ``psycopg2.sql.Composable`` for a JSONB path.
+
+    ``_jsonb_path_expr("meta", ["a", "b"])`` produces
+    ``"meta"->'a'->'b'``.
+    """
+    expr = pgsql.Identifier(column)
+    for key in (path or []):
+        expr = pgsql.SQL("{0}->{1}").format(expr, pgsql.Literal(key))
+    return expr
+
+
+def _jsonb_path_display(column: str, path: list = None) -> str:
+    """Human-readable version of the JSONB path (for diagnostics JSON)."""
+    result = column
+    for key in (path or []):
+        result = f"{result}->'{key}'"
+    return result
+
+
+# ---------------------------------------------------------------------------
+# Completions handler
+# ---------------------------------------------------------------------------
+
 class PostgresCompletionsHandler(APIHandler):
     """Handler for fetching PostgreSQL table and column completions."""
 
@@ -30,8 +156,11 @@ class PostgresCompletionsHandler(APIHandler):
         - db_url: URL-encoded PostgreSQL connection string (fallback)
         - prefix: Optional prefix to filter results
         - schema: Database schema (default: 'public')
-        - table: Optional table name to filter columns (only returns columns from this table)
-        - schema_or_table: Ambiguous identifier - backend determines if it's a schema or table
+        - table: Optional table name to filter columns
+        - schema_or_table: Ambiguous identifier
+        - jsonb_column: JSONB column for key extraction
+        - jsonb_path: JSON-encoded path array for nested JSONB
+        - connections_file: Custom path to connections.ini
         """
         if not PSYCOPG2_AVAILABLE:
             self.set_status(500)
@@ -112,6 +241,8 @@ class PostgresCompletionsHandler(APIHandler):
                 "columns": []
             }))
 
+    # -- core logic ---------------------------------------------------------
+
     def _fetch_completions(
         self,
         db_url: str,
@@ -120,144 +251,127 @@ class PostgresCompletionsHandler(APIHandler):
         table: str = None,
         schema_or_table: str = None,
         jsonb_column: str = None,
-        jsonb_path: list = None
+        jsonb_path: list = None,
     ) -> dict:
         """Fetch table and column names from PostgreSQL.
 
-        Args:
-            db_url: PostgreSQL connection string
-            schema: Database schema name
-            prefix: Filter prefix (case-insensitive)
-            table: Optional table name to filter columns (only returns columns from this table)
-            schema_or_table: Ambiguous identifier - determine if it's a schema or table
-            jsonb_column: Optional JSONB column to extract keys from
-            jsonb_path: Optional path for nested JSONB key extraction
-
-        Returns:
-            Dictionary with tables, columns, and jsonbKeys arrays
+        Results are cached server-side (120 s TTL).  Prefix filtering is
+        done in Python after the cache lookup — SQL queries fetch full
+        result sets so that subsequent keystrokes hit the cache.
         """
-        conn = None
-        try:
-            conn = psycopg2.connect(db_url)
-            cursor = conn.cursor()
+        uid = _url_id(db_url)
 
-            tables = []
-            columns = []
-            jsonb_keys = []
+        # --- JSONB key extraction ---
+        if jsonb_column:
+            cache_key = (
+                f"jsonb:{uid}:{schema}:{schema_or_table or ''}:"
+                f"{jsonb_column}:{json.dumps(jsonb_path or [])}"
+            )
+            all_keys = _cache.get(cache_key)
+            if all_keys is None:
+                with _pooled_connection(db_url) as conn:
+                    cur = conn.cursor()
+                    all_keys = self._fetch_jsonb_keys(
+                        cur, schema, schema_or_table, jsonb_column, jsonb_path,
+                    )
+                    cur.close()
+                _cache.put(cache_key, all_keys)
+            return {
+                "status": "success",
+                "tables": [],
+                "columns": [],
+                "jsonbKeys": _filter_by_prefix(all_keys, prefix),
+            }
 
-            # Handle JSONB key extraction
-            if jsonb_column:
-                jsonb_keys = self._fetch_jsonb_keys(
-                    cursor, schema, schema_or_table, jsonb_column, jsonb_path, prefix
+        # --- schema_or_table disambiguation ---
+        if schema_or_table:
+            schema_ck = f"is_schema:{uid}:{schema_or_table.lower()}"
+            is_schema = _cache.get(schema_ck)
+            if is_schema is None:
+                with _pooled_connection(db_url) as conn:
+                    cur = conn.cursor()
+                    cur.execute(
+                        "SELECT 1 FROM information_schema.schemata "
+                        "WHERE LOWER(schema_name) = %s",
+                        (schema_or_table.lower(),),
+                    )
+                    is_schema = cur.fetchone() is not None
+                    cur.close()
+                _cache.put(schema_ck, is_schema)
+
+            if is_schema:
+                return self._tables_in_schema(uid, db_url, schema_or_table, prefix)
+            return self._columns_of_table(uid, db_url, schema, schema_or_table, prefix)
+
+        # --- explicit table → columns ---
+        if table:
+            return self._columns_of_table(uid, db_url, schema, table, prefix)
+
+        # --- default: list tables in schema ---
+        return self._tables_in_schema(uid, db_url, schema, prefix)
+
+    # -- helpers ------------------------------------------------------------
+
+    def _tables_in_schema(self, uid, db_url, schema, prefix):
+        """Return tables/views in *schema*, filtered by *prefix*."""
+        cache_key = f"tables:{uid}:{schema}"
+        all_tables = _cache.get(cache_key)
+        if all_tables is None:
+            with _pooled_connection(db_url) as conn:
+                cur = conn.cursor()
+                cur.execute(
+                    "SELECT table_name, table_type "
+                    "FROM information_schema.tables "
+                    "WHERE table_schema = %s "
+                    "  AND table_type IN ('BASE TABLE', 'VIEW') "
+                    "ORDER BY table_name",
+                    (schema,),
                 )
-                cursor.close()
-                return {
-                    "status": "success",
-                    "tables": [],
-                    "columns": [],
-                    "jsonbKeys": jsonb_keys
-                }
-
-            # Handle schema_or_table: check if it's a schema first, then try as table
-            if schema_or_table:
-                # First, check if it's a schema
-                cursor.execute("""
-                    SELECT schema_name
-                    FROM information_schema.schemata
-                    WHERE LOWER(schema_name) = %s
-                """, (schema_or_table.lower(),))
-
-                is_schema = cursor.fetchone() is not None
-
-                if is_schema:
-                    # It's a schema - fetch tables and views from that schema
-                    cursor.execute("""
-                        SELECT table_name, table_type
-                        FROM information_schema.tables
-                        WHERE table_schema = %s
-                          AND table_type IN ('BASE TABLE', 'VIEW')
-                          AND LOWER(table_name) LIKE %s
-                        ORDER BY table_name
-                    """, (schema_or_table, f"{prefix}%"))
-
-                    tables = [
-                        {
-                            "name": row[0],
-                            "type": "view" if row[1] == 'VIEW' else "table"
-                        }
-                        for row in cursor.fetchall()
-                    ]
-                else:
-                    # Not a schema - treat as table name, fetch columns from default schema
-                    cursor.execute("""
-                        SELECT table_name, column_name, data_type
-                        FROM information_schema.columns
-                        WHERE table_schema = %s
-                          AND LOWER(table_name) = %s
-                          AND LOWER(column_name) LIKE %s
-                        ORDER BY ordinal_position
-                    """, (schema, schema_or_table.lower(), f"{prefix}%"))
-
-                    columns = [
-                        {
-                            "name": row[1],
-                            "table": row[0],
-                            "dataType": row[2],
-                            "type": "column"
-                        }
-                        for row in cursor.fetchall()
-                    ]
-
-            # If table is specified with explicit schema, fetch columns from that table
-            elif table:
-                cursor.execute("""
-                    SELECT table_name, column_name, data_type
-                    FROM information_schema.columns
-                    WHERE table_schema = %s
-                      AND LOWER(table_name) = %s
-                      AND LOWER(column_name) LIKE %s
-                    ORDER BY ordinal_position
-                """, (schema, table.lower(), f"{prefix}%"))
-
-                columns = [
-                    {
-                        "name": row[1],
-                        "table": row[0],
-                        "dataType": row[2],
-                        "type": "column"
-                    }
-                    for row in cursor.fetchall()
+                all_tables = [
+                    {"name": r[0], "type": "view" if r[1] == "VIEW" else "table"}
+                    for r in cur.fetchall()
                 ]
-            else:
-                # No table or schema_or_table specified - fetch tables and views from default schema
-                cursor.execute("""
-                    SELECT table_name, table_type
-                    FROM information_schema.tables
-                    WHERE table_schema = %s
-                      AND table_type IN ('BASE TABLE', 'VIEW')
-                      AND LOWER(table_name) LIKE %s
-                    ORDER BY table_name
-                """, (schema, f"{prefix}%"))
-
-                tables = [
+                cur.close()
+            _cache.put(cache_key, all_tables)
+        return {
+            "status": "success",
+            "tables": _filter_by_prefix(all_tables, prefix),
+            "columns": [],
+        }
+
+    def _columns_of_table(self, uid, db_url, schema, table, prefix):
+        """Return columns of *table*, filtered by *prefix*."""
+        cache_key = f"columns:{uid}:{schema}:{table.lower()}"
+        all_cols = _cache.get(cache_key)
+        if all_cols is None:
+            with _pooled_connection(db_url) as conn:
+                cur = conn.cursor()
+                cur.execute(
+                    "SELECT table_name, column_name, data_type "
+                    "FROM information_schema.columns "
+                    "WHERE table_schema = %s "
+                    "  AND LOWER(table_name) = %s "
+                    "ORDER BY ordinal_position",
+                    (schema, table.lower()),
+                )
+                all_cols = [
                     {
-                        "name": row[0],
-                        "type": "view" if row[1] == 'VIEW' else "table"
+                        "name": r[1],
+                        "table": r[0],
+                        "dataType": r[2],
+                        "type": "column",
                     }
-                    for row in cursor.fetchall()
+                    for r in cur.fetchall()
                 ]
+                cur.close()
+            _cache.put(cache_key, all_cols)
+        return {
+            "status": "success",
+            "tables": [],
+            "columns": _filter_by_prefix(all_cols, prefix),
+        }
 
-            cursor.close()
-
-            return {
-                "status": "success",
-                "tables": tables,
-                "columns": columns
-            }
-
-        finally:
-            if conn:
-                conn.close()
+    # -- JSONB key extraction -----------------------------------------------
 
     def _fetch_jsonb_keys(
         self,
@@ -266,79 +380,65 @@ class PostgresCompletionsHandler(APIHandler):
         table_name: str,
         jsonb_column: str,
         jsonb_path: list = None,
-        prefix: str = ''
     ) -> list:
-        """Extract unique JSONB keys from a column in a table.
+        """Extract unique JSONB keys from a column.
 
-        Args:
-            cursor: Database cursor
-            schema: Database schema
-            table_name: Table containing the JSONB column (can be None)
-            jsonb_column: Name of the JSONB column
-            jsonb_path: Optional path for nested keys (e.g., ['user', 'profile'])
-            prefix: Filter prefix for keys
-
-        Returns:
-            List of JSONB key completion items
+        Returns the full (unfiltered) list — the caller applies prefix
+        filtering.  Uses bounded sub-queries so that at most 100 rows
+        are scanned for diagnostics and 1 000 for key extraction.
         """
         try:
-            # If no table specified, find tables with this JSONB column
+            # If no table specified, find the first table with this JSONB column
             if not table_name:
-                cursor.execute("""
-                    SELECT table_name
-                    FROM information_schema.columns
-                    WHERE table_schema = %s
-                      AND LOWER(column_name) = %s
-                      AND data_type = 'jsonb'
-                    LIMIT 1
-                """, (schema, jsonb_column.lower()))
-
+                cursor.execute(
+                    "SELECT table_name "
+                    "FROM information_schema.columns "
+                    "WHERE table_schema = %s "
+                    "  AND LOWER(column_name) = %s "
+                    "  AND data_type = 'jsonb' "
+                    "LIMIT 1",
+                    (schema, jsonb_column.lower()),
+                )
                 result = cursor.fetchone()
                 if not result:
                     self.log.warning(
-                        f"JSONB completion: No JSONB column '{jsonb_column}' found "
-                        f"in schema '{schema}'. Verify the column exists and has "
-                        f"data_type='jsonb'."
+                        f"JSONB completion: No JSONB column '{jsonb_column}' "
+                        f"found in schema '{schema}'. Verify the column exists "
+                        f"and has data_type='jsonb'."
                     )
                     return []
-
                 table_name = result[0]
                 self.log.info(
                     f"JSONB completion: Found column '{jsonb_column}' in "
                     f"table '{schema}.{table_name}'"
                 )
 
-            # Build the JSONB path expression
-            if jsonb_path and len(jsonb_path) > 0:
-                # For nested paths: column->>'key1'->>'key2'
-                path_expr = jsonb_column
-                for key in jsonb_path:
-                    path_expr = f"{path_expr}->'{key}'"
-            else:
-                # For top-level keys: just the column
-                path_expr = jsonb_column
-
-            # First, check the data distribution at this path for diagnostics
-            diag_query = f"""
-                SELECT
-                    COUNT(*) as total_rows,
-                    COUNT({path_expr}) as non_null_count,
-                    COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'object' THEN 1 END) as object_count,
-                    COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'array' THEN 1 END) as array_count,
-                    COUNT(CASE WHEN jsonb_typeof({path_expr}) IN ('string', 'number', 'boolean') THEN 1 END) as scalar_count
-                FROM {schema}.{table_name}
-                LIMIT 1000
-            """
-            cursor.execute(diag_query)
+            path = _jsonb_path_expr(jsonb_column, jsonb_path)
+            sch = pgsql.Identifier(schema)
+            tbl = pgsql.Identifier(table_name)
+
+            # Diagnostic sample (100 rows) — bounded via sub-query
+            cursor.execute(
+                pgsql.SQL(
+                    "SELECT "
+                    "  COUNT(*), "
+                    "  COUNT(jval), "
+                    "  COUNT(CASE WHEN jsonb_typeof(jval) = 'object' THEN 1 END), "
+                    "  COUNT(CASE WHEN jsonb_typeof(jval) = 'array' THEN 1 END), "
+                    "  COUNT(CASE WHEN jsonb_typeof(jval) "
+                    "    IN ('string','number','boolean') THEN 1 END) "
+                    "FROM (SELECT {0} AS jval FROM {1}.{2} LIMIT 100) sub"
+                ).format(path, sch, tbl)
+            )
             diag = cursor.fetchone()
-
-            total_rows, non_null, obj_count, arr_count, scalar_count = diag
+            _, non_null, obj_count, arr_count, scalar_count = diag
 
             if non_null == 0:
                 self.log.warning(
                     f"JSONB completion: Column '{jsonb_column}' in "
                     f"'{schema}.{table_name}' has no non-NULL values at "
-                    f"path '{path_expr}'. Keys cannot be extracted from NULL data."
+                    f"path '{_jsonb_path_display(jsonb_column, jsonb_path)}'. "
+                    f"Keys cannot be extracted from NULL data."
                 )
                 return []
 
@@ -349,57 +449,61 @@ class PostgresCompletionsHandler(APIHandler):
                 if scalar_count > 0:
                     type_info.append(f"{scalar_count} scalars")
                 self.log.warning(
-                    f"JSONB completion: Path '{path_expr}' in "
+                    f"JSONB completion: Path "
+                    f"'{_jsonb_path_display(jsonb_column, jsonb_path)}' in "
                     f"'{schema}.{table_name}' contains no JSON objects "
                     f"(found: {', '.join(type_info) if type_info else 'only NULL'}). "
                     f"Keys can only be extracted from object types."
                 )
                 return []
 
-            # Query to extract unique keys
-            # LIMIT to 1000 rows for performance (sample the table)
-            query = f"""
-                SELECT DISTINCT jsonb_object_keys({path_expr})
-                FROM {schema}.{table_name}
-                WHERE {path_expr} IS NOT NULL
-                  AND jsonb_typeof({path_expr}) = 'object'
-                LIMIT 1000
-            """
-
-            cursor.execute(query)
+            # Key extraction — scan at most 1 000 qualifying rows
+            cursor.execute(
+                pgsql.SQL(
+                    "SELECT DISTINCT jsonb_object_keys(jval) "
+                    "FROM ("
+                    "  SELECT {0} AS jval FROM {1}.{2} "
+                    "  WHERE {0} IS NOT NULL "
+                    "    AND jsonb_typeof({0}) = 'object' "
+                    "  LIMIT 1000"
+                    ") sub"
+                ).format(path, sch, tbl)
+            )
             keys = cursor.fetchall()
 
             if len(keys) == 0:
                 self.log.warning(
-                    f"JSONB completion: No keys found at path '{path_expr}' in "
+                    f"JSONB completion: No keys found at path "
+                    f"'{_jsonb_path_display(jsonb_column, jsonb_path)}' in "
                     f"'{schema}.{table_name}' despite {obj_count} objects. "
                     f"Objects may be empty {{}}."
                 )
                 return []
 
-            # Filter by prefix and format results
-            result = []
-            for row in keys:
-                key = row[0]
-                if key.lower().startswith(prefix):
-                    result.append({
-                        "name": key,
-                        "type": "jsonb_key",
-                        "keyPath": (jsonb_path or []) + [key]
-                    })
-
             self.log.info(
-                f"JSONB completion: Found {len(keys)} unique keys at '{path_expr}' "
-                f"in '{schema}.{table_name}' (sampled {obj_count} objects)"
+                f"JSONB completion: Found {len(keys)} unique keys at "
+                f"'{_jsonb_path_display(jsonb_column, jsonb_path)}' in "
+                f"'{schema}.{table_name}' (sampled {obj_count} objects)"
             )
 
-            return result
+            return [
+                {
+                    "name": r[0],
+                    "type": "jsonb_key",
+                    "keyPath": (jsonb_path or []) + [r[0]],
+                }
+                for r in keys
+            ]
 
         except psycopg2.Error as e:
             self.log.error(f"JSONB key extraction error: {str(e).split(chr(10))[0]}")
             return []
 
 
+# ---------------------------------------------------------------------------
+# JSONB diagnostics handler
+# ---------------------------------------------------------------------------
+
 class JsonbDiagnosticsHandler(APIHandler):
     """Handler for diagnosing JSONB column issues."""
 
@@ -488,9 +592,7 @@ class JsonbDiagnosticsHandler(APIHandler):
         jsonb_path: list = None
     ) -> dict:
         """Get diagnostic information about JSONB columns."""
-        conn = None
-        try:
-            conn = psycopg2.connect(db_url)
+        with _pooled_connection(db_url) as conn:
             cursor = conn.cursor()
 
             result = {
@@ -502,12 +604,12 @@ class JsonbDiagnosticsHandler(APIHandler):
 
             # Find all JSONB columns in the schema
             query_params = [schema]
-            query = """
-                SELECT table_name, column_name
-                FROM information_schema.columns
-                WHERE table_schema = %s
-                  AND data_type = 'jsonb'
-            """
+            query = (
+                "SELECT table_name, column_name "
+                "FROM information_schema.columns "
+                "WHERE table_schema = %s "
+                "  AND data_type = 'jsonb'"
+            )
             if table:
                 query += " AND LOWER(table_name) = %s"
                 query_params.append(table.lower())
@@ -530,34 +632,31 @@ class JsonbDiagnosticsHandler(APIHandler):
                 actual_table = jsonb_columns[0][0]
                 actual_column = jsonb_columns[0][1]
 
-                # Build path expression
-                if jsonb_path and len(jsonb_path) > 0:
-                    path_expr = actual_column
-                    for key in jsonb_path:
-                        path_expr = f"{path_expr}->'{key}'"
-                else:
-                    path_expr = actual_column
+                path_expr = _jsonb_path_expr(actual_column, jsonb_path)
+                sch_id = pgsql.Identifier(schema)
+                tbl_id = pgsql.Identifier(actual_table)
 
                 # Get type distribution
-                diag_query = f"""
-                    SELECT
-                        COUNT(*) as total_rows,
-                        COUNT({path_expr}) as non_null_count,
-                        COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'object' THEN 1 END) as object_count,
-                        COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'array' THEN 1 END) as array_count,
-                        COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'string' THEN 1 END) as string_count,
-                        COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'number' THEN 1 END) as number_count,
-                        COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'boolean' THEN 1 END) as boolean_count,
-                        COUNT(CASE WHEN jsonb_typeof({path_expr}) = 'null' THEN 1 END) as json_null_count
-                    FROM {schema}.{actual_table}
-                """
+                diag_query = pgsql.SQL(
+                    "SELECT "
+                    "  COUNT(*) AS total_rows, "
+                    "  COUNT({0}) AS non_null_count, "
+                    "  COUNT(CASE WHEN jsonb_typeof({0}) = 'object' THEN 1 END) AS object_count, "
+                    "  COUNT(CASE WHEN jsonb_typeof({0}) = 'array' THEN 1 END) AS array_count, "
+                    "  COUNT(CASE WHEN jsonb_typeof({0}) = 'string' THEN 1 END) AS string_count, "
+                    "  COUNT(CASE WHEN jsonb_typeof({0}) = 'number' THEN 1 END) AS number_count, "
+                    "  COUNT(CASE WHEN jsonb_typeof({0}) = 'boolean' THEN 1 END) AS boolean_count, "
+                    "  COUNT(CASE WHEN jsonb_typeof({0}) = 'null' THEN 1 END) AS json_null_count "
+                    "FROM {1}.{2}"
+                ).format(path_expr, sch_id, tbl_id)
+
                 cursor.execute(diag_query)
                 diag = cursor.fetchone()
 
                 result["columnDiagnostics"] = {
                     "table": actual_table,
                     "column": actual_column,
-                    "pathExpression": path_expr,
+                    "pathExpression": _jsonb_path_display(actual_column, jsonb_path),
                     "totalRows": diag[0],
                     "nonNullCount": diag[1],
                     "typeDistribution": {
@@ -575,26 +674,24 @@ class JsonbDiagnosticsHandler(APIHandler):
                 # If there are objects, get sample keys
                 if diag[2] > 0:
                     try:
-                        key_query = f"""
-                            SELECT DISTINCT jsonb_object_keys({path_expr})
-                            FROM {schema}.{actual_table}
-                            WHERE {path_expr} IS NOT NULL
-                              AND jsonb_typeof({path_expr}) = 'object'
-                            LIMIT 20
-                        """
+                        key_query = pgsql.SQL(
+                            "SELECT DISTINCT jsonb_object_keys(jval) "
+                            "FROM ("
+                            "  SELECT {0} AS jval FROM {1}.{2} "
+                            "  WHERE {0} IS NOT NULL "
+                            "    AND jsonb_typeof({0}) = 'object' "
+                            "  LIMIT 1000"
+                            ") sub"
+                        ).format(path_expr, sch_id, tbl_id)
                         cursor.execute(key_query)
                         keys = [row[0] for row in cursor.fetchall()]
-                        result["columnDiagnostics"]["sampleKeys"] = keys
+                        result["columnDiagnostics"]["sampleKeys"] = keys[:20]
                     except psycopg2.Error:
                         result["columnDiagnostics"]["sampleKeys"] = []
 
             cursor.close()
             return result
 
-        finally:
-            if conn:
-                conn.close()
-
     def _get_recommendation(self, diag) -> str:
         """Generate a recommendation based on diagnostic data."""
         total, non_null, obj, arr, string, number, boolean, json_null = diag
@@ -628,6 +725,10 @@ class JsonbDiagnosticsHandler(APIHandler):
         return f"JSONB autocompletion should work. Found {obj} objects with extractable keys."
 
 
+# ---------------------------------------------------------------------------
+# Connections handler
+# ---------------------------------------------------------------------------
+
 class ConnectionsHandler(APIHandler):
     """Handler for listing available database connections."""
 
@@ -660,6 +761,10 @@ class ConnectionsHandler(APIHandler):
             }))
 
 
+# ---------------------------------------------------------------------------
+# Route registration
+# ---------------------------------------------------------------------------
+
 def setup_route_handlers(web_app):
     """Register route handlers with the Jupyter server."""
     host_pattern = ".*$"

{jl_db_comp-0.1.10.data → jl_db_comp-0.1.11.data}/data/share/jupyter/labextensions/jl_db_comp/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jl_db_comp",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "A JupyterLab extension to complete db queries in jupyterlab notebooks",
   "keywords": [
     "jupyter",
@@ -118,7 +118,7 @@
     "outputDir": "jl_db_comp/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.e1b6bbc28eb414e4e599.js",
+      "load": "static/remoteEntry.2f90ff333d0ac2cbb0a8.js",
       "extension": "./extension",
       "style": "./style"
     }

{jl_db_comp-0.1.10.data → jl_db_comp-0.1.11.data}/data/share/jupyter/labextensions/jl_db_comp/schemas/jl_db_comp/package.json.orig

@@ -1,6 +1,6 @@
 {
     "name": "jl_db_comp",
-    "version": "0.1.10",
+    "version": "0.1.11",
     "description": "A JupyterLab extension to complete db queries in jupyterlab notebooks",
     "keywords": [
         "jupyter",

jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/171.c84e54337cfd177ebe22.js

@@ -0,0 +1 @@
+"use strict";(self.webpackChunkjl_db_comp=self.webpackChunkjl_db_comp||[]).push([[171],{171(e,n,t){t.r(n),t.d(n,{default:()=>m});var o=t(141),s=t(931),i=t(249),a=t(125),r=t(526);async function c(e="",n={}){const t=a.ServerConnection.makeSettings(),o=r.URLExt.join(t.baseUrl,"jl-db-comp",e);let s;try{s=await a.ServerConnection.makeRequest(o,n,t)}catch(e){throw new a.ServerConnection.NetworkError(e)}let i=await s.text();if(i.length>0)try{i=JSON.parse(i)}catch(e){console.log("Not a JSON response body.",s)}if(!s.ok)throw new a.ServerConnection.ResponseError(s,i.message||i);return i}class l{constructor(e,n){this.identifier="jl_db_comp:postgres-completer",this.renderer=null,this._cache=new Map,this._cacheTTL=3e5,this._settings=null,this._notebookTracker=null,this._connectionName="",this._schema="public",this._enabled=!0,this._availableConnections=[],this._cachedKernelConfig=null,this._kernelConfigCacheTime=0,this._kernelConfigCacheTTL=3e4,this._sqlKeywords=["select","from","join","where","insert","update","delete","inner","left","right","outer","on","group","order","by","having","into","values","set"],this._notebookTracker=n||null,e&&(this._settings=e,this._loadSettings(),e.changed.connect(()=>{this._loadSettings()})),this._loadAvailableConnections()}_loadSettings(){this._settings&&(this._connectionName=this._settings.get("connectionName").composite,this._schema=this._settings.get("schema").composite,this._enabled=this._settings.get("enabled").composite)}async _loadAvailableConnections(){try{const e=await async function(){try{return await c("connections",{method:"GET"})}catch(e){if(e instanceof a.ServerConnection.ResponseError)console.error(`Failed to fetch connections: ${e.message}`);else{const n=e instanceof Error?e.message:"Unknown error";console.error(`Failed to fetch connections: ${n}`)}return{status:"error",connections:{},filePath:null,message:"Failed to fetch connections"}}}();"success"===e.status&&(this._availableConnections=Object.keys(e.connections))}catch(e){console.warn("Failed to load available connections:",e)}}async _getKernelConfig(){var e;const n=Date.now();if(this._cachedKernelConfig&&n-this._kernelConfigCacheTime<this._kernelConfigCacheTTL)return this._cachedKernelConfig;if(!this._notebookTracker)return null;const t=this._notebookTracker.currentWidget;if(!t)return null;const o=null===(e=t.sessionContext.session)||void 0===e?void 0:e.kernel;if(!o)return null;try{const e=o.requestExecute({code:"\nimport json\nimport os\nresult = {'connection': '', 'dsn_filename': ''}\n\n# Get active connection\ntry:\n    from sql.connection import ConnectionManager\n    conn = ConnectionManager.current\n    if conn:\n        for alias, c in ConnectionManager.connections.items():\n            if c is conn:\n                result['connection'] = alias\n                break\nexcept:\n    pass\n\n# Get dsn_filename from SqlMagic instance\ndsn_filename = None\ntry:\n    from sql.magic import SqlMagic\n    ip = get_ipython()\n    if ip:\n        for name, inst in ip.magics_manager.registry.items():\n            if isinstance(inst, SqlMagic):\n                dsn_filename = inst.dsn_filename\n                break\nexcept:\n    pass\n\n# Fallback: try to get from config\nif not dsn_filename:\n    try:\n        ip = get_ipython()\n        if ip and hasattr(ip, 'config'):\n            sql_config = ip.config.get('SqlMagic', {})\n            if 'dsn_filename' in sql_config:\n                dsn_filename = sql_config['dsn_filename']\n    except:\n        pass\n\n# Convert to absolute path if we have a dsn_filename\nif dsn_filename:\n    if not os.path.isabs(dsn_filename):\n        # Resolve relative to current working directory\n        dsn_filename = os.path.abspath(dsn_filename)\n    result['dsn_filename'] = dsn_filename\n\nprint(json.dumps(result))\n",silent:!0,store_history:!1}),t=await new Promise(n=>{let t="";e.onIOPub=e=>{if("stream"===e.header.msg_type){const n=e.content;"stdout"===n.name&&(t+=n.text)}},e.done.then(()=>{try{const e=JSON.parse(t.trim());n({connection:e.connection||"",dsnFilename:e.dsn_filename||""})}catch(e){n(null)}}).catch(()=>{n(null)})});return t&&(this._cachedKernelConfig=t,this._kernelConfigCacheTime=n),t}catch(e){return console.warn("Failed to get jupysql config from kernel:",e),null}}async isApplicable(e){if(!this._enabled)return!1;const n=e.editor;if(!n)return!1;const t=n.model.sharedModel.getSource();if(!t)return!1;const o=t.toLowerCase();return this._sqlKeywords.some(e=>o.includes(e))}async fetch(e,n){var t;if(!this._enabled)return{start:e.offset,end:e.offset,items:[]};const{text:o,offset:s}=e,i=this._extractContext(o,s);let r;if(i.jsonbColumn){const e=(null===(t=i.jsonbPath)||void 0===t?void 0:t.join("."))||"";r=`jsonb:${i.schemaOrTable?`${i.schemaOrTable}.`:""}${i.jsonbColumn}->${e}`.toLowerCase()}else r=i.schema&&i.tableName?`cols:${i.schema}.${i.tableName}`.toLowerCase():i.schemaOrTable?`sot:${i.schemaOrTable}`.toLowerCase():`tables:${this._schema}`.toLowerCase();const l=this._getCached(r);if(l){const n=this._filterByPrefix(l,i.prefix);return this._formatReply(n,e.offset,i.prefix)}try{const n=await this._getKernelConfig();let t,o=this._connectionName;n&&(n.dsnFilename&&(t=n.dsnFilename),!o&&n.connection&&(o=n.connection)),!o&&this._availableConnections.length>0&&(o=this._availableConnections[0]);const s=await async function(e,n="",t="public",o,s,i,r,l){try{const a=new URLSearchParams;e&&a.append("connection",e),l&&a.append("connections_file",l),n&&a.append("prefix",n),a.append("schema",t),o&&a.append("table",o),s&&a.append("schema_or_table",s),i&&(a.append("jsonb_column",i),r&&r.length>0&&a.append("jsonb_path",JSON.stringify(r)));const h=`completions?${a.toString()}`,m=await c(h,{method:"GET"});return"error"===m.status?(console.error("PostgreSQL completion error:",m.message),[]):i&&m.jsonbKeys?m.jsonbKeys:o||s?m.columns.length>0?m.columns:m.tables:[...m.tables,...m.columns]}catch(e){if(e instanceof a.ServerConnection.ResponseError){const n=e.response.status;let t=e.message;"string"==typeof t&&(t.includes("<!DOCTYPE")||t.includes("<html"))&&(t=`HTML error page (${t.substring(0,100)}...)`),console.error(`PostgreSQL completions API failed (${n}): ${t}`)}else{const n=e instanceof Error?e.message:"Unknown error";console.error(`PostgreSQL completions API failed: ${n}`)}return[]}}(o||void 0,"",i.schema||this._schema,i.tableName,i.schemaOrTable,i.jsonbColumn,i.jsonbPath,t);this._cache.set(r,{items:s,timestamp:Date.now()});const l=this._filterByPrefix(s,i.prefix);return this._formatReply(l,e.offset,i.prefix)}catch(n){return console.error("Failed to fetch PostgreSQL completions:",n),{start:e.offset,end:e.offset,items:[]}}}_extractContext(e,n){const t=e.substring(0,n);if(t.includes("->")){const e=e=>{const n=[],t=/['"]?([\w]+)['"]?\s*->/g;let o;for(;null!==(o=t.exec(e));)n.push(o[1]);const s=e.lastIndexOf("->");let i="";return i=s>=0?e.substring(s+2).trim().replace(/['"]/g,""):e.trim().replace(/['"]/g,""),{jsonbPath:n,prefix:i}},n=t.match(/([\w]+)\.([\w]+)\.([\w]+)\s*->\s*(.*)$/);if(n){const t=n[1],o=n[2],s=n[3],i=n[4],{jsonbPath:a,prefix:r}=e(i);return{schema:t,tableName:o,jsonbColumn:s,jsonbPath:a,prefix:r}}const o=t.match(/([\w]+)\.([\w]+)\s*->\s*(.*)$/);if(o){const n=o[1],t=o[2],s=o[3],{jsonbPath:i,prefix:a}=e(s);return{schemaOrTable:n,jsonbColumn:t,jsonbPath:i,prefix:a}}const s=t.match(/([\w]+)\s*->\s*(.*)$/);if(s){const n=s[1],t=s[2],{jsonbPath:o,prefix:i}=e(t);return{jsonbColumn:n,jsonbPath:o,prefix:i}}}const o=t.match(/([\w]+)\.([\w]+)\.([\w]*)$/);if(o)return{schema:o[1],tableName:o[2],prefix:o[3]};const s=t.match(/([\w]+)\.([\w]*)$/);if(s)return{schemaOrTable:s[1],prefix:s[2]};const i=t.match(/[\w]+$/),a=i?i[0]:"",r=e.toLowerCase().match(/\bfrom\s+([\w]+\.)?[\w]+/);if(r){const e=r[0].match(/\bfrom\s+(?:([\w]+)\.)?([\w]+)/);if(e){const n=e[1],t=e[2];return n?{schema:n,tableName:t,prefix:a}:{schemaOrTable:t,prefix:a}}}return{prefix:a}}_getCached(e){const n=e.toLowerCase(),t=this._cache.get(n);return t?Date.now()-t.timestamp>this._cacheTTL?(this._cache.delete(n),null):t.items:null}_formatReply(e,n,t){return{start:n-t.length,end:n,items:e.map(e=>{let n=e.name,t=e.name;"jsonb_key"===e.type&&(t=`'${e.name}'`),"column"===e.type&&e.table&&(n=`${e.name} (${e.table})`);let o,s="📊",i=e.name;return"table"===e.type?s="📋":"view"===e.type?s="👁️":"jsonb_key"===e.type&&(s="🔑",i=`0000${e.name}`),"column"===e.type&&e.dataType&&e.table?o=`${e.table}.${e.name}: ${e.dataType}`:"jsonb_key"===e.type&&e.keyPath&&(o=`JSONB key: ${e.keyPath.join(" -> ")}`),{label:`${s} ${n}`,insertText:t,sortText:i,type:e.type,documentation:o}})}}_filterByPrefix(e,n){if(!n)return e;const t=n.toLowerCase();return e.filter(e=>e.name.toLowerCase().startsWith(t))}clearCache(){this._cache.clear()}}const h="jl_db_comp:plugin",m={id:h,description:"A JupyterLab extension to complete db queries in jupyterlab notebooks",autoStart:!0,requires:[o.ICompletionProviderManager],optional:[i.ISettingRegistry,s.INotebookTracker],activate:(e,n,t,o)=>{let s;t?t.load(h).then(e=>{s=new l(e,o),n.registerProvider(s),console.log("JupyterLab extension jl_db_comp is activated!")}).catch(e=>{console.error("Failed to load settings for jl_db_comp:",e),s=new l(null,o),n.registerProvider(s),console.log("JupyterLab extension jl_db_comp is activated!")}):(s=new l(null,o),n.registerProvider(s),console.log("JupyterLab extension jl_db_comp is activated!"))}}}}]);

jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/remoteEntry.e1b6bbc28eb414e4e599.js → jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/remoteEntry.2f90ff333d0ac2cbb0a8.js

@@ -1 +1 @@
-var _JUPYTERLAB;(()=>{"use strict";var e,r,t,n,o,a,i,u,l,s,f,c,d,p,h,v,b,g,m,y={246(e,r,t){var n={"./index":()=>t.e(171).then(()=>()=>t(171)),"./extension":()=>t.e(171).then(()=>()=>t(171)),"./style":()=>t.e(728).then(()=>()=>t(728))},o=(e,r)=>(t.R=r,r=t.o(n,e)?n[e]():Promise.resolve().then(()=>{throw new Error('Module "'+e+'" does not exist in container.')}),t.R=void 0,r),a=(e,r)=>{if(t.S){var n="default",o=t.S[n];if(o&&o!==e)throw new Error("Container initialization failed as it has already been initialized with a different share scope");return t.S[n]=e,t.I(n,r)}};t.d(r,{get:()=>o,init:()=>a})}},w={};function j(e){var r=w[e];if(void 0!==r)return r.exports;var t=w[e]={id:e,exports:{}};return y[e](t,t.exports,j),t.exports}j.m=y,j.c=w,j.n=e=>{var r=e&&e.__esModule?()=>e.default:()=>e;return j.d(r,{a:r}),r},j.d=(e,r)=>{for(var t in r)j.o(r,t)&&!j.o(e,t)&&Object.defineProperty(e,t,{enumerable:!0,get:r[t]})},j.f={},j.e=e=>Promise.all(Object.keys(j.f).reduce((r,t)=>(j.f[t](e,r),r),[])),j.u=e=>e+"."+{171:"545011db1d7843ce3ab3",728:"6552504d5b9b27551bc5"}[e]+".js?v="+{171:"545011db1d7843ce3ab3",728:"6552504d5b9b27551bc5"}[e],j.g=function(){if("object"==typeof globalThis)return globalThis;try{return this||new Function("return this")()}catch(e){if("object"==typeof window)return window}}(),j.o=(e,r)=>Object.prototype.hasOwnProperty.call(e,r),e={},r="jl_db_comp:",j.l=(t,n,o,a)=>{if(e[t])e[t].push(n);else{var i,u;if(void 0!==o)for(var l=document.getElementsByTagName("script"),s=0;s<l.length;s++){var f=l[s];if(f.getAttribute("src")==t||f.getAttribute("data-webpack")==r+o){i=f;break}}i||(u=!0,(i=document.createElement("script")).charset="utf-8",j.nc&&i.setAttribute("nonce",j.nc),i.setAttribute("data-webpack",r+o),i.src=t),e[t]=[n];var c=(r,n)=>{i.onerror=i.onload=null,clearTimeout(d);var o=e[t];if(delete e[t],i.parentNode&&i.parentNode.removeChild(i),o&&o.forEach(e=>e(n)),r)return r(n)},d=setTimeout(c.bind(null,void 0,{type:"timeout",target:i}),12e4);i.onerror=c.bind(null,i.onerror),i.onload=c.bind(null,i.onload),u&&document.head.appendChild(i)}},j.r=e=>{"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},(()=>{j.S={};var e={},r={};j.I=(t,n)=>{n||(n=[]);var o=r[t];if(o||(o=r[t]={}),!(n.indexOf(o)>=0)){if(n.push(o),e[t])return e[t];j.o(j.S,t)||(j.S[t]={});var a=j.S[t],i="jl_db_comp",u=[];return"default"===t&&((e,r,t,n)=>{var o=a[e]=a[e]||{},u=o[r];(!u||!u.loaded&&(1!=!u.eager?n:i>u.from))&&(o[r]={get:()=>j.e(171).then(()=>()=>j(171)),from:i,eager:!1})})("jl_db_comp","0.1.10"),e[t]=u.length?Promise.all(u).then(()=>e[t]=1):1}}})(),(()=>{var e;j.g.importScripts&&(e=j.g.location+"");var r=j.g.document;if(!e&&r&&(r.currentScript&&"SCRIPT"===r.currentScript.tagName.toUpperCase()&&(e=r.currentScript.src),!e)){var t=r.getElementsByTagName("script");if(t.length)for(var n=t.length-1;n>-1&&(!e||!/^http(s?):/.test(e));)e=t[n--].src}if(!e)throw new Error("Automatic publicPath is not supported in this browser");e=e.replace(/^blob:/,"").replace(/#.*$/,"").replace(/\?.*$/,"").replace(/\/[^\/]+$/,"/"),j.p=e})(),t=e=>{var r=e=>e.split(".").map(e=>+e==e?+e:e),t=/^([^-+]+)?(?:-([^+]+))?(?:\+(.+))?$/.exec(e),n=t[1]?r(t[1]):[];return t[2]&&(n.length++,n.push.apply(n,r(t[2]))),t[3]&&(n.push([]),n.push.apply(n,r(t[3]))),n},n=(e,r)=>{e=t(e),r=t(r);for(var n=0;;){if(n>=e.length)return n<r.length&&"u"!=(typeof r[n])[0];var o=e[n],a=(typeof o)[0];if(n>=r.length)return"u"==a;var i=r[n],u=(typeof i)[0];if(a!=u)return"o"==a&&"n"==u||"s"==u||"u"==a;if("o"!=a&&"u"!=a&&o!=i)return o<i;n++}},o=e=>{var r=e[0],t="";if(1===e.length)return"*";if(r+.5){t+=0==r?">=":-1==r?"<":1==r?"^":2==r?"~":r>0?"=":"!=";for(var n=1,a=1;a<e.length;a++)n--,t+="u"==(typeof(u=e[a]))[0]?"-":(n>0?".":"")+(n=2,u);return t}var i=[];for(a=1;a<e.length;a++){var u=e[a];i.push(0===u?"not("+l()+")":1===u?"("+l()+" || "+l()+")":2===u?i.pop()+" "+i.pop():o(u))}return l();function l(){return i.pop().replace(/^\((.+)\)$/,"$1")}},a=(e,r)=>{if(0 in e){r=t(r);var n=e[0],o=n<0;o&&(n=-n-1);for(var i=0,u=1,l=!0;;u++,i++){var s,f,c=u<e.length?(typeof e[u])[0]:"";if(i>=r.length||"o"==(f=(typeof(s=r[i]))[0]))return!l||("u"==c?u>n&&!o:""==c!=o);if("u"==f){if(!l||"u"!=c)return!1}else if(l)if(c==f)if(u<=n){if(s!=e[u])return!1}else{if(o?s>e[u]:s<e[u])return!1;s!=e[u]&&(l=!1)}else if("s"!=c&&"n"!=c){if(o||u<=n)return!1;l=!1,u--}else{if(u<=n||f<c!=o)return!1;l=!1}else"s"!=c&&"n"!=c&&(l=!1,u--)}}var d=[],p=d.pop.bind(d);for(i=1;i<e.length;i++){var h=e[i];d.push(1==h?p()|p():2==h?p()&p():h?a(h,r):!p())}return!!p()},i=(e,r)=>e&&j.o(e,r),u=e=>(e.loaded=1,e.get()),l=e=>Object.keys(e).reduce((r,t)=>(e[t].eager&&(r[t]=e[t]),r),{}),s=(e,r,t)=>{var o=t?l(e[r]):e[r];return Object.keys(o).reduce((e,r)=>!e||!o[e].loaded&&n(e,r)?r:e,0)},f=(e,r,t,n)=>"Unsatisfied version "+t+" from "+(t&&e[r][t].from)+" of shared singleton module "+r+" (required "+o(n)+")",c=e=>{throw new Error(e)},d=e=>{"undefined"!=typeof console&&console.warn&&console.warn(e)},p=(e,r,t)=>t?t():((e,r)=>c("Shared module "+r+" doesn't exist in shared scope "+e))(e,r),h=(e=>function(r,t,n,o,a){var i=j.I(r);return i&&i.then&&!n?i.then(e.bind(e,r,j.S[r],t,!1,o,a)):e(r,j.S[r],t,n,o,a)})((e,r,t,n,o,l)=>{if(!i(r,t))return p(e,t,l);var c=s(r,t,n);return a(o,c)||d(f(r,t,c,o)),u(r[t][c])}),v={},b={125:()=>h("default","@jupyterlab/services",!1,[1,7,5,3]),141:()=>h("default","@jupyterlab/completer",!1,[1,4,5,3]),249:()=>h("default","@jupyterlab/settingregistry",!1,[1,4,5,3]),526:()=>h("default","@jupyterlab/coreutils",!1,[1,6,5,3]),931:()=>h("default","@jupyterlab/notebook",!1,[1,4,5,3])},g={171:[125,141,249,526,931]},m={},j.f.consumes=(e,r)=>{j.o(g,e)&&g[e].forEach(e=>{if(j.o(v,e))return r.push(v[e]);if(!m[e]){var t=r=>{v[e]=0,j.m[e]=t=>{delete j.c[e],t.exports=r()}};m[e]=!0;var n=r=>{delete v[e],j.m[e]=t=>{throw delete j.c[e],r}};try{var o=b[e]();o.then?r.push(v[e]=o.then(t).catch(n)):t(o)}catch(e){n(e)}}})},(()=>{var e={248:0};j.f.j=(r,t)=>{var n=j.o(e,r)?e[r]:void 0;if(0!==n)if(n)t.push(n[2]);else{var o=new Promise((t,o)=>n=e[r]=[t,o]);t.push(n[2]=o);var a=j.p+j.u(r),i=new Error;j.l(a,t=>{if(j.o(e,r)&&(0!==(n=e[r])&&(e[r]=void 0),n)){var o=t&&("load"===t.type?"missing":t.type),a=t&&t.target&&t.target.src;i.message="Loading chunk "+r+" failed.\n("+o+": "+a+")",i.name="ChunkLoadError",i.type=o,i.request=a,n[1](i)}},"chunk-"+r,r)}};var r=(r,t)=>{var n,o,[a,i,u]=t,l=0;if(a.some(r=>0!==e[r])){for(n in i)j.o(i,n)&&(j.m[n]=i[n]);u&&u(j)}for(r&&r(t);l<a.length;l++)o=a[l],j.o(e,o)&&e[o]&&e[o][0](),e[o]=0},t=self.webpackChunkjl_db_comp=self.webpackChunkjl_db_comp||[];t.forEach(r.bind(null,0)),t.push=r.bind(null,t.push.bind(t))})(),j.nc=void 0;var S=j(246);(_JUPYTERLAB=void 0===_JUPYTERLAB?{}:_JUPYTERLAB).jl_db_comp=S})();
+var _JUPYTERLAB;(()=>{"use strict";var e,r,t,n,o,a,i,u,l,f,s,d,c,p,h,v,b,g,m,y={246(e,r,t){var n={"./index":()=>t.e(171).then(()=>()=>t(171)),"./extension":()=>t.e(171).then(()=>()=>t(171)),"./style":()=>t.e(728).then(()=>()=>t(728))},o=(e,r)=>(t.R=r,r=t.o(n,e)?n[e]():Promise.resolve().then(()=>{throw new Error('Module "'+e+'" does not exist in container.')}),t.R=void 0,r),a=(e,r)=>{if(t.S){var n="default",o=t.S[n];if(o&&o!==e)throw new Error("Container initialization failed as it has already been initialized with a different share scope");return t.S[n]=e,t.I(n,r)}};t.d(r,{get:()=>o,init:()=>a})}},w={};function j(e){var r=w[e];if(void 0!==r)return r.exports;var t=w[e]={id:e,exports:{}};return y[e](t,t.exports,j),t.exports}j.m=y,j.c=w,j.n=e=>{var r=e&&e.__esModule?()=>e.default:()=>e;return j.d(r,{a:r}),r},j.d=(e,r)=>{for(var t in r)j.o(r,t)&&!j.o(e,t)&&Object.defineProperty(e,t,{enumerable:!0,get:r[t]})},j.f={},j.e=e=>Promise.all(Object.keys(j.f).reduce((r,t)=>(j.f[t](e,r),r),[])),j.u=e=>e+"."+{171:"c84e54337cfd177ebe22",728:"6552504d5b9b27551bc5"}[e]+".js?v="+{171:"c84e54337cfd177ebe22",728:"6552504d5b9b27551bc5"}[e],j.g=function(){if("object"==typeof globalThis)return globalThis;try{return this||new Function("return this")()}catch(e){if("object"==typeof window)return window}}(),j.o=(e,r)=>Object.prototype.hasOwnProperty.call(e,r),e={},r="jl_db_comp:",j.l=(t,n,o,a)=>{if(e[t])e[t].push(n);else{var i,u;if(void 0!==o)for(var l=document.getElementsByTagName("script"),f=0;f<l.length;f++){var s=l[f];if(s.getAttribute("src")==t||s.getAttribute("data-webpack")==r+o){i=s;break}}i||(u=!0,(i=document.createElement("script")).charset="utf-8",j.nc&&i.setAttribute("nonce",j.nc),i.setAttribute("data-webpack",r+o),i.src=t),e[t]=[n];var d=(r,n)=>{i.onerror=i.onload=null,clearTimeout(c);var o=e[t];if(delete e[t],i.parentNode&&i.parentNode.removeChild(i),o&&o.forEach(e=>e(n)),r)return r(n)},c=setTimeout(d.bind(null,void 0,{type:"timeout",target:i}),12e4);i.onerror=d.bind(null,i.onerror),i.onload=d.bind(null,i.onload),u&&document.head.appendChild(i)}},j.r=e=>{"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},(()=>{j.S={};var e={},r={};j.I=(t,n)=>{n||(n=[]);var o=r[t];if(o||(o=r[t]={}),!(n.indexOf(o)>=0)){if(n.push(o),e[t])return e[t];j.o(j.S,t)||(j.S[t]={});var a=j.S[t],i="jl_db_comp",u=[];return"default"===t&&((e,r,t,n)=>{var o=a[e]=a[e]||{},u=o[r];(!u||!u.loaded&&(1!=!u.eager?n:i>u.from))&&(o[r]={get:()=>j.e(171).then(()=>()=>j(171)),from:i,eager:!1})})("jl_db_comp","0.1.11"),e[t]=u.length?Promise.all(u).then(()=>e[t]=1):1}}})(),(()=>{var e;j.g.importScripts&&(e=j.g.location+"");var r=j.g.document;if(!e&&r&&(r.currentScript&&"SCRIPT"===r.currentScript.tagName.toUpperCase()&&(e=r.currentScript.src),!e)){var t=r.getElementsByTagName("script");if(t.length)for(var n=t.length-1;n>-1&&(!e||!/^http(s?):/.test(e));)e=t[n--].src}if(!e)throw new Error("Automatic publicPath is not supported in this browser");e=e.replace(/^blob:/,"").replace(/#.*$/,"").replace(/\?.*$/,"").replace(/\/[^\/]+$/,"/"),j.p=e})(),t=e=>{var r=e=>e.split(".").map(e=>+e==e?+e:e),t=/^([^-+]+)?(?:-([^+]+))?(?:\+(.+))?$/.exec(e),n=t[1]?r(t[1]):[];return t[2]&&(n.length++,n.push.apply(n,r(t[2]))),t[3]&&(n.push([]),n.push.apply(n,r(t[3]))),n},n=(e,r)=>{e=t(e),r=t(r);for(var n=0;;){if(n>=e.length)return n<r.length&&"u"!=(typeof r[n])[0];var o=e[n],a=(typeof o)[0];if(n>=r.length)return"u"==a;var i=r[n],u=(typeof i)[0];if(a!=u)return"o"==a&&"n"==u||"s"==u||"u"==a;if("o"!=a&&"u"!=a&&o!=i)return o<i;n++}},o=e=>{var r=e[0],t="";if(1===e.length)return"*";if(r+.5){t+=0==r?">=":-1==r?"<":1==r?"^":2==r?"~":r>0?"=":"!=";for(var n=1,a=1;a<e.length;a++)n--,t+="u"==(typeof(u=e[a]))[0]?"-":(n>0?".":"")+(n=2,u);return t}var i=[];for(a=1;a<e.length;a++){var u=e[a];i.push(0===u?"not("+l()+")":1===u?"("+l()+" || "+l()+")":2===u?i.pop()+" "+i.pop():o(u))}return l();function l(){return i.pop().replace(/^\((.+)\)$/,"$1")}},a=(e,r)=>{if(0 in e){r=t(r);var n=e[0],o=n<0;o&&(n=-n-1);for(var i=0,u=1,l=!0;;u++,i++){var f,s,d=u<e.length?(typeof e[u])[0]:"";if(i>=r.length||"o"==(s=(typeof(f=r[i]))[0]))return!l||("u"==d?u>n&&!o:""==d!=o);if("u"==s){if(!l||"u"!=d)return!1}else if(l)if(d==s)if(u<=n){if(f!=e[u])return!1}else{if(o?f>e[u]:f<e[u])return!1;f!=e[u]&&(l=!1)}else if("s"!=d&&"n"!=d){if(o||u<=n)return!1;l=!1,u--}else{if(u<=n||s<d!=o)return!1;l=!1}else"s"!=d&&"n"!=d&&(l=!1,u--)}}var c=[],p=c.pop.bind(c);for(i=1;i<e.length;i++){var h=e[i];c.push(1==h?p()|p():2==h?p()&p():h?a(h,r):!p())}return!!p()},i=(e,r)=>e&&j.o(e,r),u=e=>(e.loaded=1,e.get()),l=e=>Object.keys(e).reduce((r,t)=>(e[t].eager&&(r[t]=e[t]),r),{}),f=(e,r,t)=>{var o=t?l(e[r]):e[r];return Object.keys(o).reduce((e,r)=>!e||!o[e].loaded&&n(e,r)?r:e,0)},s=(e,r,t,n)=>"Unsatisfied version "+t+" from "+(t&&e[r][t].from)+" of shared singleton module "+r+" (required "+o(n)+")",d=e=>{throw new Error(e)},c=e=>{"undefined"!=typeof console&&console.warn&&console.warn(e)},p=(e,r,t)=>t?t():((e,r)=>d("Shared module "+r+" doesn't exist in shared scope "+e))(e,r),h=(e=>function(r,t,n,o,a){var i=j.I(r);return i&&i.then&&!n?i.then(e.bind(e,r,j.S[r],t,!1,o,a)):e(r,j.S[r],t,n,o,a)})((e,r,t,n,o,l)=>{if(!i(r,t))return p(e,t,l);var d=f(r,t,n);return a(o,d)||c(s(r,t,d,o)),u(r[t][d])}),v={},b={125:()=>h("default","@jupyterlab/services",!1,[1,7,5,3]),141:()=>h("default","@jupyterlab/completer",!1,[1,4,5,3]),249:()=>h("default","@jupyterlab/settingregistry",!1,[1,4,5,3]),526:()=>h("default","@jupyterlab/coreutils",!1,[1,6,5,3]),931:()=>h("default","@jupyterlab/notebook",!1,[1,4,5,3])},g={171:[125,141,249,526,931]},m={},j.f.consumes=(e,r)=>{j.o(g,e)&&g[e].forEach(e=>{if(j.o(v,e))return r.push(v[e]);if(!m[e]){var t=r=>{v[e]=0,j.m[e]=t=>{delete j.c[e],t.exports=r()}};m[e]=!0;var n=r=>{delete v[e],j.m[e]=t=>{throw delete j.c[e],r}};try{var o=b[e]();o.then?r.push(v[e]=o.then(t).catch(n)):t(o)}catch(e){n(e)}}})},(()=>{var e={248:0};j.f.j=(r,t)=>{var n=j.o(e,r)?e[r]:void 0;if(0!==n)if(n)t.push(n[2]);else{var o=new Promise((t,o)=>n=e[r]=[t,o]);t.push(n[2]=o);var a=j.p+j.u(r),i=new Error;j.l(a,t=>{if(j.o(e,r)&&(0!==(n=e[r])&&(e[r]=void 0),n)){var o=t&&("load"===t.type?"missing":t.type),a=t&&t.target&&t.target.src;i.message="Loading chunk "+r+" failed.\n("+o+": "+a+")",i.name="ChunkLoadError",i.type=o,i.request=a,n[1](i)}},"chunk-"+r,r)}};var r=(r,t)=>{var n,o,[a,i,u]=t,l=0;if(a.some(r=>0!==e[r])){for(n in i)j.o(i,n)&&(j.m[n]=i[n]);u&&u(j)}for(r&&r(t);l<a.length;l++)o=a[l],j.o(e,o)&&e[o]&&e[o][0](),e[o]=0},t=self.webpackChunkjl_db_comp=self.webpackChunkjl_db_comp||[];t.forEach(r.bind(null,0)),t.push=r.bind(null,t.push.bind(t))})(),j.nc=void 0;var S=j(246);(_JUPYTERLAB=void 0===_JUPYTERLAB?{}:_JUPYTERLAB).jl_db_comp=S})();

{jl_db_comp-0.1.10.dist-info → jl_db_comp-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jl_db_comp
-Version: 0.1.10
+Version: 0.1.11
 Summary: A JupyterLab extension to complete db queries in jupyterlab notebooks
 Project-URL: Homepage, https://github.com/Ben-Herz/jl_db_completer
 Project-URL: Bug Tracker, https://github.com/Ben-Herz/jl_db_completer/issues

jl_db_comp-0.1.11.dist-info/RECORD

@@ -0,0 +1,20 @@
+jl_db_comp/__init__.py,sha256=L_Vn61SO_SScZsd1--0ug5RN35CibN9SH-yHBpGXR7g,1102
+jl_db_comp/_version.py,sha256=YCUEZ43B30YYGak_DJCWxdpF9JOln9sgKvyOaT5e-ug,172
+jl_db_comp/connections.py,sha256=kiNxyQzvd8FlebpCnCk2sFQxlm65L5T2BSZB1H-q7kQ,5222
+jl_db_comp/routes.py,sha256=LU_6d519OFVt0mRYXUh3_jZ1n18fuE03y3BRK3sYY1w,29530
+jl_db_comp/tests/__init__.py,sha256=mdd3JucOpGsN8PiHaypFnGLdv2nemscFn0piCZc2fls,40
+jl_db_comp/tests/test_routes.py,sha256=gDoRBBtxKPkmXOCvnBeuN1w7SgpqSvrMc5YQrCIFAF4,1752
+jl_db_comp-0.1.11.data/data/etc/jupyter/jupyter_server_config.d/jl_db_comp.json,sha256=fSmY5rlApxKaWIC5WmqvKJ2xW5lLmO_ybLJKogoJcq8,85
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/package.json,sha256=Y2ZEPpRa6jKGDLqhGhLN0zsTSOKQZoKJcCksvN7WTkk,6231
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/schemas/jl_db_comp/package.json.orig,sha256=QelQ1SJ0dKluvoNXCrDSkH5kmwmDQRiHiR_Of3F2rm4,7091
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/schemas/jl_db_comp/plugin.json,sha256=PM45pSfu6jtKNdgUDNumO5-YmQIdCAGO_0x_4Ec6ldg,963
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/171.c84e54337cfd177ebe22.js,sha256=yE5UM3z9F36-IkfeEp0-IZHIp4EXDzysa6rsDnuCM7I,9395
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/728.6552504d5b9b27551bc5.js,sha256=ZVJQTVubJ1UbxRj0sICSOtdLfqM4b9Qwf9HLhKXyhb4,4402
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/remoteEntry.2f90ff333d0ac2cbb0a8.js,sha256=L5D_Mz0KwsuwqOv2qtC0fJxjIbbUlKs2RYZgex6Brt8,6791
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/style.js,sha256=kHmCPOWdRY7r0Voabh7CMyxQJA6oPseEwKF6mcW--Zs,153
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/static/third-party-licenses.json,sha256=W6N2sSD7tQihMqQk64F9xMd1Flfr2KO97esAiHUOYdM,2453
+jl_db_comp-0.1.11.data/data/share/jupyter/labextensions/jl_db_comp/install.json,sha256=AkhpkuSkRFqmwkttzIRdh8UApmBhvmiodyEPb9WylXc,181
+jl_db_comp-0.1.11.dist-info/METADATA,sha256=zHWBuGjjxVdleyvCRVtNKO-JxtnLMyEOV1ZaeKdoVH0,17252
+jl_db_comp-0.1.11.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+jl_db_comp-0.1.11.dist-info/licenses/LICENSE,sha256=Oz0oICGKssavDqCu6-PnQedLBaqF1DaZvDyJsGCSIzE,1523
+jl_db_comp-0.1.11.dist-info/RECORD,,

jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/171.545011db1d7843ce3ab3.js

@@ -1 +0,0 @@
-"use strict";(self.webpackChunkjl_db_comp=self.webpackChunkjl_db_comp||[]).push([[171],{171(e,n,t){t.r(n),t.d(n,{default:()=>m});var o=t(141),s=t(931),i=t(249),a=t(125),r=t(526);async function c(e="",n={}){const t=a.ServerConnection.makeSettings(),o=r.URLExt.join(t.baseUrl,"jl-db-comp",e);let s;try{s=await a.ServerConnection.makeRequest(o,n,t)}catch(e){throw new a.ServerConnection.NetworkError(e)}let i=await s.text();if(i.length>0)try{i=JSON.parse(i)}catch(e){console.log("Not a JSON response body.",s)}if(!s.ok)throw new a.ServerConnection.ResponseError(s,i.message||i);return i}class l{constructor(e,n){this.identifier="jl_db_comp:postgres-completer",this.renderer=null,this._cache=new Map,this._cacheTTL=3e5,this._settings=null,this._notebookTracker=null,this._connectionName="",this._schema="public",this._enabled=!0,this._availableConnections=[],this._cachedKernelConfig=null,this._kernelConfigCacheTime=0,this._kernelConfigCacheTTL=3e4,this._sqlKeywords=["select","from","join","where","insert","update","delete","inner","left","right","outer","on","group","order","by","having","into","values","set"],this._notebookTracker=n||null,e&&(this._settings=e,this._loadSettings(),e.changed.connect(()=>{this._loadSettings()})),this._loadAvailableConnections()}_loadSettings(){this._settings&&(this._connectionName=this._settings.get("connectionName").composite,this._schema=this._settings.get("schema").composite,this._enabled=this._settings.get("enabled").composite)}async _loadAvailableConnections(){try{const e=await async function(){try{return await c("connections",{method:"GET"})}catch(e){if(e instanceof a.ServerConnection.ResponseError)console.error(`Failed to fetch connections: ${e.message}`);else{const n=e instanceof Error?e.message:"Unknown error";console.error(`Failed to fetch connections: ${n}`)}return{status:"error",connections:{},filePath:null,message:"Failed to fetch connections"}}}();"success"===e.status&&(this._availableConnections=Object.keys(e.connections))}catch(e){console.warn("Failed to load available connections:",e)}}async _getKernelConfig(){var e;const n=Date.now();if(this._cachedKernelConfig&&n-this._kernelConfigCacheTime<this._kernelConfigCacheTTL)return this._cachedKernelConfig;if(!this._notebookTracker)return null;const t=this._notebookTracker.currentWidget;if(!t)return null;const o=null===(e=t.sessionContext.session)||void 0===e?void 0:e.kernel;if(!o)return null;try{const e=o.requestExecute({code:"\nimport json\nimport os\nresult = {'connection': '', 'dsn_filename': ''}\n\n# Get active connection\ntry:\n    from sql.connection import ConnectionManager\n    conn = ConnectionManager.current\n    if conn:\n        for alias, c in ConnectionManager.connections.items():\n            if c is conn:\n                result['connection'] = alias\n                break\nexcept:\n    pass\n\n# Get dsn_filename from SqlMagic instance\ndsn_filename = None\ntry:\n    from sql.magic import SqlMagic\n    ip = get_ipython()\n    if ip:\n        for name, inst in ip.magics_manager.registry.items():\n            if isinstance(inst, SqlMagic):\n                dsn_filename = inst.dsn_filename\n                break\nexcept:\n    pass\n\n# Fallback: try to get from config\nif not dsn_filename:\n    try:\n        ip = get_ipython()\n        if ip and hasattr(ip, 'config'):\n            sql_config = ip.config.get('SqlMagic', {})\n            if 'dsn_filename' in sql_config:\n                dsn_filename = sql_config['dsn_filename']\n    except:\n        pass\n\n# Convert to absolute path if we have a dsn_filename\nif dsn_filename:\n    if not os.path.isabs(dsn_filename):\n        # Resolve relative to current working directory\n        dsn_filename = os.path.abspath(dsn_filename)\n    result['dsn_filename'] = dsn_filename\n\nprint(json.dumps(result))\n",silent:!0,store_history:!1}),t=await new Promise(n=>{let t="";e.onIOPub=e=>{if("stream"===e.header.msg_type){const n=e.content;"stdout"===n.name&&(t+=n.text)}},e.done.then(()=>{try{const e=JSON.parse(t.trim());n({connection:e.connection||"",dsnFilename:e.dsn_filename||""})}catch(e){n(null)}}).catch(()=>{n(null)})});return t&&(this._cachedKernelConfig=t,this._kernelConfigCacheTime=n),t}catch(e){return console.warn("Failed to get jupysql config from kernel:",e),null}}async isApplicable(e){if(!this._enabled)return!1;const n=e.editor;if(!n)return!1;const t=n.model.sharedModel.getSource();if(!t)return!1;const o=t.toLowerCase();return this._sqlKeywords.some(e=>o.includes(e))}async fetch(e,n){var t;if(!this._enabled)return{start:e.offset,end:e.offset,items:[]};const{text:o,offset:s}=e,i=this._extractContext(o,s);let r;if(i.jsonbColumn){const e=(null===(t=i.jsonbPath)||void 0===t?void 0:t.join("."))||"";r=`${i.schemaOrTable?`${i.schemaOrTable}.`:""}${i.jsonbColumn}->${e}.${i.prefix}`.toLowerCase()}else r=i.schema&&i.tableName?`${i.schema}.${i.tableName}.${i.prefix}`.toLowerCase():i.schemaOrTable?`${i.schemaOrTable}.${i.prefix}`.toLowerCase():i.prefix.toLowerCase();const l=this._getCached(r);if(l)return this._formatReply(l,e.offset,i.prefix);try{const n=await this._getKernelConfig();let t,o=this._connectionName;n&&(n.dsnFilename&&(t=n.dsnFilename),!o&&n.connection&&(o=n.connection)),!o&&this._availableConnections.length>0&&(o=this._availableConnections[0]);const s=await async function(e,n="",t="public",o,s,i,r,l){try{const a=new URLSearchParams;e&&a.append("connection",e),l&&a.append("connections_file",l),n&&a.append("prefix",n),a.append("schema",t),o&&a.append("table",o),s&&a.append("schema_or_table",s),i&&(a.append("jsonb_column",i),r&&r.length>0&&a.append("jsonb_path",JSON.stringify(r)));const h=`completions?${a.toString()}`,m=await c(h,{method:"GET"});return"error"===m.status?(console.error("PostgreSQL completion error:",m.message),[]):i&&m.jsonbKeys?m.jsonbKeys:o||s?m.columns.length>0?m.columns:m.tables:[...m.tables,...m.columns]}catch(e){if(e instanceof a.ServerConnection.ResponseError){const n=e.response.status;let t=e.message;"string"==typeof t&&(t.includes("<!DOCTYPE")||t.includes("<html"))&&(t=`HTML error page (${t.substring(0,100)}...)`),console.error(`PostgreSQL completions API failed (${n}): ${t}`)}else{const n=e instanceof Error?e.message:"Unknown error";console.error(`PostgreSQL completions API failed: ${n}`)}return[]}}(o||void 0,i.prefix,i.schema||this._schema,i.tableName,i.schemaOrTable,i.jsonbColumn,i.jsonbPath,t);return this._cache.set(r,{items:s,timestamp:Date.now()}),this._formatReply(s,e.offset,i.prefix)}catch(n){return console.error("Failed to fetch PostgreSQL completions:",n),{start:e.offset,end:e.offset,items:[]}}}_extractContext(e,n){const t=e.substring(0,n);if(t.includes("->")){const e=e=>{const n=[],t=/['"]?([\w]+)['"]?\s*->/g;let o;for(;null!==(o=t.exec(e));)n.push(o[1]);const s=e.lastIndexOf("->");let i="";return i=s>=0?e.substring(s+2).trim().replace(/['"]/g,""):e.trim().replace(/['"]/g,""),{jsonbPath:n,prefix:i}},n=t.match(/([\w]+)\.([\w]+)\.([\w]+)\s*->\s*(.*)$/);if(n){const t=n[1],o=n[2],s=n[3],i=n[4],{jsonbPath:a,prefix:r}=e(i);return{schema:t,tableName:o,jsonbColumn:s,jsonbPath:a,prefix:r}}const o=t.match(/([\w]+)\.([\w]+)\s*->\s*(.*)$/);if(o){const n=o[1],t=o[2],s=o[3],{jsonbPath:i,prefix:a}=e(s);return{schemaOrTable:n,jsonbColumn:t,jsonbPath:i,prefix:a}}const s=t.match(/([\w]+)\s*->\s*(.*)$/);if(s){const n=s[1],t=s[2],{jsonbPath:o,prefix:i}=e(t);return{jsonbColumn:n,jsonbPath:o,prefix:i}}}const o=t.match(/([\w]+)\.([\w]+)\.([\w]*)$/);if(o)return{schema:o[1],tableName:o[2],prefix:o[3]};const s=t.match(/([\w]+)\.([\w]*)$/);if(s)return{schemaOrTable:s[1],prefix:s[2]};const i=t.match(/[\w]+$/),a=i?i[0]:"",r=e.toLowerCase().match(/\bfrom\s+([\w]+\.)?[\w]+/);if(r){const e=r[0].match(/\bfrom\s+(?:([\w]+)\.)?([\w]+)/);if(e){const n=e[1],t=e[2];return n?{schema:n,tableName:t,prefix:a}:{schemaOrTable:t,prefix:a}}}return{prefix:a}}_getCached(e){const n=e.toLowerCase(),t=this._cache.get(n);return t?Date.now()-t.timestamp>this._cacheTTL?(this._cache.delete(n),null):t.items:null}_formatReply(e,n,t){return{start:n-t.length,end:n,items:e.map(e=>{let n=e.name,t=e.name;"jsonb_key"===e.type&&(t=`'${e.name}'`),"column"===e.type&&e.table&&(n=`${e.name} (${e.table})`);let o,s="📊",i=e.name;return"table"===e.type?s="📋":"view"===e.type?s="👁️":"jsonb_key"===e.type&&(s="🔑",i=`0000${e.name}`),"column"===e.type&&e.dataType&&e.table?o=`${e.table}.${e.name}: ${e.dataType}`:"jsonb_key"===e.type&&e.keyPath&&(o=`JSONB key: ${e.keyPath.join(" -> ")}`),{label:`${s} ${n}`,insertText:t,sortText:i,type:e.type,documentation:o}})}}clearCache(){this._cache.clear()}}const h="jl_db_comp:plugin",m={id:h,description:"A JupyterLab extension to complete db queries in jupyterlab notebooks",autoStart:!0,requires:[o.ICompletionProviderManager],optional:[i.ISettingRegistry,s.INotebookTracker],activate:(e,n,t,o)=>{let s;t?t.load(h).then(e=>{s=new l(e,o),n.registerProvider(s),console.log("JupyterLab extension jl_db_comp is activated!")}).catch(e=>{console.error("Failed to load settings for jl_db_comp:",e),s=new l(null,o),n.registerProvider(s),console.log("JupyterLab extension jl_db_comp is activated!")}):(s=new l(null,o),n.registerProvider(s),console.log("JupyterLab extension jl_db_comp is activated!"))}}}}]);

jl_db_comp-0.1.10.dist-info/RECORD

@@ -1,20 +0,0 @@
-jl_db_comp/__init__.py,sha256=L_Vn61SO_SScZsd1--0ug5RN35CibN9SH-yHBpGXR7g,1102
-jl_db_comp/_version.py,sha256=aM0QVK4NJP1CO1SzDAG0tn7KvEzshOfEQ6UeJq1gd_Q,172
-jl_db_comp/connections.py,sha256=kiNxyQzvd8FlebpCnCk2sFQxlm65L5T2BSZB1H-q7kQ,5222
-jl_db_comp/routes.py,sha256=5u3w5THejkXT9-gz-qKQmJPkDJAVyU2aE0mq_nIn7sE,25798
-jl_db_comp/tests/__init__.py,sha256=mdd3JucOpGsN8PiHaypFnGLdv2nemscFn0piCZc2fls,40
-jl_db_comp/tests/test_routes.py,sha256=gDoRBBtxKPkmXOCvnBeuN1w7SgpqSvrMc5YQrCIFAF4,1752
-jl_db_comp-0.1.10.data/data/etc/jupyter/jupyter_server_config.d/jl_db_comp.json,sha256=fSmY5rlApxKaWIC5WmqvKJ2xW5lLmO_ybLJKogoJcq8,85
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/package.json,sha256=paDyPeBJsp5c0Ai3gasecQUIYca-qiQQ_HmYIaZl2J8,6231
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/schemas/jl_db_comp/package.json.orig,sha256=NkfTFsQezToYG0wniQXQhs-RxvQOO0hYP2Xx45167Y0,7091
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/schemas/jl_db_comp/plugin.json,sha256=PM45pSfu6jtKNdgUDNumO5-YmQIdCAGO_0x_4Ec6ldg,963
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/171.545011db1d7843ce3ab3.js,sha256=VFAR2x14Q846s4kdKlyFcZ2TteUdOQBEEg5PhHYVAEk,9208
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/728.6552504d5b9b27551bc5.js,sha256=ZVJQTVubJ1UbxRj0sICSOtdLfqM4b9Qwf9HLhKXyhb4,4402
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/remoteEntry.e1b6bbc28eb414e4e599.js,sha256=4ba7wo60FOTlmTdZOb0PCKd-MxRqOkleptMIRmQPFXs,6791
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/style.js,sha256=kHmCPOWdRY7r0Voabh7CMyxQJA6oPseEwKF6mcW--Zs,153
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/static/third-party-licenses.json,sha256=W6N2sSD7tQihMqQk64F9xMd1Flfr2KO97esAiHUOYdM,2453
-jl_db_comp-0.1.10.data/data/share/jupyter/labextensions/jl_db_comp/install.json,sha256=AkhpkuSkRFqmwkttzIRdh8UApmBhvmiodyEPb9WylXc,181
-jl_db_comp-0.1.10.dist-info/METADATA,sha256=AyqolcEdJ8u6q2AKCKVb8imfxwqd8yBwc81NfqyhVyU,17252
-jl_db_comp-0.1.10.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-jl_db_comp-0.1.10.dist-info/licenses/LICENSE,sha256=Oz0oICGKssavDqCu6-PnQedLBaqF1DaZvDyJsGCSIzE,1523
-jl_db_comp-0.1.10.dist-info/RECORD,,