diff --git a/python/kicad_interface.py b/python/kicad_interface.py index 2d0614f..8b77697 100644 --- a/python/kicad_interface.py +++ b/python/kicad_interface.py @@ -604,6 +604,7 @@ class KiCADInterface: "export_sch_svg": self._handle_export_sch_svg, "export_sch_dxf": self._handle_export_sch_dxf, "export_sch_hpgl": self._handle_export_sch_hpgl, + "export_sch_ps": self._handle_export_sch_ps, "generate_netlist": self._handle_generate_netlist, "sync_schematic_to_board": self._handle_sync_schematic_to_board, "list_schematic_libraries": self._handle_list_schematic_libraries, @@ -5983,6 +5984,85 @@ class KiCADInterface: logger.error(f"Error exporting schematic HPGL: {e}") return {"success": False, "message": str(e)} + def _handle_export_sch_ps(self, params: Dict[str, Any]) -> Dict[str, Any]: + """Export a schematic to PostScript via kicad-cli (`sch export ps`). + + Output is a directory (one PS per page). Exposes drawing-sheet override, + theme, B&W, exclude-drawing-sheet, default-font, no-background-color, and + page selection. schematicPath is required. + """ + import subprocess + + logger.info("Exporting schematic PostScript via kicad-cli") + try: + schematic_path = params.get("schematicPath") + output_dir = params.get("outputDir") + + if not schematic_path: + return {"success": False, "message": "schematicPath is required"} + if not os.path.exists(schematic_path): + return {"success": False, "message": f"Schematic not found: {schematic_path}"} + if not output_dir: + return {"success": False, "message": "outputDir is required"} + + kicad_cli = self._find_kicad_cli_static() + if not kicad_cli: + return {"success": False, "message": "kicad-cli not found in PATH"} + + output_dir = os.path.abspath(os.path.expanduser(output_dir)) + os.makedirs(output_dir, exist_ok=True) + + cmd = [kicad_cli, "sch", "export", "ps", "--output", output_dir] + + if params.get("drawingSheet"): + cmd += ["--drawing-sheet", params["drawingSheet"]] + for kv in params.get("defineVar", []) or []: + cmd += ["--define-var", kv] + + value_map = { + "theme": "--theme", + "defaultFont": "--default-font", + "pages": "--pages", + } + for key, flag in value_map.items(): + val = params.get(key) + if val is not None and val != "": + cmd += [flag, str(val)] + + flag_map = { + "blackAndWhite": "--black-and-white", + "excludeDrawingSheet": "--exclude-drawing-sheet", + "noBackgroundColor": "--no-background-color", + } + for key, flag in flag_map.items(): + if params.get(key): + cmd.append(flag) + + cmd.append(schematic_path) + + logger.info(f"Running: {' '.join(cmd)}") + result = subprocess.run(cmd, capture_output=True, text=True, timeout=180) + + if result.returncode != 0: + return { + "success": False, + "message": f"kicad-cli failed (exit {result.returncode}): " + f"{result.stderr.strip()}", + } + + files = sorted( + f for f in os.listdir(output_dir) if os.path.isfile(os.path.join(output_dir, f)) + ) + return {"success": True, "outputDir": output_dir, "files": files} + + except FileNotFoundError: + return {"success": False, "message": "kicad-cli not found in PATH"} + except subprocess.TimeoutExpired: + return {"success": False, "message": "kicad-cli timed out after 180 seconds"} + except Exception as e: + logger.error(f"Error exporting schematic PostScript: {e}") + return {"success": False, "message": str(e)} + def _handle_generate_netlist(self, params: Dict[str, Any]) -> Dict[str, Any]: """Generate netlist from schematic and return structured JSON. diff --git a/src/tools/export.ts b/src/tools/export.ts index 856b6aa..d5b5dfd 100644 --- a/src/tools/export.ts +++ b/src/tools/export.ts @@ -1277,5 +1277,46 @@ export function registerExportTools(server: McpServer, callKicadScript: CommandF }, ); + // ------------------------------------------------------ + // Export Schematic PostScript Tool (kicad-cli, full option set) + // ------------------------------------------------------ + server.tool( + "export_sch_ps", + "Export a schematic to PostScript via kicad-cli (`sch export ps`), one PS per page into a directory. Exposes drawing-sheet override, theme, black-and-white, exclude-drawing-sheet, default-font, no-background-color, and page selection. schematicPath is REQUIRED.", + { + schematicPath: z.string().describe("Path to the .kicad_sch (required)"), + outputDir: z.string().describe("Output directory for the PostScript files"), + drawingSheet: z.string().optional().describe("Path to a drawing sheet override"), + defineVar: z + .array(z.string()) + .optional() + .describe("Project variable overrides as 'KEY=VALUE' strings"), + theme: z.string().optional().describe("Color theme to use (default: schematic settings)"), + blackAndWhite: z.boolean().optional().describe("Black and white only"), + excludeDrawingSheet: z.boolean().optional().describe("No drawing sheet"), + defaultFont: z.string().optional().describe("Default font name"), + noBackgroundColor: z + .boolean() + .optional() + .describe("Avoid setting a background color (regardless of theme)"), + pages: z + .string() + .optional() + .describe("Comma list of page numbers to print (blank = all pages)"), + }, + async (args) => { + logger.debug(`Exporting schematic PostScript to: ${args.outputDir}`); + const result = await callKicadScript("export_sch_ps", args); + return { + content: [ + { + type: "text", + text: JSON.stringify(result), + }, + ], + }; + }, + ); + logger.info("Export tools registered"); } diff --git a/src/tools/registry.ts b/src/tools/registry.ts index 982412c..4c8bd19 100644 --- a/src/tools/registry.ts +++ b/src/tools/registry.ts @@ -75,6 +75,7 @@ export const toolCategories: ToolCategory[] = [ "export_sch_svg", "export_sch_dxf", "export_sch_hpgl", + "export_sch_ps", "export_pdf", "export_svg", "export_3d",