import { ErrorCode, McpError } from "@modelcontextprotocol/sdk/types.js";
import { z } from "zod";
import * as Charts from "../charts";
import { generateChartUrl } from "./generate";
import { ValidateError } from "./validator";
// Chart type mapping
const CHART_TYPE_MAP = {
generate_bar_chart: "bar",
generate_line_chart: "line",
generate_pie_chart: "pie",
generate_scatter_chart: "scatter",
generate_radar_chart: "radar",
generate_heatmap_chart: "heatmap",
generate_general_chart: "general",
} as const;
/**
* Call a tool to generate a chart preview link(not an image) based on the provided name and arguments.
* @param tool The name of the tool to call, e.g., "generate_bar_chart".
* @param args The arguments for the tool, which should match the expected schema for the chart type.
* @returns
*/
export async function callTool(tool: string, args: object = {}) {
const chartType = CHART_TYPE_MAP[tool as keyof typeof CHART_TYPE_MAP];
if (!chartType) {
throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${tool}.`);
}
try {
// Validate input using Zod before sending to API.
// Select the appropriate schema based on the chart type.
const schema = Charts[chartType].schema;
if (schema) {
// Use safeParse instead of parse and try-catch.
const result = z.object(schema).safeParse(args);
if (!result.success) {
throw new McpError(
ErrorCode.InvalidParams,
`Invalid parameters: ${result.error.message}`,
);
}
}
const chartPreviewUrl = await generateChartUrl(args);
return {
content: [
{
type: "text",
// text is a chart preview url, not a chart image url
text: chartPreviewUrl,
},
],
};
// biome-ignore lint/suspicious/noExplicitAny: <explanation>
} catch (error: any) {
if (error instanceof McpError) throw error;
if (error instanceof ValidateError)
throw new McpError(ErrorCode.InvalidParams, error.message);
throw new McpError(
ErrorCode.InternalError,
`Failed to generate chart: ${error?.message || "Unknown error."}`,
);
}
}
