mcp-perplexity-pro 1.1.0

package/dist/storage.js

@@ -0,0 +1,328 @@
+import { promises as fs } from 'fs';
+import path from 'path';
+import { v4 as uuidv4 } from 'uuid';
+import lockfile from 'proper-lockfile';
+export class StorageError extends Error {
+    code;
+    constructor(message, code) {
+        super(message);
+        this.code = code;
+        this.name = 'StorageError';
+    }
+}
+export class StorageManager {
+    projectRoot;
+    storagePath;
+    sessionId;
+    constructor(config) {
+        this.projectRoot = config.project_root;
+        this.storagePath = path.join(config.project_root, config.storage_path);
+        if (config.session_id) {
+            this.sessionId = config.session_id;
+        }
+    }
+    /**
+     * Ensures the storage directory exists
+     */
+    async ensureStorageDirectory() {
+        try {
+            await fs.mkdir(this.storagePath, { recursive: true });
+        }
+        catch (error) {
+            throw new StorageError(`Failed to create storage directory: ${this.storagePath}`, 'STORAGE_INIT_ERROR');
+        }
+    }
+    /**
+     * Gets the full path for a conversation file
+     */
+    getConversationPath(chatId) {
+        // Include session ID if provided for multi-session support
+        const filename = this.sessionId ? `${this.sessionId}_${chatId}.json` : `${chatId}.json`;
+        return path.join(this.storagePath, filename);
+    }
+    /**
+     * Gets the path for a research report
+     */
+    getReportPath(reportId) {
+        const filename = this.sessionId
+            ? `${this.sessionId}_report_${reportId}.md`
+            : `report_${reportId}.md`;
+        return path.join(this.storagePath, filename);
+    }
+    /**
+     * Executes a function with file locking for thread safety
+     */
+    async withLock(filePath, operation) {
+        try {
+            // Ensure directory exists for both the file and lock
+            const fileDir = path.dirname(filePath);
+            await fs.mkdir(fileDir, { recursive: true });
+            // Ensure the target file exists (create empty file if it doesn't exist)
+            // This is needed for proper-lockfile to work correctly
+            try {
+                await fs.access(filePath);
+            }
+            catch {
+                // File doesn't exist, create empty file
+                await fs.writeFile(filePath, '', 'utf-8');
+            }
+            // Acquire lock with timeout
+            await lockfile.lock(filePath, {
+                stale: 10000, // 10 seconds
+                retries: 3,
+            });
+            try {
+                return await operation();
+            }
+            finally {
+                // Always release the lock
+                await lockfile.unlock(filePath);
+            }
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.includes('EEXIST')) {
+                throw new StorageError('Storage operation timed out - another process may be accessing the file', 'LOCK_TIMEOUT');
+            }
+            throw error;
+        }
+    }
+    /**
+     * Creates a new conversation
+     */
+    async createConversation(title, model, initialMessage) {
+        await this.ensureStorageDirectory();
+        const chatId = uuidv4();
+        const now = new Date().toISOString();
+        const conversation = {
+            metadata: {
+                id: chatId,
+                title,
+                created_at: now,
+                updated_at: now,
+                message_count: initialMessage ? 1 : 0,
+                model,
+            },
+            messages: initialMessage ? [initialMessage] : [],
+        };
+        const filePath = this.getConversationPath(chatId);
+        await this.withLock(filePath, async () => {
+            await fs.writeFile(filePath, JSON.stringify(conversation, null, 2), 'utf-8');
+        });
+        return chatId;
+    }
+    /**
+     * Adds a message to an existing conversation
+     */
+    async addMessage(chatId, message) {
+        const filePath = this.getConversationPath(chatId);
+        await this.withLock(filePath, async () => {
+            try {
+                const data = await fs.readFile(filePath, 'utf-8');
+                const conversation = JSON.parse(data);
+                conversation.messages.push(message);
+                conversation.metadata.updated_at = new Date().toISOString();
+                conversation.metadata.message_count = conversation.messages.length;
+                await fs.writeFile(filePath, JSON.stringify(conversation, null, 2), 'utf-8');
+            }
+            catch (error) {
+                if (error instanceof Error && error.message.includes('ENOENT')) {
+                    throw new StorageError(`Conversation not found: ${chatId}`, 'CONVERSATION_NOT_FOUND');
+                }
+                throw new StorageError(`Failed to add message to conversation: ${chatId}`, 'WRITE_ERROR');
+            }
+        });
+    }
+    /**
+     * Retrieves a conversation by ID
+     */
+    async getConversation(chatId) {
+        const filePath = this.getConversationPath(chatId);
+        try {
+            const data = await fs.readFile(filePath, 'utf-8');
+            return JSON.parse(data);
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.includes('ENOENT')) {
+                throw new StorageError(`Conversation not found: ${chatId}`, 'CONVERSATION_NOT_FOUND');
+            }
+            throw new StorageError(`Failed to read conversation: ${chatId}`, 'READ_ERROR');
+        }
+    }
+    /**
+     * Lists all conversations in the current project
+     */
+    async listConversations() {
+        await this.ensureStorageDirectory();
+        try {
+            const files = await fs.readdir(this.storagePath);
+            const conversations = [];
+            // Filter for conversation files (not reports)
+            const conversationFiles = files.filter(file => {
+                if (this.sessionId) {
+                    return (file.startsWith(`${this.sessionId}_`) &&
+                        file.endsWith('.json') &&
+                        !file.includes('report_'));
+                }
+                return file.endsWith('.json') && !file.includes('report_');
+            });
+            // Read metadata from each conversation
+            for (const file of conversationFiles) {
+                try {
+                    const filePath = path.join(this.storagePath, file);
+                    const data = await fs.readFile(filePath, 'utf-8');
+                    const conversation = JSON.parse(data);
+                    conversations.push(conversation.metadata);
+                }
+                catch (error) {
+                    // Skip corrupted files but don't fail the entire operation
+                    console.warn(`Warning: Could not read conversation file ${file}:`, error);
+                }
+            }
+            // Sort by updated_at descending (most recent first)
+            return conversations.sort((a, b) => new Date(b.updated_at).getTime() - new Date(a.updated_at).getTime());
+        }
+        catch (error) {
+            throw new StorageError('Failed to list conversations', 'LIST_ERROR');
+        }
+    }
+    /**
+     * Deletes a conversation
+     */
+    async deleteConversation(chatId) {
+        const filePath = this.getConversationPath(chatId);
+        try {
+            await fs.unlink(filePath);
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.includes('ENOENT')) {
+                throw new StorageError(`Conversation not found: ${chatId}`, 'CONVERSATION_NOT_FOUND');
+            }
+            throw new StorageError(`Failed to delete conversation: ${chatId}`, 'DELETE_ERROR');
+        }
+    }
+    /**
+     * Saves a research report to the project directory
+     */
+    async saveReport(content, title) {
+        await this.ensureStorageDirectory();
+        const reportId = uuidv4();
+        const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+        const filename = `${timestamp}_${title.replace(/[^a-zA-Z0-9]/g, '_')}.md`;
+        const reportPath = path.join(this.storagePath, filename);
+        const reportContent = `# ${title}
+
+**Generated:** ${new Date().toLocaleString()}  
+**Report ID:** ${reportId}
+
+---
+
+${content}
+`;
+        try {
+            await fs.writeFile(reportPath, reportContent, 'utf-8');
+            return reportId;
+        }
+        catch (error) {
+            throw new StorageError('Failed to save research report', 'SAVE_REPORT_ERROR');
+        }
+    }
+    /**
+     * Gets storage statistics for the current project
+     */
+    async getStorageStats() {
+        await this.ensureStorageDirectory();
+        try {
+            const conversations = await this.listConversations();
+            let totalMessages = 0;
+            let storageSize = 0;
+            let lastActivity = null;
+            // Calculate total messages and find last activity
+            for (const conv of conversations) {
+                totalMessages += conv.message_count;
+                if (!lastActivity || new Date(conv.updated_at) > new Date(lastActivity)) {
+                    lastActivity = conv.updated_at;
+                }
+            }
+            // Calculate storage size
+            const files = await fs.readdir(this.storagePath);
+            for (const file of files) {
+                try {
+                    const filePath = path.join(this.storagePath, file);
+                    const stats = await fs.stat(filePath);
+                    storageSize += stats.size;
+                }
+                catch (error) {
+                    // Skip files we can't access
+                }
+            }
+            return {
+                total_conversations: conversations.length,
+                total_messages: totalMessages,
+                storage_size_bytes: storageSize,
+                last_activity: lastActivity,
+            };
+        }
+        catch (error) {
+            throw new StorageError('Failed to get storage statistics', 'STATS_ERROR');
+        }
+    }
+    /**
+     * Handles storage errors and creates structured error responses
+     */
+    static handleError(error) {
+        if (error instanceof StorageError) {
+            switch (error.code) {
+                case 'CONVERSATION_NOT_FOUND':
+                    return {
+                        error: {
+                            type: 'storage_error',
+                            message: error.message,
+                            details: {
+                                suggestion: 'Check the conversation ID and try again, or list conversations to see available IDs',
+                            },
+                        },
+                    };
+                case 'LOCK_TIMEOUT':
+                    return {
+                        error: {
+                            type: 'storage_error',
+                            message: error.message,
+                            details: {
+                                suggestion: 'Try again in a moment - another operation may be in progress',
+                            },
+                        },
+                    };
+                case 'STORAGE_INIT_ERROR':
+                    return {
+                        error: {
+                            type: 'storage_error',
+                            message: error.message,
+                            details: {
+                                suggestion: 'Check that the project_root path exists and is writable',
+                            },
+                        },
+                    };
+                default:
+                    return {
+                        error: {
+                            type: 'storage_error',
+                            message: error.message,
+                            details: {
+                                suggestion: 'Check file permissions and available disk space',
+                            },
+                        },
+                    };
+            }
+        }
+        return {
+            error: {
+                type: 'storage_error',
+                message: error instanceof Error ? error.message : 'Unknown storage error',
+                details: {
+                    suggestion: 'Check the storage path configuration and file permissions',
+                },
+            },
+        };
+    }
+}
+//# sourceMappingURL=storage.js.map

