blun-king-cli 6.0.0 → 6.1.0

package/bin/blun.js

@@ -1,100 +1,134 @@
 #!/usr/bin/env node
-const path = require('path');
-const pkg = require('../package.json');
+const path = require("path");
+const pkg = require("../package.json");
 
 const args = process.argv.slice(2);
-const command = args[0] || 'chat';
+const command = args[0] || "chat";
 
 async function main() {
-  const ui = require('../lib/ui');
-  const auth = require('../lib/auth');
-  const client = require('../lib/client');
+  const ui = require("../lib/ui");
+  const auth = require("../lib/auth");
+  const client = require("../lib/client");
+  const workspace = require("../lib/workspace");
 
   switch (command) {
-    case 'version':
-    case '--version':
-    case '-v':
-      console.log('@blun/cli v' + pkg.version);
+    case "version":
+    case "--version":
+    case "-v":
+      console.log("@blun/cli v" + pkg.version);
       break;
 
-    case 'help':
-    case '--help':
-    case '-h':
+    case "help":
+    case "--help":
+    case "-h":
       ui.renderBanner();
-      ui.renderBox('BLUN CLI Commands', [
-        'blun              Start interactive chat (default)',
-        'blun chat          Interactive chat mode',
-        'blun ask "..."     One-shot question',
-        'blun login         Device login flow',
-        'blun logout        Remove stored credentials',
-        'blun auth --api-key KEY   Authenticate with API key',
-        'blun status        Show system health',
-        'blun version       Show version',
-        'blun help          Show this help'
-      ].join('\n'), 'cyan');
+      ui.renderBox("BLUN CLI Commands", [
+        "blun              Start interactive chat (default)",
+        "blun chat          Interactive chat mode",
+        "blun ask \"...\"     One-shot question",
+        "blun login         Device login flow",
+        "blun logout        Remove stored credentials",
+        "blun auth --api-key KEY   Authenticate with API key",
+        "blun status        Show system health",
+        "blun init           Initialize BLUN project in current dir",
+        "blun version       Show version",
+        "blun help          Show this help"
+      ].join("\n"), "cyan");
       break;
 
-    case 'login':
-      try { await auth.login(); } catch (e) { ui.renderResponse(e.message, 'error'); process.exit(1); }
+    case "login":
+      try { await auth.login(); } catch (e) { ui.renderResponse(e.message, "error"); process.exit(1); }
       break;
 
-    case 'logout':
+    case "logout":
       auth.logout();
       break;
 
-    case 'auth':
-      if (args[1] === '--api-key' && args[2]) {
-        try { await auth.loginWithApiKey(args[2]); } catch (e) { ui.renderResponse(e.message, 'error'); process.exit(1); }
+    case "auth":
+      if (args[1] === "--api-key" && args[2]) {
+        try { await auth.loginWithApiKey(args[2]); } catch (e) { ui.renderResponse(e.message, "error"); process.exit(1); }
       } else {
-        console.error('Usage: blun auth --api-key <KEY>');
+        console.error("Usage: blun auth --api-key <KEY>");
         process.exit(1);
       }
       break;
 
-    case 'ask': {
-      const question = args.slice(1).join(' ');
-      if (!question) { console.error('Usage: blun ask "your question"'); process.exit(1); }
+    case "ask": {
+      const question = args.slice(1).join(" ");
+      if (!question) { console.error("Usage: blun ask \"your question\""); process.exit(1); }
       try {
         auth.ensureAuth();
-        const spinner = ui.renderSpinner('Thinking...');
+        const spinner = ui.renderSpinner("Thinking...");
         const res = await client.sendMessage(question);
         ui.stopSpinner(true);
         const text = res.data?.response || res.data?.text || res.data?.message || JSON.stringify(res.data);
-        ui.renderResponse(text, 'agent');
+        ui.renderResponse(text, "agent");
       } catch (e) {
-        ui.renderResponse(e.message, 'error');
+        ui.renderResponse(e.message, "error");
         process.exit(1);
       }
       break;
     }
 
-    case 'status':
+    case "status":
       try {
-        const spinner = ui.renderSpinner('Checking health...');
+        const spinner = ui.renderSpinner("Checking health...");
         const res = await client.getHealth();
         ui.stopSpinner(true);
-        if (res.status === 200 && typeof res.data === 'object') {
+        if (res.status === 200 && typeof res.data === "object") {
           const d = res.data;
           ui.renderTable(
-            ['Metric', 'Value'],
-            Object.entries(d).map(([k, v]) => [k, typeof v === 'object' ? JSON.stringify(v) : String(v)])
+            ["Metric", "Value"],
+            Object.entries(d).map(([k, v]) => [k, typeof v === "object" ? JSON.stringify(v) : String(v)])
           );
         } else {
-          ui.renderResponse('Health check returned: ' + res.status, 'error');
+          ui.renderResponse("Health check returned: " + res.status, "error");
         }
       } catch (e) {
-        ui.renderResponse('Health check failed: ' + e.message, 'error');
+        ui.renderResponse("Health check failed: " + e.message, "error");
         process.exit(1);
       }
       break;
 
-    case 'chat':
+    case "init":
+      workspace.initProject(process.cwd());
+      break;
+
+    case "chat":
     default: {
-      const { startChat } = require('../lib/chat');
+      // Determine initial folder: path arg or cwd
+      let initialFolder = process.cwd();
+      if (command !== "chat" && command !== undefined) {
+        // default case — command might be a path
+        try {
+          const fs = require("fs");
+          const resolved = path.resolve(command);
+          if (fs.existsSync(resolved) && fs.statSync(resolved).isDirectory()) {
+            initialFolder = resolved;
+          }
+        } catch {}
+      }
+      // Also check args[1] for "blun chat /path"
+      if (args[1]) {
+        try {
+          const fs = require("fs");
+          const resolved = path.resolve(args[1]);
+          if (fs.existsSync(resolved) && fs.statSync(resolved).isDirectory()) {
+            initialFolder = resolved;
+          }
+        } catch {}
+      }
+
+      // Workspace trust flow
+      const trustedFolder = await workspace.selectFolder(initialFolder);
+      process.chdir(trustedFolder);
+      workspace.loadProjectConfig(trustedFolder);
+
+      const { startChat } = require("../lib/chat");
       await startChat();
       break;
     }
   }
 }
 
-main().catch((e) => { console.error('Fatal:', e.message); process.exit(1); });
+main().catch((e) => { console.error("Fatal:", e.message); process.exit(1); });

package/lib/workspace.js

@@ -0,0 +1,110 @@
+const fs = require("fs");
+const path = require("path");
+const os = require("os");
+const readline = require("readline");
+const chalk = require("chalk");
+
+const TRUST_FILE = path.join(os.homedir(), ".blun", "trusted-folders.json");
+
+function loadTrusted() {
+  try {
+    return JSON.parse(fs.readFileSync(TRUST_FILE, "utf8"));
+  } catch {
+    return [];
+  }
+}
+
+function saveTrusted(folders) {
+  const dir = path.dirname(TRUST_FILE);
+  if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+  fs.writeFileSync(TRUST_FILE, JSON.stringify(folders, null, 2));
+}
+
+function isTrusted(folder) {
+  return loadTrusted().includes(path.resolve(folder));
+}
+
+function trustFolder(folder) {
+  const abs = path.resolve(folder);
+  const trusted = loadTrusted();
+  if (!trusted.includes(abs)) {
+    trusted.push(abs);
+    saveTrusted(trusted);
+  }
+  return abs;
+}
+
+function prompt(question) {
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise((resolve) => rl.question(question, (a) => { rl.close(); resolve(a.trim()); }));
+}
+
+async function selectFolder(initialFolder) {
+  let folder = path.resolve(initialFolder);
+
+  while (true) {
+    console.log();
+    console.log(chalk.bold.cyan("  Workspace: ") + chalk.white(folder));
+    console.log();
+
+    if (isTrusted(folder)) {
+      console.log(chalk.green("  ✓ Working in " + folder + " (trusted)"));
+      return folder;
+    }
+
+    console.log(chalk.yellow("  Do you trust the files in this folder?"));
+    console.log();
+    console.log("    " + chalk.bold("1)") + " Yes, I trust this folder");
+    console.log("    " + chalk.bold("2)") + " No, exit");
+    console.log("    " + chalk.bold("3)") + " Choose a different folder");
+    console.log();
+
+    const answer = await prompt(chalk.cyan("  Your choice (1/2/3): "));
+
+    if (answer === "1") {
+      trustFolder(folder);
+      console.log(chalk.green("\n  ✓ Folder trusted and saved.\n"));
+      return folder;
+    } else if (answer === "2") {
+      console.log(chalk.red("\n  Exiting.\n"));
+      process.exit(0);
+    } else if (answer === "3") {
+      const newPath = await prompt(chalk.cyan("  Enter folder path: "));
+      if (!newPath) continue;
+      const resolved = path.resolve(newPath);
+      if (!fs.existsSync(resolved)) {
+        console.log(chalk.red("  Folder does not exist: " + resolved));
+        continue;
+      }
+      folder = resolved;
+    } else {
+      console.log(chalk.red("  Invalid choice."));
+    }
+  }
+}
+
+function loadProjectConfig(folder) {
+  const mdPath = path.join(folder, "BLUN.md");
+  if (fs.existsSync(mdPath)) {
+    const content = fs.readFileSync(mdPath, "utf8");
+    console.log(chalk.green("  Found project config: BLUN.md"));
+    return content;
+  }
+  return null;
+}
+
+function initProject(folder) {
+  const dir = path.resolve(folder);
+  const blunDir = path.join(dir, ".blun");
+  if (!fs.existsSync(blunDir)) fs.mkdirSync(blunDir, { recursive: true });
+
+  const mdPath = path.join(dir, "BLUN.md");
+  if (!fs.existsSync(mdPath)) {
+    fs.writeFileSync(mdPath, `# Project Name\n\nYour project name here.\n\n## Description\n\nBrief description of the project.\n\n## Rules\n\n- Add project-specific rules here\n\n## Context\n\n- Add relevant context for the AI assistant\n`);
+  }
+  console.log(chalk.green("  ✓ Initialized BLUN project in " + dir));
+  console.log(chalk.dim("    Created: .blun/ directory"));
+  console.log(chalk.dim("    Created: BLUN.md template"));
+}
+
+module.exports = { trustFolder, isTrusted, selectFolder, initProject, loadProjectConfig };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blun-king-cli",
-  "version": "6.0.0",
+  "version": "6.1.0",
   "description": "BLUN AI Assistant - Command Line Interface",
   "bin": { "blun": "./bin/blun.js" },
   "files": ["bin/", "lib/", "README.md"],