viewgate-mcp 1.0.48 → 1.0.50

package/dist/index.js

@@ -53,6 +53,7 @@ function createMcpServer(apiKey, personalKey) {
         step: 0,
         startedAt: Date.now(),
         lastActivityAt: Date.now(),
+        lockTokens: {},
     };
     function resetGuard() {
         guard.flow = "idle";
@@ -60,6 +61,8 @@ function createMcpServer(apiKey, personalKey) {
         guard.lastTool = undefined;
         guard.startedAt = Date.now();
         guard.lastActivityAt = Date.now();
+        guard.lockTokens = {};
+        guard.backlogVersion = undefined;
     }
     function getToolFlow(toolName) {
         switch (toolName) {
@@ -99,6 +102,7 @@ function createMcpServer(apiKey, personalKey) {
             // Implicit flow reset if we are starting a new flow with a "start tool"
             switch (toolName) {
                 case "get_ui_components":
+                case "mark_ui_component_generated":
                     guard.flow = "ui_components";
                     guard.step = 1;
                     break;
@@ -139,18 +143,13 @@ function createMcpServer(apiKey, personalKey) {
         }
         if (guard.flow === "ui_components") {
             if (toolName === "get_ui_components") {
-                if (guard.step > 2)
-                    throw new Error("TOOL_CALL_BLOCKED: cannot restart flow at this step");
                 guard.step = 1;
             }
             else if (toolName === "generate_ui_components") {
-                if (guard.step !== 1)
-                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
                 guard.step = 2;
             }
             else if (toolName === "mark_ui_component_generated") {
-                if (guard.step !== 2)
-                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                // Allow finishing if it was started in this flow
                 resetGuard();
             }
             else {
@@ -533,7 +532,13 @@ function createMcpServer(apiKey, personalKey) {
                     if (!Array.isArray(rawAnnotations)) {
                         return { content: [{ type: "text", text: "Error: Invalid format from backend" }], isError: true };
                     }
-                    console.error(`[MCP] Received ${rawAnnotations.length} annotations from backend.`);
+                    // Point: Lease Collection | Store lockTokens in the session guard
+                    rawAnnotations.forEach((ann) => {
+                        if (ann._id && ann.lockToken) {
+                            guard.lockTokens[ann._id] = ann.lockToken;
+                        }
+                    });
+                    console.error(`[MCP] Received ${rawAnnotations.length} annotations. Tokens stored: ${Object.keys(guard.lockTokens).length}`);
                     // Local filtering fallback (for older backend versions or additional client-side safety)
                     if (search) {
                         const lowSearch = search.toLowerCase();
@@ -617,7 +622,11 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                             'x-personal-key': personalKey || ''
                         },
                         body: JSON.stringify({
-                            results: results.map((r) => ({ ...r, status: 'applied' }))
+                            results: results.map((r) => ({
+                                ...r,
+                                status: 'applied',
+                                lockToken: guard.lockTokens[r.id] // Attach the session-locked token
+                            }))
                         })
                     });
                     if (!response.ok)
@@ -634,11 +643,18 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                     const response = await fetch(url, {
                         method,
                         headers: { 'Content-Type': 'application/json', 'x-api-key': apiKey, 'x-mcp-tool-name': toolName, ...(personalKey ? { 'x-personal-key': personalKey } : {}) },
-                        body: args.results ? JSON.stringify({ results: args.results, force: args.force }) : undefined
+                        body: args.results ? JSON.stringify({
+                            results: args.results,
+                            force: args.force,
+                            backlogVersion: guard.backlogVersion
+                        }) : undefined
                     });
                     if (!response.ok)
                         throw new Error(`Backend responded with ${response.status}`);
                     const data = (await response.json());
+                    // Point: Version Storing
+                    if (data.backlogVersion)
+                        guard.backlogVersion = data.backlogVersion;
                     const langHint = data.preferredLanguage === 'es'
                         ? "\n*** [INSTRUCTION: Project is in SPANISH. Provide all comments, appliedChanges, and aiAnalysis in SPANISH ONLY.] ***\n\n\n\n"
                         : (data.preferredLanguage === 'en' ? "\n*** [INSTRUCTION: Provide all comments and analysis in English.] ***\n\n\n\n" : "");
@@ -667,15 +683,17 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                             ...(personalKey ? { 'x-personal-key': personalKey } : {})
                         }
                     });
-                    if (!response.ok) {
-                        const errorBody = await response.text();
-                        throw new Error(`Backend responded with ${response.status}: ${errorBody}`);
-                    }
                     const rawData = (await response.json());
                     let rawImprovements = Array.isArray(rawData) ? rawData : (rawData?.data || rawData?.improvements || []);
                     if (!Array.isArray(rawImprovements)) {
                         return { content: [{ type: "text", text: "Error: Invalid format from backend" }], isError: true };
                     }
+                    // Point: Lease Collection UI | Store lockTokens for UI tickets
+                    rawImprovements.forEach((ann) => {
+                        if (ann._id && ann.lockToken) {
+                            guard.lockTokens[ann._id] = ann.lockToken;
+                        }
+                    });
                     const improvementsWithTips = rawImprovements.map((ann) => {
                         return {
                             ...ann,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viewgate-mcp",
-  "version": "1.0.48",
+  "version": "1.0.50",
   "main": "dist/index.js",
   "bin": {
     "viewgate-mcp": "./dist/index.js"