@zivis/mcp 0.1.10 → 0.1.12

package/dist/tools/generate-diagram.js

@@ -1,161 +0,0 @@
-import { z } from "zod";
-import { readFile } from "fs/promises";
-import { join, relative, resolve } from "path";
-import { sanitizeResponse } from "../sanitize.js";
-function resolveWithinProjectDir(projectDir, candidate) {
-    const base = resolve(projectDir);
-    const target = resolve(base, candidate);
-    const rel = relative(base, target);
-    if (rel.startsWith(".."))
-        return null;
-    return target;
-}
-export const GENERATE_DIAGRAM_NAME = "zivis_generate_diagram";
-export const GENERATE_DIAGRAM_DESCRIPTION = `Generate a visual architecture diagram from a Docker Compose file.
-
-Reads a local docker-compose.yml and sends it to the ZIVIS platform to generate
-a positioned canvas diagram with:
-- Services mapped to color-coded nodes (database, cache, app, proxy, etc.)
-- Docker networks mapped to boundary groups
-- depends_on relationships mapped to connections
-- Auto-positioned grid layout grouped by network
-
-The diagram is saved to your organization and can be viewed/edited in the diagram editor.
-
-Supported formats: docker-compose (docker-compose.yml, compose.yml)
-Coming soon: Kubernetes manifests, Terraform configs`;
-export const GENERATE_DIAGRAM_SCHEMA = {
-    file_path: z
-        .string()
-        .optional()
-        .describe("Path to the infrastructure file (e.g., docker-compose.yml). " +
-        "If not provided, searches the current directory for docker-compose.yml, " +
-        "docker-compose.yaml, compose.yml, or compose.yaml."),
-    source_format: z
-        .enum(["docker-compose"])
-        .default("docker-compose")
-        .describe("The format of the source file. Currently only docker-compose is supported."),
-    name: z
-        .string()
-        .optional()
-        .describe("Custom name for the generated diagram. Defaults to 'Docker Compose Architecture'."),
-    project_dir: z
-        .string()
-        .optional()
-        .describe("Project directory to search for compose files. Defaults to current working directory."),
-};
-const COMPOSE_FILE_CANDIDATES = [
-    "docker-compose.yml",
-    "docker-compose.yaml",
-    "compose.yml",
-    "compose.yaml",
-];
-export function createGenerateDiagramHandler(apiClient) {
-    return async (params) => {
-        const projectDir = params.project_dir || process.cwd();
-        const sourceFormat = params.source_format || "docker-compose";
-        let filePath = params.file_path;
-        let content = null;
-        if (filePath) {
-            const resolved = resolveWithinProjectDir(projectDir, filePath);
-            if (!resolved) {
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: `Error: file_path must resolve inside project_dir (got: ${filePath})`,
-                        },
-                    ],
-                    isError: true,
-                };
-            }
-            try {
-                content = await readFile(resolved, "utf-8");
-            }
-            catch {
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: `Error: File not found: ${resolved}`,
-                        },
-                    ],
-                    isError: true,
-                };
-            }
-        }
-        else {
-            for (const candidate of COMPOSE_FILE_CANDIDATES) {
-                try {
-                    content = await readFile(join(projectDir, candidate), "utf-8");
-                    filePath = candidate;
-                    break;
-                }
-                catch {
-                }
-            }
-            if (!content) {
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: "Error: No docker-compose file found. Searched for: " +
-                                COMPOSE_FILE_CANDIDATES.join(", "),
-                        },
-                    ],
-                    isError: true,
-                };
-            }
-        }
-        if (content.length > 500 * 1024) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: "Error: File too large (max 500KB)",
-                    },
-                ],
-                isError: true,
-            };
-        }
-        try {
-            const data = await apiClient.post("/api/diagrams/generate", {
-                content,
-                sourceFormat,
-                name: params.name,
-            });
-            const summary = {
-                diagram_id: data.id,
-                name: data.name,
-                type: data.diagramType,
-                source_file: filePath,
-                stats: {
-                    nodes: data.nodes?.length || 0,
-                    connections: data.connections?.length || 0,
-                    boundaries: data.boundaries?.length || 0,
-                },
-                nodes: (data.nodes || []).map((n) => ({
-                    name: n.name,
-                    tags: n.tags,
-                })),
-                boundaries: (data.boundaries || []).map((b) => b.label),
-            };
-            const sanitized = sanitizeResponse(summary);
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify(sanitized, null, 2),
-                    },
-                ],
-            };
-        }
-        catch (err) {
-            const message = err instanceof Error ? err.message : "Failed to generate diagram";
-            return {
-                content: [{ type: "text", text: `Error: ${message}` }],
-                isError: true,
-            };
-        }
-    };
-}