viewgate-mcp 1.0.47 → 1.0.49

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) {
@@ -68,7 +71,6 @@ function createMcpServer(apiKey, personalKey) {
             case "mark_ui_component_generated":
                 return "ui_components";
             case "get_annotations":
-            case "mark_annotation_ready":
             case "mark_annotations_as_live":
                 return "annotations";
             case "get_ui_improvements":
@@ -124,10 +126,13 @@ function createMcpServer(apiKey, personalKey) {
                     break;
                 default:
                     if (guard.flow === "idle") {
-                        throw new Error("TOOL_CALL_BLOCKED: tool not allowed in idle");
+                        throw new Error(`TOOL_CALL_BLOCKED: tool '${toolName}' not allowed in idle. Please start a workflow with 'get_annotations', 'get_ui_components', or 'get_ui_improvements'.`);
                     }
                     else {
-                        throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+                        const nextTool = guard.flow === "annotations" ? "mark_annotations_as_live" :
+                            guard.flow === "ui_components" ? "mark_ui_component_generated" :
+                                guard.flow === "ui_improvements" ? "mark_annotations_as_live" : "the next workflow step";
+                        throw new Error(`TOOL_CALL_BLOCKED: tool '${toolName}' not allowed while in '${guard.flow}' workflow. You must call '${nextTool}' to complete the current task before starting another.`);
                     }
             }
             guard.startedAt = now;
@@ -160,16 +165,12 @@ function createMcpServer(apiKey, personalKey) {
                 // Allow re-fetching at any time during the flow
                 guard.step = 1;
             }
-            else if (toolName === "mark_annotation_ready") {
+            else if (toolName === "mark_annotations_as_live") {
                 if (guard.step !== 1)
                     throw new Error("TOOL_CALL_BLOCKED: unexpected step");
                 // Stop here and reset flow as per USER_REQUEST (MCP only reaches 'applied')
                 resetGuard();
             }
-            else if (toolName === "mark_annotations_as_live") {
-                // Optional step if the model decides to use it, but no longer part of the required chain
-                resetGuard();
-            }
             else {
                 throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
             }
@@ -178,14 +179,11 @@ function createMcpServer(apiKey, personalKey) {
             if (toolName === "get_ui_improvements") {
                 guard.step = 1;
             }
-            else if (toolName === "mark_annotation_ready") {
+            else if (toolName === "mark_annotations_as_live") {
                 if (guard.step !== 1)
                     throw new Error("TOOL_CALL_BLOCKED: unexpected step");
                 resetGuard();
             }
-            else if (toolName === "mark_annotations_as_live") {
-                resetGuard();
-            }
             else {
                 throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
             }
@@ -290,8 +288,8 @@ function createMcpServer(apiKey, personalKey) {
                     },
                 },
                 {
-                    name: "mark_annotation_ready",
-                    description: "Mark as ready/applied. Use internal IDs. IMPORTANT: appliedChanges must be in the project's preferredLanguage (e.g. SPANISH).",
+                    name: "mark_annotations_as_live",
+                    description: "Mark tickets (functional or UI/UX) as applied. Use this tool after fixing the code to submit your changes. IMPORTANT: appliedChanges must be in the project's preferredLanguage (e.g. SPANISH).",
                     inputSchema: {
                         type: "object",
                         properties: {
@@ -310,21 +308,6 @@ function createMcpServer(apiKey, personalKey) {
                         required: ["results"]
                     },
                 },
-                {
-                    name: "mark_annotations_as_live",
-                    description: "Mark as live (ready_for_review). IDs required.",
-                    inputSchema: {
-                        type: "object",
-                        properties: {
-                            ids: {
-                                type: "array",
-                                items: { type: "string" },
-                                description: "Internal IDs."
-                            }
-                        },
-                        required: ["ids"]
-                    },
-                },
                 {
                     name: "planning",
                     description: "Planning tool for backlog tickets. Fetch tickets or submit analysis. IMPORTANT: aiAnalysis must be in the project's preferredLanguage (e.g. SPANISH).",
@@ -553,7 +536,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();
@@ -608,6 +597,12 @@ Endpoints: ${ann.backendEndpoints?.length ? ann.backendEndpoints.join(', ') : 'N
 Feedback previo: ${ann.corrections && Array.isArray(ann.corrections) && ann.corrections.length > 0 ? ann.corrections[ann.corrections.length - 1].text : 'Ninguno'}
 Tarea: ${ann.message}
 Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
+
+### FLUJO DE TRABAJO OBLIGATORIO ###
+1. Analiza el código y el problema.
+2. Aplica la solución técnica necesaria.
+3. AL TERMINAR, DEBES LLAMAR A 'mark_annotations_as_live' con el id '${ann._id}' y un resumen de tus cambios en 'appliedChanges' (en ESPAÑOL).
+4. El estado final del ticket será 'applied'. NO intentes usar otros estados o herramientas hasta completar este paso.
 (RECUERDA: Toda tu respuesta, comentarios y análisis DEBEN estar en ESPAÑOL)`
                         };
                     });
@@ -618,10 +613,10 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                         content: [{ type: "text", text: langHint + JSON.stringify({ preferredLanguage: rawData.preferredLanguage || 'en', annotations: annotationsWithTips }, null, 2) }],
                     };
                 }
-                case "mark_annotation_ready": {
+                case "mark_annotations_as_live": {
                     const args = argsAny;
                     const results = args.results;
-                    console.error(`[mark_annotation_ready] Submitting results for ${results?.length} items`);
+                    console.error(`[mark_annotations_as_live] Submitting results for ${results?.length} items as applied`);
                     const response = await fetch(`${BACKEND_URL}/api/mcp/annotations/batch-ready`, {
                         method: 'PATCH',
                         headers: {
@@ -631,7 +626,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)
@@ -641,27 +640,6 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                         content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
                     };
                 }
-                case "mark_annotations_as_live": {
-                    const args = argsAny;
-                    const ids = args.ids;
-                    console.error(`[mark_annotations_as_live] Marking ${ids?.length} items as live`);
-                    const response = await fetch(`${BACKEND_URL}/api/mcp/annotations/mark-as-live`, {
-                        method: 'POST',
-                        headers: {
-                            'Content-Type': 'application/json',
-                            'x-api-key': apiKey,
-                            'x-mcp-tool-name': toolName,
-                            'x-personal-key': personalKey || ''
-                        },
-                        body: JSON.stringify({ ids })
-                    });
-                    if (!response.ok)
-                        throw new Error(`Backend responded with ${response.status}`);
-                    const data = (await response.json());
-                    return {
-                        content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
-                    };
-                }
                 case "planning": {
                     const args = argsAny;
                     const url = args.results ? `${BACKEND_URL}/api/mcp/annotations/batch-planning` : `${BACKEND_URL}/api/mcp/backlog`;
@@ -669,11 +647,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" : "");
@@ -681,6 +666,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                 }
                 case "get_ui_improvements": {
                     const args = argsAny;
+                    const agentId = personalKey || 'mcp-agent-ui-' + (apiKey?.slice(-6) || 'generic');
                     const fetchUrl = new URL(`${BACKEND_URL}/api/mcp/ui-improvements`);
                     if (args.limit)
                         fetchUrl.searchParams.append("limit", args.limit.toString());
@@ -690,6 +676,10 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                         fetchUrl.searchParams.append("key", args.key);
                     if (args.ids)
                         fetchUrl.searchParams.append("ids", args.ids);
+                    // Add lock and agentId parameters to enable blocking flow
+                    fetchUrl.searchParams.append("lock", "true");
+                    fetchUrl.searchParams.append("agentId", agentId);
+                    console.error(`[get_ui_improvements] Fetching with lock: ${fetchUrl.toString()}`);
                     const response = await fetch(fetchUrl, {
                         headers: {
                             'x-api-key': apiKey,
@@ -697,15 +687,39 @@ 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 };
                     }
-                    const data = (await response.json());
-                    const langHint = data.preferredLanguage === 'es'
-                        ? "\n*** [INSTRUCTION: Project is in SPANISH. Provide all CSS/Visual changes in SPANISH comments if requested.] ***\n\n\n\n"
-                        : "";
-                    return { content: [{ type: "text", text: langHint + JSON.stringify(data, null, 2) }] };
+                    // 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,
+                            _ia_fix_instruction: `### INSTRUCCIONES PARA LA IA (MEJORA UI - IMPORTANTE) ###
+A continuación se presenta un ticket de MEJORA VISUAL/UX que requiere tu atención.
+DEBES RESPONDER EN ESPAÑOL.
+Proyecto: ${rawData.name || 'ViewGate'}
+Tarea: ${ann.message}
+Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
+
+### FLUJO DE TRABAJO OBLIGATORIO ###
+1. Analiza el diseño y los estilos CSS actuales.
+2. Aplica los cambios visuales solicitados (STRICTLY LIMITED TO VISUAL/CSS CHANGES).
+3. AL TERMINAR, DEBES LLAMAR A 'mark_annotations_as_live' con el id '${ann._id}' y un resumen de tus cambios en 'appliedChanges' (en ESPAÑOL).
+4. El estado final del ticket será 'applied'. NO intentes usar otros estados o herramientas hasta completar este paso.
+(RECUERDA: Toda tu respuesta, comentarios y análisis DEBEN estar en ESPAÑOL)`
+                        };
+                    });
+                    const langHint = rawData.preferredLanguage === 'es'
+                        ? "\n*** [INSTRUCTION: Project is in SPANISH. Provide all comments and changes in SPANISH ONLY.] ***\n\n"
+                        : (rawData.preferredLanguage === 'en' ? "\n*** [INSTRUCTION: Provide all comments and analysis in English.] ***\n\n" : "");
+                    return { content: [{ type: "text", text: langHint + JSON.stringify({ preferredLanguage: rawData.preferredLanguage || 'en', improvements: improvementsWithTips }, null, 2) }] };
                 }
                 case "sync_endpoints": {
                     const args = argsAny;

package/package.json

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