runcell 0.1.16a0__py3-none-any.whl → 0.1.16a2__py3-none-any.whl

d5m_ai/_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.16a0'
+__version__ = VERSION = '0.1.16a2'

d5m_ai/completion/__init__.py

@@ -65,6 +65,11 @@ class CompletionHandler(APIHandler):
                 mime = body.get('mime', 'text/x-python')
                 path = body.get('path')
                 cell_id = body.get('cell_id')
+                mode = body.get('mode')
+                region_text = body.get('region_text')
+                cursor_offset = body.get('cursor_offset')
+                region_start_line = body.get('region_start_line')
+                region_start_column = body.get('region_start_column')
             except KeyError as e:
                 raise HTTPError(400, f"Missing required field: {e}")
             
@@ -76,7 +81,12 @@ class CompletionHandler(APIHandler):
                 "language": language,
                 "mime": mime,
                 "path": path,
-                "cell_id": cell_id
+                "cell_id": cell_id,
+                "mode": mode,
+                "region_text": region_text,
+                "cursor_offset": cursor_offset,
+                "region_start_line": region_start_line,
+                "region_start_column": region_start_column
             }
 
             # Forward request to remote server

d5m_ai/empty_display/__init__.py

@@ -0,0 +1,3 @@
+from .empty_display_handler import EmptyDisplayProxyHandler
+
+__all__ = ["EmptyDisplayProxyHandler"]

d5m_ai/empty_display/empty_display_handler.py

@@ -0,0 +1,137 @@
+"""
+Empty display proxy handler.
+
+Forwards empty state display requests to the SaaS backend to avoid CORS issues.
+"""
+
+import asyncio
+import aiohttp
+import certifi
+import json
+import logging
+import os
+import ssl
+from typing import Optional
+from urllib.parse import urlparse
+
+from jupyter_server.base.handlers import APIHandler
+from tornado import web
+
+from ..auth.token_handler import get_current_user_token_string
+
+logging.basicConfig(level=logging.INFO)
+
+DEFAULT_SAAS_URL = "https://www.runcell.dev"
+ALLOWED_HOST_SUFFIXES = (".runcell.dev",)
+ALLOWED_HOSTS = {"runcell.dev", "www.runcell.dev", "localhost", "127.0.0.1"}
+
+
+def _is_allowed_host(hostname: str) -> bool:
+    if not hostname:
+        return False
+    hostname = hostname.lower()
+    if hostname in ALLOWED_HOSTS:
+        return True
+    return any(hostname.endswith(suffix) for suffix in ALLOWED_HOST_SUFFIXES)
+
+
+class EmptyDisplayProxyHandler(APIHandler):
+    """Proxy handler for empty display requests."""
+
+    def _create_ssl_context(self):
+        try:
+            certifi_ca_bundle = certifi.where()
+            return ssl.create_default_context(cafile=certifi_ca_bundle)
+        except Exception as error:
+            logging.warning(f"[EMPTY-DISPLAY-PROXY] Failed to use certifi CA bundle: {error}")
+            return ssl.create_default_context()
+
+    def _resolve_saas_url(self, saas_url: Optional[str]) -> str:
+        candidate = saas_url or os.environ.get("D5M_SAAS_URL") or DEFAULT_SAAS_URL
+        candidate = candidate.strip()
+
+        parsed = urlparse(candidate)
+        if not parsed.scheme or not parsed.netloc:
+            raise ValueError("Invalid SaaS URL")
+
+        if not _is_allowed_host(parsed.hostname or ""):
+            raise ValueError("SaaS URL host not allowed")
+
+        return f"{parsed.scheme}://{parsed.netloc}".rstrip("/")
+
+    def _extract_token(self) -> Optional[str]:
+        local_token = get_current_user_token_string()
+        if local_token:
+            return local_token
+
+        auth_header = self.request.headers.get("Authorization", "")
+        if auth_header.startswith("Bearer "):
+            return auth_header[7:]
+
+        return None
+
+    @web.authenticated
+    async def post(self):
+        try:
+            try:
+                body = json.loads(self.request.body or b"{}")
+            except json.JSONDecodeError:
+                self.set_status(400)
+                self.finish(json.dumps({"error": "Invalid JSON in request body"}))
+                return
+
+            ping_status = body.get("ping_status") or {}
+            payload = {
+                "runcell_version": body.get("runcell_version", ""),
+                "ping_status": {
+                    "local": bool(ping_status.get("local")),
+                    "remote": bool(ping_status.get("remote"))
+                }
+            }
+
+            if "logged_in" in body:
+                payload["logged_in"] = bool(body.get("logged_in"))
+            if "user_id" in body:
+                payload["user_id"] = body.get("user_id")
+
+            try:
+                saas_url = self._resolve_saas_url(body.get("saas_url"))
+            except ValueError as error:
+                self.set_status(400)
+                self.finish(json.dumps({"error": str(error)}))
+                return
+
+            user_token = self._extract_token()
+            headers = {"Content-Type": "application/json"}
+            if user_token:
+                headers["Authorization"] = f"Bearer {user_token}"
+
+            ssl_context = self._create_ssl_context()
+            timeout = aiohttp.ClientTimeout(total=10)
+
+            async with aiohttp.ClientSession(timeout=timeout) as session:
+                url = f"{saas_url}/api/plugin-auth/empty_display"
+                logging.info(f"[EMPTY-DISPLAY-PROXY] Forwarding request to: {url}")
+                async with session.post(url, json=payload, headers=headers, ssl=ssl_context) as response:
+                    response_text = await response.text()
+                    if response.status == 200:
+                        try:
+                            response_data = json.loads(response_text)
+                            self.finish(json.dumps(response_data))
+                        except json.JSONDecodeError:
+                            logging.error("[EMPTY-DISPLAY-PROXY] Invalid JSON response")
+                            self.set_status(502)
+                            self.finish(json.dumps({"error": "Invalid response from server"}))
+                    else:
+                        logging.error(f"[EMPTY-DISPLAY-PROXY] Remote error {response.status}: {response_text}")
+                        self.set_status(response.status)
+                        self.finish(response_text)
+
+        except asyncio.TimeoutError:
+            logging.error("[EMPTY-DISPLAY-PROXY] Request timeout")
+            self.set_status(408)
+            self.finish(json.dumps({"error": "Request timeout"}))
+        except Exception as error:
+            logging.error(f"[EMPTY-DISPLAY-PROXY] Unexpected error: {error}")
+            self.set_status(500)
+            self.finish(json.dumps({"error": f"Internal server error: {error}"}))

