nabla-cli 0.2.0__tar.gz → 0.2.1__tar.gz

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nabla-cli
-Version: 0.2.0
+Version: 0.2.1
 Summary: Call-site harvester: scan OpenAI call sites, record real traffic, emit site profiles
 License: MIT
 Requires-Python: >=3.11

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/nabla/__init__.py

@@ -1,3 +1,3 @@
 """nabla — call-site harvester CLI (pre-Phase-0)."""
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/nabla/cli.py

@@ -15,6 +15,7 @@ from .profile import build_profile, find_site
 from .recorder import RecordSource, ZeroTrafficError, record
 from .scanner import scan
 from .schema_resolver import resolve_schema
+from .term import init_terminal, style
 
 _PROVIDER_DESCRIPTIONS = {
     "openai": "real OpenAI API, uses your OPENAI_API_KEY",
@@ -114,7 +115,32 @@ def _build_parser() -> argparse.ArgumentParser:
     return parser
 
 
+def _ok(msg: str) -> None:
+    print(style(msg, "green"))
+
+
+def _warn(msg: str) -> None:
+    print(style(msg, "yellow", stream=sys.stderr), file=sys.stderr)
+
+
+def _fail(msg: str) -> None:
+    print(style(msg, "red", stream=sys.stderr), file=sys.stderr)
+
+
+def _unconfigured() -> bool:
+    """True when no provider is set up anywhere — the state where record/
+    profile will fail and the user should be nudged toward `nabla init`."""
+    try:
+        provider = apply_to_env()
+    except Exception:
+        provider = None
+    return provider is None and not (
+        os.environ.get("OPENAI_API_KEY") or os.environ.get("NABLA_UPSTREAM_API_KEY")
+    )
+
+
 def main(argv: list[str] | None = None) -> int:
+    init_terminal()
     # LLM-produced text (goal descriptions, error bodies) can contain
     # characters legacy Windows consoles (cp1252) can't encode; degrade to
     # "?" instead of crashing mid-command.
@@ -145,13 +171,42 @@ def main(argv: list[str] | None = None) -> int:
     return commands[args.command](args)
 
 
+def _read_api_key(prompt: str) -> str:
+    """Read a secret with visible masking: one `*` per character, backspace
+    supported, pasting shows a `*` per pasted character — so the user can
+    see that their input (typed or pasted) actually arrived. Falls back to
+    fully hidden input where character-level reading isn't available."""
+    if sys.platform == "win32" and sys.stdin.isatty():
+        import msvcrt
+
+        print(prompt, end="", flush=True)
+        chars: list[str] = []
+        while True:
+            ch = msvcrt.getwch()
+            if ch in ("\r", "\n"):
+                print()
+                break
+            if ch == "\x03":  # Ctrl+C
+                raise KeyboardInterrupt
+            if ch == "\b":
+                if chars:
+                    chars.pop()
+                    print("\b \b", end="", flush=True)
+            elif ch >= " ":
+                chars.append(ch)
+                print("*", end="", flush=True)
+        return "".join(chars).strip()
+    return getpass(f"{prompt} (input is hidden — paste and press Enter): ").strip()
+
+
 def _cmd_init(args) -> int:
     provider = args.provider
     if provider is None:
         names = list(PROVIDERS)
-        print("Pick a provider:")
+        print(style("Pick a provider:", "bold"))
         for i, name in enumerate(names, start=1):
-            print(f"  {i}) {name} — {_PROVIDER_DESCRIPTIONS.get(name, '')}")
+            print(f"  {style(f'{i})', 'cyan')} {style(name, 'bold')} — "
+                  f"{_PROVIDER_DESCRIPTIONS.get(name, '')}")
         choice = input("> ").strip()
         if choice.isdigit() and 1 <= int(choice) <= len(names):
             provider = names[int(choice) - 1]
@@ -161,25 +216,25 @@ def _cmd_init(args) -> int:
     api_key = args.api_key
     if api_key is None:
         key_env = PROVIDERS.get(provider, {}).get("api_key_env", "API key")
-        api_key = getpass(f"Paste your {key_env}: ").strip()
+        api_key = _read_api_key(f"Paste your {key_env}: ")
 
     if not api_key:
-        print("nabla init: no API key given", file=sys.stderr)
+        _fail("nabla init: no API key given")
         return 2
 
     try:
         path = save_config(provider, api_key)
     except ValueError as exc:
-        print(f"nabla init: {exc}", file=sys.stderr)
+        _fail(f"nabla init: {exc}")
         return 2
 
     masked = f"...{api_key[-4:]}" if len(api_key) >= 4 else "..."
-    print(f"nabla init: saved to {path} (key: {masked})")
+    _ok(f"nabla init: saved to {path} (key: {masked})")
     try:
         _ensure_scripts_on_path()
     except Exception:
         pass  # PATH help is best-effort; setup itself already succeeded
-    print("Setup complete — try: nabla scan")
+    print(style("Setup complete — try: nabla scan", "cyan"))
     return 0
 
 
@@ -245,11 +300,26 @@ def _cmd_scan(args) -> int:
     if args.json:
         print(json.dumps(rows, indent=2))
         return 0
-    print(f"{'SYMBOL':<22} {'FILE':<28} {'KIND':<6} {'VERIFIABILITY':<22} FLAGS")
+    print(style(f"{'SYMBOL':<22} {'FILE':<28} {'KIND':<6} {'VERIFIABILITY':<22} FLAGS", "bold"))
+    supported_count = 0
     for r in rows:
-        print(f"{r['symbol']:<22} {r['file']:<28} {r['kind']:<6} "
-              f"{r['verifiability']:<22} {','.join(r['flags']) or '-'}")
+        verif = r["verifiability"]
+        if verif in ("json_schema", "pydantic_parse"):
+            verif_color = "green"
+            if not r["flags"]:
+                supported_count += 1
+        elif verif in ("tool_call", "json_object_no_schema"):
+            verif_color = "yellow"
+        else:
+            verif_color = "dim"
+        flags = ",".join(r["flags"]) or "-"
+        print(f"{style(r['symbol'], 'cyan')}{' ' * max(1, 22 - len(r['symbol']))}"
+              f"{r['file']:<28} {r['kind']:<6} "
+              f"{style(f'{verif:<22}', verif_color)} {flags}")
     print(f"\n{len(rows)} call sites found.")
+    if _unconfigured():
+        _warn("tip: run `nabla init` once to set up a provider — `nabla record` "
+              "and `nabla profile` need one")
     return 0
 
 
@@ -282,7 +352,7 @@ def _resolve_site(args, sites):
 
     if len(supported) == 1:
         site = supported[0]
-        print(f"nabla: using site {site.symbol} (the only supported call site)")
+        print(style(f"nabla: using site {site.symbol} (the only supported call site)", "dim"))
         return site, None
 
     candidates = supported or sites
@@ -307,14 +377,11 @@ def _cmd_record(args) -> int:
     if err is not None:
         return err
 
-    if apply_to_env() is None and not (
-        os.environ.get("OPENAI_API_KEY") or os.environ.get("NABLA_UPSTREAM_API_KEY")
-    ):
-        print(
+    if _unconfigured():
+        _warn(
             "nabla record: warning — no API key configured (OPENAI_API_KEY / "
             "NABLA_UPSTREAM_API_KEY unset, and no saved config). Run `nabla init` "
-            "first unless this call site never reaches a real upstream.",
-            file=sys.stderr,
+            "first unless this call site never reaches a real upstream."
         )
 
     source_kind = None  # written to the JSONL; may differ from source.kind
@@ -338,17 +405,17 @@ def _cmd_record(args) -> int:
     try:
         examples = record(site, source, args.repo)
     except ZeroTrafficError as exc:
-        print(f"nabla record: {exc}", file=sys.stderr)
+        _fail(f"nabla record: {exc}")
         return 1
     out = args.out or f"{site.symbol}.recorded.jsonl"
     with open(out, "w", encoding="utf-8") as f:
         for ex in examples:
             f.write(json.dumps({**ex.to_dict(), "_source_kind": source_kind or source.kind}) + "\n")
-    print(f"nabla record: {len(examples)} examples -> {out}")
+    _ok(f"nabla record: {len(examples)} examples -> {out}")
     if args.out is None and args.site is None:
-        print(f"next: nabla profile {args.repo}")
+        print(style(f"next: nabla profile {args.repo}", "cyan"))
     else:
-        print(f"next: nabla profile {args.repo} --site {site.symbol} --examples {out}")
+        print(style(f"next: nabla profile {args.repo} --site {site.symbol} --examples {out}", "cyan"))
     return 0
 
 
@@ -425,12 +492,12 @@ def _cmd_profile(args) -> int:
     out = args.out or f"{site.symbol}.profile.json"
     Path(out).write_text(json.dumps(profile, indent=2), encoding="utf-8")
     supported = profile["classification"]["supported"]
-    print(f"nabla profile: wrote {out} "
-          f"(site {profile['site_id'][:12]}…, supported={supported}, "
-          f"{len(examples)} examples)")
+    _ok(f"nabla profile: wrote {out} "
+        f"(site {profile['site_id'][:12]}…, supported={supported}, "
+        f"{len(examples)} examples)")
     if len(examples) < 20:
-        print(f"nabla profile: WARNING — {len(examples)} examples < 20; "
-              "Phase-0-ready profiles need >=20 (cli-plan §5.4)", file=sys.stderr)
+        _warn(f"nabla profile: WARNING — {len(examples)} examples < 20; "
+              "Phase-0-ready profiles need >=20 (cli-plan §5.4)")
     print(f"goal: {goal.description[:100]}")
     return 0
 

nabla_cli-0.2.1/nabla/term.py

@@ -0,0 +1,53 @@
+"""Minimal terminal styling — ANSI colors with zero dependencies.
+
+Colors are applied only when the target stream is a real terminal and
+NO_COLOR is unset, so piped/captured output stays plain (tests, --json
+consumers, CI logs). init_terminal() switches Windows consoles into VT
+mode, which cmd.exe does not enable by default.
+"""
+
+from __future__ import annotations
+
+import os
+import sys
+
+_CODES = {
+    "bold": "1",
+    "dim": "2",
+    "red": "31",
+    "green": "32",
+    "yellow": "33",
+    "cyan": "36",
+}
+
+
+def init_terminal() -> None:
+    if sys.platform != "win32":
+        return
+    try:
+        import ctypes
+
+        kernel32 = ctypes.windll.kernel32
+        for handle_id in (-11, -12):  # stdout, stderr
+            handle = kernel32.GetStdHandle(handle_id)
+            mode = ctypes.c_ulong()
+            if kernel32.GetConsoleMode(handle, ctypes.byref(mode)):
+                kernel32.SetConsoleMode(handle, mode.value | 0x0004)  # ENABLE_VIRTUAL_TERMINAL_PROCESSING
+    except Exception:
+        pass  # styling is cosmetic; never let it break a command
+
+
+def supports_color(stream=None) -> bool:
+    stream = stream or sys.stdout
+    if os.environ.get("NO_COLOR"):
+        return False
+    return hasattr(stream, "isatty") and stream.isatty()
+
+
+def style(text: str, *names: str, stream=None) -> str:
+    """Wrap text in ANSI codes when the stream supports it. Apply AFTER any
+    column padding — escape codes are invisible but count toward len()."""
+    if not names or not supports_color(stream):
+        return text
+    codes = ";".join(_CODES[n] for n in names)
+    return f"\x1b[{codes}m{text}\x1b[0m"

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/nabla_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nabla-cli
-Version: 0.2.0
+Version: 0.2.1
 Summary: Call-site harvester: scan OpenAI call sites, record real traffic, emit site profiles
 License: MIT
 Requires-Python: >=3.11

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/nabla_cli.egg-info/SOURCES.txt

@@ -12,6 +12,7 @@ nabla/recorder.py
 nabla/samplegen.py
 nabla/scanner.py
 nabla/schema_resolver.py
+nabla/term.py
 nabla/schemas/site_profile.schema.json
 nabla_cli.egg-info/PKG-INFO
 nabla_cli.egg-info/SOURCES.txt

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nabla-cli"
-version = "0.2.0"
+version = "0.2.1"
 description = "Call-site harvester: scan OpenAI call sites, record real traffic, emit site profiles"
 readme = "README.md"
 license = { text = "MIT" }

{nabla_cli-0.2.0 → nabla_cli-0.2.1}/tests/test_cli_ux.py

@@ -330,3 +330,21 @@ def test_ensure_scripts_on_path_non_windows_prints_instruction(monkeypatch, caps
     out = capsys.readouterr().out
     assert "PATH" in out
     assert "python -m nabla" in out
+
+
+def test_scan_prints_init_tip_when_unconfigured(capsys):
+    rc = main(["scan", FIXTURE_REPO])
+    assert rc == 0
+    assert "nabla init" in capsys.readouterr().err
+
+
+def test_scan_json_output_stays_clean_json(capsys):
+    rc = main(["scan", FIXTURE_REPO, "--json"])
+    assert rc == 0
+    json.loads(capsys.readouterr().out)  # must parse — no tip/color on stdout
+
+
+def test_style_is_plain_when_not_a_tty(capsys):
+    from nabla.term import style
+
+    assert style("hello", "red") == "hello"  # captured stdout is not a tty