@coderule/mcp 2.1.0 → 2.2.1

package/dist/mcp-cli.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
-import fs5 from 'fs/promises';
-import path2 from 'path';
+import fs5, { mkdir, writeFile } from 'fs/promises';
+import path2, { dirname } from 'path';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { createHash } from 'crypto';
 import envPaths from 'env-paths';
@@ -2397,6 +2397,145 @@ ${statusText}`;
       return { content: [{ type: "text", text }] };
     }
   );
+  server.registerTool(
+    "dump_bush",
+    {
+      title: "Dump Raw Index (Debug)",
+      description: "Debug tool. Downloads the raw HDF5 bush file for the current snapshot and saves it to the specified path. Returns an HDF5 bush reader skill with format documentation and Python code examples.",
+      inputSchema: {
+        filePath: z.string().min(1, "File path is required").refine((p) => p.endsWith(".h5"), "File must have .h5 extension").describe(
+          "Full file name to save (use base dir + any name with .h5 extension), e.g. /tmp/snapshot.h5"
+        )
+      },
+      annotations: {
+        readOnlyHint: false,
+        destructiveHint: false,
+        idempotentHint: true,
+        openWorldHint: true
+      }
+    },
+    async ({ filePath }) => {
+      const deadline = Date.now() + runtime.config.maxQueryWaitMs;
+      while (Date.now() < deadline) {
+        if (isLocalStable()) break;
+        await sleep3(250);
+      }
+      if (!isLocalStable()) {
+        const statusText2 = formatStatus(
+          await collectIndexingStatus(runtime, runner)
+        );
+        return {
+          content: [
+            {
+              type: "text",
+              text: `Indexer not ready (local hashing/dirty). Current status:
+
+${statusText2}`
+            }
+          ],
+          isError: true
+        };
+      }
+      let currentHash = computeSnapshot(runtime.filesRepo).snapshotHash;
+      let lastStatus;
+      while (Date.now() < deadline) {
+        if (!isLocalStable()) {
+          while (Date.now() < deadline) {
+            if (isLocalStable()) break;
+            await sleep3(250);
+          }
+          if (!isLocalStable()) break;
+          currentHash = computeSnapshot(runtime.filesRepo).snapshotHash;
+          continue;
+        }
+        try {
+          const status = await withDeadline2(
+            runtime.clients.sync.checkSnapshotStatus(currentHash),
+            deadline
+          );
+          lastStatus = status.status;
+          if (status.status === "READY") {
+            try {
+              const bushBuffer = await withDeadline2(
+                runtime.clients.sync.downloadSnapshotBush(currentHash),
+                deadline
+              );
+              await mkdir(dirname(filePath), { recursive: true });
+              await writeFile(filePath, Buffer.from(bushBuffer));
+              runtime.logger.info(
+                {
+                  filePath,
+                  snapshotHash: currentHash,
+                  bytes: bushBuffer.byteLength
+                },
+                "HDF5 bush file saved"
+              );
+              const skill = await withDeadline2(
+                runtime.clients.sync.getHdf5BushReaderSkill(),
+                deadline
+              );
+              return {
+                content: [
+                  {
+                    type: "text",
+                    text: `HDF5 bush saved to ${filePath} (${bushBuffer.byteLength} bytes, snapshot ${currentHash}).
+
+${skill}`
+                  }
+                ]
+              };
+            } catch (error) {
+              const message = error instanceof Error ? error.message : "Unknown error";
+              runtime.logger.error(
+                { err: error, snapshotHash: currentHash },
+                "Bush download failed"
+              );
+              return {
+                content: [
+                  {
+                    type: "text",
+                    text: `Failed to download bush for snapshot ${currentHash}: ${message}`
+                  }
+                ],
+                isError: true
+              };
+            }
+          }
+          if (status.status === "FAILED") {
+            while (Date.now() < deadline) {
+              if (isLocalStable()) break;
+              await sleep3(250);
+            }
+            if (!isLocalStable()) break;
+            currentHash = computeSnapshot(runtime.filesRepo).snapshotHash;
+            continue;
+          }
+          runtime.outbox.enqueueSnapshot(runtime.config.rootId, 0);
+        } catch {
+        }
+        const newHash = computeSnapshot(runtime.filesRepo).snapshotHash;
+        if (newHash !== currentHash) {
+          currentHash = newHash;
+          continue;
+        }
+        await sleep3(500);
+      }
+      const statusText = formatStatus(
+        await collectIndexingStatus(runtime, runner)
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Snapshot not READY before deadline (last status: ${lastStatus ?? "unknown"}). Current status:
+
+${statusText}`
+          }
+        ],
+        isError: true
+      };
+    }
+  );
   return server;
 }