d5m_ai/extension.py

@@ -13,8 +13,10 @@ from .sandbox.shell_exec import ShellExecHandler
 from .chat.code_apply import CodeApplyHandler
 from .mention.mention_file_handler import MentionFileHandler
 from .auth.token_handler import TokenHandler
-from .ping.ping_handler import PingProxyHandler
+from .ping.ping_handler import PingProxyHandler, PingLocalHandler
+from .empty_display import EmptyDisplayProxyHandler
 from .title import TitleGenerationHandler
+from .quick_fix import QuickFixApplyHandler
 
 from .completion import CompletionHandler
 from .edit import AIEditChatHandler
@@ -25,6 +27,9 @@ from .agent import AIJLAgentProxyHandler
 from .agent.config import IMAGE_DIR
 from .image import get_image_handlers
 from .upgrade import UpgradeEnvironmentHandler, UpgradeHandler
+from .gitignore_handler import GitIgnoreHandler
+from .git_status_handler import GitStatusHandler
+from .search_replace_handler import SearchReplaceHandler
 
 # Static file handler for serving image files
 class ImageFileHandler(web.StaticFileHandler):
@@ -62,9 +67,15 @@ class D5MAIExtensionApp(ExtensionApp):
             (r"/api/d5m_ai/mention/files", MentionFileHandler),
             (r"/api/d5m_ai/auth/token", TokenHandler),
             (r"/api/d5m_ai/ping", PingProxyHandler),
