@coderule/mcp 2.0.0 → 2.0.2

package/dist/mcp-cli.cjs

@@ -1441,6 +1441,21 @@ async function runInitialSyncPipeline(runtime, options) {
       hashLogger.debug("Hasher processed batch");
     }
   }
+  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(
@@ -2266,6 +2281,9 @@ function printUsage() {
   console.log(
     "  --clean, --reindex           Remove existing local state before running"
   );
+  console.log(
+    "  --exit                       Exit after reindex (use with --reindex)"
+  );
   console.log(
     "  --inline-hasher              Force inline hashing (debug only)"
   );
@@ -2313,6 +2331,7 @@ function parseArgs(argv) {
   let token = process.env.CODERULE_TOKEN;
   let rootPath;
   let clean = false;
+  let exit = false;
   let inlineHasher = false;
   const env = {};
   const args = [...argv];
@@ -2326,6 +2345,10 @@ function parseArgs(argv) {
       clean = true;
       continue;
     }
+    if (arg === "--exit") {
+      exit = true;
+      continue;
+    }
     if (arg === "--inline-hasher") {
       inlineHasher = true;
       continue;
@@ -2382,7 +2405,7 @@ function parseArgs(argv) {
       "Missing token. Provide via argument or CODERULE_TOKEN environment variable."
     );
   }
-  return { token, rootPath, clean, inlineHasher, env };
+  return { token, rootPath, clean, exit, inlineHasher, env };
 }
 async function ensureClean(configToken, rootPath) {
   const config = await resolveConfig({ token: configToken, rootPath });
@@ -2435,6 +2458,25 @@ async function main() {
       rootPath: options.rootPath
     });
     const runner = new ServiceRunner(runtime);
+    if (options.exit && options.clean) {
+      try {
+        const initial = await runInitialSyncPipeline(runtime, {
+          blockUntilReady: true
+        });
+        runtime.logger.info(
+          {
+            snapshotHash: initial.snapshotHash,
+            filesCount: initial.filesCount
+          },
+          "Reindex completed; exiting"
+        );
+      } finally {
+        await runtime.hasher.close();
+        runtime.clients.close();
+        runtime.db.close();
+      }
+      return;
+    }
     try {
       await runner.prepareWatcher(true);
       const server = createMcpServer({ runtime, runner });