codestrain 0.1.3__tar.gz → 0.1.4__tar.gz

{codestrain-0.1.3 → codestrain-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codestrain
-Version: 0.1.3
+Version: 0.1.4
 Summary: Your AI coding recovery score, from the terminal.
 Project-URL: Homepage, https://codestrain.dev
 Project-URL: Repository, https://github.com/codestrain/codestrain-cli

{codestrain-0.1.3 → codestrain-0.1.4}/codestrain_cli.py

@@ -707,6 +707,49 @@ def print_project_breakdown(project_stats, anonymize=False):
     print()
 
 
+# ── Share encoding ───────────────────────────────────────────────────────────
+#
+# `codestrain --share` produces a single self-contained URL that anyone can
+# open to see the same anonymized report. No server, no upload, no cookies —
+# the whole report is gzip+base64-encoded into the URL query string. Decoded
+# client-side by codestrain.dev/s/index.html.
+
+SHARE_BASE_URL = "https://codestrain.dev/s/"
+
+
+def build_share_url(text: str, base: str = SHARE_BASE_URL) -> str:
+    """Encode `text` into a self-contained share URL.
+
+    Pipeline: utf-8 → gzip (level 9, deterministic) → base64-urlsafe (strip `=`
+    padding). The frontend reverses this with DecompressionStream('gzip').
+    Average codestrain --all output (~1.5 KB raw) compresses to ~600 B → ~800 B
+    base64 → final URL ~ 850 B. Well within all known browser URL limits
+    (Chrome / Safari / Firefox / curl all accept up to ~32 KB).
+    """
+    import base64
+    import gzip
+    payload = gzip.compress(text.encode("utf-8"), compresslevel=9, mtime=0)
+    encoded = base64.urlsafe_b64encode(payload).decode("ascii").rstrip("=")
+    return f"{base.rstrip('/')}/?d={encoded}"
+
+
+def _copy_to_clipboard(text: str) -> bool:
+    """Best-effort clipboard copy. Returns True on success, False if no
+    suitable system command is available. Never raises.
+    """
+    import shutil
+    import subprocess
+    for cmd in (["pbcopy"], ["xclip", "-selection", "clipboard"], ["wl-copy"]):
+        if shutil.which(cmd[0]):
+            try:
+                p = subprocess.run(cmd, input=text.encode("utf-8"), timeout=2)
+                if p.returncode == 0:
+                    return True
+            except (OSError, subprocess.TimeoutExpired):
+                continue
+    return False
+
+
 # ── Main ─────────────────────────────────────────────────────────────────────
 
 def main():
@@ -766,9 +809,21 @@ examples:
         help="Logo variant: auto (default, adapts to terminal width), "
              "big (5-line ASCII), small (one-line), or none (skip logo)",
     )
+    parser.add_argument(
+        "--share",
+        action="store_true",
+        help="Generate a shareable codestrain.dev URL with anonymized stats "
+             "(implies --anonymize --no-color; nothing is uploaded — all data "
+             "is encoded in the URL itself)",
+    )
 
     args = parser.parse_args()
 
+    # --share is a shortcut that anonymizes, strips color, then prints a URL.
+    if args.share:
+        args.anonymize = True
+        args.no_color = True
+
     if args.no_color:
         global _colors_on
         _colors_on = False
@@ -852,23 +907,43 @@ examples:
             project_stats[project_name] = []
         project_stats[project_name].append(stats)
 
-    # Display results
-    print_header_adaptive(args.logo)
-
-    time_label = "Today" if not args.all else "All Time"
-    if args.project and not args.anonymize:
-        time_label += f" (project: {args.project})"
-    elif args.project and args.anonymize:
-        time_label += " (filtered)"
-
-    print_divider(time_label)
-    print()
-    print_session_summary(all_stats)
-
-    if len(project_stats) > 1 and not args.no_breakdown:
-        print_project_breakdown(project_stats, anonymize=args.anonymize)
+    # Display results — capture stdout into a buffer if --share, so we can
+    # encode the rendered report into a URL after the run.
+    import contextlib
+    import io as _io
+
+    def _render_to(stream):
+        with contextlib.redirect_stdout(stream):
+            print_header_adaptive(args.logo)
+            time_label = "Today" if not args.all else "All Time"
+            if args.project and not args.anonymize:
+                time_label += f" (project: {args.project})"
+            elif args.project and args.anonymize:
+                time_label += " (filtered)"
+            print_divider(time_label)
+            print()
+            print_session_summary(all_stats)
+            if len(project_stats) > 1 and not args.no_breakdown:
+                print_project_breakdown(project_stats, anonymize=args.anonymize)
+            print()
 