+            (r"/api/d5m_ai/ping_local", PingLocalHandler),
+            (r"/api/d5m_ai/empty_display", EmptyDisplayProxyHandler),
             (r"/api/d5m_ai/generate_title", TitleGenerationHandler),
+            (r"/api/d5m_ai/quick_fix_apply", QuickFixApplyHandler),
             (r"/api/d5m_ai/upgrade/info", UpgradeEnvironmentHandler),
             (r"/api/d5m_ai/upgrade", UpgradeHandler),
+            (r"/api/d5m_ai/gitignore", GitIgnoreHandler),
+            (r"/api/d5m_ai/git-status", GitStatusHandler),
+            (r"/api/d5m_ai/search_replace", SearchReplaceHandler),
             # Static file handler for the temporary image files
             (r"/d5m-ai/images/(.*)", ImageFileHandler, {"path": IMAGE_DIR}),
         ]

d5m_ai/git_status_handler.py

@@ -0,0 +1,225 @@
+import asyncio
+import json
+import os
+import time
+from typing import Dict, Optional
+
+from jupyter_server.base.handlers import APIHandler
+import tornado.web
+
+
+_CACHE_TTL_SECONDS = 2.0
+_cache_timestamp = 0.0
+_cache_root: Optional[str] = None
+_cache_payload: Optional[Dict[str, object]] = None
+_cache_lock = asyncio.Lock()
+
+
+def _status_priority(status: str) -> int:
+    if status == "modified":
+        return 3
+    if status == "added":
+        return 2
+    if status == "deleted":
+        return 1
+    return 0
+
+
+def _merge_status(current: Optional[str], incoming: str) -> str:
+    if not current:
+        return incoming
+    return incoming if _status_priority(incoming) > _status_priority(current) else current
+
+
+def _map_git_status(code: str) -> Optional[str]:
+    if code == "??":
+        return "added"
+    if "M" in code or "R" in code or "C" in code or "U" in code:
+        return "modified"
+    if "A" in code:
+        return "added"
+    if "D" in code:
+        return "deleted"
+    return None
+
+
+def _parse_git_status(output: str) -> Dict[str, str]:
+    entries = output.split("\0")
+    statuses: Dict[str, str] = {}
+    index = 0
+
+    while index < len(entries):
+        token = entries[index]
+        index += 1
+
+        if not token or len(token) < 3:
+            continue
+
+        status_code = token[:2]
+        path = token[3:]
+
+        if status_code[0] in ("R", "C") or status_code[1] in ("R", "C"):
+            if index < len(entries):
+                path = entries[index]
+                index += 1
+
+        status = _map_git_status(status_code)
+        if not status or not path:
+            continue
+
+        statuses[path] = _merge_status(statuses.get(path), status)
+
+    return statuses
+
+
+async def _get_repo_root(root_dir: str) -> Optional[str]:
+    try:
+        process = await asyncio.create_subprocess_exec(
+            "git",
+            "rev-parse",
+            "--show-toplevel",
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+            cwd=root_dir
+        )
+    except Exception:
+        return None
+
+    stdout, stderr = await process.communicate()
+    if process.returncode != 0:
+        return None
+
+    output = stdout.decode("utf-8", errors="replace").strip()
+    return output or None
+
+
+def _normalize_rel_path(path: str) -> str:
+    return path.replace(os.sep, "/").lstrip("./")
+
+
+def _adjust_status_paths(
+    statuses: Dict[str, str],
+    root_dir: str,
+    repo_root: str
+) -> Dict[str, str]:
+    root_dir = os.path.abspath(root_dir)
+    repo_root = os.path.abspath(repo_root)
+
+    if root_dir == repo_root:
+        return statuses
+
+    try:
+        repo_rel_to_root = os.path.relpath(repo_root, root_dir)
+    except ValueError:
+        return statuses
+
+    repo_rel_to_root = _normalize_rel_path(repo_rel_to_root)
+    if repo_rel_to_root and not repo_rel_to_root.startswith(".."):
+        adjusted: Dict[str, str] = {}
+        for path, status in statuses.items():
+            adjusted_path = f"{repo_rel_to_root}/{path}" if path else repo_rel_to_root
+            adjusted[adjusted_path] = _merge_status(adjusted.get(adjusted_path), status)
+        return adjusted
+
+    server_rel_to_repo = _normalize_rel_path(os.path.relpath(root_dir, repo_root))
+    if not server_rel_to_repo or server_rel_to_repo.startswith(".."):
+        return statuses
+
+    adjusted: Dict[str, str] = {}
+    prefix = f"{server_rel_to_repo}/"
+    for path, status in statuses.items():
+        if path.startswith(prefix):
+            trimmed = path[len(prefix):]
+            if trimmed:
+                adjusted[trimmed] = _merge_status(adjusted.get(trimmed), status)
+    return adjusted
+
+
+async def _run_git_status(root_dir: str) -> Dict[str, object]:
+    try:
+        process = await asyncio.create_subprocess_exec(
+            "git",
+            "status",
+            "--porcelain=1",
+            "-z",
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+            cwd=root_dir
+        )
+    except Exception as exc:
+        return {
+            "success": False,
+            "repo": False,
+            "statuses": {},
+            "error": str(exc)
+        }
+
+    stdout, stderr = await process.communicate()
+    stderr_text = stderr.decode("utf-8", errors="replace").strip()
+
+    if process.returncode != 0:
+        if "not a git repository" in stderr_text.lower():
+            return {
+                "success": True,
+                "repo": False,
+                "statuses": {}
+            }
+        return {
+            "success": False,
+            "repo": False,
+            "statuses": {},
+            "error": stderr_text or "Failed to run git status."
+        }
+
+    output = stdout.decode("utf-8", errors="replace")
+    repo_root = await _get_repo_root(root_dir)
+    statuses = _parse_git_status(output)
+    if repo_root:
+        statuses = _adjust_status_paths(statuses, root_dir, repo_root)
+    return {
+        "success": True,
+        "repo": True,
+        "statuses": statuses
+    }
+
+
+async def _get_cached_status(root_dir: str) -> Dict[str, object]:
+    global _cache_timestamp, _cache_payload, _cache_root
+    now = time.monotonic()
+
+    if _cache_payload and _cache_root == root_dir and (now - _cache_timestamp) < _CACHE_TTL_SECONDS:
+        return _cache_payload
+
+    async with _cache_lock:
+        now = time.monotonic()
+        if _cache_payload and _cache_root == root_dir and (now - _cache_timestamp) < _CACHE_TTL_SECONDS:
+            return _cache_payload
+
+        payload = await _run_git_status(root_dir)
+        _cache_payload = payload
+        _cache_root = root_dir
+        _cache_timestamp = time.monotonic()
+        return payload
+
+
+class GitStatusHandler(APIHandler):
+    @tornado.web.authenticated
+    async def get(self):
+        await self._handle()
+
+    @tornado.web.authenticated
+    async def post(self):
+        await self._handle()
+
+    async def _handle(self):
+        root_dir = getattr(self.contents_manager, "root_dir", None)
+        if not root_dir:
+            root_dir = (
+                self.settings.get("server_root_dir")
+                or self.settings.get("root_dir")
+                or os.getcwd()
+            )
+
+        root_dir = os.path.abspath(root_dir)
+        payload = await _get_cached_status(root_dir)
+        self.finish(json.dumps(payload))

