import socket
from typing import Optional
from mcp.server.fastmcp import FastMCP

mcp = FastMCP("BDB grandMA3 MCP")

OSC_IP = "127.0.0.1"
OSC_PORT = 8000

def send_osc_message(address: str, argument: Optional[str] = None) -> str:
    """Send a simple OSC string message."""
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        
        # Build basic OSC message
        addr_bytes = address.encode('utf-8') + b'\x00'
        while len(addr_bytes) % 4 != 0:
            addr_bytes += b'\x00'
            
        if argument is not None:
            tags = b',s\x00\x00'
            arg_bytes = argument.encode('utf-8') + b'\x00'
            while len(arg_bytes) % 4 != 0:
                arg_bytes += b'\x00'
            msg = addr_bytes + tags + arg_bytes
        else:
            tags = b',\x00\x00\x00'
            msg = addr_bytes + tags
            
        sock.sendto(msg, (OSC_IP, OSC_PORT))
        return f"Successfully sent OSC message {address} {argument or ''}"
    except Exception as e:
        return f"Failed to send OSC message: {e}"

@mcp.tool()
def grandma3_ping() -> str:
    """Check if grandMA3 MCP is ready."""
    return "grandMA3 MCP is active. Sending commands via OSC."

@mcp.tool()
def execute_command(command: str) -> str:
    """Execute an arbitrary grandMA3 command via OSC. Requires grandMA3 to be listening for OSC /cmd on port 8000."""
    return send_osc_message("/cmd", command)

@mcp.tool()
def execute_macro(macro_number: int) -> str:
    """Execute a specific macro in grandMA3."""
    return send_osc_message("/cmd", f"Go Macro {macro_number}")

@mcp.tool()
def patch_fixture(fixture_id: int, name: str, fixture_type: str, address: str) -> str:
    """Patch a fixture via grandMA3 command line."""
    cmd = f'Assign Fixture {fixture_id} Name "{name}" /Type="{fixture_type}" /Patch="{address}"'
    return send_osc_message("/cmd", cmd)

if __name__ == "__main__":
    mcp.run()