@coderule/mcp 1.6.0 → 1.6.2

package/dist/index.js

@@ -834,6 +834,9 @@ var Hasher = class {
       await this.worker.terminate();
     }
   }
+  async yieldToEventLoop() {
+    await new Promise((resolve) => setImmediate(resolve));
+  }
   resolveAbsolutePath(record) {
     if (path.isAbsolute(record.display_path)) {
       return record.display_path;
@@ -904,6 +907,9 @@ var Hasher = class {
       const absPath = this.resolveAbsolutePath(record);
       const exists = await this.ensureExists(absPath, record);
       if (!exists) {
+        if (this.inlineMode) {
+          await this.yieldToEventLoop();
+        }
         continue;
       }
       try {
@@ -928,6 +934,9 @@ var Hasher = class {
           failures.push(record.id);
         }
       }
+      if (this.inlineMode) {
+        await this.yieldToEventLoop();
+      }
     }
     if (successes.length) {
       this.log.debug({ count: successes.length }, "Hashing succeeded");
@@ -1224,7 +1233,7 @@ async function publishSnapshot(rootPath, filesRepo, snapshotsRepo, syncClient, l
 }
 
 // src/service/InitialSync.ts
-async function runInitialSyncPipeline(runtime) {
+async function runInitialSyncPipeline(runtime, options) {
   const inventoryLogger = runtime.logger.child({ scope: "inventory" });
   await runInventory({
     rootPath: runtime.config.rootPath,
@@ -1240,19 +1249,37 @@ async function runInitialSyncPipeline(runtime) {
       hashLogger.debug("Hasher processed batch");
     }
   }
-  const syncLogger = runtime.logger.child({ scope: "snapshot" });
-  const result = await publishSnapshot(
-    runtime.config.rootPath,
-    runtime.filesRepo,
-    runtime.snapshotsRepo,
-    runtime.clients.sync,
-    syncLogger,
-    {
-      maxAttempts: runtime.config.maxSnapshotAttempts,
-      uploadChunkSize: runtime.config.uploadChunkSize
-    }
+  if (options?.blockUntilReady !== false) {
+    const syncLogger = runtime.logger.child({ scope: "snapshot" });
+    const result = await publishSnapshot(
+      runtime.config.rootPath,
+      runtime.filesRepo,
+      runtime.snapshotsRepo,
+      runtime.clients.sync,
+      syncLogger,
+      {
+        maxAttempts: runtime.config.maxSnapshotAttempts,
+        uploadChunkSize: runtime.config.uploadChunkSize
+      }
+    );
+    return result;
+  }
+  const computation = computeSnapshot(runtime.filesRepo);
+  const createdAt = Date.now();
+  runtime.snapshotsRepo.insert(
+    computation.snapshotHash,
+    computation.filesCount,
+    computation.totalSize,
+    createdAt
   );
-  return result;
+  runtime.outbox.enqueueSnapshot(runtime.config.rootId, 0);
+  return {
+    snapshotHash: computation.snapshotHash,
+    filesCount: computation.filesCount,
+    totalSize: computation.totalSize,
+    status: "READY",
+    createdAt
+  };
 }
 async function createChokidarWatcher(options, usePolling) {
   const log = options.logger.child({
@@ -1716,7 +1743,9 @@ function awaitShutdownSignals() {
 async function runInitialSync(params) {
   const runtime = await bootstrap(params);
   try {
-    const result = await runInitialSyncPipeline(runtime);
+    const result = await runInitialSyncPipeline(runtime, {
+      blockUntilReady: true
+    });
     runtime.logger.info(
       {
         snapshotHash: result.snapshotHash,
@@ -1740,7 +1769,9 @@ async function runService(params) {
     await runner.prepareWatcher(true);
     let initialCreatedAt;
     try {
-      const initial = await runInitialSyncPipeline(runtime);
+      const initial = await runInitialSyncPipeline(runtime, {
+        blockUntilReady: false
+      });
       runtime.logger.info(
         {
           snapshotHash: initial.snapshotHash,