vigthoria-cli 1.6.53 → 1.6.54

package/dist/commands/chat.js

@@ -47,6 +47,7 @@ const api_js_1 = require("../utils/api.js");
 const tools_js_1 = require("../utils/tools.js");
 const session_js_1 = require("../utils/session.js");
 const bridge_client_js_1 = require("../utils/bridge-client.js");
+const workspace_stream_js_1 = require("../utils/workspace-stream.js");
 const DEFAULT_V3_AGENT_TIMEOUT_MS = (() => {
     const rawValue = process.env.VIGTHORIA_AGENT_TIMEOUT_MS || process.env.V3_AGENT_TIMEOUT_MS || '3900000';
     const parsed = Number.parseInt(rawValue, 10);
@@ -1442,6 +1443,17 @@ class ChatCommand {
             targetPath: this.currentProjectPath,
             ...runtimeContext,
         };
+        // Start workspace watcher for bidirectional real-time sync
+        let watcher = null;
+        if (this.currentProjectPath && fs.existsSync(this.currentProjectPath)) {
+            watcher = new workspace_stream_js_1.WorkspaceWatcher({
+                workspaceRoot: this.currentProjectPath,
+                onFileChange: (relativePath, content, action) => {
+                    this.logger.debug(`Local change detected: ${action} ${relativePath}`);
+                },
+            });
+            watcher.start();
+        }
         try {
             const workflowPromise = this.api.runV3AgentWorkflow(executionPrompt, {
                 workspace: { path: this.currentProjectPath },
@@ -1484,6 +1496,7 @@ class ChatCommand {
                     }
                     this.logger.warn('Falling back to legacy CLI agent loop');
                     this.logger.debug(`V3 agent workflow returned an incomplete result: ${previewGate?.error || 'workspace changes were not fully validated'}`);
+                    watcher?.stop();
                     return false;
                 }
                 const errorMessage = `V3 agent workflow returned an incomplete result and legacy fallback is disabled. ${previewGate?.error || 'Workspace changes were not fully validated.'}`;
@@ -1504,6 +1517,7 @@ class ChatCommand {
                         metadata: { executionPath: 'v3-agent', previewGate },
                     }, null, 2));
                 }
+                watcher?.stop();
                 return true;
             }
             if (!this.jsonOutput && previewGate?.required && previewGate?.passed !== true && workspaceHasOutput) {
@@ -1543,9 +1557,11 @@ class ChatCommand {
                 }
             }
             this.messages.push({ role: 'assistant', content: response.content || 'V3 agent workflow completed.' });
