xiaotime 0.2.1 → 0.3.0

package/dist/ws.js

@@ -1,64 +1,40 @@
 "use strict";
 /**
  * WebSocket connection and main message loop.
+ *
+ * SPRINT-CLI-STATE-MACHINE-1 (#6163 / closes #6127) — drives the
+ * SessionUI state machine from incoming WS events so the user gets
+ * a spinner while the model is generating / a tool is running, and
+ * approval prompts don't race the main rl on keystrokes.
  */
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.runSession = runSession;
 const ws_1 = __importDefault(require("ws"));
-const readline = __importStar(require("readline"));
 const config_js_1 = require("./config.js");
 const auth_js_1 = require("./auth.js");
 const tools_js_1 = require("./tools.js");
 const display_js_1 = require("./display.js");
+const ui_state_js_1 = require("./ui_state.js");
 async function runSession(sessionId) {
     const wsUrl = `${config_js_1.WS_URL}/terminal/sessions/${sessionId}/stream`;
     return new Promise((resolve, reject) => {
         const ws = new ws_1.default(wsUrl);
         let connected = false;
-        let waitingForToolResult = false;
-        let pendingResolve = null;
-        // ── Readline for user input ───────────────────────────────────
-        const rl = readline.createInterface({
-            input: process.stdin,
-            output: process.stdout,
-            terminal: true,
-        });
+        const ui = new ui_state_js_1.SessionUI();
+        let firstTokenThisTurn = true;
+        // #6169 review §1 — guard against concurrent tool_call messages.
+        // The server-side WS handler dispatches tool calls serially (one
+        // tool_call → wait for tool_result → next tool_call), but a buggy
+        // server or future protocol change could break that invariant.
+        // Overlapping executions would corrupt the SessionUI state
+        // machine (running_tool vs awaiting_approval) and possibly skip
+        // the approval gate on the second call. Track the in-flight
+        // tool_use_id; reject any second tool_call that arrives before
+        // the first one's tool_result was sent.
+        let inFlightToolUseId = null;
         ws.on("open", () => {
             // Auth
             ws.send(JSON.stringify({ type: "auth", token: (0, auth_js_1.getToken)() }));
@@ -72,51 +48,99 @@ async function runSession(sessionId) {
                     connected = true;
                     (0, tools_js_1.setSessionCwd)(process.cwd());
                     (0, display_js_1.printWelcome)(msg.session_id, msg.session_name, msg.resumed, msg.message_count);
-                    promptUser();
+                    startUserPromptLoop();
                     break;
                 case "text_delta":
+                    if (firstTokenThisTurn) {
+                        ui.onFirstToken();
+                        firstTokenThisTurn = false;
+                    }
                     (0, display_js_1.printTextDelta)(msg.content);
                     break;
-                case "tool_call":
-                    waitingForToolResult = true;
+                case "tool_call": {
+                    // #6169 review §1 — reject overlapping tool calls. The
+                    // server-side WS handler is supposed to dispatch one
+                    // tool_call at a time and wait for its tool_result before
+                    // sending the next; a protocol violation here would
+                    // corrupt the SessionUI state. Reject the second tool by
+                    // returning an is_error tool_result and logging — leaves
+                    // the first execution untouched.
+                    if (inFlightToolUseId !== null) {
+                        const errMsg = `Protocol violation: tool_call (id=${msg.tool_use_id}) arrived ` +
+                            `while another tool (id=${inFlightToolUseId}) is still in flight. ` +
+                            `CLI rejects overlapping executions; the server WS handler should ` +
+                            `serialize tool dispatch.`;
+                        console.error(`\n[cli] ${errMsg}\n`);
+                        ws.send(JSON.stringify({
+                            type: "tool_result",
+                            tool_use_id: msg.tool_use_id,
+                            content: errMsg,
+                            is_error: true,
+                        }));
+                        break;
+                    }
+                    inFlightToolUseId = msg.tool_use_id;
+                    // executeTool drives its own state transitions when an
+                    // approval is needed (running_tool ↔ awaiting_approval).
+                    ui.setState("running_tool");
                     (0, tools_js_1.executeTool)({
                         tool_use_id: msg.tool_use_id,
                         name: msg.name,
                         input: msg.input,
-                    }).then((result) => {
-                        waitingForToolResult = false;
+                    }, ui).then((result) => {
+                        inFlightToolUseId = null;
                         ws.send(JSON.stringify({
                             type: "tool_result",
                             tool_use_id: result.tool_use_id,
                             content: result.content,
                             is_error: result.is_error,
                         }));
+                        // Back to generating; the next text_delta will clear any
+                        // spinner if onFirstToken hasn't yet been called for this
+                        // generation segment.
+                        ui.setState("generating");
+                        firstTokenThisTurn = true;
                     });
                     break;
+                }
                 case "message_stop":
+                    ui.setState("idle");
                     (0, display_js_1.printTurnEnd)(msg.cost_usd || 0);
-                    promptUser();
+                    startUserPromptLoop();
                     break;
                 case "error":
+                    ui.setState("idle");
                     (0, display_js_1.printError)(msg.message);
                     if (!connected) {
                         ws.close();
                         reject(new Error(msg.message));
                     }
                     else {
-                        promptUser();
+                        startUserPromptLoop();
                     }
                     break;
+                case "server_tool_call":
+                    // #6509 PR3 — visibility line for the 15 server-side tools
+                    // in terminal_orchestrator/main.py SERVER_TOOLS. Clear the
+                    // spinner first so the [server:<name>] line doesn't get
+                    // overwritten by the spinner's next frame, and pin
+                    // firstTokenThisTurn=false so a subsequent text_delta
+                    // doesn't double-clear (onFirstToken is idempotent but
+                    // the bookkeeping flag should reflect reality).
+                    ui.onFirstToken();
+                    firstTokenThisTurn = false;
+                    (0, display_js_1.printServerToolCall)(msg.name, msg.input_summary || "");
+                    break;
                 case "pong":
                     break;
             }
         });
         ws.on("close", () => {
-            rl.close();
+            ui.getRl().close();
             resolve();
         });
         ws.on("error", (err) => {
-            rl.close();
+            ui.getRl().close();
             reject(err);
         });
         // ── Ping keepalive ────────────────────────────────────────────
@@ -126,24 +150,33 @@ async function runSession(sessionId) {
             }
         }, 30000);
         ws.on("close", () => clearInterval(pingInterval));
-        // ── User input ────────────────────────────────────────────────
-        function promptUser() {
-            if (!connected || waitingForToolResult)
+        // ── User input loop ────────────────────────────────────────────
+        async function startUserPromptLoop() {
+            if (!connected)
                 return;
             (0, display_js_1.printPrompt)();
-            rl.once("line", (line) => {
-                const text = line.trim();
-                if (!text) {
-                    promptUser();
-                    return;
-                }
-                if (ws.readyState === ws_1.default.OPEN) {
-                    ws.send(JSON.stringify({ type: "message", content: text }));
-                }
-            });
+            const text = await ui.promptLine();
+            if (!text) {
+                startUserPromptLoop();
+                return;
+            }
+            if (ws.readyState === ws_1.default.OPEN) {
+                // Mark generating BEFORE sending the message — gives the user
+                // an immediate spinner while we wait for the first token. The
+                // server may take 1-3s before the first text_delta lands;
+                // pre-#6163 this was a silent wait that looked like a hang.
+                ui.setState("generating");
+                firstTokenThisTurn = true;
+                ws.send(JSON.stringify({ type: "message", content: text }));
+            }
         }
-        // Ctrl+C
-        rl.on("close", () => {
+        // Ctrl+C — registered via onRlClose (not getRl().on) so the listener
+        // survives the rl close-and-recreate cycle in promptApprovalMenu
+        // (SPRINT-CLI-APPROVAL-MENU-1 / #6509 PR1). The intra-menu close is
+        // suppressed inside promptApprovalMenu via removeAllListeners before
+        // calling rl.close(), so this listener only fires for genuine
+        // user-initiated Ctrl+C at the main prompt.
+        ui.onRlClose(() => {
             console.log("\nSession paused. Resume with: xiaotime dev --session " + sessionId);
             ws.close();
         });

package/dist/ws.js.map

@@ -1 +1 @@
-{"version":3,"file":"ws.js","sourceRoot":"","sources":["../src/ws.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAeH,gCA0HC;AAvID,4CAA2B;AAC3B,mDAAqC;AACrC,2CAAqC;AACrC,uCAAqC;AACrC,yCAAwD;AACxD,6CAMsB;AAEf,KAAK,UAAU,UAAU,CAAC,SAAiB;IAChD,MAAM,KAAK,GAAG,GAAG,kBAAM,sBAAsB,SAAS,SAAS,CAAC;IAEhE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,EAAE,GAAG,IAAI,YAAS,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,oBAAoB,GAAG,KAAK,CAAC;QACjC,IAAI,cAAc,GAAsC,IAAI,CAAC;QAE7D,iEAAiE;QACjE,MAAM,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;YAClC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,IAAI;SACf,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,OAAO;YACP,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAA,kBAAQ,GAAE,EAAE,CAAC,CAAC,CAAC;YAC7D,mBAAmB;YACnB,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,GAAsB,EAAE,EAAE;YAChD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;YAEvC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,WAAW;oBACd,SAAS,GAAG,IAAI,CAAC;oBACjB,IAAA,wBAAa,EAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;oBAC7B,IAAA,yBAAY,EACV,GAAG,CAAC,UAAU,EACd,GAAG,CAAC,YAAY,EAChB,GAAG,CAAC,OAAO,EACX,GAAG,CAAC,aAAa,CAClB,CAAC;oBACF,UAAU,EAAE,CAAC;oBACb,MAAM;gBAER,KAAK,YAAY;oBACf,IAAA,2BAAc,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC5B,MAAM;gBAER,KAAK,WAAW;oBACd,oBAAoB,GAAG,IAAI,CAAC;oBAC5B,IAAA,sBAAW,EAAC;wBACV,WAAW,EAAE,GAAG,CAAC,WAAW;wBAC5B,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,KAAK,EAAE,GAAG,CAAC,KAAK;qBACjB,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;wBACjB,oBAAoB,GAAG,KAAK,CAAC;wBAC7B,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;4BACrB,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,MAAM,CAAC,WAAW;4BAC/B,OAAO,EAAE,MAAM,CAAC,OAAO;4BACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;yBAC1B,CAAC,CAAC,CAAC;oBACN,CAAC,CAAC,CAAC;oBACH,MAAM;gBAER,KAAK,cAAc;oBACjB,IAAA,yBAAY,EAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC;oBAChC,UAAU,EAAE,CAAC;oBACb,MAAM;gBAER,KAAK,OAAO;oBACV,IAAA,uBAAU,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACxB,IAAI,CAAC,SAAS,EAAE,CAAC;wBACf,EAAE,CAAC,KAAK,EAAE,CAAC;wBACX,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;oBACjC,CAAC;yBAAM,CAAC;wBACN,UAAU,EAAE,CAAC;oBACf,CAAC;oBACD,MAAM;gBAER,KAAK,MAAM;oBACT,MAAM;YACV,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACrB,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,iEAAiE;QACjE,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;YACpC,IAAI,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;gBACrC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC,CAAC;QAElD,iEAAiE;QACjE,SAAS,UAAU;YACjB,IAAI,CAAC,SAAS,IAAI,oBAAoB;gBAAE,OAAO;YAC/C,IAAA,wBAAW,GAAE,CAAC;YAEd,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBACvB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,UAAU,EAAE,CAAC;oBACb,OAAO;gBACT,CAAC;gBACD,IAAI,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;oBACrC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,SAAS;QACT,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,OAAO,CAAC,GAAG,CAAC,wDAAwD,GAAG,SAAS,CAAC,CAAC;YAClF,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"ws.js","sourceRoot":"","sources":["../src/ws.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;;;AAgBH,gCAgMC;AA9MD,4CAA2B;AAC3B,2CAAqC;AACrC,uCAAqC;AACrC,yCAAwD;AACxD,6CAOsB;AACtB,+CAA0C;AAEnC,KAAK,UAAU,UAAU,CAAC,SAAiB;IAChD,MAAM,KAAK,GAAG,GAAG,kBAAM,sBAAsB,SAAS,SAAS,CAAC;IAEhE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,EAAE,GAAG,IAAI,YAAS,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,MAAM,EAAE,GAAG,IAAI,uBAAS,EAAE,CAAC;QAC3B,IAAI,kBAAkB,GAAG,IAAI,CAAC;QAC9B,iEAAiE;QACjE,iEAAiE;QACjE,kEAAkE;QAClE,+DAA+D;QAC/D,2DAA2D;QAC3D,gEAAgE;QAChE,4DAA4D;QAC5D,+DAA+D;QAC/D,wCAAwC;QACxC,IAAI,iBAAiB,GAAkB,IAAI,CAAC;QAE5C,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,OAAO;YACP,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAA,kBAAQ,GAAE,EAAE,CAAC,CAAC,CAAC;YAC7D,mBAAmB;YACnB,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,GAAsB,EAAE,EAAE;YAChD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;YAEvC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,WAAW;oBACd,SAAS,GAAG,IAAI,CAAC;oBACjB,IAAA,wBAAa,EAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;oBAC7B,IAAA,yBAAY,EACV,GAAG,CAAC,UAAU,EACd,GAAG,CAAC,YAAY,EAChB,GAAG,CAAC,OAAO,EACX,GAAG,CAAC,aAAa,CAClB,CAAC;oBACF,mBAAmB,EAAE,CAAC;oBACtB,MAAM;gBAER,KAAK,YAAY;oBACf,IAAI,kBAAkB,EAAE,CAAC;wBACvB,EAAE,CAAC,YAAY,EAAE,CAAC;wBAClB,kBAAkB,GAAG,KAAK,CAAC;oBAC7B,CAAC;oBACD,IAAA,2BAAc,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC5B,MAAM;gBAER,KAAK,WAAW,CAAC,CAAC,CAAC;oBACjB,uDAAuD;oBACvD,qDAAqD;oBACrD,0DAA0D;oBAC1D,oDAAoD;oBACpD,yDAAyD;oBACzD,yDAAyD;oBACzD,iCAAiC;oBACjC,IAAI,iBAAiB,KAAK,IAAI,EAAE,CAAC;wBAC/B,MAAM,MAAM,GACV,qCAAqC,GAAG,CAAC,WAAW,YAAY;4BAChE,0BAA0B,iBAAiB,wBAAwB;4BACnE,mEAAmE;4BACnE,0BAA0B,CAAC;wBAC7B,OAAO,CAAC,KAAK,CAAC,WAAW,MAAM,IAAI,CAAC,CAAC;wBACrC,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;4BACb,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,GAAG,CAAC,WAAW;4BAC5B,OAAO,EAAE,MAAM;4BACf,QAAQ,EAAE,IAAI;yBACf,CAAC,CACH,CAAC;wBACF,MAAM;oBACR,CAAC;oBACD,iBAAiB,GAAG,GAAG,CAAC,WAAW,CAAC;oBACpC,uDAAuD;oBACvD,yDAAyD;oBACzD,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;oBAC5B,IAAA,sBAAW,EACT;wBACE,WAAW,EAAE,GAAG,CAAC,WAAW;wBAC5B,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,KAAK,EAAE,GAAG,CAAC,KAAK;qBACjB,EACD,EAAE,CACH,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;wBAChB,iBAAiB,GAAG,IAAI,CAAC;wBACzB,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;4BACb,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,MAAM,CAAC,WAAW;4BAC/B,OAAO,EAAE,MAAM,CAAC,OAAO;4BACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;yBAC1B,CAAC,CACH,CAAC;wBACF,yDAAyD;wBACzD,0DAA0D;wBAC1D,sBAAsB;wBACtB,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;wBAC1B,kBAAkB,GAAG,IAAI,CAAC;oBAC5B,CAAC,CAAC,CAAC;oBACH,MAAM;gBACR,CAAC;gBAED,KAAK,cAAc;oBACjB,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACpB,IAAA,yBAAY,EAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC;oBAChC,mBAAmB,EAAE,CAAC;oBACtB,MAAM;gBAER,KAAK,OAAO;oBACV,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACpB,IAAA,uBAAU,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACxB,IAAI,CAAC,SAAS,EAAE,CAAC;wBACf,EAAE,CAAC,KAAK,EAAE,CAAC;wBACX,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;oBACjC,CAAC;yBAAM,CAAC;wBACN,mBAAmB,EAAE,CAAC;oBACxB,CAAC;oBACD,MAAM;gBAER,KAAK,kBAAkB;oBACrB,2DAA2D;oBAC3D,2DAA2D;oBAC3D,wDAAwD;oBACxD,mDAAmD;oBACnD,sDAAsD;oBACtD,uDAAuD;oBACvD,gDAAgD;oBAChD,EAAE,CAAC,YAAY,EAAE,CAAC;oBAClB,kBAAkB,GAAG,KAAK,CAAC;oBAC3B,IAAA,gCAAmB,EAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;oBACvD,MAAM;gBAER,KAAK,MAAM;oBACT,MAAM;YACV,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACrB,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,iEAAiE;QACjE,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;YACpC,IAAI,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;gBACrC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC,CAAC;QAElD,kEAAkE;QAClE,KAAK,UAAU,mBAAmB;YAChC,IAAI,CAAC,SAAS;gBAAE,OAAO;YACvB,IAAA,wBAAW,GAAE,CAAC;YACd,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,UAAU,EAAE,CAAC;YACnC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,mBAAmB,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;YACD,IAAI,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;gBACrC,8DAA8D;gBAC9D,8DAA8D;gBAC9D,0DAA0D;gBAC1D,4DAA4D;gBAC5D,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBAC1B,kBAAkB,GAAG,IAAI,CAAC;gBAC1B,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAED,qEAAqE;QACrE,iEAAiE;QACjE,oEAAoE;QACpE,qEAAqE;QACrE,8DAA8D;QAC9D,4CAA4C;QAC5C,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE;YAChB,OAAO,CAAC,GAAG,CACT,wDAAwD,GAAG,SAAS,CACrE,CAAC;YACF,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xiaotime",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Xiaotime Developer — persistent-memory AI coding assistant",
   "bin": {
     "xiaotime": "./dist/index.js"
@@ -12,6 +12,8 @@
   "scripts": {
     "build": "tsc",
     "dev": "ts-node src/index.ts",
+    "test": "vitest run",
+    "test:watch": "vitest",
     "prepublishOnly": "npm run build"
   },
   "dependencies": {
@@ -23,7 +25,8 @@
     "@types/ws": "^8.5.12",
     "@types/node": "^22.0.0",
     "typescript": "^5.7.0",
-    "ts-node": "^10.9.2"
+    "ts-node": "^10.9.2",
+    "vitest": "^2.1.0"
   },
   "engines": {
     "node": ">=18.0.0"