@vtstech/pi-status 1.1.2 → 1.1.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vtstech/pi-status",
-  "version": "1.1.2",
+  "version": "1.1.4",
   "description": "System monitor / status bar extension for Pi Coding Agent",
   "main": "status.js",
   "keywords": ["pi-extensions"],
@@ -14,7 +14,7 @@
     "url": "https://github.com/VTSTech/pi-coding-agent"
   },
   "dependencies": {
-    "@vtstech/pi-shared": "1.1.2"
+    "@vtstech/pi-shared": "1.1.4"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"

package/status.js

@@ -1,33 +1,34 @@
 // .build-npm/status/status.temp.ts
+import * as fs from "node:fs";
+import { execSync } from "node:child_process";
 import os from "node:os";
-import { execSync as gitExecSync } from "node:child_process";
 import { getOllamaBaseUrl, fetchModelContextLength, readModelsJson } from "@vtstech/pi-shared/ollama";
 import { fmtBytes, fmtDur } from "@vtstech/pi-shared/format";
+import { debugLog } from "@vtstech/pi-shared/debug";
+var STATUS_UPDATE_INTERVAL_MS = 5e3;
+var TOOL_TIMER_INTERVAL_MS = 1e3;
 function status_temp_default(pi) {
   let lastResponseTime = null;
   let agentStartTime = null;
   let updateInterval = null;
+  let toolTimerInterval = null;
   let currentCtx = null;
   let ctxUi = null;
+  let ctxTheme = null;
   let prevCpuInfo = getCpuSnapshot();
   let lastPayload = null;
-  let tuiRef = null;
-  let gitBranchCache = "";
   let cpuUsage = 0;
   let memUsed = 0;
   let memTotal = 0;
   let swapUsed = 0;
   let swapTotal = 0;
   let hasSwap = false;
-  let ollamaLoaded = "";
   let footerModel = "";
-  let footerThinking = "";
-  let footerCtxPct = "";
   let footerNativeCtx = "";
   let nativeCtxModel = "";
   let isLocalProvider = true;
-  let lastUpstream = 0;
-  let lastDownstream = 0;
+  let versionsText = "";
+  let cachedPromptText = null;
   let securityFlashTool = "";
   let securityFlashUntil = 0;
   let activeTool = "";
@@ -65,18 +66,20 @@ function status_temp_default(pi) {
     return { used, total };
   }
   function getSwap() {
+    if (process.platform !== "linux") {
+      debugLog("status", "swap detection skipped: not a Linux platform");
+      return null;
+    }
     try {
-      const out = gitExecSync("cat /proc/meminfo", { encoding: "utf-8", timeout: 3e3 });
+      const out = fs.readFileSync("/proc/meminfo", "utf-8");
       const swapTotal2 = Number(out.match(/SwapTotal:\s+(\d+)/)?.[1]) * 1024;
       const swapFree = Number(out.match(/SwapFree:\s+(\d+)/)?.[1]) * 1024;
       if (swapTotal2 > 0) return { used: swapTotal2 - swapFree, total: swapTotal2 };
-    } catch {
+    } catch (err) {
+      debugLog("status", "failed to read /proc/meminfo", err);
     }
     return null;
   }
-  let ollamaLoadedCache = "";
-  let ollamaLoadedLastCheck = 0;
-  const OLLAMA_LOADED_INTERVAL = 15e3;
   function detectLocalProvider(modelsJson) {
     const isLocalUrl = (url) => url.includes("localhost") || url.includes("127.0.0.1") || url.includes("0.0.0.0");
     try {
@@ -116,68 +119,70 @@ function status_temp_default(pi) {
     }
     return footerNativeCtx;
   }
-  async function fetchOllamaLoadedModel() {
-    try {
-      const ollamaBase = getOllamaBaseUrl();
-      const res = await fetch(`${ollamaBase}/api/ps`, {
-        signal: AbortSignal.timeout(5e3)
-      });
-      if (!res.ok) return "";
-      const data = await res.json();
-      const models = data?.models || [];
-      if (Array.isArray(models) && models.length > 0) {
-        return models[0].name || models[0].model || "";
-      }
-    } catch {
-    }
-    return "";
-  }
-  function getOllamaLoadedModel() {
-    const now = Date.now();
-    if (now - ollamaLoadedLastCheck < OLLAMA_LOADED_INTERVAL) return ollamaLoadedCache;
-    ollamaLoadedLastCheck = now;
-    fetchOllamaLoadedModel().then((loaded) => {
-      ollamaLoadedCache = loaded;
-    }).catch(() => {
-      ollamaLoadedCache = "";
-    });
-    return ollamaLoadedCache;
-  }
   function extractParams(payload) {
     const params = [];
     if (payload.temperature !== void 0) params.push(`temp:${payload.temperature}`);
     if (payload.top_p !== void 0) params.push(`top_p:${payload.top_p}`);
     if (payload.top_k !== void 0) params.push(`top_k:${payload.top_k}`);
-    if (payload.max_completion_tokens !== void 0) params.push(`max:${payload.max_completion_tokens}`);
-    else if (payload.max_tokens !== void 0) params.push(`max:${payload.max_tokens}`);
     if (payload.num_predict !== void 0) params.push(`predict:${payload.num_predict}`);
     if (payload.num_ctx !== void 0) params.push(`ctx:${payload.num_ctx}`);
     if (payload.reasoning_effort !== void 0) params.push(`think:${payload.reasoning_effort}`);
     return params;
   }
-  function fmtTk(n) {
-    if (n >= 1e3) return `${(n / 1e3).toFixed(1)}k`;
-    return String(n);
-  }
-  function getPwd() {
-    const cwd = process.cwd();
-    if (cwd.startsWith(os.homedir())) return "~" + cwd.slice(os.homedir().length);
-    return cwd;
-  }
-  function getGitBranch() {
-    if (gitBranchCache) return gitBranchCache;
-    try {
-      const branch = gitExecSync("git rev-parse --abbrev-ref HEAD 2>/dev/null", {
-        encoding: "utf-8",
-        timeout: 3e3
-      }).trim();
-      if (branch) gitBranchCache = branch;
-    } catch {
+  function flushStatus() {
+    if (!ctxUi) return;
+    const theme = ctxTheme;
+    const dim2 = (s) => theme?.fg?.("dim", s) ?? s;
+    const green2 = (s) => theme?.fg?.("success", s) ?? s;
+    ctxUi.setStatus("status-cpu", isLocalProvider ? `${dim2("CPU")} ${green2(cpuUsage.toFixed(0) + "%")}` : void 0);
+    ctxUi.setStatus("status-ram", isLocalProvider ? `${dim2("RAM")} ${green2(fmtBytes(memUsed) + "/" + fmtBytes(memTotal))}` : void 0);
+    ctxUi.setStatus(
+      "status-swap",
+      isLocalProvider && hasSwap && swapUsed > 0 ? `${dim2("Swap")} ${green2(fmtBytes(swapUsed) + "/" + fmtBytes(swapTotal))}` : void 0
+    );
+    ctxUi.setStatus(
+      "status-native-ctx",
+      footerNativeCtx ? `${dim2("CtxMax:")}${green2(footerNativeCtx)}` : void 0
+    );
+    if (lastPayload) {
+      const rawMax = lastPayload.max_completion_tokens ?? lastPayload.max_tokens;
+      if (rawMax !== void 0) {
+        const formatted = rawMax >= 1e3 ? `${(rawMax / 1e3).toFixed(rawMax % 1e3 === 0 ? 0 : 1)}k` : String(rawMax);
+        ctxUi.setStatus("status-resp-max", `${dim2("RespMax:")}${green2(formatted)}`);
+      } else {
+        ctxUi.setStatus("status-resp-max", void 0);
+      }
+    } else {
+      ctxUi.setStatus("status-resp-max", void 0);
+    }
+    ctxUi.setStatus(
+      "status-resp",
+      lastResponseTime !== null ? `${dim2("Resp")} ${green2(fmtDur(lastResponseTime))}` : void 0
+    );
+    if (lastPayload) {
+      const params = extractParams(lastPayload);
+      ctxUi.setStatus("status-params", params.length > 0 ? dim2(params.join(" ")) : void 0);
+    } else {
+      ctxUi.setStatus("status-params", void 0);
+    }
+    const now = Date.now();
+    if (securityFlashTool && now < securityFlashUntil) {
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))} ${dim2("(blocked: " + securityFlashTool + ")")}`);
+    } else if (blockedCount > 0) {
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))}`);
+    } else {
+      ctxUi.setStatus("status-sec", void 0);
+    }
+    if (activeTool && activeToolStart > 0) {
+      const elapsed = performance.now() - activeToolStart;
+      ctxUi.setStatus("status-tool", `${green2(">")} ${dim2(activeTool + ":")} ${green2(fmtDur(elapsed))}`);
+    } else {
+      ctxUi.setStatus("status-tool", void 0);
+    }
+    ctxUi.setStatus("system-prompt", cachedPromptText ?? void 0);
+    if (versionsText) {
+      ctxUi.setStatus("status-versions", dim2(versionsText));
     }
-    return gitBranchCache;
-  }
-  function incrementBlockedCount() {
-    blockedCount++;
   }
   function updateMetrics() {
     cpuUsage = getCpuUsage();
@@ -192,154 +197,79 @@ function status_temp_default(pi) {
     } else {
       hasSwap = false;
     }
-    ollamaLoaded = getOllamaLoadedModel();
     const modelsJson = readModelsJson();
     isLocalProvider = modelsJson ? detectLocalProvider(modelsJson) : false;
     if (currentCtx) {
       footerModel = currentCtx.model?.id || "";
-      footerThinking = pi.getThinkingLevel?.() ?? "";
-      const usage = currentCtx.getContextUsage?.();
-      if (usage && usage.contextWindow > 0) {
-        const pctVal = (usage.tokens / usage.contextWindow * 100).toFixed(1);
-        footerCtxPct = `${pctVal}%/${(usage.contextWindow / 1e3).toFixed(0)}k`;
-      } else {
-        footerCtxPct = "";
-      }
       const modelId = currentCtx.model?.id || "";
-      if (modelId && isLocalProvider) {
+      if (modelId) {
         getNativeModelCtx(modelId);
       }
     }
+    flushStatus();
   }
   pi.on("session_start", async (_event, ctx) => {
     currentCtx = ctx;
     ctxUi = ctx.ui;
+    ctxTheme = ctx.ui.theme;
     prevCpuInfo = getCpuSnapshot();
+    try {
+      const out = execSync("pi -v 2>&1", { encoding: "utf-8", timeout: 5e3 }).trim();
+      if (out) versionsText = `pi:${out}`;
+    } catch {
+    }
     updateMetrics();
-    ctx.ui.setFooter((tui, theme, footerData) => {
-      tuiRef = tui;
-      const dim = (s) => theme?.fg?.("dim", s) ?? s;
-      const red = (s) => theme?.fg?.("error", s) ?? s;
-      const yellow = (s) => theme?.fg?.("yellow", s) ?? s;
-      const sep = dim(" \xB7 ");
-      const truncateLine = (line, maxW) => {
-        const ellipsis = dim("...");
-        const visible = line.replace(/\x1b\[[0-9;]*m/g, "");
-        if (visible.length > maxW) {
-          let vis = 0, cut = 0;
-          for (let i = 0; i < line.length && vis < maxW - 3; i++) {
-            if (line[i] === "\x1B") {
-              while (i < line.length && line[i] !== "m") i++;
-            } else {
-              vis++;
-            }
-            cut = i + 1;
-          }
-          return line.slice(0, cut) + ellipsis;
-        }
-        return line;
-      };
-      return {
-        render(width) {
-          const lines = [];
-          let branch = "";
-          try {
-            branch = footerData?.getGitBranch?.() || "";
-          } catch {
-          }
-          if (!branch) branch = getGitBranch();
-          const line1Parts = [];
-          if (footerModel) line1Parts.push(`conf:${footerModel}`);
-          line1Parts.push(getPwd());
-          if (footerThinking && footerThinking !== "off") line1Parts.push(dim(footerThinking));
-          if (isLocalProvider) {
-            line1Parts.push(dim(`CPU ${cpuUsage.toFixed(0)}%`));
-          }
-          let line1 = truncateLine(line1Parts.join(sep), width);
-          lines.push(line1);
-          const line2Parts = [];
-          if (ollamaLoaded) line2Parts.push(`load:${ollamaLoaded}`);
-          if (footerNativeCtx) line2Parts.push(`M:${footerNativeCtx}`);
-          if (footerCtxPct) line2Parts.push(`S:${footerCtxPct}`);
-          if (isLocalProvider) {
-            line2Parts.push(`RAM ${fmtBytes(memUsed)}/${fmtBytes(memTotal)}`);
-            if (hasSwap && swapUsed > 0) {
-              line2Parts.push(`Swap ${fmtBytes(swapUsed)}/${fmtBytes(swapTotal)}`);
-            }
-          }
-          if (lastUpstream > 0 || lastDownstream > 0) {
-            line2Parts.push(dim(`\u2191${fmtTk(lastUpstream)} \u2193${fmtTk(lastDownstream)}`));
-          }
-          if (lastResponseTime !== null) line2Parts.push(`Resp ${fmtDur(lastResponseTime)}`);
-          if (lastPayload) {
-            const params = extractParams(lastPayload);
-            if (params.length > 0) line2Parts.push(...params.map((p) => dim(p)));
-          }
-          const now = Date.now();
-          if (securityFlashTool && now < securityFlashUntil) {
-            line2Parts.push(red(`BLOCKED:${securityFlashTool}`));
-          }
-          if (blockedCount > 0) {
-            line2Parts.push(red(`SEC:${blockedCount}`));
-          }
-          let line2 = truncateLine(line2Parts.join(sep), width);
-          if (line2) lines.push(line2);
-          if (activeTool && activeToolStart > 0) {
-            const elapsed = performance.now() - activeToolStart;
-            lines.push(`${yellow("\u23F3")} ${activeTool}: ${fmtDur(elapsed)}`);
-          }
-          return lines;
-        },
-        invalidate() {
-        },
-        dispose() {
-        }
-      };
-    });
     if (updateInterval) clearInterval(updateInterval);
-    updateInterval = setInterval(() => {
-      updateMetrics();
-      if (tuiRef) tuiRef.requestRender();
-    }, 3e3);
+    updateInterval = setInterval(updateMetrics, STATUS_UPDATE_INTERVAL_MS);
   });
-  pi.on("session_shutdown", async () => {
-    if (updateInterval) clearInterval(updateInterval);
-    updateInterval = null;
-    tuiRef = null;
-    if (ctxUi) {
-      ctxUi.setFooter(void 0);
-      ctxUi = null;
+  pi.on("session_shutdown", async (_event, ctx) => {
+    if (updateInterval) {
+      clearInterval(updateInterval);
+      updateInterval = null;
     }
+    if (toolTimerInterval) {
+      clearInterval(toolTimerInterval);
+      toolTimerInterval = null;
+    }
+    ctxUi = null;
     currentCtx = null;
+    const ui = ctx?.ui;
+    if (ui) {
+      ui.setStatus("status-cpu", void 0);
+      ui.setStatus("status-ram", void 0);
+      ui.setStatus("status-swap", void 0);
+      ui.setStatus("status-native-ctx", void 0);
+      ui.setStatus("status-resp", void 0);
+      ui.setStatus("status-resp-max", void 0);
+      ui.setStatus("status-params", void 0);
+      ui.setStatus("system-prompt", void 0);
+      ui.setStatus("status-sec", void 0);
+      ui.setStatus("status-tool", void 0);
+      ui.setStatus("status-versions", void 0);
+    }
     securityFlashTool = "";
     securityFlashUntil = 0;
     activeTool = "";
     activeToolStart = 0;
     blockedCount = 0;
-    lastUpstream = 0;
-    lastDownstream = 0;
+    lastResponseTime = null;
+    lastPayload = null;
+    versionsText = "";
+    cachedPromptText = null;
   });
   pi.on("before_provider_request", (event) => {
     lastPayload = event.payload;
   });
-  function captureUsage(event) {
-    if (event?.message?.role !== "assistant") return;
-    const usage = event?.message?.usage ?? // normalised Pi usage
-    event?.usage ?? // alternative path
-    null;
-    if (!usage) return;
-    const inp = usage.input ?? usage.promptTokens ?? usage.prompt_tokens;
-    const out = usage.output ?? usage.completionTokens ?? usage.completion_tokens;
-    if (inp != null) lastUpstream = inp;
-    if (out != null) lastDownstream = out;
-    if (tuiRef) tuiRef.requestRender();
-  }
-  pi.on("message_end", captureUsage);
-  pi.on("turn_end", captureUsage);
-  pi.on("agent_start", async () => {
+  pi.on("agent_start", async (_event, ctx) => {
     agentStartTime = performance.now();
-    lastUpstream = 0;
-    lastDownstream = 0;
+    try {
+      const prompt = ctx.getSystemPrompt();
+      const chr = prompt.length;
+      const tok = prompt.split(/\s+/).filter(Boolean).length;
+      cachedPromptText = `${dim("Prompt:")} ${green(`${chr} chr ${tok} tok`)}`;
+    } catch {
+    }
+    flushStatus();
   });
   pi.on("agent_end", async () => {
     if (agentStartTime !== null) {
@@ -348,29 +278,40 @@ function status_temp_default(pi) {
     }
     activeTool = "";
     activeToolStart = 0;
+    stopToolTimer();
     updateMetrics();
-    if (tuiRef) tuiRef.requestRender();
   });
+  function startToolTimer() {
+    if (toolTimerInterval) return;
+    toolTimerInterval = setInterval(flushStatus, TOOL_TIMER_INTERVAL_MS);
+  }
+  function stopToolTimer() {
+    if (toolTimerInterval) {
+      clearInterval(toolTimerInterval);
+      toolTimerInterval = null;
+    }
+  }
   pi.on("tool_call", (event) => {
     if (!event) return;
     const isBlocked = event.blocked === true || event.blocked === "true" || event.result?.blocked === true || event.error?.includes("blocked");
     if (isBlocked) {
       securityFlashTool = event.tool ?? event.name ?? "unknown";
       securityFlashUntil = Date.now() + 3e3;
-      incrementBlockedCount();
-      if (tuiRef) tuiRef.requestRender();
+      blockedCount++;
+      flushStatus();
     }
   });
   pi.on("tool_execution_start", (event) => {
     if (!event) return;
     activeTool = event.tool ?? event.name ?? "tool";
     activeToolStart = performance.now();
-    if (tuiRef) tuiRef.requestRender();
+    startToolTimer();
   });
   pi.on("tool_execution_end", () => {
     activeTool = "";
     activeToolStart = 0;
-    if (tuiRef) tuiRef.requestRender();
+    stopToolTimer();
+    flushStatus();
   });
 }
 export {