package/dist/storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.js","sourceRoot":"","sources":["../src/storage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,QAAQ,MAAM,iBAAiB,CAAC;AAUvC,MAAM,OAAO,YAAa,SAAQ,KAAK;IAG5B;IAFT,YACE,OAAe,EACR,IAAa;QAEpB,KAAK,CAAC,OAAO,CAAC,CAAC;QAFR,SAAI,GAAJ,IAAI,CAAS;QAGpB,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,OAAO,cAAc;IACjB,WAAW,CAAS;IACpB,WAAW,CAAS;IACpB,SAAS,CAAU;IAE3B,YAAY,MAAc;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,YAAY,CAAC;QACvC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACvE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,sBAAsB;QAClC,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,YAAY,CACpB,uCAAuC,IAAI,CAAC,WAAW,EAAE,EACzD,oBAAoB,CACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,MAAc;QACxC,2DAA2D;QAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,IAAI,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,OAAO,CAAC;QACxF,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,QAAgB;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS;YAC7B,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,WAAW,QAAQ,KAAK;YAC3C,CAAC,CAAC,UAAU,QAAQ,KAAK,CAAC;QAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,QAAQ,CAAI,QAAgB,EAAE,SAA2B;QACrE,IAAI,CAAC;YACH,qDAAqD;YACrD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACvC,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE7C,wEAAwE;YACxE,uDAAuD;YACvD,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC5B,CAAC;YAAC,MAAM,CAAC;gBACP,wCAAwC;gBACxC,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;YAC5C,CAAC;YAED,4BAA4B;YAC5B,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAC5B,KAAK,EAAE,KAAK,EAAE,aAAa;gBAC3B,OAAO,EAAE,CAAC;aACX,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,OAAO,MAAM,SAAS,EAAE,CAAC;YAC3B,CAAC;oBAAS,CAAC;gBACT,0BAA0B;gBAC1B,MAAM,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/D,MAAM,IAAI,YAAY,CACpB,yEAAyE,EACzE,cAAc,CACf,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CACtB,KAAa,EACb,KAAsB,EACtB,cAAwB;QAExB,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAEpC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,MAAM,YAAY,GAAiB;YACjC,QAAQ,EAAE;gBACR,EAAE,EAAE,MAAM;gBACV,KAAK;gBACL,UAAU,EAAE,GAAG;gBACf,UAAU,EAAE,GAAG;gBACf,aAAa,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrC,KAAK;aACN;YACD,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE;SACjD,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAElD,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,OAAgB;QAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAElD,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YACvC,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAClD,MAAM,YAAY,GAAiB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEpD,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpC,YAAY,CAAC,QAAQ,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;gBAC5D,YAAY,CAAC,QAAQ,CAAC,aAAa,GAAG,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAEnE,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAC/E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC/D,MAAM,IAAI,YAAY,CAAC,2BAA2B,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC;gBACxF,CAAC;gBACD,MAAM,IAAI,YAAY,CAAC,0CAA0C,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC;YAC5F,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,MAAc;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAElD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAClD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAiB,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/D,MAAM,IAAI,YAAY,CAAC,2BAA2B,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC;YACxF,CAAC;YACD,MAAM,IAAI,YAAY,CAAC,gCAAgC,MAAM,EAAE,EAAE,YAAY,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB;QACrB,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAEpC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACjD,MAAM,aAAa,GAAmB,EAAE,CAAC;YAEzC,8CAA8C;YAC9C,MAAM,iBAAiB,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;gBAC5C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,OAAO,CACL,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC;wBACrC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;wBACtB,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAC1B,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAC7D,CAAC,CAAC,CAAC;YAEH,uCAAuC;YACvC,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,CAAC;gBACrC,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;oBACnD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBAClD,MAAM,YAAY,GAAiB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACpD,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;gBAC5C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,2DAA2D;oBAC3D,OAAO,CAAC,IAAI,CAAC,6CAA6C,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC5E,CAAC;YACH,CAAC;YAED,oDAAoD;YACpD,OAAO,aAAa,CAAC,IAAI,CACvB,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAC9E,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,YAAY,CAAC,8BAA8B,EAAE,YAAY,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,MAAc;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAElD,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/D,MAAM,IAAI,YAAY,CAAC,2BAA2B,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC;YACxF,CAAC;YACD,MAAM,IAAI,YAAY,CAAC,kCAAkC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC;QACrF,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,OAAe,EAAE,KAAa;QAC7C,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAEpC,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC;QAC1B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,KAAK,CAAC;QAC1E,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,KAAK,KAAK;;iBAEnB,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE;iBAC3B,QAAQ;;;;EAIvB,OAAO;CACR,CAAC;QAEE,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;YACvD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,YAAY,CAAC,gCAAgC,EAAE,mBAAmB,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe;QAMnB,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAEpC,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrD,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,YAAY,GAAkB,IAAI,CAAC;YAEvC,kDAAkD;YAClD,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBACjC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC;gBACpC,IAAI,CAAC,YAAY,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;oBACxE,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC;gBACjC,CAAC;YACH,CAAC;YAED,yBAAyB;YACzB,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACjD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;oBACnD,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACtC,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC;gBAC5B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,6BAA6B;gBAC/B,CAAC;YACH,CAAC;YAED,OAAO;gBACL,mBAAmB,EAAE,aAAa,CAAC,MAAM;gBACzC,cAAc,EAAE,aAAa;gBAC7B,kBAAkB,EAAE,WAAW;gBAC/B,aAAa,EAAE,YAAY;aAC5B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,YAAY,CAAC,kCAAkC,EAAE,aAAa,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,WAAW,CAAC,KAAc;QAC/B,IAAI,KAAK,YAAY,YAAY,EAAE,CAAC;YAClC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,wBAAwB;oBAC3B,OAAO;wBACL,KAAK,EAAE;4BACL,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE;gCACP,UAAU,EACR,qFAAqF;6BACxF;yBACF;qBACF,CAAC;gBAEJ,KAAK,cAAc;oBACjB,OAAO;wBACL,KAAK,EAAE;4BACL,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE;gCACP,UAAU,EAAE,8DAA8D;6BAC3E;yBACF;qBACF,CAAC;gBAEJ,KAAK,oBAAoB;oBACvB,OAAO;wBACL,KAAK,EAAE;4BACL,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE;gCACP,UAAU,EAAE,yDAAyD;6BACtE;yBACF;qBACF,CAAC;gBAEJ;oBACE,OAAO;wBACL,KAAK,EAAE;4BACL,IAAI,EAAE,eAAe;4BACrB,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE;gCACP,UAAU,EAAE,iDAAiD;6BAC9D;yBACF;qBACF,CAAC;YACN,CAAC;QACH,CAAC;QAED,OAAO;YACL,KAAK,EAAE;gBACL,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB;gBACzE,OAAO,EAAE;oBACP,UAAU,EAAE,2DAA2D;iBACxE;aACF;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/streaming-wrapper.d.ts

