token-cut-cli 0.1.5__tar.gz → 0.1.6__tar.gz

{token_cut_cli-0.1.5 → token_cut_cli-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-cut-cli
-Version: 0.1.5
+Version: 0.1.6
 Summary: Verify Token-Cut API integration from the terminal (no repo clone required)
 Project-URL: Homepage, https://www.token-cut.com
 Project-URL: Documentation, https://www.token-cut.com

{token_cut_cli-0.1.5 → token_cut_cli-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "token-cut-cli"
-version = "0.1.5"
+version = "0.1.6"
 description = "Verify Token-Cut API integration from the terminal (no repo clone required)"
 readme = "README.md"
 requires-python = ">=3.10"

token_cut_cli-0.1.6/src/token_cut_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.6"

{token_cut_cli-0.1.5 → token_cut_cli-0.1.6}/src/token_cut_cli/mcp_server.py

@@ -1,4 +1,4 @@
-"""Token-Cut MCP server — shipped in token-cut-cli (calls your cloud API with cc_live_ key)."""
+"""Token-Cut MCP server — any MCP host (Cursor, Claude, Copilot, …)."""
 
 from __future__ import annotations
 
@@ -7,10 +7,22 @@ import os
 import sys
 
 from token_cut_cli import __version__
-from token_cut_cli.client import DEFAULT_API_URL, api_base, api_key
+from token_cut_cli.client import api_base, api_key
+from token_cut_cli.workspace import detect_workspace, workspace_forward_headers
 
 
-def _post(path: str, body: dict) -> dict:
+def _resolve_repo(args: dict) -> str | None:
+    for val in (
+        args.get("repo_path"),
+        os.getenv("TOKEN_CUT_REPO_PATH") or os.getenv("CONTEXT_CUT_REPO_PATH"),
+        detect_workspace(),
+    ):
+        if val and str(val).strip():
+            return str(val).strip()
+    return None
+
+
+def _post(path: str, body: dict, *, repo: str | None = None) -> dict:
     import urllib.request
 
     base = api_base().rstrip("/")
@@ -18,10 +30,14 @@ def _post(path: str, body: dict) -> dict:
     if not key:
         raise RuntimeError("Set TOKEN_CUT_API_KEY=cc_live_…")
 
+    headers = {"Content-Type": "application/json", "x-api-key": key}
+    if repo:
+        headers.update(workspace_forward_headers(repo))
+
     req = urllib.request.Request(
         f"{base}{path}",
         data=json.dumps(body).encode(),
-        headers={"Content-Type": "application/json", "x-api-key": key},
+        headers=headers,
         method="POST",
     )
     timeout = int(os.getenv("TOKEN_CUT_HTTP_TIMEOUT", "360"))
@@ -39,14 +55,17 @@ def _tool_list() -> dict:
             {
                 "name": "optimize_prompt",
                 "description": (
-                    "Optimize a prompt (Token-Cut cloud). Auto-builds graph.json when missing. "
-                    "Requires cc_live_ API key. IDE still runs the LLM on your plan."
+                    "Optimize a prompt. Open workspace auto-detected from the host IDE. "
+                    "Requires cc_live_ API key (or local API URL). IDE still runs the LLM."
                 ),
                 "inputSchema": {
                     "type": "object",
                     "properties": {
                         "prompt": {"type": "string"},
-                        "repo_path": {"type": "string"},
+                        "repo_path": {
+                            "type": "string",
+                            "description": "Optional override; default is IDE open folder.",
+                        },
                         "repo_url": {"type": "string"},
                         "auto_build_graph": {"type": "boolean"},
                     },
@@ -68,10 +87,6 @@ def _tool_list() -> dict:
     }
 
 
-def _env_repo_path() -> str | None:
-    return os.getenv("TOKEN_CUT_REPO_PATH") or os.getenv("CONTEXT_CUT_REPO_PATH")
-
-
 def _env_repo_url() -> str | None:
     return os.getenv("TOKEN_CUT_REPO_URL") or os.getenv("CONTEXT_CUT_REPO_URL")
 
@@ -82,16 +97,18 @@ def _handle_tools_call(msg: dict) -> None:
     args = params.get("arguments", {})
     try:
         if name == "optimize_prompt":
+            repo = _resolve_repo(args)
             data = _post(
                 "/v1/optimize",
                 {
                     "prompt": args["prompt"],
-                    "repo_path": args.get("repo_path") or _env_repo_path(),
+                    "repo_path": repo,
                     "repo_url": args.get("repo_url") or _env_repo_url(),
                     "model": "gpt-4o",
                     "use_graph": True,
                     "auto_build_graph": args.get("auto_build_graph", True),
                 },
+                repo=repo,
             )
             graph_line = ""
             if data.get("graph_ready"):
@@ -105,12 +122,14 @@ def _handle_tools_call(msg: dict) -> None:
                 f"{data['optimized_prompt'][:8000]}"
             )
         elif name == "build_repo_graph":
+            repo = _resolve_repo(args)
             data = _post(
                 "/v1/graph/build",
                 {
-                    "repo_path": args.get("repo_path"),
+                    "repo_path": repo,
                     "repo_url": args.get("repo_url"),
                 },
+                repo=repo,
             )
             text = json.dumps(data, indent=2)
         else:

token_cut_cli-0.1.6/src/token_cut_cli/workspace.py

@@ -0,0 +1,96 @@
+"""IDE-agnostic open-folder detection for MCP (pip-installed CLI)."""
+
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+# Keep in sync with api/app/workspace.py IDE_WORKSPACE_ENV_KEYS
+IDE_WORKSPACE_ENV_KEYS = (
+    "CURSOR_WORKSPACE_FOLDER",
+    "CURSOR_PROJECT_DIR",
+    "CURSOR_WORKSPACE",
+    "VSCODE_CWD",
+    "VSCODE_WORKSPACE_FOLDER",
+    "VSCODE_WORKSPACE",
+    "CODE_WORKSPACE_FOLDER",
+    "WINDSURF_WORKSPACE_FOLDER",
+    "CODEIUM_WORKSPACE",
+    "CLAUDE_PROJECT_ROOT",
+    "CLAUDE_CODE_PROJECT_DIR",
+    "CLAUDE_PROJECT_DIR",
+    "ZED_WORKSPACE",
+    "ZED_ROOT",
+    "IDEA_INITIAL_DIRECTORY",
+    "PROJECT_DIR",
+    "PROJECT_ROOT",
+    "WORKSPACE_FOLDER",
+    "WORKSPACE",
+    "REPO_ROOT",
+    "INIT_CWD",
+    "GITHUB_WORKSPACE",
+    "PWD",
+)
+
+
+def _find_git_root(start: Path) -> Path | None:
+    current = start.resolve()
+    for _ in range(32):
+        if (current / ".git").exists():
+            return current
+        parent = current.parent
+        if parent == current:
+            break
+        current = parent
+    return None
+
+
+def _looks_like_repo(path: Path) -> bool:
+    if not path.is_dir():
+        return False
+    if (path / ".git").is_dir() or (path / ".git").is_file():
+        return True
+    markers = ("package.json", "pyproject.toml", "go.mod", "Cargo.toml", "requirements.txt")
+    return any((path / m).exists() for m in markers)
+
+
+def detect_workspace() -> str | None:
+    candidates: list[Path] = []
+    for key in IDE_WORKSPACE_ENV_KEYS:
+        v = os.getenv(key)
+        if v:
+            candidates.append(Path(v).expanduser())
+    env_repo = os.getenv("TOKEN_CUT_REPO_PATH") or os.getenv("CONTEXT_CUT_REPO_PATH")
+    if env_repo:
+        candidates.append(Path(env_repo).expanduser())
+    try:
+        candidates.append(Path.cwd())
+    except OSError:
+        pass
+
+    seen: set[str] = set()
+    for raw in candidates:
+        try:
+            p = raw.resolve()
+        except OSError:
+            continue
+        key = str(p)
+        if key in seen:
+            continue
+        seen.add(key)
+        root = _find_git_root(p) if p.is_dir() else None
+        if root and _looks_like_repo(root):
+            return str(root)
+        if _looks_like_repo(p):
+            return str(p)
+    return None
+
+
+def workspace_forward_headers(repo_path: str) -> dict[str, str]:
+    path = str(repo_path).strip()
+    return {
+        "x-repo-path": path,
+        "x-workspace-path": path,
+        "x-workspace-folder": path,
+        "x-project-root": path,
+    }

token_cut_cli-0.1.5/src/token_cut_cli/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.5"