MCP Tools 파일 분할 및 간소화 - mcp_main.py 파일에 포함된 tool 함수들을 다수의 tools/*.py 개별 파일로 이동 및 정리

Author: call518

src/mcp_openstack_ops/mcp_main.py

@@ -0,0 +1,18 @@
+"""Tool registration utilities for OpenStack MCP."""
+
+import importlib
+import pkgutil
+from typing import Iterable
+
+def _iter_tool_modules() -> Iterable[str]:
+    """Yield importable tool module names within this package."""
+    for module_info in pkgutil.iter_modules(__path__):  # type: ignore[name-defined]
+        name = module_info.name
+        if name.startswith('_'):
+            continue
+        yield name
+
+def register_all_tools() -> None:
+    """Import every tool module so decorators register with FastMCP."""
+    for module_name in sorted(_iter_tool_modules()):
+        importlib.import_module(f"{__name__}.{module_name}")

src/mcp_openstack_ops/tools/__init__.py

@@ -0,0 +1,35 @@
+"""Tool implementation for get_availability_zones."""
+
+import json
+from datetime import datetime
+from ..functions import get_availability_zones as _get_availability_zones
+from ..mcp_main import (
+    logger,
+    mcp,
+)
+
+@mcp.tool()
+async def get_availability_zones() -> str:
+    """
+    List availability zones and their status
+    
+    Returns:
+        JSON string with availability zones information
+    """
+    try:
+        logger.info("Getting availability zones")
+        
+        zones_result = _get_availability_zones()
+        
+        response = {
+            "timestamp": datetime.now().isoformat(),
+            "operation": "get_availability_zones",
+            "result": zones_result
+        }
+        
+        return json.dumps(response, indent=2, ensure_ascii=False)
+        
+    except Exception as e:
+        error_msg = f"Error: Failed to get availability zones - {str(e)}"
+        logger.error(error_msg)
+        return error_msg

src/mcp_openstack_ops/tools/get_availability_zones.py

@@ -0,0 +1,45 @@
+"""Tool implementation for get_floating_ip_pools."""
+
+import json
+from datetime import datetime
+from ..functions import get_floating_ip_pools as _get_floating_ip_pools
+from ..mcp_main import (
+    logger,
+    mcp,
+)
+
+@mcp.tool()
+async def get_floating_ip_pools() -> str:
+    """
+    Get list of floating IP pools (external networks).
+    
+    Functions:
+    - List all external networks that can provide floating IPs
+    - Show available and used IP counts for each pool
+    - Display network configuration for floating IP allocation
+    - Provide pool capacity and utilization information
+    
+    Use when user requests:
+    - "Show floating IP pools"
+    - "List available floating IP networks"
+    - "Check floating IP capacity"
+    - "What external networks are available?"
+    
+    Returns:
+        List of floating IP pools with capacity information in JSON format.
+    """
+    try:
+        result_data = _get_floating_ip_pools()
+        
+        result = {
+            "timestamp": datetime.now().isoformat(),
+            "pools": result_data,
+            "total_pools": len(result_data)
+        }
+        
+        return json.dumps(result, indent=2)
+        
+    except Exception as e:
+        error_msg = f"Error: Failed to get floating IP pools - {str(e)}"
+        logger.error(error_msg)
+        return error_msg

src/mcp_openstack_ops/tools/get_floating_ip_pools.py

@@ -0,0 +1,42 @@
+"""Tool implementation for get_floating_ips."""
+
+import json
+from datetime import datetime
+from ..functions import get_floating_ips as _get_floating_ips
+from ..mcp_main import (
+    logger,
+    mcp,
+)
+
+@mcp.tool()
+async def get_floating_ips() -> str:
+    """
+    Get list of floating IPs with their associations.
+    
+    Functions:
+    - Query floating IPs and their current status
+    - Display associated fixed IPs and ports
+    - Show floating IP pool and router associations
+    - Provide floating IP allocation and usage information
+    
+    Use when user requests floating IP information, external connectivity queries, or IP management tasks.
+    
+    Returns:
+        List of floating IPs with detailed association information in JSON format.
+    """
+    try:
+        logger.info("Fetching floating IPs")
+        floating_ips = _get_floating_ips()
+        
+        result = {
+            "timestamp": datetime.now().isoformat(),
+            "total_floating_ips": len(floating_ips),
+            "floating_ips": floating_ips
+        }
+        
+        return json.dumps(result, indent=2, ensure_ascii=False)
+        
+    except Exception as e:
+        error_msg = f"Error: Failed to fetch floating IPs - {str(e)}"
+        logger.error(error_msg)
+        return error_msg

src/mcp_openstack_ops/tools/get_floating_ips.py

@@ -0,0 +1,42 @@
+"""Tool implementation for get_heat_stacks."""
+
+import json
+from datetime import datetime
+from ..functions import get_heat_stacks as _get_heat_stacks
+from ..mcp_main import (
+    logger,
+    mcp,
+)
+
+@mcp.tool()
+async def get_heat_stacks() -> str:
+    """
+    Get list of Heat orchestration stacks.
+    
+    Functions:
+    - Query Heat stacks and their current status
+    - Display stack creation and update timestamps
+    - Show stack templates and resource information
+    - Provide orchestration deployment information
+    
+    Use when user requests stack information, orchestration queries, or infrastructure-as-code status.
+    
+    Returns:
+        List of Heat stacks with detailed information in JSON format.
+    """
+    try:
+        logger.info("Fetching Heat stacks")
+        stacks = _get_heat_stacks()
+        
+        result = {
+            "timestamp": datetime.now().isoformat(),
+            "total_stacks": len(stacks),
+            "stacks": stacks
+        }
+        
+        return json.dumps(result, indent=2, ensure_ascii=False)
+        
+    except Exception as e:
+        error_msg = f"Error: Failed to fetch Heat stacks - {str(e)}"
+        logger.error(error_msg)
+        return error_msg

src/mcp_openstack_ops/tools/get_heat_stacks.py

@@ -0,0 +1,43 @@
+"""Tool implementation for get_hypervisor_details."""
+
+import json
+from datetime import datetime
+from ..functions import get_hypervisor_details as _get_hypervisor_details
+from ..mcp_main import (
+    logger,
+    mcp,
+)
+
+@mcp.tool()
+async def get_hypervisor_details(
+    hypervisor_name: str = "all"
+) -> str:
+    """
+    Get detailed information about hypervisors
+    
+    Args:
+        hypervisor_name: Name/ID of specific hypervisor or "all" for all hypervisors
+    
+    Returns:
+        JSON string with hypervisor details and statistics
+    """
+    try:
+        logger.info(f"Getting hypervisor details for: {hypervisor_name}")
+        
+        hypervisor_result = _get_hypervisor_details(hypervisor_name=hypervisor_name)
+        
+        response = {
+            "timestamp": datetime.now().isoformat(),
+            "operation": "get_hypervisor_details",
+            "parameters": {
+                "hypervisor_name": hypervisor_name
+            },
+            "result": hypervisor_result
+        }
+        
+        return json.dumps(response, indent=2, ensure_ascii=False)
+        
+    except Exception as e:
+        error_msg = f"Error: Failed to get hypervisor details - {str(e)}"
+        logger.error(error_msg)
+        return error_msg

src/mcp_openstack_ops/tools/get_hypervisor_details.py

@@ -0,0 +1,43 @@
+"""Tool implementation for get_image_detail_list."""
+
+import json
+from datetime import datetime
+from ..functions import get_image_detail_list as _get_image_detail_list
+from ..mcp_main import (
+    logger,
+    mcp,
+)
+
+@mcp.tool()
+async def get_image_detail_list() -> str:
+    """
+    Get detailed list of all images with comprehensive metadata.
+    
+    Functions:
+    - List all images available in the project
+    - Show image status, size, and format information
+    - Display image properties and metadata
+    - Provide ownership and visibility details
+    
+    Use when user requests image listing, image information, or image metadata details.
+    
+    Returns:
+        Comprehensive image list in JSON format with detailed metadata, properties, and status information.
+    """
+    try:
+        logger.info("Fetching detailed image list")
+        images = _get_image_detail_list()
+        
+        response = {
+            "timestamp": datetime.now().isoformat(),
+            "images": images,
+            "count": len(images),
+            "operation": "list_images_detailed"
+        }
+        
+        return json.dumps(response, indent=2, ensure_ascii=False)
+        
+    except Exception as e:
+        error_msg = f"Error: Failed to fetch detailed image list - {str(e)}"
+        logger.error(error_msg)
+        return error_msg