Grasshopper MCP Workflow

"""
Grasshopper MCP 通信客戶端

提供統一的 Grasshopper MCP 通信接口
"""

import socket
import json
from typing import Dict, Any, Optional
from threading import Lock


class GrasshopperClient:
    """Grasshopper MCP 通信客戶端"""
    
    def __init__(self, host: str = "localhost", port: int = 8080):
        """
        初始化 Grasshopper 客戶端
        
        Args:
            host: Grasshopper MCP 服務器地址
            port: Grasshopper MCP 服務器端口
        """
        self.host = host
        self.port = port
        self._print_lock = Lock()  # 線程安全的打印鎖
    
    def send_command(self, command_type: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
        """
        向 Grasshopper MCP 發送命令
        
        Args:
            command_type: 命令類型（如 "add_component", "connect_components"）
            params: 命令參數字典
        
        Returns:
            響應字典，包含 success 和 data/error 字段
        """
        if params is None:
            params = {}
        
        command = {
            "type": command_type,
            "parameters": params
        }
        
        try:
            # 連接到 Grasshopper MCP
            client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            client.connect((self.host, self.port))
            
            # 發送命令
            command_json = json.dumps(command, ensure_ascii=False)
            client.sendall((command_json + "\n").encode("utf-8"))
            
            # 接收響應
            response_data = b""
            while True:
                chunk = client.recv(4096)
                if not chunk:
                    break
                response_data += chunk
                if response_data.endswith(b"\n"):
                    break
            
            # 處理可能的 BOM
            response_str = response_data.decode("utf-8-sig").strip()
            response = json.loads(response_str)
            client.close()
            return response
        except Exception as e:
            return {
                "success": False,
                "error": f"與 Grasshopper 通信時出錯: {str(e)}"
            }
    
    def safe_print(self, *args, **kwargs):
        """線程安全的打印函數"""
        with self._print_lock:
            print(*args, **kwargs)
    
    def extract_component_id(self, response: Dict[str, Any]) -> Optional[str]:
        """
        從響應中提取組件 ID
        
        Args:
            response: Grasshopper MCP 響應字典
        
        Returns:
            組件 ID 字符串，如果提取失敗則返回 None
        """
        if not response.get("success"):
            return None
        
        # 格式1: response.data.id
        data = response.get("data")
        if data and isinstance(data, dict):
            comp_id = data.get("id")
            if comp_id:
                return str(comp_id)
        
        # 格式2: response.result.id
        result = response.get("result")
        if result and isinstance(result, dict):
            comp_id = result.get("id")
            if comp_id:
                return str(comp_id)
        
        # 格式3: response.data 直接是 ID
        if data and isinstance(data, str):
            return data
        
        # 格式4: response.result 直接是 ID
        if result and isinstance(result, str):
            return result
        
        return None