@vibegrid/mcp 0.4.0-beta.5 → 0.4.0-beta.6

package/dist/index.js

@@ -177,6 +177,7 @@ function createSchema() {
       agent_type TEXT PRIMARY KEY,
       command TEXT NOT NULL,
       args TEXT NOT NULL DEFAULT '[]',
+      headless_args TEXT,
       fallback_command TEXT,
       fallback_args TEXT
     );
@@ -414,6 +415,18 @@ function migrateSchema(d) {
     })();
     logger_default.info("[database] migrated schema to version 4 (worktree name)");
   }
+  if (version < 5) {
+    d.transaction(() => {
+      const agentCols = d.prepare("PRAGMA table_info(agent_commands)").all();
+      if (!agentCols.some((c) => c.name === "headless_args")) {
+        d.exec("ALTER TABLE agent_commands ADD COLUMN headless_args TEXT");
+      }
+      d.prepare(
+        "INSERT OR REPLACE INTO schema_meta (key, value) VALUES ('schema_version', '5')"
+      ).run();
+    })();
+    logger_default.info("[database] migrated schema to version 5 (headless args)");
+  }
 }
 function verifySchema(d) {
   const expectedByTable = {
@@ -443,6 +456,12 @@ function verifySchema(d) {
         ddl: "ALTER TABLE sessions ADD COLUMN sort_order INTEGER NOT NULL DEFAULT 0"
       },
       { column: "worktree_name", ddl: "ALTER TABLE sessions ADD COLUMN worktree_name TEXT" }
+    ],
+    agent_commands: [
+      {
+        column: "headless_args",
+        ddl: "ALTER TABLE agent_commands ADD COLUMN headless_args TEXT"
+      }
     ]
   };
   for (const [table, columns] of Object.entries(expectedByTable)) {
@@ -541,6 +560,7 @@ function loadAgentCommands(d) {
     result[r.agent_type] = {
       command: r.command,
       args: JSON.parse(r.args),
+      ...r.headless_args != null && { headlessArgs: JSON.parse(r.headless_args) },
       ...r.fallback_command != null && { fallbackCommand: r.fallback_command },
       ...r.fallback_args != null && { fallbackArgs: JSON.parse(r.fallback_args) }
     };
@@ -617,7 +637,7 @@ function saveConfig(config) {
     }
     d.prepare("DELETE FROM agent_commands").run();
     const insertAgent = d.prepare(
-      "INSERT INTO agent_commands (agent_type, command, args, fallback_command, fallback_args) VALUES (?, ?, ?, ?, ?)"
+      "INSERT INTO agent_commands (agent_type, command, args, headless_args, fallback_command, fallback_args) VALUES (?, ?, ?, ?, ?, ?)"
     );
     if (config.agentCommands) {
       for (const [agentType, cmd] of Object.entries(config.agentCommands)) {
@@ -626,6 +646,7 @@ function saveConfig(config) {
             agentType,
             cmd.command,
             JSON.stringify(cmd.args),
+            cmd.headlessArgs ? JSON.stringify(cmd.headlessArgs) : null,
             cmd.fallbackCommand ?? null,
             cmd.fallbackArgs ? JSON.stringify(cmd.fallbackArgs) : null
           );
@@ -1072,6 +1093,7 @@ var ConfigManager = class {
   changeCallbacks = [];
   dbWatcher = null;
   debounceTimer = null;
+  cachedConfig = null;
   init() {
     initDatabase();
   }
@@ -1080,8 +1102,11 @@ var ConfigManager = class {
     closeDatabase();
   }
   loadConfig() {
+    if (this.cachedConfig) return this.cachedConfig;
     try {
-      return loadConfig();
+      const config = loadConfig();
+      this.cachedConfig = config;
+      return config;
     } catch (err) {
       logger_default.error("[config-manager] loadConfig failed, returning defaults:", err);
       return {
@@ -1101,6 +1126,7 @@ var ConfigManager = class {
   saveConfig(config) {
     try {
       saveConfig(config);
+      this.cachedConfig = null;
     } catch (err) {
       logger_default.error("[config-manager] saveConfig failed:", err);
       throw err;
@@ -1112,6 +1138,7 @@ var ConfigManager = class {
   }
   /** Notify all registered callbacks (call after main-process config mutations) */
   notifyChanged() {
+    this.cachedConfig = null;
     const config = this.loadConfig();
     for (const cb of this.changeCallbacks) {
       cb(config);
@@ -1548,32 +1575,128 @@ import { z as z4 } from "zod";
 import fs3 from "fs";
 import path4 from "path";
 import os3 from "os";
+import { execFileSync } from "child_process";
 import { WebSocket } from "ws";
 var PORT_FILE = path4.join(os3.homedir(), ".vibegrid", "ws-port");
 var TIMEOUT_MS = 1e4;
+var IS_WIN = process.platform === "win32";
+var PORT_FILE_MISSING_MSG = IS_WIN ? `VibeGrid port file not found (~/.vibegrid/ws-port).
+The app may be running but the port file was deleted (e.g. by another instance shutting down).
+To fix, find the VibeGrid process and its listening port:
+  powershell -c "Get-NetTCPConnection -State Listen -OwningProcess (Get-Process VibeGrid).Id | Select LocalPort"
+Then write the WS port to the file:
+  echo {"port":<PORT>,"pid":<PID>} > %USERPROFILE%\\.vibegrid\\ws-port
+Or restart VibeGrid to regenerate it.` : `VibeGrid port file not found (~/.vibegrid/ws-port).
+The app may be running but the port file was deleted (e.g. by another instance shutting down).
+To fix, run:  lsof -iTCP -sTCP:LISTEN -P | grep VibeGrid
+Then write the WS port (the one on *:<port>) to the file:
+  echo '{"port":<PORT>,"pid":<PID>}' > ~/.vibegrid/ws-port
+Or restart VibeGrid to regenerate it.`;
+var PORT_FILE_INVALID_MSG = `VibeGrid port file exists but contains invalid data (~/.vibegrid/ws-port).
+Delete it and restart VibeGrid, or overwrite it with the correct port:
+  ${IS_WIN ? "del %USERPROFILE%\\.vibegrid\\ws-port" : "rm ~/.vibegrid/ws-port"}`;
 var rpcId = 0;
+var cachedPort = null;
+var cacheTimestamp = 0;
+var CACHE_TTL_MS = 5e3;
+var EXEC_OPTS = {
+  encoding: "utf-8",
+  timeout: 5e3,
+  stdio: ["pipe", "pipe", "pipe"]
+};
+function discoverPort() {
+  try {
+    if (IS_WIN) {
+      const taskOut = execFileSync(
+        "tasklist",
+        ["/FI", "IMAGENAME eq VibeGrid.exe", "/FO", "CSV", "/NH"],
+        EXEC_OPTS
+      );
+      const pidMatch = taskOut.match(/"VibeGrid\.exe","(\d+)"/);
+      if (!pidMatch) return null;
+      const pid = pidMatch[1];
+      const lines = execFileSync("netstat", ["-ano"], EXEC_OPTS).split("\n");
+      let fallback = null;
+      for (const line of lines) {
+        if (!line.includes("LISTENING") || !line.trim().endsWith(pid)) continue;
+        const m = line.match(/(?:0\.0\.0\.0|127\.0\.0\.1):(\d+)/);
+        if (!m) continue;
+        if (line.includes("0.0.0.0")) return parseInt(m[1], 10);
+        fallback ??= parseInt(m[1], 10);
+      }
+      return fallback;
+    } else {
+      const lines = execFileSync("lsof", ["-iTCP", "-sTCP:LISTEN", "-P", "-n"], EXEC_OPTS).split(
+        "\n"
+      );
+      let fallback = null;
+      for (const line of lines) {
+        if (!line.includes("VibeGrid")) continue;
+        if (line.includes("*:")) {
+          const m = line.match(/\*:(\d+)/);
+          if (m) return parseInt(m[1], 10);
+        }
+        if (!fallback) {
+          const m = line.match(/:(\d+)\s/);
+          if (m) fallback = parseInt(m[1], 10);
+        }
+      }
+      return fallback;
+    }
+  } catch {
+  }
+  return null;
+}
+function discoverAndHeal() {
+  const now = Date.now();
+  if (cachedPort && now - cacheTimestamp < CACHE_TTL_MS) return { port: cachedPort };
+  const discovered = discoverPort();
+  cachedPort = discovered;
+  cacheTimestamp = now;
+  if (discovered) {
+    try {
+      fs3.mkdirSync(path4.dirname(PORT_FILE), { recursive: true });
+      fs3.writeFileSync(PORT_FILE, JSON.stringify({ port: discovered }), "utf-8");
+    } catch {
+    }
+    return { port: discovered };
+  }
+  return { port: null, reason: "missing" };
+}
 function readPort() {
   try {
     const raw = fs3.readFileSync(PORT_FILE, "utf-8").trim();
-    if (!raw) return null;
+    if (!raw) return { port: null, reason: "invalid" };
     if (raw.startsWith("{")) {
       const parsed = JSON.parse(raw);
-      const port2 = parsed?.port;
-      return typeof port2 === "number" && Number.isFinite(port2) && port2 > 0 ? port2 : null;
+      const p2 = parsed?.port;
+      const pid = parsed?.pid;
+      if (typeof p2 !== "number" || !Number.isFinite(p2) || p2 <= 0) {
+        return { port: null, reason: "invalid" };
+      }
+      if (typeof pid === "number" && Number.isInteger(pid) && pid > 0) {
+        try {
+          process.kill(pid, 0);
+        } catch (err) {
+          if (err.code === "EPERM") return { port: p2 };
+          return discoverAndHeal();
+        }
+      }
+      return { port: p2 };
     }
-    const port = parseInt(raw, 10);
-    return Number.isFinite(port) && port > 0 ? port : null;
+    const p = parseInt(raw, 10);
+    return Number.isFinite(p) && p > 0 ? { port: p } : { port: null, reason: "invalid" };
   } catch {
-    return null;
+    return discoverAndHeal();
   }
 }
 async function rpcCall(method, params) {
-  const port = readPort();
-  if (!port) {
-    throw new Error("VibeGrid app is not running. Start VibeGrid to use session management tools.");
+  const result = readPort();
+  if (!result.port) {
+    throw new Error(result.reason === "invalid" ? PORT_FILE_INVALID_MSG : PORT_FILE_MISSING_MSG);
   }
   return new Promise((resolve, reject) => {
-    const ws = new WebSocket(`ws://127.0.0.1:${port}/ws`);
+    const ws = new WebSocket(`ws://127.0.0.1:${result.port}/ws`);
     const id = ++rpcId;
     const timer = setTimeout(() => {
       ws.close();
@@ -1603,12 +1726,12 @@ async function rpcCall(method, params) {
   });
 }
 async function rpcNotify(method, params) {
-  const port = readPort();
-  if (!port) {
-    throw new Error("VibeGrid app is not running. Start VibeGrid to use session management tools.");
+  const result = readPort();
+  if (!result.port) {
+    throw new Error(result.reason === "invalid" ? PORT_FILE_INVALID_MSG : PORT_FILE_MISSING_MSG);
   }
   return new Promise((resolve, reject) => {
-    const ws = new WebSocket(`ws://127.0.0.1:${port}/ws`);
+    const ws = new WebSocket(`ws://127.0.0.1:${result.port}/ws`);
     ws.on("open", () => {
       ws.send(JSON.stringify({ jsonrpc: "2.0", method, params }));
       ws.close();
@@ -1844,12 +1967,12 @@ function registerSessionTools(server) {
     "Send input to a running terminal session. Requires the VibeGrid app to be running.",
     {
       id: V.id.describe("Session ID"),
-      data: z4.string().max(5e4, "Data must be 50000 characters or less").describe("Data to write (text input to send to the agent)")
+      data: z4.string().max(5e4, "Data must be 50000 characters or less").describe("Data to write (text input to send to the agent)"),
+      raw: z4.boolean().optional().describe("Send data as-is without appending carriage return (for raw terminal control)")
     },
     async (args) => {
       try {
-        const trimmed = args.data.replace(/[\r\n]+$/, "");
-        const data = trimmed + "\r";
+        const data = args.raw ? args.data : args.data.replace(/[\r\n]+$/, "") + "\r";
         await rpcNotify("terminal:write", { id: args.id, data });
         return { content: [{ type: "text", text: `Wrote to session: ${args.id}` }] };
       } catch (err) {
@@ -1865,6 +1988,73 @@ function registerSessionTools(server) {
       }
     }
   );
+  const KEY_MAP = {
+    enter: "\r",
+    escape: "\x1B",
+    esc: "\x1B",
+    tab: "	",
+    "shift+tab": "\x1B[Z",
+    up: "\x1B[A",
+    down: "\x1B[B",
+    left: "\x1B[D",
+    right: "\x1B[C",
+    backspace: "\x7F",
+    delete: "\x1B[3~",
+    home: "\x1B[H",
+    end: "\x1B[F",
+    "ctrl+c": "",
+    "ctrl+d": "",
+    "ctrl+x": "",
+    "ctrl+z": ""
+  };
+  server.tool(
+    "send_key",
+    "Send a single keystroke or key combo to a terminal session without appending Enter. Use for TUI interactions like selecting menu options (1, 2, y, n), pressing Escape, Ctrl+C, arrow keys, etc.",
+    {
+      id: V.id.describe("Session ID"),
+      key: z4.string().min(1).max(20).describe(
+        "Key to send: single char (1, y, n), named key (enter, escape, tab, up, down, left, right, backspace, delete, home, end), or combo (ctrl+c, ctrl+d, ctrl+x, ctrl+z, shift+tab)"
+      )
+    },
+    async (args) => {
+      const key = args.key.toLowerCase().trim();
+      let data = KEY_MAP[key];
+      if (!data) {
+        const ctrlMatch = key.match(/^ctrl\+([a-z])$/);
+        if (ctrlMatch) {
+          data = String.fromCharCode(ctrlMatch[1].toUpperCase().charCodeAt(0) - 64);
+        } else if (args.key.length === 1) {
+          data = args.key;
+        } else {
+          return {
+            content: [
+              {
+                type: "text",
+                text: `Unknown key: "${args.key}". Supported: single chars (1, y, n), named keys (${Object.keys(KEY_MAP).join(", ")}), or ctrl+<letter>.`
+              }
+            ],
+            isError: true
+          };
+        }
+      }
+      try {
+        await rpcNotify("terminal:write", { id: args.id, data });
+        return {
+          content: [{ type: "text", text: `Sent key "${args.key}" to session: ${args.id}` }]
+        };
+      } catch (err) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: `Error sending key to terminal: ${err instanceof Error ? err.message : err}`
+            }
+          ],
+          isError: true
+        };
+      }
+    }
+  );
   server.tool(
     "list_session_events",
     "List session lifecycle events (created, exited, task_linked, renamed, archived, unarchived). Use for post-mortem analysis and multi-agent coordination.",
@@ -2025,7 +2215,7 @@ function registerWorkflowTools(server) {
       const workflow = {
         id: crypto2.randomUUID(),
         name: args.name,
-        icon: args.icon ?? "zap",
+        icon: args.icon ?? "Zap",
         iconColor: args.icon_color ?? "#6366f1",
         nodes,
         edges,
@@ -2280,7 +2470,7 @@ console.warn = (...args) => _origError("[mcp:warn]", ...args);
 console.error = (...args) => _origError("[mcp:error]", ...args);
 async function main() {
   configManager.init();
-  const version = true ? "0.4.0-beta.5" : createRequire(import.meta.url)("../package.json").version;
+  const version = true ? "0.4.0-beta.6" : createRequire(import.meta.url)("../package.json").version;
   const server = createMcpServer(version);
   const transport = new StdioServerTransport();
   await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibegrid/mcp",
-  "version": "0.4.0-beta.5",
+  "version": "0.4.0-beta.6",
   "description": "VibeGrid MCP server — task management, git, and workflow tools for AI coding agents",
   "type": "module",
   "license": "MIT",