@@ -0,0 +1,63 @@
+import type { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import type { Config } from './types.js';
+/**
+ * Universal async tool wrapper that creates async jobs and streams progress to Claude Code
+ */
+export declare class StreamingWrapper {
+    private server;
+    private config;
+    constructor(server: Server, config: Config);
+    /**
+     * Execute a tool call with streaming content updates (not just progress)
+     */
+    executeWithStreaming(toolName: string, params: any, progressToken?: string | number): Promise<any>;
+    /**
+     * Create async job based on tool type
+     */
+    private createAsyncJob;
+    /**
+     * Execute tool synchronously (fallback for no progress token)
+     */
+    private executeSynchronously;
+    /**
+     * Start streaming actual content from async job (not just progress)
+     */
+    private startContentStreaming;
+    /**
+     * Stream a chunk of content to Claude Code
+     */
+    private streamContentChunk;
+    /**
+     * Legacy method - keeping for backwards compatibility
+     */
+    private startProgressStreaming;
+    /**
+     * Send progress notification to Claude Code
+     */
+    private sendProgressNotification;
+    /**
+     * Handle job completion and send final result
+     */
+    private handleJobCompletion;
+    /**
+     * Handle job failure
+     */
+    private handleJobFailure;
+    /**
+     * Process completed job based on tool type
+     */
+    private processCompletedJob;
+    /**
+     * Get status message for progress updates
+     */
+    private getStatusMessage;
+    /**
+     * Cancel streaming for a job
+     */
+    cancelStreaming(jobId: string): void;
+    /**
+     * Get active streaming jobs
+     */
+    getActiveStreams(): string[];
+}
+//# sourceMappingURL=streaming-wrapper.d.ts.map

package/dist/streaming-wrapper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-wrapper.d.ts","sourceRoot":"","sources":["../src/streaming-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAGxE,OAAO,KAAK,EAAE,MAAM,EAA2B,MAAM,YAAY,CAAC;AAqBlE;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;IAK1C;;OAEG;IACG,oBAAoB,CACxB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,GAAG,EACX,aAAa,CAAC,EAAE,MAAM,GAAG,MAAM,GAC9B,OAAO,CAAC,GAAG,CAAC;IAqDf;;OAEG;YACW,cAAc;IA2C5B;;OAEG;YACW,oBAAoB;IAwBlC;;OAEG;YACW,qBAAqB;IAgGnC;;OAEG;YACW,kBAAkB;IA8BhC;;OAEG;YACW,sBAAsB;IA6FpC;;OAEG;YACW,wBAAwB;IAgCtC;;OAEG;YACW,mBAAmB;IA8BjC;;OAEG;YACW,gBAAgB;IAO9B;;OAEG;YACW,mBAAmB;IAqDjC;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAiBxB;;OAEG;IACI,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI3C;;OAEG;IACI,gBAAgB,IAAI,MAAM,EAAE;CAGpC"}