d5m_ai/gitignore_handler.py

@@ -0,0 +1,120 @@
+import asyncio
+import json
+import os
+from jupyter_server.base.handlers import APIHandler
+import tornado.web
+
+
+async def _get_repo_root(root_dir: str):
+    try:
+        process = await asyncio.create_subprocess_exec(
+            "git",
+            "rev-parse",
+            "--show-toplevel",
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+            cwd=root_dir,
+        )
+    except Exception:
+        return None
+
+    stdout, _stderr = await process.communicate()
+    if process.returncode != 0:
+        return None
+
+    output = stdout.decode("utf-8", errors="replace").strip()
+    return output or None
+
+
+def _normalize_rel_path(path: str) -> str:
+    return path.replace(os.sep, "/").lstrip("./")
+
+
+class GitIgnoreHandler(APIHandler):
+    @tornado.web.authenticated
+    async def get(self):
+        root_dir = getattr(self.contents_manager, "root_dir", None)
+        if not root_dir:
+            root_dir = (
+                self.settings.get("server_root_dir")
+                or self.settings.get("root_dir")
+                or os.getcwd()
+            )
+
+        dir_arg = self.get_argument("dir", default="", strip=True)
+        if dir_arg in ("", ".", "/"):
+            target_dir = root_dir
+        else:
+            dir_arg = dir_arg.lstrip("/").lstrip("\\")
+            target_dir = os.path.abspath(os.path.join(root_dir, dir_arg))
+
+        root_dir = os.path.abspath(root_dir)
+        try:
+            if os.path.commonpath([root_dir, target_dir]) != root_dir:
+                raise ValueError("Path is outside server root.")
+        except ValueError:
+            self.set_status(400)
+            self.finish(json.dumps({
+                "success": False,
+                "error": "Invalid directory path."
+            }))
+            return
+
+        gitignore_path = os.path.join(target_dir, ".gitignore")
+        root_content = None
+        root_prefix = None
+        is_root_request = dir_arg in ("", ".", "/")
+
+        if is_root_request:
+            repo_root = await _get_repo_root(root_dir)
+            if repo_root:
+                repo_root = os.path.abspath(repo_root)
+                if repo_root != root_dir:
+                    try:
+                        rel_from_repo = os.path.relpath(root_dir, repo_root)
+                    except ValueError:
+                        rel_from_repo = None
+                    if rel_from_repo and not rel_from_repo.startswith(".."):
+                        rel_from_repo = _normalize_rel_path(rel_from_repo)
+                        root_gitignore_path = os.path.join(repo_root, ".gitignore")
+                        if os.path.isfile(root_gitignore_path):
+                            try:
+                                with open(root_gitignore_path, "r", encoding="utf-8") as handle:
+                                    root_content = handle.read()
+                                root_prefix = rel_from_repo or None
+                            except Exception:
+                                root_content = None
+                                root_prefix = None
+
+        if not os.path.isfile(gitignore_path):
+            payload = {
+                "success": True,
+                "found": False
+            }
+            if root_content:
+                payload["root_content"] = root_content
+                if root_prefix:
+                    payload["root_prefix"] = root_prefix
+            self.finish(json.dumps(payload))
+            return
+
+        try:
+            with open(gitignore_path, "r", encoding="utf-8") as handle:
+                content = handle.read()
+            payload = {
+                "success": True,
+                "found": True,
+                "content": content,
+                "dir": dir_arg
+            }
+            if root_content:
+                payload["root_content"] = root_content
+                if root_prefix:
+                    payload["root_prefix"] = root_prefix
+            self.finish(json.dumps(payload))
+        except Exception as exc:
+            self.set_status(500)
+            self.finish(json.dumps({
+                "success": False,
+                "error": str(exc)
+            }))

