sourcecode 1.35.17__tar.gz → 1.35.18__tar.gz

{sourcecode-1.35.17 → sourcecode-1.35.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sourcecode
-Version: 1.35.17
+Version: 1.35.18
 Summary: Persistent structural context and ultra-fast repeated analysis for AI coding agents
 License-File: LICENSE
 Keywords: agents,ai,codebase,context,developer-tools,llm

{sourcecode-1.35.17 → sourcecode-1.35.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sourcecode"
-version = "1.35.17"
+version = "1.35.18"
 description = "Persistent structural context and ultra-fast repeated analysis for AI coding agents"
 readme = "README.md"
 requires-python = ">=3.9"

{sourcecode-1.35.17 → sourcecode-1.35.18}/src/sourcecode/__init__.py

@@ -1,3 +1,3 @@
 """sourcecode — Deterministic codebase context maps for AI coding agents."""
 
-__version__ = "1.35.17"
+__version__ = "1.35.18"

{sourcecode-1.35.17 → sourcecode-1.35.18}/src/sourcecode/cli.py

@@ -167,6 +167,11 @@ Cold scan: 2–10s depending on repo size. Warm cache: 0.3–0.6s.
   sourcecode --compact --git-context    include git hotspots and uncommitted files
   sourcecode --agent                    full structured JSON for AI agents
 
+[bold]Auth commands:[/bold]
+  auth login                   [dim]# authenticate via browser (device code)[/dim]
+  auth status                  [dim]# show current plan and auth state[/dim]
+  auth logout                  [dim]# remove local credentials[/dim]
+
 [bold]Cache commands:[/bold]
   cache status                 [dim]# cache size, hit keys, last-warmed timestamp[/dim]
   cache warm                   [dim]# pre-build cache ahead of an agent session[/dim]
@@ -221,8 +226,8 @@ _SUBCOMMANDS: frozenset[str] = frozenset(
         "repo-ir", "mcp", "endpoints", "impact",
         # Enterprise workflow commands
         "onboard", "modernize", "fix-bug", "review-pr",
-        # License
-        "activate",
+        # License / auth
+        "activate", "auth",
         # Cache observability
         "cache",
         # RIS bootstrap
@@ -512,6 +517,9 @@ app.add_typer(mcp_app, name="mcp")
 cache_app = typer.Typer(help="Cache inspection and management.", rich_markup_mode="rich")
 app.add_typer(cache_app, name="cache")
 
+auth_app = typer.Typer(help="Authentication: login, status, logout.", rich_markup_mode="rich")
+app.add_typer(auth_app, name="auth")
+
 
 def _maybe_ask_consent() -> None:
     """Show first-run consent prompt once, on interactive TTYs only."""
@@ -4799,6 +4807,73 @@ def activate_cmd(
     _activate(license_key)
 
 
+# ---------------------------------------------------------------------------
+# Auth commands (device-flow login / status / logout)
+# ---------------------------------------------------------------------------
+
+@auth_app.command("login")
+def auth_login_cmd() -> None:
+    """Authenticate via browser (device code flow).
+
+    \b
+    The CLI shows a URL. Open it in your browser, log in with your account,
+    and the CLI completes authentication automatically.
+    Credentials are stored in ~/.sourcecode/license.json (30-min cache; Supabase is source of truth).
+
+    \b
+    Examples:
+      sourcecode auth login
+    """
+    from sourcecode.license import auth_login as _auth_login
+    _auth_login()
+
+
+@auth_app.command("status")
+def auth_status_cmd() -> None:
+    """Show current authentication and plan status."""
+    import json as _json
+    try:
+        from sourcecode.license import _license_data as _ld, is_pro as _ip
+    except Exception:
+        _ld = None
+        _ip = False
+
+    if not _ld:
+        out: dict = {"status": "unauthenticated", "pro": False}
+        sys.stdout.write(_json.dumps(out, ensure_ascii=False) + "\n")
+        sys.stdout.flush()
+        return
+
+    out = {
+        "status": "authenticated",
+        "auth_method": _ld.get("auth_method", "license_key"),
+        "email": _ld.get("email", ""),
+        "plan": _ld.get("plan", "unknown"),
+        "plan_status": _ld.get("status", "unknown"),
+        "pro": _ip,
+        "validated_at": _ld.get("validated_at") or _ld.get("activated_at") or "",
+    }
+    sys.stdout.write(_json.dumps(out, indent=2, ensure_ascii=False) + "\n")
+    sys.stdout.flush()
+
+
+@auth_app.command("logout")
+def auth_logout_cmd() -> None:
+    """Remove local credentials (does not cancel your subscription)."""
+    import json as _json
+    _lf = Path.home() / ".sourcecode" / "license.json"
+    if _lf.exists():
+        try:
+            _lf.unlink()
+            out: dict = {"status": "logged_out", "message": "Local credentials removed."}
+        except Exception as _exc:
+            out = {"status": "error", "message": str(_exc)}
+    else:
+        out = {"status": "logged_out", "message": "No local credentials found."}
+    sys.stdout.write(_json.dumps(out, ensure_ascii=False) + "\n")
+    sys.stdout.flush()
+
+
 @app.command("version")
 def version_cmd() -> None:
     """Show version and exit.

{sourcecode-1.35.17 → sourcecode-1.35.18}/src/sourcecode/license.py

@@ -42,8 +42,17 @@ if _SUPABASE_URL != _DEFAULT_SUPABASE_URL:
 _LICENSE_DIR: Path = Path.home() / ".sourcecode"
 _LICENSE_FILE: Path = _LICENSE_DIR / "license.json"
 _DELTA_RUNS_FILE: Path = _LICENSE_DIR / "delta_runs.json"
-_CACHE_TTL_SECONDS: int = 86400  # 24 hours
+_CACHE_TTL_SECONDS: int = 1800  # 30 minutes default; CI env overrides to 24h (see _get_cache_ttl)
+_CACHE_TTL_CI_SECONDS: int = 86400  # 24 hours — CI containers must not re-validate mid-run
+
+
+def _get_cache_ttl() -> int:
+    """Return TTL in seconds. CI containers get 24h to avoid mid-run network calls."""
+    return _CACHE_TTL_CI_SECONDS if os.environ.get("SOURCECODE_CI") else _CACHE_TTL_SECONDS
 _DELTA_FREE_LIMIT: int = 30
+_DEVICE_POLL_INTERVAL_S: float = 2.5
+_DEVICE_POLL_TIMEOUT_S: float = 300.0  # 5-minute window for user to complete browser auth
+_AUTH_BASE_URL: str = "https://sourcecode.dev"
 _LICENSE_KEY_RE = re.compile(r"^[A-Za-z0-9_\-]{1,200}$")
 
 # ---------------------------------------------------------------------------
@@ -213,6 +222,78 @@ def _call_get_license(license_key: str) -> Optional[dict]:
         return None  # Network error — caller decides what to do
 
 
+def _generate_device_code() -> str:
+    """Generate a human-readable device code: XXXX-XXXX-XXXX."""
+    import uuid
+    raw = uuid.uuid4().hex.upper()
+    return f"{raw[:4]}-{raw[4:8]}-{raw[8:12]}"
+
+
+def _call_device_check(device_code: str) -> Optional[dict]:
+    """Poll /device-check edge function. Returns dict or None on network error.
+
+    Expected responses:
+      {"status": "pending"}
+      {"status": "complete", "device_token": "...", "email": "...", "plan": "pro", ...}
+      {"status": "error", "message": "..."}
+    """
+    import urllib.error
+    import urllib.request
+
+    if not _SUPABASE_ANON_KEY:
+        return None
+
+    url = f"{_SUPABASE_URL}/functions/v1/device-check"
+    body = json.dumps({"device_code": device_code}).encode("utf-8")
+    req = urllib.request.Request(url, data=body, method="POST")
+    req.add_header("apikey", _SUPABASE_ANON_KEY)
+    req.add_header("Authorization", f"Bearer {_SUPABASE_ANON_KEY}")
+    req.add_header("Content-Type", "application/json")
+    req.add_header("Accept", "application/json")
+    try:
+        with urllib.request.urlopen(req, timeout=8) as resp:
+            return json.loads(resp.read().decode("utf-8"))
+    except urllib.error.HTTPError as exc:
+        try:
+            return json.loads(exc.read().decode("utf-8", errors="replace"))
+        except Exception:
+            return {"status": "error", "message": f"HTTP {exc.code}"}
+    except Exception:
+        return None
+
+
+def _call_get_user_plan(device_token: str) -> Optional[dict]:
+    """Fetch current plan/status for an authenticated device token.
+
+    Expected response:
+      {"valid": true, "plan": "pro", "status": "active", "features": [...], "email": "..."}
+      {"valid": false, "error": "token_revoked"}
+    """
+    import urllib.error
+    import urllib.request
+
+    if not _SUPABASE_ANON_KEY:
+        return None
+
+    url = f"{_SUPABASE_URL}/functions/v1/get-user-plan"
+    body = json.dumps({"device_token": device_token}).encode("utf-8")
+    req = urllib.request.Request(url, data=body, method="POST")
+    req.add_header("apikey", _SUPABASE_ANON_KEY)
+    req.add_header("Authorization", f"Bearer {_SUPABASE_ANON_KEY}")
+    req.add_header("Content-Type", "application/json")
+    req.add_header("Accept", "application/json")
+    try:
+        with urllib.request.urlopen(req, timeout=8) as resp:
+            return json.loads(resp.read().decode("utf-8"))
+    except urllib.error.HTTPError as exc:
+        try:
+            return json.loads(exc.read().decode("utf-8", errors="replace"))
+        except Exception:
+            return {"valid": False, "error": f"HTTP {exc.code}"}
+    except Exception:
+        return None
+
+
 def _maybe_revalidate() -> None:
     """Re-validate cached license if stale. Mutates globals; never raises."""
     global _license_data, is_pro
@@ -220,18 +301,55 @@ def _maybe_revalidate() -> None:
     if not _license_data:
         return
 
-    validated_at_str = _license_data.get("validated_at") or _license_data.get("activated_at")
+    validated_at_str = (
+        _license_data.get("validated_at")
+        or _license_data.get("activated_at")
+        or _license_data.get("authenticated_at")
+    )
     if validated_at_str:
         try:
             validated_at = datetime.fromisoformat(validated_at_str)
             if validated_at.tzinfo is None:
                 validated_at = validated_at.replace(tzinfo=timezone.utc)
             age = (datetime.now(timezone.utc) - validated_at).total_seconds()
-            if age < _CACHE_TTL_SECONDS:
+            if age < _get_cache_ttl():
                 return
         except Exception:
             pass
 
+    auth_method = _license_data.get("auth_method")
+
+    if auth_method == "device_flow":
+        device_token = _license_data.get("device_token")
+        if not device_token:
+            return
+        result = _call_get_user_plan(device_token)
+        if result is None:
+            return  # Network error — keep cached (offline-first)
+        if not result.get("valid", True):
+            _license_data = None
+            is_pro = False
+            try:
+                if _LICENSE_FILE.exists():
+                    _LICENSE_FILE.unlink()
+            except Exception:
+                pass
+            return
+        _license_data["plan"] = result.get("plan", "free")
+        _license_data["status"] = result.get("status", "active")
+        _license_data["features"] = result.get("features", [])
+        _license_data["validated_at"] = datetime.now(timezone.utc).isoformat()
+        is_pro = (
+            _license_data.get("plan") == "pro"
+            and _license_data.get("status", "active") != "inactive"
+        )
+        try:
+            _write_license_file(_license_data)
+        except Exception:
+            pass
+        return
+
+    # Key-based auth (existing flow / legacy)
     key = _license_data.get("license_key")
     if not key:
         return
@@ -266,6 +384,7 @@ def _init() -> None:
     is_pro = (
         _license_data is not None
         and _license_data.get("plan") == "pro"
+        and _license_data.get("status", "active") != "inactive"
     )
 
 
@@ -354,7 +473,105 @@ def require_pro(feature_name: str) -> None:
 
 
 # ---------------------------------------------------------------------------
-# Activation
+# Device-flow authentication
+# ---------------------------------------------------------------------------
+
+def _finish_device_auth(result: dict) -> None:
+    """Persist device-flow credentials and emit success JSON. Exits on error."""
+    global _license_data, is_pro
+
+    device_token = result.get("device_token") or result.get("access_token") or ""
+    email = result.get("email", "")
+    plan = result.get("plan", "free")
+    plan_status = (
+        result.get("status_detail")
+        or result.get("user_status")
+        or result.get("status", "active")
+    )
+    features = result.get("features") or []
+
+    if not device_token:
+        sys.stderr.write("\n")
+        _fail("auth_error", "Authentication completed but no session token received. Contact support.")
+
+    _LICENSE_DIR.mkdir(parents=True, exist_ok=True)
+    now = datetime.now(timezone.utc).isoformat()
+    data: dict = {
+        "auth_method": "device_flow",
+        "device_token": device_token,
+        "email": email,
+        "plan": plan,
+        "status": plan_status,
+        "features": features,
+        "authenticated_at": now,
+        "validated_at": now,
+    }
+    _write_license_file(data)
+    _license_data = data
+    is_pro = plan == "pro" and plan_status != "inactive"
+
+    sys.stderr.write(f"\n  Authenticated as {email}.  Plan: {plan}\n\n")
+    sys.stderr.flush()
+
+    output: dict = {"status": "authenticated", "email": email, "plan": plan, "pro": is_pro}
+    if not is_pro:
+        output["upgrade_hint"] = "https://sourcecode.dev/pricing"
+    else:
+        output["features"] = features
+    sys.stdout.write(json.dumps(output, ensure_ascii=False) + "\n")
+    sys.stdout.flush()
+
+
+def auth_login() -> None:
+    """Device code authentication flow.
+
+    Shows a browser URL; polls the backend every 2.5 s until the user
+    completes authentication or the 5-minute window expires.
+    Writes credentials to ~/.sourcecode/license.json on success.
+    Exits 0 on success, 1 on any failure.
+    """
+    import time
+
+    device_code = _generate_device_code()
+    activate_url = f"{_AUTH_BASE_URL}/activate?code={device_code}"
+
+    sys.stderr.write(f"\n  Open this URL to authenticate:\n  {activate_url}\n\n  Waiting")
+    sys.stderr.flush()
+
+    deadline = time.monotonic() + _DEVICE_POLL_TIMEOUT_S
+    _tick = 0
+
+    while time.monotonic() < deadline:
+        time.sleep(_DEVICE_POLL_INTERVAL_S)
+        _tick += 1
+        if _tick % 4 == 0:
+            sys.stderr.write(".")
+            sys.stderr.flush()
+
+        result = _call_device_check(device_code)
+        if result is None:
+            continue  # network blip — keep polling
+
+        status = result.get("status")
+        if status == "pending":
+            continue
+
+        if status == "complete":
+            _finish_device_auth(result)
+            return
+
+        if status == "error" or result.get("error"):
+            sys.stderr.write("\n")
+            _fail("auth_error", result.get("message") or result.get("error") or "Authentication failed.")
+
+        # Unknown status — keep polling
+
+    sys.stderr.write("\n")
+    _fail("auth_timeout", "Authentication timed out after 5 minutes. Please try again.")
+
+
+# ---------------------------------------------------------------------------
+# Activation (key-based — legacy / direct key entry)
 # ---------------------------------------------------------------------------
 
 def activate_license(license_key: str) -> None: