@j0hanz/code-review-analyst-mcp 1.4.1 → 1.4.2

package/dist/lib/tool-factory.d.ts

@@ -1,5 +1,6 @@
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { ZodRawShapeCompat } from '@modelcontextprotocol/sdk/server/zod-compat.js';
+import type { CallToolResult } from '@modelcontextprotocol/sdk/types.js';
 import { z } from 'zod';
 import { type DiffSlot } from './diff-store.js';
 import { createErrorToolResponse } from './tool-response.js';
@@ -17,6 +18,22 @@ export interface PromptParts {
 export interface ToolExecutionContext {
     readonly diffSlot: DiffSlot | undefined;
 }
+type ProgressToken = string | number;
+interface ProgressNotificationParams {
+    progressToken: ProgressToken;
+    progress: number;
+    total?: number;
+    message?: string;
+}
+interface ProgressExtra {
+    _meta?: {
+        progressToken?: unknown;
+    };
+    sendNotification: (notification: {
+        method: 'notifications/progress';
+        params: ProgressNotificationParams;
+    }) => Promise<void>;
+}
 export interface StructuredToolTaskConfig<TInput extends object = Record<string, unknown>, TResult extends object = Record<string, unknown>, TFinal extends TResult = TResult> {
     /** Tool name registered with the MCP server (e.g. 'analyze_pr_impact'). */
     name: string;
@@ -57,4 +74,9 @@ export interface StructuredToolTaskConfig<TInput extends object = Record<string,
     /** Builds the system instruction and user prompt from parsed tool input. */
     buildPrompt: (input: TInput, ctx: ToolExecutionContext) => PromptParts;
 }
+export declare function wrapToolHandler<TInput, TResult extends CallToolResult>(options: {
+    toolName: string;
+    progressContext?: (input: TInput) => string;
+}, handler: (input: TInput, extra: ProgressExtra) => Promise<TResult> | TResult): (input: TInput, extra: ProgressExtra) => Promise<TResult>;
 export declare function registerStructuredToolTask<TInput extends object, TResult extends object = Record<string, unknown>, TFinal extends TResult = TResult>(server: McpServer, config: StructuredToolTaskConfig<TInput, TResult, TFinal>): void;
+export {};

package/dist/lib/tool-factory.js

@@ -218,6 +218,43 @@ function createGeminiLogger(server, taskId) {
         }
     };
 }
+export function wrapToolHandler(options, handler) {
+    return async (input, extra) => {
+        const context = normalizeProgressContext(options.progressContext?.(input));
+        // Start progress (0/1)
+        await sendTaskProgress(extra, {
+            current: 0,
+            total: 1,
+            message: formatProgressStep(options.toolName, context, 'starting'),
+        });
+        try {
+            const result = await handler(input, extra);
+            // End progress (1/1)
+            const outcome = result.isError ? 'failed' : 'completed';
+            const success = !result.isError;
+            await sendTaskProgress(extra, {
+                current: 1,
+                total: 1,
+                message: formatProgressCompletion(options.toolName, context, outcome, success),
+            });
+            return result;
+        }
+        catch (error) {
+            // Progress is best-effort; must never mask the original error.
+            try {
+                await sendTaskProgress(extra, {
+                    current: 1,
+                    total: 1,
+                    message: formatProgressCompletion(options.toolName, context, 'failed', false),
+                });
+            }
+            catch {
+                // Swallow progress delivery errors so the original error propagates.
+            }
+            throw error;
+        }
+    };
+}
 export function registerStructuredToolTask(server, config) {
     const responseSchema = createGeminiResponseSchema({
         geminiSchema: config.geminiSchema,

package/dist/tools/generate-diff.js

@@ -3,6 +3,7 @@ import { z } from 'zod';
 import { cleanDiff, isEmptyDiff, NOISY_EXCLUDE_PATHSPECS, } from '../lib/diff-cleaner.js';
 import { computeDiffStatsFromFiles, parseDiffFiles, } from '../lib/diff-parser.js';
 import { DIFF_RESOURCE_URI, storeDiff } from '../lib/diff-store.js';
+import { wrapToolHandler } from '../lib/tool-factory.js';
 import { createErrorToolResponse, createToolResponse, } from '../lib/tool-response.js';
 const GIT_TIMEOUT_MS = 30_000;
 const GIT_MAX_BUFFER = 10 * 1024 * 1024; // 10 MB
@@ -29,6 +30,9 @@ export function registerGenerateDiffTool(server) {
                 .enum(['unstaged', 'staged'])
                 .describe('"unstaged": working-tree changes not yet staged. "staged": changes added to the index with git add.'),
         },
+    }, wrapToolHandler({
+        toolName: 'generate_diff',
+        progressContext: (input) => input.mode,
     }, (input) => {
         const { mode } = input;
         const args = buildGitArgs(mode);
@@ -67,5 +71,5 @@ export function registerGenerateDiffTool(server) {
                 message: summary,
             },
         }, summary);
-    });
+    }));
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@j0hanz/code-review-analyst-mcp",
-  "version": "1.4.1",
+  "version": "1.4.2",
   "mcpName": "io.github.j0hanz/code-review-analyst",
   "description": "Gemini-powered MCP server for code review analysis.",
   "type": "module",