d5m_ai/labextension/package.json

@@ -1,6 +1,6 @@
 {
   "name": "runcell",
-  "version": "0.1.16-alpha.",
+  "version": "0.1.16-alpha.2",
   "description": "AI Agent for Jupyter",
   "keywords": [
     "jupyter",
@@ -71,10 +71,12 @@
     "@jupyterlab/apputils": "^4.5.1",
     "@jupyterlab/completer": "^4.4.2",
     "@jupyterlab/coreutils": "^6.0.0",
+    "@jupyterlab/filebrowser": "^4.0.0",
     "@jupyterlab/launcher": "^4.0.0",
     "@jupyterlab/mainmenu": "^4.0.0",
     "@jupyterlab/notebook": "^4.0.0",
     "@jupyterlab/settingregistry": "^4.0.0",
+    "@jupyterlab/translation": "^4.0.0",
     "@jupyterlab/ui-components": "^4.0.0",
     "@lumino/widgets": "^2.7.1",
     "@mui/material": "^5.15.0",
@@ -85,11 +87,13 @@
     "@radix-ui/react-select": "^2.2.4",
     "@radix-ui/react-slot": "^1.2.2",
     "@radix-ui/react-tabs": "^1.1.12",
+    "@radix-ui/react-toggle": "^1.1.10",
     "@radix-ui/react-tooltip": "^1.2.6",
     "@uiw/react-codemirror": "^4.23.11",
     "class-variance-authority": "^0.7.1",
     "clsx": "^2.1.1",
     "embla-carousel-react": "^8.6.0",
+    "ignore": "^5.2.0",
     "lucide-react": "^0.507.0",
     "mobx": "^6.13.7",
     "mobx-react-lite": "^4.1.0",
@@ -152,8 +156,9 @@
     "extension": true,
     "outputDir": "d5m_ai/labextension",
     "schemaDir": "schema",
+    "webpackConfig": "webpack.config.js",
     "_build": {
-      "load": "static/remoteEntry.d0ed4a0bca8c2e002f18.js",
+      "load": "static/remoteEntry.c31036ba7b40ed91b072.js",
       "extension": "./extension",
       "style": "./style"
     }

d5m_ai/labextension/schemas/runcell/file-browser-toggle.json

@@ -0,0 +1,22 @@
+{
+    "jupyter.lab.shortcuts": [],
+    "title": "Runcell File Browser Toggle",
+    "description": "Settings for the file browser view switcher.",
+    "type": "object",
+    "properties": {
+        "singleClickToUnfold": {
+            "type": "boolean",
+            "title": "Single click to unfold folders",
+            "description": "If enabled, single click will expand or collapse folders in tree view.",
+            "default": true
+        },
+        "defaultView": {
+            "type": "string",
+            "title": "Default file browser view",
+            "description": "Controls which view is shown when JupyterLab starts.",
+            "enum": ["list", "tree"],
+            "default": "list"
+        }
+    },
+    "additionalProperties": false
+}

d5m_ai/labextension/schemas/runcell/package.json.orig

@@ -1,6 +1,6 @@
 {
     "name": "runcell",
-    "version": "0.1.16-alpha.",
+    "version": "0.1.16-alpha.2",
     "description": "AI Agent for Jupyter",
     "keywords": [
         "jupyter",
@@ -71,10 +71,12 @@
         "@jupyterlab/apputils": "^4.5.1",
         "@jupyterlab/completer": "^4.4.2",
         "@jupyterlab/coreutils": "^6.0.0",
+        "@jupyterlab/filebrowser": "^4.0.0",
         "@jupyterlab/launcher": "^4.0.0",
         "@jupyterlab/mainmenu": "^4.0.0",
         "@jupyterlab/notebook": "^4.0.0",
         "@jupyterlab/settingregistry": "^4.0.0",
+        "@jupyterlab/translation": "^4.0.0",
         "@jupyterlab/ui-components": "^4.0.0",
         "@lumino/widgets": "^2.7.1",
         "@mui/material": "^5.15.0",
@@ -85,11 +87,13 @@
         "@radix-ui/react-select": "^2.2.4",
         "@radix-ui/react-slot": "^1.2.2",
         "@radix-ui/react-tabs": "^1.1.12",
+        "@radix-ui/react-toggle": "^1.1.10",
         "@radix-ui/react-tooltip": "^1.2.6",
         "@uiw/react-codemirror": "^4.23.11",
         "class-variance-authority": "^0.7.1",
         "clsx": "^2.1.1",
         "embla-carousel-react": "^8.6.0",
+        "ignore": "^5.2.0",
         "lucide-react": "^0.507.0",
         "mobx": "^6.13.7",
         "mobx-react-lite": "^4.1.0",
@@ -151,7 +155,8 @@
     "jupyterlab": {
         "extension": true,
         "outputDir": "d5m_ai/labextension",
-        "schemaDir": "schema"
+        "schemaDir": "schema",
+        "webpackConfig": "webpack.config.js"
     },
     "eslintIgnore": [
         "node_modules",