viberag 0.1.0

package/dist/mcp/index.js

@@ -0,0 +1,66 @@
+#!/usr/bin/env node
+/**
+ * VibeRAG MCP Server Entry Point
+ *
+ * Exposes VibeRAG functionality via Model Context Protocol.
+ * Uses current working directory as project root.
+ * Includes file watcher for automatic incremental indexing.
+ *
+ * Usage with Claude Code:
+ *   claude mcp add viberag -- npx viberag-mcp
+ */
+import { createMcpServer } from './server.js';
+import { configExists, Indexer } from '../rag/index.js';
+// Use current working directory as project root (same behavior as CLI)
+const projectRoot = process.cwd();
+const { server, startWatcher, stopWatcher } = createMcpServer(projectRoot);
+// Handle shutdown signals
+async function shutdown(signal) {
+    console.error(`[viberag-mcp] Received ${signal}, shutting down...`);
+    await stopWatcher();
+    process.exit(0);
+}
+process.on('SIGINT', () => shutdown('SIGINT'));
+process.on('SIGTERM', () => shutdown('SIGTERM'));
+// Start the server, then start the watcher
+server.start({
+    transportType: 'stdio',
+});
+// Start watcher and sync index after server is running
+// Use setImmediate to ensure server.start() completes first
+setImmediate(async () => {
+    try {
+        // Start watcher first (will queue any changes during sync)
+        await startWatcher();
+    }
+    catch (error) {
+        // Watcher errors shouldn't crash the server
+        console.error('[viberag-mcp] Failed to start watcher:', error instanceof Error ? error.message : error);
+    }
+    // Sync index on startup if project is initialized
+    // This catches any changes made while MCP server was offline
+    try {
+        if (await configExists(projectRoot)) {
+            console.error('[viberag-mcp] Running startup sync...');
+            const indexer = new Indexer(projectRoot);
+            try {
+                const stats = await indexer.index({ force: false });
+                if (stats.filesNew > 0 ||
+                    stats.filesModified > 0 ||
+                    stats.filesDeleted > 0) {
+                    console.error(`[viberag-mcp] Startup sync complete: ${stats.filesNew} new, ${stats.filesModified} modified, ${stats.filesDeleted} deleted`);
+                }
+                else {
+                    console.error('[viberag-mcp] Startup sync: index up to date');
+                }
+            }
+            finally {
+                indexer.close();
+            }
+        }
+    }
+    catch (error) {
+        // Sync errors shouldn't crash the server
+        console.error('[viberag-mcp] Startup sync failed:', error instanceof Error ? error.message : error);
+    }
+});

package/dist/mcp/server.d.ts

@@ -0,0 +1,25 @@
+/**
+ * MCP Server for VibeRAG
+ *
+ * Exposes RAG functionality via Model Context Protocol.
+ * Tools: codebase_search, codebase_parallel_search, viberag_index, viberag_status, viberag_watch_status
+ *
+ * Includes file watcher for automatic incremental indexing.
+ */
+import { FastMCP } from 'fastmcp';
+import { FileWatcher } from './watcher.js';
+/**
+ * MCP server with file watcher.
+ */
+export interface McpServerWithWatcher {
+    server: FastMCP;
+    watcher: FileWatcher;
+    /** Start the watcher (call after server.start) */
+    startWatcher: () => Promise<void>;
+    /** Stop the watcher (call before exit) */
+    stopWatcher: () => Promise<void>;
+}
+/**
+ * Create and configure the MCP server with file watcher.
+ */
+export declare function createMcpServer(projectRoot: string): McpServerWithWatcher;