KiCAD-MCP-Server

import glob
import logging

# Symbol class might not be directly importable in the current version
import os
from typing import Any, Dict, List, Optional

logger = logging.getLogger(__name__)


class LibraryManager:
    """Manage symbol libraries"""

    @staticmethod
    def list_available_libraries(search_paths: Optional[List[str]] = None) -> Dict[str, List[str]]:
        """List all available symbol libraries"""
        if search_paths is None:
            # Default library paths based on common KiCAD installations
            # This would need to be configured for the specific environment
            search_paths = [
                # KiCAD 8/9 single-file libraries
                "C:/Program Files/KiCad/*/share/kicad/symbols/*.kicad_sym",
                "/usr/share/kicad/symbols/*.kicad_sym",
                "/Applications/KiCad/KiCad.app/Contents/SharedSupport/symbols/*.kicad_sym",
                os.path.expanduser("~/Documents/KiCad/*/symbols/*.kicad_sym"),
                # KiCAD 10 per-symbol directory libraries
                "C:/Program Files/KiCad/*/share/kicad/symbols/*.kicad_symdir/*.kicad_sym",
                "/usr/share/kicad/symbols/*.kicad_symdir/*.kicad_sym",
                "/Applications/KiCad/KiCad.app/Contents/SharedSupport/symbols/*.kicad_symdir/*.kicad_sym",
                os.path.expanduser("~/Documents/KiCad/*/symbols/*.kicad_symdir/*.kicad_sym"),
            ]

        libraries = []
        for path_pattern in search_paths:
            try:
                # Use glob to find all matching files
                matching_libs = glob.glob(path_pattern, recursive=True)
                libraries.extend(matching_libs)
            except Exception as e:
                logger.error(f"Error searching for libraries at {path_pattern}: {e}")

        # Extract library names from paths
        library_names = [os.path.splitext(os.path.basename(lib))[0] for lib in libraries]
        logger.info(
            f"Found {len(library_names)} libraries: {', '.join(library_names[:10])}{'...' if len(library_names) > 10 else ''}"
        )

        # Return both full paths and library names
        return {"paths": libraries, "names": library_names}

    @staticmethod
    def get_symbol_details(library_path: str, symbol_name: str) -> Dict[str, Any]:
        """Get detailed information about a symbol"""
        try:
            logger.warning(
                f"Attempted to get details for symbol {symbol_name} in library {library_path}. This requires advanced implementation."
            )
            return {}
        except Exception as e:
            logger.error(f"Error getting symbol details for {symbol_name} in {library_path}: {e}")
            return {}

    @staticmethod
    def search_symbols(query: str, search_paths: Optional[List[str]] = None) -> List[Any]:
        """Search for symbols matching criteria"""
        try:
            # This would typically involve:
            # 1. Getting a list of all libraries using list_available_libraries
            # 2. For each library, getting a list of all symbols
            # 3. Filtering symbols based on the query

            # For now, this is a placeholder implementation
            libraries = LibraryManager.list_available_libraries(search_paths)

            results = []
            logger.warning(
                f"Searched for symbols matching '{query}'. This requires advanced implementation."
            )
            return results
        except Exception as e:
            logger.error(f"Error searching for symbols matching '{query}': {e}")
            return []

    @staticmethod
    def get_default_symbol_for_component_type(
        component_type: str, search_paths: Optional[List[str]] = None
    ) -> Dict[str, str]:
        """Get a recommended default symbol for a given component type"""
        # This method provides a simplified way to get a symbol for common component types
        # It's useful when the user doesn't specify a particular library/symbol

        # Define common mappings from component type to library/symbol
        common_mappings = {
            "resistor": {"library": "Device", "symbol": "R"},
            "capacitor": {"library": "Device", "symbol": "C"},
            "inductor": {"library": "Device", "symbol": "L"},
            "diode": {"library": "Device", "symbol": "D"},
            "led": {"library": "Device", "symbol": "LED"},
            "transistor_npn": {"library": "Device", "symbol": "Q_NPN_BCE"},
            "transistor_pnp": {"library": "Device", "symbol": "Q_PNP_BCE"},
            "opamp": {"library": "Amplifier_Operational", "symbol": "OpAmp_Dual_Generic"},
            "microcontroller": {"library": "MCU_Module", "symbol": "Arduino_UNO_R3"},
            # Add more common components as needed
        }

        # Normalize input to lowercase
        component_type_lower = component_type.lower()

        # Try direct match first
        if component_type_lower in common_mappings:
            return common_mappings[component_type_lower]

        # Try partial matches
        for key, value in common_mappings.items():
            if component_type_lower in key or key in component_type_lower:
                return value

        # Default fallback
        return {"library": "Device", "symbol": "R"}


if __name__ == "__main__":
    # Example Usage (for testing)
    # List available libraries
    libraries = LibraryManager.list_available_libraries()
    # Get default symbol for a component type
    resistor_sym = LibraryManager.get_default_symbol_for_component_type("resistor")
    print(f"Default symbol for resistor: {resistor_sym['library']}/{resistor_sym['symbol']}")

    # Try a partial match
    cap_sym = LibraryManager.get_default_symbol_for_component_type("cap")
    print(f"Default symbol for 'cap': {cap_sym['library']}/{cap_sym['symbol']}")