aiterm-mcp 0.7.1 → 0.9.0

package/dist/grok-stop-hook.js

@@ -0,0 +1,120 @@
+#!/usr/bin/env node
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+const LAUNCH_ID_RE = /^[0-9a-f]{32}$/;
+const SESSION_RE = /^[A-Za-z0-9_-]{1,64}$/;
+const MAX_STDIN_BYTES = 1024 * 1024;
+function fail(message) {
+    process.stderr.write(`aiterm grok-stop-hook: ${message}\n`);
+    process.exit(0);
+}
+function noop() {
+    process.exit(0);
+}
+function hasAitermEnv() {
+    return !!(process.env.AITERM_AGENT_KIND ||
+        process.env.AITERM_SESSION_ID ||
+        process.env.AITERM_AGENT_SESSION_ID ||
+        process.env.AITERM_AGENT_LAUNCH_ID);
+}
+function uid() {
+    if (typeof process.getuid !== "function")
+        fail("POSIX getuid が使えません");
+    return process.getuid();
+}
+function runtimeStateBase() {
+    const xdg = process.env.XDG_RUNTIME_DIR;
+    if (xdg) {
+        try {
+            if (fs.statSync(xdg).isDirectory())
+                return xdg;
+        }
+        catch {
+            /* XDG_RUNTIME_DIR が壊れている CI/非 login 環境では os.tmpdir() に戻す */
+        }
+    }
+    return os.tmpdir();
+}
+function secureAgentsDir() {
+    const root = path.join(runtimeStateBase(), `aiterm-mcp-${uid()}`);
+    const agents = path.join(root, "agents");
+    const rst = fs.lstatSync(root);
+    if (!rst.isDirectory() || rst.isSymbolicLink() || rst.uid !== uid() || (rst.mode & 0o077) !== 0) {
+        fail(`agent state root が安全ではありません: ${root}`);
+    }
+    const ast = fs.lstatSync(agents);
+    if (!ast.isDirectory() || ast.isSymbolicLink() || ast.uid !== uid() || (ast.mode & 0o077) !== 0) {
+        fail(`agent state dir が安全ではありません: ${agents}`);
+    }
+    return agents;
+}
+function str(v) {
+    return typeof v === "string" ? v : null;
+}
+async function readStdin() {
+    const chunks = [];
+    let total = 0;
+    for await (const chunk of process.stdin) {
+        const b = Buffer.isBuffer(chunk) ? chunk : Buffer.from(String(chunk));
+        total += b.length;
+        if (total > MAX_STDIN_BYTES)
+            fail("payload が大きすぎます");
+        chunks.push(b);
+    }
+    return Buffer.concat(chunks).toString("utf8");
+}
+function appendEvent(file, event) {
+    const nofollow = fs.constants.O_NOFOLLOW ?? 0;
+    const line = JSON.stringify(event) + "\n";
+    if (Buffer.byteLength(line, "utf8") > 64 * 1024)
+        fail("event line が大きすぎます");
+    const fd = fs.openSync(file, fs.constants.O_CREAT | fs.constants.O_APPEND | fs.constants.O_WRONLY | nofollow, 0o600);
+    try {
+        const st = fs.fstatSync(fd);
+        if (!st.isFile() || st.uid !== uid() || st.nlink !== 1 || (st.mode & 0o077) !== 0) {
+            fail(`event file が安全ではありません: ${file}`);
+        }
+        fs.writeSync(fd, line, undefined, "utf8");
+    }
+    finally {
+        fs.closeSync(fd);
+    }
+}
+async function main() {
+    if (!hasAitermEnv())
+        noop();
+    const kind = process.env.AITERM_AGENT_KIND;
+    const session = process.env.AITERM_SESSION_ID || process.env.AITERM_AGENT_SESSION_ID || "";
+    const launchId = process.env.AITERM_AGENT_LAUNCH_ID || "";
+    if (kind !== "grok" && kind !== "composer")
+        fail(`AITERM_AGENT_KIND が grok/composer ではありません: ${kind ?? ""}`);
+    if (!SESSION_RE.test(session))
+        fail(`session id が不正です: ${session}`);
+    if (!LAUNCH_ID_RE.test(launchId))
+        fail(`launch id が不正です: ${launchId}`);
+    let payload = {};
+    const input = await readStdin();
+    if (input.trim()) {
+        try {
+            payload = JSON.parse(input);
+        }
+        catch {
+            payload = {};
+        }
+    }
+    const agents = secureAgentsDir();
+    const eventFile = path.join(agents, `${session}.${launchId}.events.jsonl`);
+    appendEvent(eventFile, {
+        type: "agent_done",
+        vendor: kind,
+        aiterm_session: session,
+        launch_id: launchId,
+        vendor_session_id: str(payload.sessionId),
+        turn_id: str(payload.promptId),
+        reason: str(payload.reason) ?? str(payload.hookEventName) ?? "stop",
+        done_status: "turn_done",
+        at: new Date().toISOString(),
+    });
+}
+main().catch((e) => fail(e instanceof Error ? e.message : String(e)));