+            watcher?.stop();
             return true;
         }
         catch (error) {
+            watcher?.stop();
             if (rescueEligible && !this.api.hasAgentWorkspaceOutput(workspaceContext)) {
                 const rescued = await this.tryCommandLevelSaaSRescue(executionPrompt, prompt, workspaceContext, routingPolicy, spinner, error);
                 if (rescued) {

package/dist/utils/api.js

@@ -17,6 +17,7 @@ const https_1 = __importDefault(require("https"));
 const net_1 = __importDefault(require("net"));
 const path_1 = __importDefault(require("path"));
 const ws_1 = __importDefault(require("ws"));
+const workspace_stream_js_1 = require("./workspace-stream.js");
 class CLIError extends Error {
     category;
     statusCode;
@@ -2636,6 +2637,19 @@ document.addEventListener('DOMContentLoaded', () => {
                     serverWorkspaceRoot = event.workspace_root.trim();
                 }
                 this.captureV3AgentStreamMutation(event, streamedFiles, serverWorkspaceRoot);
+                // Real-time workspace streaming: apply file mutations to local disk immediately
+                if (event.type === 'file_mutation') {
+                    const localRoot = context.projectPath || context.workspacePath || context.targetPath;
+                    if (localRoot && typeof event.path === 'string') {
+                        (0, workspace_stream_js_1.applyFileMutation)(event, localRoot);
+                        if (typeof event.content === 'string') {
+                            const relPath = this.normalizeAgentWorkspaceRelativePath(event.path, serverWorkspaceRoot || undefined);
+                            if (relPath) {
+                                streamedFiles[relPath] = event.content;
+                            }
+                        }
+                    }
+                }
                 // Empty workspace guard: if the remote agent lists its root
                 // and finds nothing while our local workspace has files, the
                 // workspace was not hydrated. Abort early with a clear error

package/dist/utils/workspace-stream.d.ts

@@ -0,0 +1,75 @@
+/**
+ * Workspace Stream — Real-time bidirectional file sync between CLI and V3.
+ *
+ * Architecture:
+ *   1. WorkspaceWatcher — chokidar-based file watcher that detects local changes
+ *      and pushes them to V3 via WebSocket.
+ *   2. MutationApplier — receives file_mutation events from V3's SSE stream
+ *      and applies them to the local workspace IN REAL TIME (not at completion).
+ *   3. WorkspaceWSClient — WebSocket client connected to V3's /ws/workspace
+ *      for bidirectional file sync outside the SSE stream.
+ *
+ * Event types (V3 → CLI via SSE):
+ *   {"type": "file_mutation", "path": "rel/path", "content": "...", "action": "write"|"edit", "tool": "write_file"|"edit_file"}
+ *
+ * Event types (CLI → V3 via /ws/workspace):
+ *   {"type": "bind", "context_id": "...", "workspace_root": "..."}
+ *   {"type": "file_sync", "path": "rel/path", "content": "...", "action": "write"|"delete"}
+ *   {"type": "file_batch", "files": [{path, content}, ...]}
+ */
+/**
+ * Apply a file_mutation event from V3's SSE stream to the local workspace.
+ * Call this from the SSE event handler for real-time file application.
+ */
+export declare function applyFileMutation(event: {
+    type: string;
+    path: string;
+    content: string;
+    action: string;
+}, workspaceRoot: string): boolean;
+export interface WatcherOptions {
+    workspaceRoot: string;
+    onFileChange?: (relativePath: string, content: string | null, action: 'write' | 'delete') => void;
+}
+export declare class WorkspaceWatcher {
+    private watcher;
+    private workspaceRoot;
+    private onFileChange;
+    private _ready;
+    constructor(options: WatcherOptions);
+    start(): void;
+    stop(): void;
+    get isReady(): boolean;
+    private _handleChange;
+}
+export interface WorkspaceWSOptions {
+    serverUrl: string;
+    token: string;
+    contextId: string;
+    workspaceRoot: string;
+    onMutation?: (event: any) => void;
+}
+export declare class WorkspaceWSClient {
+    private ws;
+    private opts;
+    private reconnectTimer;
+    private _connected;
+    private _queue;
+    private _maxQueue;
+    constructor(opts: WorkspaceWSOptions);
+    connect(): void;
+    disconnect(): void;
+    /**
+     * Push a local file change to V3's workspace.
+     */
+    syncFile(relativePath: string, content: string | null, action: 'write' | 'delete'): void;
+    /**
+     * Push multiple files at once.
+     */
+    syncBatch(files: Array<{
+        path: string;
+        content: string;
+    }>): void;
+    get isConnected(): boolean;
+    private _send;
+}

package/dist/utils/workspace-stream.js

@@ -0,0 +1,301 @@
+"use strict";
+/**
+ * Workspace Stream — Real-time bidirectional file sync between CLI and V3.
+ *
+ * Architecture:
+ *   1. WorkspaceWatcher — chokidar-based file watcher that detects local changes
+ *      and pushes them to V3 via WebSocket.
+ *   2. MutationApplier — receives file_mutation events from V3's SSE stream
+ *      and applies them to the local workspace IN REAL TIME (not at completion).
+ *   3. WorkspaceWSClient — WebSocket client connected to V3's /ws/workspace
+ *      for bidirectional file sync outside the SSE stream.
+ *
+ * Event types (V3 → CLI via SSE):
+ *   {"type": "file_mutation", "path": "rel/path", "content": "...", "action": "write"|"edit", "tool": "write_file"|"edit_file"}
+ *
+ * Event types (CLI → V3 via /ws/workspace):
+ *   {"type": "bind", "context_id": "...", "workspace_root": "..."}
+ *   {"type": "file_sync", "path": "rel/path", "content": "...", "action": "write"|"delete"}
+ *   {"type": "file_batch", "files": [{path, content}, ...]}
+ */
+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.WorkspaceWSClient = exports.WorkspaceWatcher = void 0;
+exports.applyFileMutation = applyFileMutation;
+const chokidar = __importStar(require("chokidar"));
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const ws_1 = __importDefault(require("ws"));
+const logger_js_1 = require("./logger.js");
+const logger = new logger_js_1.Logger();
+logger.setVerbose(!!process.env.VIGTHORIA_DEBUG);
+// Files/dirs to ignore in the watcher
+const IGNORE_PATTERNS = [
+    '**/node_modules/**',
+    '**/.git/**',
+    '**/__pycache__/**',
+    '**/.venv/**',
+    '**/dist/**',
+    '**/build/**',
+    '**/.next/**',
+    '**/.cache/**',
+    '**/.vigthoria/**',
+    '**/coverage/**',
+    '**/*.pyc',
+];
+const BINARY_EXTENSIONS = new Set([
+    '.png', '.jpg', '.jpeg', '.gif', '.ico', '.svg', '.webp', '.bmp',
+    '.mp3', '.mp4', '.wav', '.ogg', '.webm', '.avi',
+    '.zip', '.tar', '.gz', '.bz2', '.7z', '.rar',
+    '.woff', '.woff2', '.ttf', '.eot', '.otf',
+    '.pdf', '.doc', '.docx', '.xls', '.xlsx',
+    '.pyc', '.pyo', '.so', '.dll', '.exe', '.bin',
+    '.db', '.sqlite', '.sqlite3',
+]);
+const MAX_SYNC_FILE_BYTES = 500 * 1024; // 500 KB
+// ── Mutation Applier ─────────────────────────────────────────
+/**
+ * Tracks files that V3 has mutated so the watcher doesn't echo them back.
+ */
+const _v3MutatedFiles = new Set();
+const _v3MuteTimeout = 2000; // 2s mute window after V3 writes a file
+/**
+ * Apply a file_mutation event from V3's SSE stream to the local workspace.
+ * Call this from the SSE event handler for real-time file application.
+ */
+function applyFileMutation(event, workspaceRoot) {
+    if (event.type !== 'file_mutation')
+        return false;
+    if (!event.path || !workspaceRoot)
+        return false;
+    const absPath = path_1.default.resolve(workspaceRoot, event.path);
+    // Safety: ensure the resolved path is within the workspace
+    if (!absPath.startsWith(path_1.default.resolve(workspaceRoot) + path_1.default.sep) && absPath !== path_1.default.resolve(workspaceRoot)) {
+        logger.warn(`Refusing to apply mutation outside workspace: ${event.path}`);
+        return false;
+    }
+    try {
+        if (event.action === 'delete') {
+            if (fs_1.default.existsSync(absPath)) {
+                fs_1.default.unlinkSync(absPath);
+                logger.debug(`Deleted: ${event.path}`);
+            }
+        }
+        else if (typeof event.content === 'string') {
+            fs_1.default.mkdirSync(path_1.default.dirname(absPath), { recursive: true });
+            fs_1.default.writeFileSync(absPath, event.content, 'utf8');
+            // Mute the watcher for this file to prevent echo
+            _v3MutatedFiles.add(absPath);
+            setTimeout(() => _v3MutatedFiles.delete(absPath), _v3MuteTimeout);
+            logger.debug(`Applied: ${event.path} (${event.action})`);
+        }
+        return true;
+    }
+    catch (err) {
+        logger.error(`Failed to apply mutation for ${event.path}: ${err}`);
+        return false;
+    }
+}
+class WorkspaceWatcher {
+    watcher = null;
+    workspaceRoot;
+    onFileChange;
+    _ready = false;
+    constructor(options) {
+        this.workspaceRoot = path_1.default.resolve(options.workspaceRoot);
+        this.onFileChange = options.onFileChange;
+    }
+    start() {
+        if (this.watcher)
+            return;
+        this.watcher = chokidar.watch(this.workspaceRoot, {
+            ignored: IGNORE_PATTERNS,
+            persistent: true,
+            ignoreInitial: true,
+            awaitWriteFinish: { stabilityThreshold: 300, pollInterval: 100 },
+        });
+        this.watcher
+            .on('ready', () => {
+            this._ready = true;
+            logger.debug('Workspace watcher ready');
+        })
+            .on('add', (filePath) => this._handleChange(filePath, 'write'))
+            .on('change', (filePath) => this._handleChange(filePath, 'write'))
+            .on('unlink', (filePath) => this._handleChange(filePath, 'delete'));
+    }
+    stop() {
+        if (this.watcher) {
+            this.watcher.close();
+            this.watcher = null;
+            this._ready = false;
+        }
+    }
+    get isReady() {
+        return this._ready;
+    }
+    _handleChange(filePath, action) {
+        // Skip if this file was just written by V3 (echo prevention)
+        if (_v3MutatedFiles.has(filePath))
+            return;
+        const ext = path_1.default.extname(filePath).toLowerCase();
+        if (BINARY_EXTENSIONS.has(ext))
+            return;
+        const relativePath = path_1.default.relative(this.workspaceRoot, filePath);
+        if (relativePath.startsWith('..'))
+            return; // safety
+        if (action === 'delete') {
+            this.onFileChange?.(relativePath, null, 'delete');
+            return;
+        }
+        try {
+            const stat = fs_1.default.statSync(filePath);
+            if (stat.size > MAX_SYNC_FILE_BYTES)
+                return;
+            const content = fs_1.default.readFileSync(filePath, 'utf8');
+            this.onFileChange?.(relativePath, content, 'write');
+        }
+        catch {
+            // File might have been deleted between event and read
+        }
+    }
+}
+exports.WorkspaceWatcher = WorkspaceWatcher;
+class WorkspaceWSClient {
+    ws = null;
+    opts;
+    reconnectTimer = null;
+    _connected = false;
+    _queue = [];
+    _maxQueue = 200;
+    constructor(opts) {
+        this.opts = opts;
+    }
+    connect() {
+        if (this.ws)
+            return;
+        const url = `${this.opts.serverUrl}/ws/workspace?token=${encodeURIComponent(this.opts.token)}`;
+        this.ws = new ws_1.default(url);
+        this.ws.on('open', () => {
+            this._connected = true;
+            // Bind to workspace
+            this._send({
+                type: 'bind',
+                context_id: this.opts.contextId,
+                workspace_root: this.opts.workspaceRoot,
+            });
+            // Flush queued messages
+            while (this._queue.length > 0) {
+                const msg = this._queue.shift();
+                this._send(msg);
+            }
+            logger.debug('WS workspace connected');
+        });
+        this.ws.on('message', (data) => {
+            try {
+                const event = JSON.parse(data.toString());
+                if (event.type === 'file_mutation') {
+                    this.opts.onMutation?.(event);
+                }
+            }
+            catch {
+                // ignore parse errors
+            }
+        });
+        this.ws.on('close', () => {
+            this._connected = false;
+            this.ws = null;
+            // Auto-reconnect after 3s
+            this.reconnectTimer = setTimeout(() => this.connect(), 3000);
+        });
+        this.ws.on('error', () => {
+            // error handler to prevent crash, close event handles reconnect
+        });
+    }
+    disconnect() {
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+        if (this.ws) {
+            this.ws.close();
+            this.ws = null;
+        }
+        this._connected = false;
+    }
+    /**
+     * Push a local file change to V3's workspace.
+     */
+    syncFile(relativePath, content, action) {
+        const msg = {
+            type: 'file_sync',
+            path: relativePath,
+            content: content || '',
+            action,
+        };
+        if (this._connected) {
+            this._send(msg);
+        }
+        else if (this._queue.length < this._maxQueue) {
+            this._queue.push(msg);
+        }
+    }
+    /**
+     * Push multiple files at once.
+     */
+    syncBatch(files) {
+        const msg = { type: 'file_batch', files };
+        if (this._connected) {
+            this._send(msg);
+        }
+        else if (this._queue.length < this._maxQueue) {
+            this._queue.push(msg);
+        }
+    }
+    get isConnected() {
+        return this._connected;
+    }
+    _send(msg) {
+        try {
+            this.ws?.send(JSON.stringify(msg));
+        }
+        catch {
+            // connection lost
+        }
+    }
+}
+exports.WorkspaceWSClient = WorkspaceWSClient;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vigthoria-cli",
-  "version": "1.6.53",
+  "version": "1.6.54",
   "description": "Vigthoria Coder CLI - AI-powered terminal coding assistant",
   "main": "dist/index.js",
   "files": [
@@ -61,6 +61,7 @@
     "archiver": "^6.0.1",
     "axios": "^1.6.0",
     "chalk": "^5.3.0",
+    "chokidar": "^5.0.0",
     "commander": "^11.1.0",
     "conf": "^12.0.0",
     "diff": "^5.1.0",
@@ -86,4 +87,4 @@
   "engines": {
     "node": ">=18.0.0"
   }
-}
+}