viewgate-mcp 1.0.43 → 1.0.45

package/dist/index.js

@@ -46,6 +46,187 @@ function createMcpServer(apiKey, personalKey) {
             tools: {},
         },
     });
+    const FLOW_TTL_MS = 10 * 60 * 1000;
+    const guard = {
+        flow: "idle",
+        step: 0,
+        startedAt: Date.now(),
+        lastActivityAt: Date.now(),
+    };
+    function resetGuard() {
+        guard.flow = "idle";
+        guard.step = 0;
+        guard.lastTool = undefined;
+        guard.startedAt = Date.now();
+        guard.lastActivityAt = Date.now();
+    }
+    function getToolFlow(toolName) {
+        switch (toolName) {
+            case "get_ui_components":
+            case "generate_ui_components":
+            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":
+                return "ui_improvements";
+            case "planning":
+                return "planning";
+            case "sync_endpoints":
+            case "get_synced_endpoints":
+                return "sync";
+            case "get_ai_resolved_tickets":
+                return "idle";
+            default:
+                return "idle";
+        }
+    }
+    function enforceWorkflow(toolName, args) {
+        const now = Date.now();
+        if (guard.flow !== "idle" && now - guard.lastActivityAt > FLOW_TTL_MS) {
+            resetGuard();
+        }
+        const desiredFlow = getToolFlow(toolName);
+        if (toolName === "get_ai_resolved_tickets") {
+            if (guard.flow !== "idle") {
+                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+            }
+            guard.lastTool = toolName;
+            guard.lastActivityAt = now;
+            return;
+        }
+        if (guard.flow === "idle") {
+            switch (toolName) {
+                case "get_ui_components":
+                    guard.flow = "ui_components";
+                    guard.step = 1;
+                    break;
+                case "get_annotations":
+                    guard.flow = "annotations";
+                    guard.step = 1;
+                    break;
+                case "get_ui_improvements":
+                    guard.flow = "ui_improvements";
+                    guard.step = 1;
+                    break;
+                case "planning":
+                    if (args?.results) {
+                        throw new Error("TOOL_CALL_BLOCKED: planning submit requires prior fetch");
+                    }
+                    guard.flow = "planning";
+                    guard.step = 1;
+                    break;
+                case "sync_endpoints":
+                    guard.flow = "sync";
+                    guard.step = 1;
+                    break;
+                default:
+                    throw new Error("TOOL_CALL_BLOCKED: tool not allowed in idle");
+            }
+            guard.startedAt = now;
+            guard.lastTool = toolName;
+            guard.lastActivityAt = now;
+            return;
+        }
+        if (desiredFlow !== guard.flow) {
+            throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+        }
+        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");
+                resetGuard();
+            }
+            else {
+                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+            }
+        }
+        else if (guard.flow === "annotations") {
+            if (toolName === "get_annotations") {
+                if (guard.step > 2)
+                    throw new Error("TOOL_CALL_BLOCKED: cannot restart flow at this step");
+                guard.step = 1;
+            }
+            else if (toolName === "mark_annotation_ready") {
+                if (guard.step !== 1)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                guard.step = 2;
+            }
+            else if (toolName === "mark_annotations_as_live") {
+                if (guard.step !== 2)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                resetGuard();
+            }
+            else {
+                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+            }
+        }
+        else if (guard.flow === "ui_improvements") {
+            if (toolName === "get_ui_improvements") {
+                if (guard.step > 2)
+                    throw new Error("TOOL_CALL_BLOCKED: cannot restart flow at this step");
+                guard.step = 1;
+            }
+            else if (toolName === "mark_annotation_ready") {
+                if (guard.step !== 1)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                guard.step = 2;
+            }
+            else if (toolName === "mark_annotations_as_live") {
+                if (guard.step !== 2)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                resetGuard();
+            }
+            else {
+                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+            }
+        }
+        else if (guard.flow === "planning") {
+            if (toolName !== "planning") {
+                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+            }
+            const isSubmit = Array.isArray(args?.results);
+            if (!isSubmit) {
+                if (guard.step !== 1)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                guard.step = 1;
+            }
+            else {
+                if (guard.step !== 1)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                resetGuard();
+            }
+        }
+        else if (guard.flow === "sync") {
+            if (toolName === "sync_endpoints") {
+                if (guard.step !== 1)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                guard.step = 2;
+            }
+            else if (toolName === "get_synced_endpoints") {
+                if (guard.step !== 2)
+                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                resetGuard();
+            }
+            else {
+                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+            }
+        }
+        guard.lastTool = toolName;
+        guard.lastActivityAt = now;
+    }
     server.setRequestHandler(ListToolsRequestSchema, async () => {
         return {
             tools: [
@@ -62,7 +243,7 @@ function createMcpServer(apiKey, personalKey) {
                 },
                 {
                     name: "generate_ui_components",
-                    description: "Generate real UI components into /components (no overwrite). Also generates and uploads a preview, then marks the component as generated.",
+                    description: "Fetch pending UI component specs and return clear instructions for the LLM to implement a real, functional component. Does not write files, upload previews, or mark components as generated.",
                     inputSchema: {
                         type: "object",
                         properties: {
@@ -99,6 +280,19 @@ function createMcpServer(apiKey, personalKey) {
                         },
                     },
                 },
+                {
+                    name: "get_ui_improvements",
+                    description: "Fetch pending UI/UX improvement tickets. STRICTLY LIMITED TO VISUAL/CSS CHANGES. Do not add functional logic or modify backend integrations.",
+                    inputSchema: {
+                        type: "object",
+                        properties: {
+                            limit: { type: "number", description: "Max results.", default: 5 },
+                            search: { type: "string", description: "Search message/file." },
+                            key: { type: "string", description: "VG-XXXX key." },
+                            ids: { type: "string", description: "Internal IDs." }
+                        },
+                    },
+                },
                 {
                     name: "mark_annotation_ready",
                     description: "Mark as ready/applied. Use internal IDs. IMPORTANT: appliedChanges must be in the project's preferredLanguage (e.g. SPANISH).",
@@ -203,10 +397,12 @@ function createMcpServer(apiKey, personalKey) {
     server.setRequestHandler(CallToolRequestSchema, async (request) => {
         const toolName = request.params.name;
         console.error(`[MCP] Handling tool call: ${toolName}`);
+        const argsAny = request.params.arguments;
         try {
+            enforceWorkflow(toolName, argsAny);
             switch (toolName) {
                 case "get_ui_components": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     const limit = args.limit || 1;
                     const status = args.status || 'pending';
                     const fetchUrl = new URL(`${BACKEND_URL}/api/mcp/components`);
@@ -215,6 +411,7 @@ function createMcpServer(apiKey, personalKey) {
                     const response = await fetch(fetchUrl, {
                         headers: {
                             'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
                             ...(personalKey ? { 'x-personal-key': personalKey } : {})
                         }
                     });
@@ -226,7 +423,7 @@ function createMcpServer(apiKey, personalKey) {
                     return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
                 }
                 case "generate_ui_components": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     const limit = Math.min(args.limit || 1, 10);
                     const fetchUrl = new URL(`${BACKEND_URL}/api/mcp/components`);
                     fetchUrl.searchParams.append("limit", limit.toString());
@@ -234,6 +431,7 @@ function createMcpServer(apiKey, personalKey) {
                     const response = await fetch(fetchUrl, {
                         headers: {
                             'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
                             ...(personalKey ? { 'x-personal-key': personalKey } : {})
                         }
                     });
@@ -256,11 +454,15 @@ function createMcpServer(apiKey, personalKey) {
                             requiredProps,
                             commonProps,
                             figmaUrl: item.figmaUrl,
+                            htmlContent: item.htmlContent,
+                            cssContent: item.cssContent,
+                            sourceType: item.sourceType,
                             constraints: {
                                 mustBeFunctional: true,
                                 mustSupportRequiredProps: true,
                                 mustSupportCommonProps: true,
                                 avoidBreakingChanges: true,
+                                useProvidedHtmlAndCssIfAvailable: true
                             }
                         };
                         results.push({
@@ -285,7 +487,7 @@ function createMcpServer(apiKey, personalKey) {
                     };
                 }
                 case "mark_ui_component_generated": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     if (!args?.id || !args?.code) {
                         throw new Error("id and code are required");
                     }
@@ -294,6 +496,7 @@ function createMcpServer(apiKey, personalKey) {
                         headers: {
                             'Content-Type': 'application/json',
                             'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
                             ...(personalKey ? { 'x-personal-key': personalKey } : {})
                         },
                         body: JSON.stringify({ code: args.code, props: args.props })
@@ -306,7 +509,7 @@ function createMcpServer(apiKey, personalKey) {
                     return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
                 }
                 case "get_annotations": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     const limit = args.limit || 3;
                     // Strictly enforce allowed statuses: pending and bug_fixing
                     const allowedStatuses = ['pending', 'bug_fixing'];
@@ -341,6 +544,7 @@ function createMcpServer(apiKey, personalKey) {
                     const response = await fetch(fetchUrl, {
                         headers: {
                             'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
                             ...(personalKey ? { 'x-personal-key': personalKey } : {})
                         }
                     });
@@ -419,7 +623,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                     };
                 }
                 case "mark_annotation_ready": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     const results = args.results;
                     console.error(`[mark_annotation_ready] Submitting results for ${results?.length} items`);
                     const response = await fetch(`${BACKEND_URL}/api/mcp/annotations/batch-ready`, {
@@ -427,6 +631,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                         headers: {
                             'Content-Type': 'application/json',
                             'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
                             'x-personal-key': personalKey || ''
                         },
                         body: JSON.stringify({
@@ -441,7 +646,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                     };
                 }
                 case "mark_annotations_as_live": {
-                    const args = request.params.arguments;
+                    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`, {
@@ -449,6 +654,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                         headers: {
                             'Content-Type': 'application/json',
                             'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
                             'x-personal-key': personalKey || ''
                         },
                         body: JSON.stringify({ ids })
@@ -461,12 +667,12 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                     };
                 }
                 case "planning": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     const url = args.results ? `${BACKEND_URL}/api/mcp/annotations/batch-planning` : `${BACKEND_URL}/api/mcp/backlog`;
                     const method = args.results ? 'PATCH' : 'GET';
                     const response = await fetch(url, {
                         method,
-                        headers: { 'Content-Type': 'application/json', 'x-api-key': apiKey, ...(personalKey ? { 'x-personal-key': personalKey } : {}) },
+                        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
                     });
                     if (!response.ok)
@@ -477,11 +683,39 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                         : (data.preferredLanguage === 'en' ? "\n*** [INSTRUCTION: Provide all comments and analysis in English.] ***\n\n\n\n" : "");
                     return { content: [{ type: "text", text: langHint + JSON.stringify(data, null, 2) }] };
                 }
+                case "get_ui_improvements": {
+                    const args = argsAny;
+                    const fetchUrl = new URL(`${BACKEND_URL}/api/mcp/ui-improvements`);
+                    if (args.limit)
+                        fetchUrl.searchParams.append("limit", args.limit.toString());
+                    if (args.search)
+                        fetchUrl.searchParams.append("search", args.search);
+                    if (args.key)
+                        fetchUrl.searchParams.append("key", args.key);
+                    if (args.ids)
+                        fetchUrl.searchParams.append("ids", args.ids);
+                    const response = await fetch(fetchUrl, {
+                        headers: {
+                            'x-api-key': apiKey,
+                            'x-mcp-tool-name': toolName,
+                            ...(personalKey ? { 'x-personal-key': personalKey } : {})
+                        }
+                    });
+                    if (!response.ok) {
+                        const errorBody = await response.text();
+                        throw new Error(`Backend responded with ${response.status}: ${errorBody}`);
+                    }
+                    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) }] };
+                }
                 case "sync_endpoints": {
-                    const args = request.params.arguments;
+                    const args = argsAny;
                     const response = await fetch(`${BACKEND_URL}/api/mcp/sync-endpoints`, {
                         method: 'POST',
-                        headers: { 'Content-Type': 'application/json', 'x-api-key': apiKey, ...(personalKey ? { 'x-personal-key': personalKey } : {}) },
+                        headers: { 'Content-Type': 'application/json', 'x-api-key': apiKey, 'x-mcp-tool-name': toolName, ...(personalKey ? { 'x-personal-key': personalKey } : {}) },
                         body: JSON.stringify(args)
                     });
                     if (!response.ok)
@@ -491,7 +725,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                 }
                 case "get_synced_endpoints": {
                     const response = await fetch(`${BACKEND_URL}/api/projects/endpoints`, {
-                        headers: { 'x-api-key': apiKey, ...(personalKey ? { 'x-personal-key': personalKey } : {}) }
+                        headers: { 'x-api-key': apiKey, 'x-mcp-tool-name': toolName, ...(personalKey ? { 'x-personal-key': personalKey } : {}) }
                     });
                     if (!response.ok)
                         throw new Error(`Backend responded with ${response.status}`);
@@ -500,7 +734,7 @@ Lang: ${rawData.preferredLanguage === 'es' ? 'ES' : 'EN'}
                 }
                 case "get_ai_resolved_tickets": {
                     const response = await fetch(`${BACKEND_URL}/api/mcp/resolved-tickets`, {
-                        headers: { 'x-api-key': apiKey, ...(personalKey ? { 'x-personal-key': personalKey } : {}) }
+                        headers: { 'x-api-key': apiKey, 'x-mcp-tool-name': toolName, ...(personalKey ? { 'x-personal-key': personalKey } : {}) }
                     });
                     if (!response.ok)
                         throw new Error(`Backend responded with ${response.status}`);

package/package.json

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