kicad-mcp/kicad_mcp/tools/drc_impl/cli_drc.py

"""
Design Rule Check (DRC) implementation using KiCad command-line interface.
"""
import os
import json
import subprocess
import tempfile
from typing import Dict, Any, Optional
from mcp.server.fastmcp import Context

from kicad_mcp.utils.logger import Logger
from kicad_mcp.config import system

# Create logger for this module
logger = Logger()

async def run_drc_via_cli(pcb_file: str, ctx: Context) -> Dict[str, Any]:
    """Run DRC using KiCad command line tools.
    
    Args:
        pcb_file: Path to the PCB file (.kicad_pcb)
        ctx: MCP context for progress reporting
        
    Returns:
        Dictionary with DRC results
    """
    results = {
        "success": False,
        "method": "cli",
        "pcb_file": pcb_file
    }
    
    try:
        # Create a temporary directory for the output
        with tempfile.TemporaryDirectory() as temp_dir:
            # Output file for DRC report
            output_file = os.path.join(temp_dir, "drc_report.json")
            
            # Find kicad-cli executable
            kicad_cli = find_kicad_cli()
            if not kicad_cli:
                logger.error("kicad-cli not found in PATH or common installation locations")
                results["error"] = "kicad-cli not found. Please ensure KiCad 9.0+ is installed and kicad-cli is available."
                return results
            
            # Report progress 
            await ctx.report_progress(50, 100)
            ctx.info("Running DRC using KiCad CLI...")
            
            # Build the DRC command
            cmd = [
                kicad_cli, 
                "pcb", 
                "drc",
                "--format", "json",
                "--output", output_file,
                pcb_file
            ]
            
            logger.info(f"Running command: {' '.join(cmd)}")
            process = subprocess.run(cmd, capture_output=True, text=True)
            
            # Check if the command was successful
            if process.returncode != 0:
                logger.error(f"DRC command failed with code {process.returncode}")
                logger.error(f"Error output: {process.stderr}")
                results["error"] = f"DRC command failed: {process.stderr}"
                return results
            
            # Check if the output file was created
            if not os.path.exists(output_file):
                logger.error("DRC report file not created")
                results["error"] = "DRC report file not created"
                return results
            
            # Read the DRC report
            with open(output_file, 'r') as f:
                try:
                    drc_report = json.load(f)
                except json.JSONDecodeError:
                    logger.error("Failed to parse DRC report JSON")
                    results["error"] = "Failed to parse DRC report JSON"
                    return results
            
            # Process the DRC report
            violations = drc_report.get("violations", [])
            violation_count = len(violations)
            logger.info(f"DRC completed with {violation_count} violations")
            await ctx.report_progress(70, 100)
            ctx.info(f"DRC completed with {violation_count} violations")
            
            # Categorize violations by type
            error_types = {}
            for violation in violations:
                error_type = violation.get("message", "Unknown")
                if error_type not in error_types:
                    error_types[error_type] = 0
                error_types[error_type] += 1
            
            # Create success response
            results = {
                "success": True,
                "method": "cli",
                "pcb_file": pcb_file,
                "total_violations": violation_count,
                "violation_categories": error_types,
                "violations": violations
            }
            
            await ctx.report_progress(90, 100)
            return results
            
    except Exception as e:
        logger.error(f"Error in CLI DRC: {str(e)}", exc_info=True)
        results["error"] = f"Error in CLI DRC: {str(e)}"
        return results


def find_kicad_cli() -> Optional[str]:
    """Find the kicad-cli executable in the system PATH.
    
    Returns:
        Path to kicad-cli if found, None otherwise
    """
    # Check if kicad-cli is in PATH
    try:
        if system == "Windows":
            # On Windows, check for kicad-cli.exe
            result = subprocess.run(["where", "kicad-cli.exe"], capture_output=True, text=True)
            if result.returncode == 0:
                return result.stdout.strip().split("\n")[0]
        else:
            # On Unix-like systems, use which
            result = subprocess.run(["which", "kicad-cli"], capture_output=True, text=True)
            if result.returncode == 0:
                return result.stdout.strip()
    
    except Exception as e:
        logger.error(f"Error finding kicad-cli: {str(e)}")
    
    # If we get here, kicad-cli is not in PATH
    # Try common installation locations
    if system == "Windows":
        # Common Windows installation path
        potential_paths = [
            r"C:\Program Files\KiCad\bin\kicad-cli.exe",
            r"C:\Program Files (x86)\KiCad\bin\kicad-cli.exe"
        ]
    elif system == "Darwin":  # macOS
        # Common macOS installation paths
        potential_paths = [
            "/Applications/KiCad/KiCad.app/Contents/MacOS/kicad-cli",
            "/Applications/KiCad/kicad-cli"
        ]
    else:  # Linux and other Unix-like systems
        # Common Linux installation paths
        potential_paths = [
            "/usr/bin/kicad-cli",
            "/usr/local/bin/kicad-cli",
            "/opt/kicad/bin/kicad-cli"
        ]
    
    # Check each potential path
    for path in potential_paths:
        if os.path.exists(path) and os.access(path, os.X_OK):
            return path
    
    # If still not found, return None
    return None