blun-king-cli 2.5.2 → 2.6.1

package/bin/blun.js

@@ -495,9 +495,17 @@ async function sendChat(message) {
     chatHistory.push({ role: "user", content: message });
     chatHistory.push({ role: "assistant", content: resp.data.answer });
 
+    // Token tracking
+    var inTok = Math.ceil(message.length / 3.5);
+    var outTok = Math.ceil((resp.data.answer || "").length / 3.5);
+    sessionCost.requests++;
+    sessionCost.inputTokensEst += inTok;
+    sessionCost.outputTokensEst += outTok;
+
     var q = resp.data.quality || {};
+    var tokInfo = C.dim + (inTok + outTok) + " tok" + C.reset;
     var meta = resp.data.task_type + "/" + resp.data.role + " " + BOX.dot + " score: " + (q.score || "?") +
-      (q.retried ? " " + BOX.dot + " retried" : "");
+      (q.retried ? " " + BOX.dot + " retried" : "") + " " + BOX.dot + " " + tokInfo;
     printAnswer(resp.data.answer, meta);
 
     // Auto Memory Dream Mode
@@ -2258,36 +2266,82 @@ async function main() {
       console.log("");
       console.log(C.yellow + "  " + BOX.bot + " Working in: " + C.brightWhite + process.cwd() + C.reset);
       console.log(C.gray + "  Do you trust this folder? BLUN King will read/write files here." + C.reset);
-      console.log(C.gray + "  [y] Yes, trust  [n] Choose another folder  [a] Always trust" + C.reset);
+      console.log("");
+
+      var trustOptions = [
+        { label: "Yes, trust this folder", action: "trust" },
+        { label: "Always trust this folder", action: "always" },
+        { label: "Choose another folder", action: "choose" }
+      ];
+      var trustSel = 0;
+
+      // Arrow key menu for trust selection
+      var trustChoice = await new Promise(function(resolve) {
+        function renderTrustMenu() {
+          // Move up and clear previous menu
+          if (trustSel >= 0) process.stdout.write("\x1b[3A\r");
+          trustOptions.forEach(function(opt, i) {
+            var prefix = i === trustSel ? C.green + C.bold + "  \u276F " : "    ";
+            var color = i === trustSel ? C.brightWhite + C.bold : C.gray;
+            process.stdout.write("\x1b[2K" + prefix + color + (i + 1) + ". " + opt.label + C.reset + "\n");
+          });
+        }
+        // Initial render
+        console.log(""); console.log(""); console.log("");
+        renderTrustMenu();
 
-      var trustAnswer = await new Promise(function(resolve) {
         process.stdin.setRawMode(true);
         process.stdin.resume();
         process.stdin.setEncoding("utf8");
-        process.stdin.once("data", function(key) {
-          process.stdin.setRawMode(false);
-          resolve(key.toLowerCase());
-        });
+        function onKey(key) {
+          if (key === "\x1b[A") { trustSel = Math.max(0, trustSel - 1); renderTrustMenu(); return; }
+          if (key === "\x1b[B") { trustSel = Math.min(2, trustSel + 1); renderTrustMenu(); return; }
+          if (key === "1") { trustSel = 0; process.stdin.removeListener("data", onKey); process.stdin.setRawMode(false); resolve(trustOptions[0].action); return; }
+          if (key === "2") { trustSel = 1; process.stdin.removeListener("data", onKey); process.stdin.setRawMode(false); resolve(trustOptions[1].action); return; }
+          if (key === "3") { trustSel = 2; process.stdin.removeListener("data", onKey); process.stdin.setRawMode(false); resolve(trustOptions[2].action); return; }
+          if (key === "\r" || key === "\n") {
+            process.stdin.removeListener("data", onKey);
+            process.stdin.setRawMode(false);
+            resolve(trustOptions[trustSel].action);
+            return;
+          }
+        }
+        process.stdin.on("data", onKey);
       });
 
-      if (trustAnswer === "a") {
+      if (trustChoice === "always") {
         trustedDirs.push(process.cwd());
         fs.writeFileSync(trustFile, JSON.stringify(trustedDirs, null, 2));
         printSuccess("Folder trusted permanently.");
-      } else if (trustAnswer === "n") {
-        var rlDir = readline.createInterface({ input: process.stdin, output: process.stdout });
-        var newDir = await new Promise(function(resolve) {
-          rlDir.question(C.brightBlue + "  Enter path: " + C.reset, resolve);
-        });
-        rlDir.close();
-        if (newDir.trim() && fs.existsSync(newDir.trim())) {
-          config.workdir = newDir.trim();
+      } else if (trustChoice === "choose") {
+        // Open Windows folder picker if available
+        var newDir = null;
+        if (process.platform === "win32") {
+          try {
+            printInfo("Opening folder picker...");
+            var psCmd = 'powershell -NoProfile -Command "Add-Type -AssemblyName System.Windows.Forms; $f = New-Object System.Windows.Forms.FolderBrowserDialog; $f.Description = \'Choose BLUN King workspace\'; $f.ShowNewFolderButton = $true; if($f.ShowDialog() -eq \'OK\'){ Write-Output $f.SelectedPath } else { Write-Output \'CANCELLED\' }"';
+            newDir = require("child_process").execSync(psCmd, { encoding: "utf8", timeout: 60000 }).trim();
+            if (newDir === "CANCELLED") newDir = null;
+          } catch(e) { newDir = null; }
+        }
+        // Fallback to text input
+        if (!newDir) {
+          var rlDir = readline.createInterface({ input: process.stdin, output: process.stdout });
+          newDir = await new Promise(function(resolve) {
+            rlDir.question(C.brightBlue + "  Enter path: " + C.reset, resolve);
+          });
+          rlDir.close();
+          newDir = newDir ? newDir.trim() : null;
+        }
+        if (newDir && fs.existsSync(newDir)) {
+          config.workdir = newDir;
           try { fs.writeFileSync(path.join(CONFIG_DIR, "last-workdir.json"), JSON.stringify({ dir: config.workdir, ts: new Date().toISOString() })); } catch(e) {}
-        } else {
+          printSuccess("Workspace: " + config.workdir);
+        } else if (newDir) {
           printError("Invalid path, using current directory.");
         }
       }
-      // y or anything else = trust for this session
+      // trust = trust for this session only
       console.log("");
     }
   }
@@ -2396,7 +2450,9 @@ async function main() {
     var permColor = isDangerous ? C.red + C.bold : permMode === "deny" ? C.red : C.yellow;
     var modelName = typeof config.model === "string" ? config.model : (config.model && config.model.name ? config.model.name : "default");
     var wdShort = config.workdir ? path.basename(config.workdir) : "~";
-    lines.push(permColor + "  " + permIcon + " " + permLabel + C.reset + C.dim + "  \u2502  " + C.reset + C.cyan + modelName + C.reset + C.dim + "  \u2502  " + C.reset + C.dim + wdShort + C.reset);
+    var totalTok = sessionCost.inputTokensEst + sessionCost.outputTokensEst;
+    var tokStr = totalTok > 1000 ? (totalTok / 1000).toFixed(1) + "k" : String(totalTok);
+    lines.push(permColor + "  " + permIcon + " " + permLabel + C.reset + C.dim + "  \u2502  " + C.reset + C.cyan + modelName + C.reset + C.dim + "  \u2502  " + C.reset + C.dim + wdShort + C.reset + C.dim + "  \u2502  " + C.reset + C.yellow + tokStr + " tok" + C.reset);
 
     // Menu
     if (menuVisible && menuItems.length > 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blun-king-cli",
-  "version": "2.5.2",
+  "version": "2.6.1",
   "description": "BLUN King CLI — Premium KI Console",
   "bin": {
     "blun": "./bin/blun.js"