package/dist/index.js

@@ -12,7 +12,13 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
 import * as core from "./core.js";
-const server = new McpServer({ name: "aiterm", version: "0.4.0" });
+import { createRequire } from "node:module";
+// package.json の version を実行時に読み、MCP initialize で配るサーバ版と一致させる。
+// createRequire を使うのは、import 属性 `with { type: "json" }` が Node 18.20+ 限定で
+// engines "node >=18"（18.0〜18.19）を SyntaxError で壊し、旧 `assert` 構文は逆に Node 22 で
+// 除去済み＝どちらの静的構文も 18〜22 全域を満たせないため（実行時 require が唯一全域で動く）。
+const pkg = createRequire(import.meta.url)("../package.json");
+const server = new McpServer({ name: "aiterm", version: pkg.version });
 function ok(s) {
     return { content: [{ type: "text", text: s }] };
 }
@@ -38,18 +44,42 @@ server.registerTool("pty_open", {
     }
 });
 server.registerTool("pty_send", {
-    description: "セッションへテキスト(コマンド)を送る。送信後の出力は pty_read で取得する。",
+    description: "セッションへテキスト(コマンド)を送る。通常は送信だけ行い、出力は pty_read で取得する。" +
+        "agent_done:true で起動した Codex/Grok/Composer セッションは wait:'agent_done' で Stop hook まで待てる。",
     inputSchema: {
         session_id: z.string(),
         text: z.string().describe("送る文字列（コマンド）"),
         enter: z.boolean().default(true).describe("末尾で Enter を送る"),
-        mark: z.boolean().default(false).describe("完了 sentinel(終了コード付き)で包む"),
+        wait: z
+            .enum(["none", "agent_done"])
+            .default("none")
+            .describe("none=従来通り送信のみ。agent_done=agent Stop hook まで待って最終画面を返す"),
+        timeout: z.number().default(600).describe("wait:'agent_done' の最大待ち秒数"),
+        screen: z.boolean().default(true).describe("wait:'agent_done' の返り値を描画済みスクリーンにする"),
+        lines: z.number().int().nullish().describe("wait:'agent_done' で返す末尾 N 行"),
+        mark: z
+            .boolean()
+            .default(false)
+            .describe("完了 sentinel(終了コード付き)で包む。pty_read(wait:true) が until 無しでも自動検出して" +
+            "完了確定する（ネスト中や非シェル前面でも効く確実な完了検出。手で until を組む必要なし）"),
         force: z.boolean().default(false).describe("破壊的コマンドゲートを越える"),
         rtk: z.boolean().default(false).describe("既知コマンドを rtk 形へ委譲して送る（rtk 不在なら素通し）"),
         raw: z.boolean().default(false).describe("送信前サニタイズを無効化"),
     },
-}, async ({ session_id, text, enter, mark, force, rtk, raw }) => {
+}, async ({ session_id, text, enter, wait, timeout, screen, lines, mark, force, rtk, raw }) => {
     try {
+        if (wait === "agent_done") {
+            return ok(await core.sendAndWaitAgentDone(session_id, text, {
+                enter,
+                mark,
+                force,
+                rtk,
+                raw,
+                timeout,
+                screen,
+                lines: lines ?? null,
+            }));
+        }
         return ok(core.send(session_id, text, { enter, mark, force, rtk, raw }));
     }
     catch (e) {
@@ -61,8 +91,18 @@ server.registerTool("pty_read", {
         "削減: 制御文字除去 / 反復圧縮 / head+tail 折りたたみ＋復元ヒント＋メタ併記。",
     inputSchema: {
         session_id: z.string(),
-        wait: z.boolean().default(false).describe("完了まで待つ（4層: dead/until/出力静止∧シェル復帰/timeout）"),
-        until: z.string().nullish().describe("この正規表現が出たら完了とみなす"),
+        wait: z
+            .boolean()
+            .default(false)
+            .describe("完了まで待つ（dead / mark sentinel 自動検出 / until / 出力静止∧シェル復帰 / timeout）"),
+        until: z
+            .string()
+            .nullish()
+            .describe("この文字列が出たら完了とみなす（既定はリテラル部分一致。`$ ` や `[..]` もそのまま探せる）"),
+        until_regex: z
+            .boolean()
+            .default(false)
+            .describe("until を正規表現として扱う（既定 false＝リテラル部分一致。メタ文字を使いたい時のみ true）"),
         timeout: z.number().default(10).describe("wait の最大待ち秒数"),
         screen: z.boolean().default(false).describe("描画済みスクリーン(TUI 向け)"),
         full: z.boolean().default(false).describe("増分でなく全文"),
@@ -71,20 +111,23 @@ server.registerTool("pty_read", {
         raw: z.boolean().default(false).describe("削減せず生テキスト"),
         rtk: z.boolean().default(false).describe("直前コマンド別の自前 reducer(git/grep/pytest 等)で縮約"),
     },
-}, async ({ session_id, wait, until, timeout, screen, full, lines, line_range, raw, rtk }) => {
+}, async ({ session_id, wait, until, until_regex, timeout, screen, full, lines, line_range, raw, rtk }) => {
     try {
         let range = null;
         if (line_range) {
             const idx = line_range.indexOf(":");
             const lo = idx < 0 ? line_range : line_range.slice(0, idx);
             const hi = idx < 0 ? "" : line_range.slice(idx + 1);
-            // 不正/空の上端は「末尾まで」(null) に倒す。"5:abc" を空に潰さず "5:" と同じく 5 行目以降にする。
+            // 不正/空/負の上端は「末尾まで」(null) に倒す。"5:abc" を空に潰さず "5:" と同じく 5 行目以降に。
+            // 下端は負値を 0 にクランプする（"-3:5" が負 slice にならないように・C12）。
+            const loN = Math.max(0, parseInt(lo, 10) || 0);
             const hiN = hi ? parseInt(hi, 10) : NaN;
-            range = [parseInt(lo, 10) || 0, Number.isNaN(hiN) ? null : hiN];
+            range = [loN, Number.isNaN(hiN) || hiN < 0 ? null : hiN];
         }
         const out = await core.readOutput(session_id, {
             wait,
             until: until ?? null,
+            untilRegex: until_regex,
             timeout,
             screen,
             full,
@@ -152,14 +195,19 @@ function registerAgentTool(toolName, kind, desc, grokLike) {
                 .describe(agentEffortDesc(grokLike)),
             cwd: z.string().nullish().describe("作業ディレクトリ（対象リポのルート等・任意）"),
             session_name: z.string().nullish().describe("セッション名（省略で自動採番）"),
+            agent_done: z
+                .boolean()
+                .default(false)
+                .describe("managed Stop hook を有効化し、pty_send(wait:'agent_done') を使えるようにする"),
         },
-    }, async ({ prompt, reasoning_effort, cwd, session_name }) => {
+    }, async ({ prompt, reasoning_effort, cwd, session_name, agent_done }) => {
         try {
             const [sid, hint] = core.openAgent(kind, {
                 prompt: prompt ?? undefined,
                 reasoning_effort: reasoning_effort ?? undefined,
                 cwd: cwd ?? undefined,
                 session_name: session_name ?? undefined,
+                agent_done: agent_done ?? false,
             });
             return ok(`session_id: ${sid}\n${hint}`);
         }

package/dist/rtk.js

@@ -28,16 +28,23 @@ function truncate(s, n) {
         return "...";
     return cp.slice(0, n - 3).join("") + "...";
 }
-const PROMPT_TAIL = /[$#%>]\s*$/;
+// プロンプト行の判定（C3）: 行全体が「プロンプト前置文字(語/@ : ~ / \ [] . -)＋プロンプト記号($#%>)＋末尾空白」
+// の形のときだけプロンプトとみなす。末尾記号だけを見ると `</div>` `>>>` `echo x >` 等の本文行を誤除去する。
+const PROMPT_TAIL = /^[\w@.:~\/\\\[\]-]*[$#%>]\s*$/;
 /** 観測ログ片からエコー行と前後のプロンプト行を落とし、コマンド出力本体だけ残す（発見的）。 */
 export function stripShellFrame(text, command) {
     const lines = text.split("\n");
     const cmd = command.trim();
     let start = 0;
     if (cmd) {
-        for (let i = 0; i < lines.length; i++)
-            if (lines[i].includes(cmd))
+        // 最初の一致（＝コマンドエコー行）だけを落とす。最後の一致まで進めると、出力本体に cmd 文字列が
+        // 再出現（cat したファイル内・commit メッセージ内 等）した場合に本体を丸ごと捨ててしまう（C2）。
+        for (let i = 0; i < lines.length; i++) {
+            if (lines[i].includes(cmd)) {
                 start = i + 1;
+                break;
+            }
+        }
     }
     let end = lines.length;
     while (end > start) {
@@ -77,7 +84,10 @@ export function reducePytest(output) {
             flush();
             continue;
         }
-        if (t.startsWith("===") && (t.includes("passed") || t.includes("failed") || t.includes("skipped"))) {
+        if (t.startsWith("===") &&
+            (t.includes("passed") || t.includes("failed") || t.includes("skipped") || t.includes("error"))) {
+            // 収集エラーのみ（`=== 1 error in Xs ===`）も要約行として拾う。拾わないと全ゼロ扱いで
+            // "No tests collected"（無害誤読）に潰れる。"ERRORS" セクション見出しは大文字ゆえ非該当。
             summaryLine = t;
             continue;
         }
@@ -85,7 +95,7 @@ export function reducePytest(output) {
             !t.startsWith("===") &&
             !t.startsWith("FAILED") &&
             !t.startsWith("ERROR") &&
-            (t.includes(" passed") || t.includes(" failed") || t.includes(" skipped")) &&
+            (t.includes(" passed") || t.includes(" failed") || t.includes(" skipped") || t.includes(" error")) &&
             t.includes(" in ")) {
             summaryLine = t;
             continue;
@@ -114,13 +124,16 @@ export function reducePytest(output) {
         }
     }
     flush();
-    const [p, f, s, xf, xp] = parsePytestCounts(summaryLine);
-    if (p === 0 && f === 0 && s === 0 && xf === 0 && xp === 0)
+    const [p, f, s, xf, xp, e] = parsePytestCounts(summaryLine);
+    if (p === 0 && f === 0 && s === 0 && xf === 0 && xp === 0 && e === 0)
         return "Pytest: No tests collected";
-    const extras = s > 0 || xf > 0 || xp > 0 || xfailLines.length > 0;
+    // error(収集/内部エラー)は失敗の一種＝緑扱いにしない。extras に含め、"N passed" 早期 return を止める。
+    const extras = s > 0 || xf > 0 || xp > 0 || e > 0 || xfailLines.length > 0;
     if (f === 0 && p > 0 && !extras)
         return `Pytest: ${p} passed`;
     let head = `Pytest: ${p} passed, ${f} failed`;
+    if (e > 0)
+        head += `, ${e} error${e === 1 ? "" : "s"}`;
     if (s > 0)
         head += `, ${s} skipped`;
     if (xf > 0)
@@ -183,7 +196,7 @@ export function reducePytest(output) {
     return out.join("\n").trim();
 }
 function parsePytestCounts(summary) {
-    let p = 0, f = 0, s = 0, xf = 0, xp = 0;
+    let p = 0, f = 0, s = 0, xf = 0, xp = 0, e = 0;
     for (const part of summary.split(",")) {
         const words = part.split(/\s+/).filter(Boolean);
         for (let i = 0; i < words.length; i++) {
@@ -193,6 +206,7 @@ function parsePytestCounts(summary) {
             if (Number.isNaN(n))
                 continue;
             const w = words[i];
+            // "error"/"errors" は passed/failed/skipped/xfailed/xpassed のいずれの部分文字列でもないので順不同で安全。
             if (w.includes("xpassed"))
                 xp = n;
             else if (w.includes("xfailed"))
@@ -203,9 +217,11 @@ function parsePytestCounts(summary) {
                 f = n;
             else if (w.includes("skipped"))
                 s = n;
+            else if (w.includes("error"))
+                e = n;
         }
     }
-    return [p, f, s, xf, xp];
+    return [p, f, s, xf, xp, e];
 }
 // ---------------------------------------------------------------- grep
 const GREP_LINE = /^(.*?):(\d+):(.*)$/;
@@ -310,7 +326,8 @@ export function reduceGitLog(output) {
             else if (t && !subject && (ln.startsWith("    ") || ln.startsWith("\t")))
                 subject = t;
             else if (t && (ln.startsWith("    ") || ln.startsWith("\t"))) {
-                if (!t.startsWith("Signed-off-by:") && !t.startsWith("Co-authored-by:"))
+                // trailer は大小文字が実装依存（本リポ規約は Co-Authored-By:）。大小無視で除外する（C5）。
+                if (!/^(signed-off-by|co-authored-by):/i.test(t))
                     body.push(t);
             }
         }
@@ -355,13 +372,25 @@ function applyFilter(rule, output) {
     return body;
 }
 // ---------------------------------------------------------------- ルーティング
+const WRAPPERS = new Set(["sudo", "env", "command", "exec"]);
+const RUNNERS = new Set(["uv", "poetry", "pdm", "rye", "hatch", "pipenv"]);
+// [verb, sub, stripped]。stripped は前置(ラッパー/環境代入/フラグ/ランナー run)を除いた実コマンド。
+// 誤分類は classify で null→generic に落ちるだけなので、読み飛ばしは積極的で安全（C4）。
 function basenameCmd(command) {
     const toks = command.trim().split(/\s+/).filter(Boolean);
     let i = 0;
-    while (i < toks.length && (toks[i].includes("=") || ["sudo", "env", "command", "exec"].includes(toks[i])))
+    // 前置ラッパー(sudo/env/…)・環境代入(FOO=1)・そのフラグ(-E/-i 等)を読み飛ばす
+    while (i < toks.length && (toks[i].includes("=") || WRAPPERS.has(toks[i]) || toks[i].startsWith("-")))
         i++;
+    // ランナー( uv/poetry run <cmd> )は run と共に読み飛ばして実コマンドへ
+    if (i + 1 < toks.length && RUNNERS.has(toks[i]) && toks[i + 1] === "run") {
+        i += 2;
+        while (i < toks.length && (toks[i].includes("=") || toks[i].startsWith("-")))
+            i++;
+    }
+    const stripped = toks.slice(i).join(" ");
     if (i >= toks.length)
-        return ["", ""];
+        return ["", "", ""];
     const verb = toks[i].split("/").pop();
     let sub = "";
     for (const t of toks.slice(i + 1)) {
@@ -370,10 +399,10 @@ function basenameCmd(command) {
             break;
         }
     }
-    return [verb, sub];
+    return [verb, sub, stripped];
 }
 export function classify(command) {
-    const [verb, sub] = basenameCmd(command);
+    const [verb, sub, stripped] = basenameCmd(command);
     if (verb === "git") {
         if (sub === "status")
             return "git-status";
@@ -385,10 +414,12 @@ export function classify(command) {
         return "grep";
     if (verb === "pytest" || verb === "py.test")
         return "pytest";
-    if (verb === "python" && command.includes("pytest"))
+    // python3 / python3.11 等のバージョン付きも許容（C4: `python3 -m pytest` を拾う）
+    if (/^python[0-9.]*$/.test(verb) && command.includes("pytest"))
         return "pytest";
+    // FILTERS は basename 経由の stripped で判定（C6: `sudo df` `sudo make` も分類できる）
     for (const rule of FILTERS)
-        if (rule.match.test(command.trim()))
+        if (rule.match.test(stripped))
             return rule.name;
     return null;
 }
@@ -398,7 +429,11 @@ const REDUCERS = {
     grep: reduceGrep,
     pytest: (o) => reducePytest(o),
 };
-/** コマンドに応じた reducer を観測出力へ適用。返り値 [reducedText|null, reducerName|null]。 */
+/**
+ * コマンドに応じた reducer を観測出力へ適用。返り値 [reducedText|null, reducerName|null]。
+ * 前提(C13): `output` は制御文字除去済み（ANSI/CR 等）であること。色付き/生 PTY 出力を直接渡すと
+ * 誤パースし得る。呼び手 core.readOutput は `stripShellFrame(stripControl(text), cmd)` で前処理してから渡す。
+ */
 export function reduce(command, output) {
     const name = classify(command);
     if (name === null)

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "aiterm-mcp",
-  "version": "0.7.1",
+  "version": "0.9.0",
   "mcpName": "io.github.kitepon-rgb/aiterm-mcp",
-  "description": "AI-driven persistent terminal as a local stdio MCP server (tmux-backed). Holds one local PTY; SSH and containers are just commands you send into it. Token-reducing reads.",
+  "description": "AI-driven persistent terminal as a local stdio MCP server (tmux-backed). Holds one local PTY; SSH and containers are just commands you send into it. Also launches interactive Codex/Grok/Composer agent TUIs in a persistent terminal. Token-reducing reads.",
   "keywords": [
     "mcp",
     "mcp-server",
@@ -15,6 +15,9 @@
     "claude-code",
     "cursor",
     "ai-agent",
+    "agent",
+    "codex",
+    "grok",
     "devtools"
   ],
   "license": "MIT",