codebyplan 1.10.3 → 1.11.0

package/templates/hooks/cbp-statusline.py

@@ -0,0 +1,331 @@
+#!/usr/bin/env python3
+# @hook: NOT-A-HOOK (statusLine renderer, invoked via the cbp-statusline.sh dispatcher)
+# Claude Code Status Line — python renderer.
+# Byte-identical output to the bash renderer in cbp-statusline.sh and the node
+# renderer in cbp-statusline.mjs. See that .sh file for the full option/env/seam
+# contract. Selected via .codebyplan/statusline.local.json -> {"renderer":"python"}.
+#
+# Reads stdin JSON; resolves .codebyplan/ root from CBP_STATUSLINE_ROOT (set by the
+# dispatcher) or argv[1], else the script's ../.. directory. Never throws to stdout.
+
+import json
+import math
+import os
+import subprocess
+import sys
+import time
+
+
+def _get(obj, keys, dflt):
+    # jq `//` semantics: default on null/missing (and false → false).
+    cur = obj
+    for k in keys:
+        if not isinstance(cur, dict):
+            return dflt
+        cur = cur.get(k)
+    return dflt if cur is None else cur
+
+
+def main():
+    raw_input = sys.stdin.read()
+    try:
+        data = json.loads(raw_input)
+    except Exception:
+        data = {}
+    if not isinstance(data, dict):
+        data = {}
+
+    # ---- Root resolution (matches the dispatcher contract) -------------------
+    root = os.environ.get("CBP_STATUSLINE_ROOT") or (sys.argv[1] if len(sys.argv) > 1 else "")
+    if not root:
+        root = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", ".."))
+
+    MODEL_ID = _get(data, ["model", "id"], "")
+    MODEL_NAME = _get(data, ["model", "display_name"], "")
+    CWD = _get(data, ["cwd"], "")
+    WS_CURRENT_DIR = _get(data, ["workspace", "current_dir"], "")
+    WS_REPO_HOST = _get(data, ["workspace", "repo", "host"], "")
+    WS_REPO_OWNER = _get(data, ["workspace", "repo", "owner"], "")
+    WS_REPO_NAME = _get(data, ["workspace", "repo", "name"], "")
+    COST = _get(data, ["cost", "total_cost_usd"], 0)
+    DURATION = _get(data, ["cost", "total_duration_ms"], 0)
+    API_DURATION = _get(data, ["cost", "total_api_duration_ms"], 0)
+    LINES_ADD = _get(data, ["cost", "total_lines_added"], 0)
+    LINES_DEL = _get(data, ["cost", "total_lines_removed"], 0)
+    CTX_SIZE = _get(data, ["context_window", "context_window_size"], 200000)
+    CTX_PCT = _get(data, ["context_window", "used_percentage"], 0)
+    CUR_IN = _get(data, ["context_window", "current_usage", "input_tokens"], 0)
+    CUR_OUT = _get(data, ["context_window", "current_usage", "output_tokens"], 0)
+    CACHE_CREATE = _get(data, ["context_window", "current_usage", "cache_creation_input_tokens"], 0)
+    CACHE_READ = _get(data, ["context_window", "current_usage", "cache_read_input_tokens"], 0)
+    EXCEEDS_200K = _get(data, ["exceeds_200k_tokens"], False)
+    EFFORT = _get(data, ["effort", "level"], "")
+    THINKING = _get(data, ["thinking", "enabled"], False)
+    RATE_5H_PCT = _get(data, ["rate_limits", "five_hour", "used_percentage"], "")
+    RATE_5H_RESETS = _get(data, ["rate_limits", "five_hour", "resets_at"], 0)
+    RATE_7D_PCT = _get(data, ["rate_limits", "seven_day", "used_percentage"], "")
+    RATE_7D_RESETS = _get(data, ["rate_limits", "seven_day", "resets_at"], 0)
+    SESSION_NAME = _get(data, ["session_name"], "")
+    OUTPUT_STYLE = _get(data, ["output_style", "name"], "")
+    VIM_MODE = _get(data, ["vim", "mode"], "")
+    AGENT_NAME = _get(data, ["agent", "name"], "")
+    PR_NUMBER = _get(data, ["pr", "number"], "")
+    PR_URL = _get(data, ["pr", "url"], "")
+    PR_REVIEW_STATE = _get(data, ["pr", "review_state"], "")
+    WT_NAME = _get(data, ["worktree", "name"], "")
+    WT_PATH = _get(data, ["worktree", "path"], "")
+    WT_BRANCH = _get(data, ["worktree", "branch"], "")
+    WT_ORIG_BRANCH = _get(data, ["worktree", "original_branch"], "")
+
+    # ---- Config: line toggles + no_color -------------------------------------
+    cfg = {
+        "identity": True, "context": True, "cost": True,
+        "rate_limits": True, "repo_pr": True, "worktree": True, "no_color": False,
+    }
+    try:
+        with open(os.path.join(root, ".codebyplan", "statusline.json"), "r", encoding="utf-8") as fh:
+            parsed = json.load(fh)
+        if isinstance(parsed, dict):
+            if isinstance(parsed.get("no_color"), bool):
+                cfg["no_color"] = parsed["no_color"]
+            lines = parsed.get("lines")
+            if isinstance(lines, dict):
+                for k in ["identity", "context", "cost", "rate_limits", "repo_pr", "worktree"]:
+                    if isinstance(lines.get(k), bool):
+                        cfg[k] = lines[k]
+    except Exception:
+        pass  # absent / invalid → keep defaults
+
+    def should_show(env_suffix, cfg_value):
+        # env HIDE=1 > config false > default show
+        if os.environ.get("CBP_STATUSLINE_HIDE_" + env_suffix) == "1":
+            return False
+        if cfg_value is False:
+            return False
+        return True
+
+    # ---- Colour setup (env > config) -----------------------------------------
+    no_color = (
+        (os.environ.get("NO_COLOR") not in (None, ""))
+        or os.environ.get("CBP_STATUSLINE_NO_COLOR") == "1"
+        or cfg["no_color"] is True
+    )
+    if no_color:
+        RST = DIM = BOLD = GREEN = YELLOW = RED = CYAN = MAGENTA = BLUE = ""
+    else:
+        RST = "\x1b[0m"
+        DIM = "\x1b[2m"
+        BOLD = "\x1b[1m"
+        GREEN = "\x1b[32m"
+        YELLOW = "\x1b[33m"
+        RED = "\x1b[31m"
+        CYAN = "\x1b[36m"
+        MAGENTA = "\x1b[35m"
+        BLUE = "\x1b[34m"
+
+    # ---- Numeric helpers (integer round-half-up; cross-runtime identical) -----
+    def num_str(n):
+        try:
+            x = float(n)
+        except (TypeError, ValueError):
+            return str(n)
+        if x.is_integer():
+            return str(int(x))
+        return str(n)
+
+    def fmt_k(val):
+        v = float(val)
+        if v >= 1000000:
+            t = math.floor((v + 50000) / 100000)
+            return "%d.%dM" % (t // 10, t % 10)
+        elif v >= 1000:
+            t = math.floor((v + 50) / 100)
+            return "%d.%dK" % (t // 10, t % 10)
+        return str(int(v))
+
+    def fmt_cost(c):
+        n = math.floor(float(c) * 10000 + 0.5)
+        return "$%d.%04d" % (n // 10000, n % 10000)
+
+    def fmt_dur(ms):
+        secs = math.trunc(float(ms) / 1000)
+        if secs >= 3600:
+            return "%dh%dm" % (secs // 3600, (secs % 3600) // 60)
+        if secs >= 60:
+            return "%dm%ds" % (secs // 60, secs % 60)
+        return "%ds" % secs
+
+    def cbp_now():
+        nv = os.environ.get("CBP_STATUSLINE_NOW")
+        if nv not in (None, ""):
+            return math.trunc(float(nv))
+        return math.floor(time.time())
+
+    def fmt_rel_time(epoch):
+        delta = math.trunc(float(epoch)) - cbp_now()
+        if delta <= 0:
+            return "now"
+        if delta >= 86400:
+            return "%dd" % (delta // 86400)
+        if delta >= 3600:
+            return "%dh" % (delta // 3600)
+        return "%dm" % (delta // 60)
+
+    def gte(v, t):
+        try:
+            return float(v) >= t
+        except (TypeError, ValueError):
+            return False
+
+    # ---- Folder + branch -----------------------------------------------------
+    folder = ""
+    if CWD:
+        folder = os.path.basename(CWD)
+    elif WS_CURRENT_DIR:
+        folder = os.path.basename(WS_CURRENT_DIR)
+    branch = WT_BRANCH
+    if not branch and CWD:
+        try:
+            res = subprocess.run(
+                ["git", "-C", CWD, "rev-parse", "--abbrev-ref", "HEAD"],
+                stdout=subprocess.PIPE, stderr=subprocess.DEVNULL, text=True,
+            )
+            branch = res.stdout.rstrip() if res.returncode == 0 else ""
+        except Exception:
+            branch = ""
+
+    out = []
+
+    # ===== LINE 1 — Identity =====
+    if should_show("IDENTITY", cfg["identity"]):
+        l1 = ""
+        if folder:
+            l1 = "%s%s%s%s" % (BOLD, BLUE, folder, RST)
+            if branch:
+                l1 += " %s⎇%s%s%s%s" % (DIM, RST, CYAN, branch, RST)
+            l1 += " "
+        if WT_NAME:
+            l1 += "%swt:%s%s%s%s " % (DIM, RST, MAGENTA, WT_NAME, RST)
+        elif SESSION_NAME:
+            l1 += "%ssession:%s%s%s%s " % (DIM, RST, MAGENTA, SESSION_NAME, RST)
+        elif AGENT_NAME:
+            l1 += "%sagent:%s%s%s%s " % (DIM, RST, MAGENTA, AGENT_NAME, RST)
+        if MODEL_NAME:
+            l1 += "%s%s%s%s" % (BOLD, CYAN, MODEL_NAME, RST)
+        elif MODEL_ID:
+            l1 += "%s%s%s%s" % (BOLD, CYAN, MODEL_ID, RST)
+        if EFFORT:
+            l1 += " %seffort:%s%s" % (DIM, RST, EFFORT)
+        if THINKING is True:
+            l1 += " %sthinking:on%s" % (YELLOW, RST)
+        if OUTPUT_STYLE and OUTPUT_STYLE != "default":
+            l1 += " %sstyle:%s%s" % (DIM, RST, OUTPUT_STYLE)
+        if VIM_MODE:
+            l1 += " %s[%s]%s" % (DIM, VIM_MODE, RST)
+        if l1:
+            out.append(l1)
+
+    # ===== LINE 2 — Context window =====
+    if should_show("CONTEXT", cfg["context"]):
+        if gte(CTX_PCT, 75):
+            bar_color = RED
+        elif gte(CTX_PCT, 50):
+            bar_color = YELLOW
+        else:
+            bar_color = GREEN
+
+        filled = math.floor((math.trunc(float(CTX_PCT)) + 4) / 5)
+        if filled > 20:
+            filled = 20
+        empty = 20 - filled
+        bar = ("▓" * filled) + ("░" * empty)
+
+        l2 = "%s%s%s %s%s%%%s%s/%s%s" % (
+            bar_color, bar, RST, bar_color, num_str(CTX_PCT), RST, DIM, fmt_k(CTX_SIZE), RST,
+        )
+        l2 += " %sin:%s%s%s%s %sout:%s%s%s%s %scache_cr:%s%s %scache_rd:%s%s" % (
+            DIM, RST, BLUE, fmt_k(CUR_IN), RST,
+            DIM, RST, MAGENTA, fmt_k(CUR_OUT), RST,
+            DIM, RST, fmt_k(CACHE_CREATE),
+            DIM, RST, fmt_k(CACHE_READ),
+        )
+        if EXCEEDS_200K is True:
+            l2 += " %s⚠ 200k+%s" % (YELLOW, RST)
+        out.append(l2)
+
+    # ===== LINE 3 — Cost =====
+    if should_show("COST", cfg["cost"]):
+        l3 = "%s%s%s %sdur:%s%s %sapi:%s%s %s+%s%s %s-%s%s %slines%s" % (
+            GREEN, fmt_cost(COST), RST,
+            DIM, RST, fmt_dur(DURATION),
+            DIM, RST, fmt_dur(API_DURATION),
+            GREEN, num_str(LINES_ADD), RST,
+            RED, num_str(LINES_DEL), RST,
+            DIM, RST,
+        )
+        out.append(l3)
+
+    # ===== LINE 4 — Rate limits =====
+    if should_show("RATE_LIMITS", cfg["rate_limits"]):
+        has_5h = RATE_5H_PCT != "" and str(RATE_5H_RESETS) != "0"
+        has_7d = RATE_7D_PCT != "" and str(RATE_7D_RESETS) != "0"
+        if has_5h or has_7d:
+            l4 = ""
+            if has_5h:
+                if gte(RATE_5H_PCT, 80):
+                    c5 = RED
+                elif gte(RATE_5H_PCT, 60):
+                    c5 = YELLOW
+                else:
+                    c5 = GREEN
+                l4 = "%s5h:%s%s%s%%%s %s(resets in %s)%s" % (
+                    DIM, RST, c5, num_str(RATE_5H_PCT), RST, DIM, fmt_rel_time(RATE_5H_RESETS), RST,
+                )
+            if has_7d:
+                if gte(RATE_7D_PCT, 80):
+                    c7 = RED
+                elif gte(RATE_7D_PCT, 60):
+                    c7 = YELLOW
+                else:
+                    c7 = GREEN
+                seg7 = "%s7d:%s%s%s%%%s %s(resets in %s)%s" % (
+                    DIM, RST, c7, num_str(RATE_7D_PCT), RST, DIM, fmt_rel_time(RATE_7D_RESETS), RST,
+                )
+                l4 = ("%s %s|%s %s" % (l4, DIM, RST, seg7)) if l4 else seg7
+            out.append(l4)
+
+    # ===== LINE 5 — Repo / PR =====
+    if should_show("REPO_PR", cfg["repo_pr"]):
+        l5 = ""
+        if WS_REPO_HOST:
+            l5 = "%s%s/%s/%s%s" % (DIM, WS_REPO_HOST, WS_REPO_OWNER, WS_REPO_NAME, RST)
+        if PR_NUMBER != "":
+            pr_seg = "%sPR%s %s#%s%s" % (DIM, RST, CYAN, num_str(PR_NUMBER), RST)
+            if PR_REVIEW_STATE:
+                pr_seg += " %s%s%s" % (DIM, PR_REVIEW_STATE, RST)
+            if PR_URL:
+                pr_seg += " %s%s%s" % (BLUE, PR_URL, RST)
+            l5 = ("%s %s|%s %s" % (l5, DIM, RST, pr_seg)) if l5 else pr_seg
+        if l5:
+            out.append(l5)
+
+    # ===== LINE 6 — Worktree =====
+    if should_show("WORKTREE", cfg["worktree"]):
+        if WT_NAME:
+            l6 = "%s%s%s %s@%s %s%s%s" % (MAGENTA, WT_NAME, RST, DIM, RST, CYAN, WT_BRANCH, RST)
+            if WT_ORIG_BRANCH and WT_ORIG_BRANCH != WT_BRANCH:
+                l6 += " %s(was: %s)%s" % (DIM, WT_ORIG_BRANCH, RST)
+            wt_path_disp = WT_PATH
+            if len(WT_PATH) > 60:
+                wt_path_disp = "..." + WT_PATH[-57:]
+            l6 += " %s%s%s" % (DIM, wt_path_disp, RST)
+            out.append(l6)
+
+    sys.stdout.write(("\n".join(out) + "\n") if out else "")
+
+
+try:
+    main()
+except Exception:
+    # Statusline must never error to stdout.
+    sys.exit(0)