import { z } from "zod";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { withRevitConnection } from "../utils/ConnectionManager.js";
export function registerSendCodeToRevitTool(server: McpServer) {
server.tool(
"send_code_to_revit",
"Send C# code to Revit for execution. The code will be inserted into a template with access to the Revit Document and parameters. Your code should be written to work within the Execute method of the template.",
{
code: z
.string()
.describe(
"The C# code to execute in Revit. This code will be inserted into the Execute method of a template with access to Document and parameters."
),
parameters: z
.array(z.any())
.optional()
.describe(
"Optional execution parameters that will be passed to your code"
),
},
async (args, extra) => {
const params = {
code: args.code,
parameters: args.parameters || [],
};
try {
const response = await withRevitConnection(async (revitClient) => {
return await revitClient.sendCommand("send_code_to_revit", params);
});
return {
content: [
{
type: "text",
text: `Code execution successful!\nResult: ${JSON.stringify(
response,
null,
2
)}`,
},
],
};
} catch (error) {
return {
content: [
{
type: "text",
text: `Code execution failed: ${
error instanceof Error ? error.message : String(error)
}`,
},
],
};
}
}
);
}