-    print()
+    if args.share:
+        buf = _io.StringIO()
+        _render_to(buf)
+        report_text = buf.getvalue()
+        # Print the report to stdout so the user sees what they're sharing
+        sys.stdout.write(report_text)
+        share_url = build_share_url(report_text)
+        print()
+        print(f"  Shareable URL ({len(share_url)} chars):")
+        print(f"  {share_url}")
+        print()
+        # Best-effort clipboard copy on macOS / Linux
+        if _copy_to_clipboard(share_url):
+            print("  (copied to clipboard)")
+            print()
+    else:
+        _render_to(sys.stdout)
 
 
 if __name__ == "__main__":

{codestrain-0.1.3 → codestrain-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codestrain"
-version = "0.1.3"
+version = "0.1.4"
 description = "Your AI coding recovery score, from the terminal."
 requires-python = ">=3.9"
 readme = "README.md"

{codestrain-0.1.3 → codestrain-0.1.4}/tests/smoke.sh

@@ -107,6 +107,13 @@ out=$("$PY" "$CLI" --path "$FIXTURES" --all --project projectA --no-color 2>&1)
 # format_tokens compacts to "3.8K"-style — assert the K
 contains "$out" "3.8K" "expected ~3800 input tokens for projectA (compact = 3.8K)"
 
+# 9. --share emits a codestrain.dev/s/ URL and implies --anonymize.
+out=$("$PY" "$CLI" --path "$FIXTURES" --all --share 2>&1)
+contains "$out" "https://codestrain.dev/s/?d=" "--share prints shareable URL"
+contains "$out" "project-1"                    "--share implies --anonymize (project-1)"
+not_contains "$out" "projectA"                 "--share scrubs real project names"
+not_contains "$out" "projectB"                 "--share scrubs real project names"
+
 # ----------------------------------------------------------------------------
 # summary
 # ----------------------------------------------------------------------------

{codestrain-0.1.3 → codestrain-0.1.4}/tests/test_unit.py

@@ -145,3 +145,54 @@ def test_format_cost_dollars(cli, cost, expected_prefix):
 def test_format_tokens_compact(cli, n, expected_substr):
     out = cli.format_tokens(n)
     assert expected_substr in out
+
+
+# ─── Share encoding ─────────────────────────────────────────────────────────
+
+def test_share_url_round_trip(cli):
+    """gzip + base64-urlsafe encoder must round-trip a real report string."""
+    import base64
+    import gzip
+    sample = (
+        "--- All Time ----\n"
+        "  Sessions: 1454\n"
+        "  Duration: 137h 21m\n"
+        "  Cost: $21948.61\n"
+    )
+    url = cli.build_share_url(sample)
+    assert url.startswith("https://codestrain.dev/s/?d=")
+    encoded = url.split("?d=", 1)[1]
+    # Restore stripped padding + base64-urlsafe → bytes → gunzip → text.
+    pad = "=" * (-len(encoded) % 4)
+    raw = base64.urlsafe_b64decode(encoded + pad)
+    decoded = gzip.decompress(raw).decode("utf-8")
+    assert decoded == sample
+
+
+def test_share_url_unicode_safe(cli):
+    """Non-ASCII content (Cyrillic, emoji, currency) must survive the trip."""
+    import base64
+    import gzip
+    sample = "Сессий: 1454 · стоимость €18 950 / $21 948 · 🟢 recovery 82%"
+    url = cli.build_share_url(sample)
+    encoded = url.split("?d=", 1)[1]
+    pad = "=" * (-len(encoded) % 4)
+    raw = base64.urlsafe_b64decode(encoded + pad)
+    assert gzip.decompress(raw).decode("utf-8") == sample
+
+
+def test_share_url_compact_for_typical_report(cli):
+    """A realistic 1.5 KB report should encode to under ~1 KB URL."""
+    # 30 lines × 50 chars = 1500 bytes (close to real --all output).
+    sample = "\n".join(f"  line {i:02d}: project-{i} 12h 34m $123.45" for i in range(30))
+    url = cli.build_share_url(sample)
+    # Comfortably under 32 KB (the conservative browser URL limit).
+    assert len(url) < 2_000
+    # Should compress reasonably — the line template is highly repetitive.
+    assert len(url) < len(sample)
+
+
+def test_share_url_deterministic(cli):
+    """Same input → same URL across calls (mtime=0 in gzip header)."""
+    text = "hello world\n"
+    assert cli.build_share_url(text) == cli.build_share_url(text)