@vtstech/pi-status 1.1.4-dev → 1.1.5

package/README.md

@@ -18,12 +18,12 @@ CPU/RAM/Swap are only shown when using a local Ollama provider (not for cloud/re
 
 **Example (local Ollama):**
 ```
-CtxMax:41k RespMax:16.4k Resp 2m3s CPU 12% RAM 2.2G/15.1G Prompt: 2840 chr 393 tok pi:0.66.1
+CtxMax:41k RespMax:16.4k Resp 2m3s CPU 12% RAM 2.2G/15.1G SEC:MAX Prompt: 2840 chr 393 tok pi:0.66.1
 ```
 
-**Example (cloud provider):**
+**Example (cloud provider, basic mode):**
 ```
-CtxMax:128k RespMax:16.4k Resp 1m22s Prompt: 2840 chr 393 tok pi:0.66.1
+CtxMax:128k RespMax:16.4k Resp 1m22s SEC:BASIC Prompt: 2840 chr 393 tok pi:0.66.1
 ```
 
 ## Status Slots
@@ -39,7 +39,7 @@ Slots are updated every 5 seconds (1 second for active tool timing). Render orde
 | **RAM** | Used/total system memory | Local Ollama only |
 | **Swap** | Used/total swap space | Local only, when active |
 | **Generation params** | Temperature, top_p, top_k, num_predict, context size, reasoning_effort (dimmed) | After first provider request |
-| **SEC** | Session-scoped blocked tool count + 3s flash on block event | When blocks occur |
+| **SEC** | Security mode indicator (`SEC:BASIC`/`SEC:MAX`) + session-scoped blocked count + 3s flash on block event | Always shown |
 | **Active tool** | Live elapsed timer with `>` indicator | While a tool is running |
 | **Prompt** | System prompt size as `chars chr tokens tok` | After first agent start |
 | **Pi version** | `pi:0.66.1` (dimmed, always last) | Always shown |

package/package.json

@@ -1,11 +1,9 @@
 {
   "name": "@vtstech/pi-status",
-  "version": "1.1.4-dev",
+  "version": "1.1.5",
   "description": "System monitor / status bar extension for Pi Coding Agent",
   "main": "status.js",
-  "keywords": [
-    "pi-extensions"
-  ],
+  "keywords": ["pi-extensions"],
   "license": "MIT",
   "access": "public",
   "type": "module",
@@ -16,14 +14,12 @@
     "url": "https://github.com/VTSTech/pi-coding-agent"
   },
   "dependencies": {
-    "@vtstech/pi-shared": "1.1.4"
+    "@vtstech/pi-shared": "1.1.5"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"
   },
   "pi": {
-    "extensions": [
-      "./status.js"
-    ]
+    "extensions": ["./status.js"]
   }
 }

package/status.js

@@ -5,6 +5,7 @@ import os from "node:os";
 import { getOllamaBaseUrl, fetchModelContextLength, readModelsJson } from "@vtstech/pi-shared/ollama";
 import { fmtBytes, fmtDur } from "@vtstech/pi-shared/format";
 import { debugLog } from "@vtstech/pi-shared/debug";
+import { getSecurityMode } from "@vtstech/pi-shared/security";
 var STATUS_UPDATE_INTERVAL_MS = 5e3;
 var TOOL_TIMER_INTERVAL_MS = 1e3;
 function status_temp_default(pi) {
@@ -165,13 +166,14 @@ function status_temp_default(pi) {
     } else {
       ctxUi.setStatus("status-params", void 0);
     }
+    const secMode = getSecurityMode();
     const now = Date.now();
     if (securityFlashTool && now < securityFlashUntil) {
-      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))} ${dim2("(blocked: " + securityFlashTool + ")")}`);
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))} ${dim2("(" + secMode.toUpperCase() + ")")} ${dim2("(blocked: " + securityFlashTool + ")")}`);
     } else if (blockedCount > 0) {
-      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))}`);
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(String(blockedCount))} ${dim2("(" + secMode.toUpperCase() + ")")}`);
     } else {
-      ctxUi.setStatus("status-sec", void 0);
+      ctxUi.setStatus("status-sec", `${dim2("SEC:")}${green2(secMode.toUpperCase())}`);
     }
     if (activeTool && activeToolStart > 0) {
       const elapsed = performance.now() - activeToolStart;
@@ -259,15 +261,42 @@ function status_temp_default(pi) {
   });
   pi.on("before_provider_request", (event) => {
     lastPayload = event.payload;
+    measurePromptFromPayload(lastPayload);
   });
+  function measurePromptFromPayload(payload) {
+    if (!payload || cachedPromptText) return;
+    const theme = ctxTheme;
+    const dim2 = (s) => theme?.fg?.("dim", s) ?? s;
+    const green2 = (s) => theme?.fg?.("success", s) ?? s;
+    try {
+      const messages = payload.messages;
+      if (!messages?.length) return;
+      const sysMsg = messages.find((m) => m.role === "system") ?? messages[0];
+      if (!sysMsg?.content) return;
+      const chr = sysMsg.content.length;
+      const tok = sysMsg.content.split(/\s+/).filter(Boolean).length;
+      cachedPromptText = `${dim2("Prompt:")} ${green2(`${chr} chr ${tok} tok`)}`;
+      debugLog("status", `system prompt measured from payload: ${chr} chars, ~${tok} words`);
+      flushStatus();
+    } catch (err) {
+      debugLog("status", "failed to measure prompt from payload", err);
+    }
+  }
   pi.on("agent_start", async (_event, ctx) => {
     agentStartTime = performance.now();
     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 {
+      if (prompt) {
+        const chr = prompt.length;
+        const tok = prompt.split(/\s+/).filter(Boolean).length;
+        cachedPromptText = `${dim("Prompt:")} ${green(`${chr} chr ${tok} tok`)}`;
+        debugLog("status", `system prompt measured via getSystemPrompt(): ${chr} chars, ~${tok} words`);
+      }
+    } catch (err) {
+      debugLog("status", "getSystemPrompt() not available, will measure from payload", err);
+    }
+    if (!cachedPromptText && lastPayload) {
+      measurePromptFromPayload(lastPayload);
     }
     flushStatus();
   });