"""
Touch simulation tools - Simulate taps on the Solar2D simulator.
"""
import json
import os
import tempfile
from pathlib import Path
from mcp.types import Tool, TextContent
from utils import find_main_lua
# Tool definitions
SIMULATE_TAP_TOOL = Tool(
name="simulate_tap",
description="Simulate a tap/click in the Solar2D simulator. Specify a bounding box using percentages and the tool taps the center. Example: a button spanning 30-50% horizontally and 60-70% vertically would use left=30, right=50, top=60, bottom=70.",
inputSchema={
"type": "object",
"properties": {
"project_path": {
"type": "string",
"description": "Path to the project directory or main.lua file"
},
"left": {
"type": "number",
"description": "Left edge of target as percentage (0=left edge of screen)"
},
"right": {
"type": "number",
"description": "Right edge of target as percentage (100=right edge of screen)"
},
"top": {
"type": "number",
"description": "Top edge of target as percentage (0=top of screen)"
},
"bottom": {
"type": "number",
"description": "Bottom edge of target as percentage (100=bottom of screen)"
}
},
"required": ["project_path", "left", "right", "top", "bottom"]
}
)
GET_DISPLAY_INFO_TOOL = Tool(
name="get_display_info",
description="Get the Solar2D display coordinate system. Call this before tapping to understand how screenshot pixels map to tap coordinates. Screenshots are captured at contentWidth x contentHeight. Tap coordinates use the same content space.",
inputSchema={
"type": "object",
"properties": {
"project_path": {
"type": "string",
"description": "Path to the project directory or main.lua file"
}
},
"required": ["project_path"]
}
)
SIMULATE_DRAG_TOOL = Tool(
name="simulate_drag",
description="Simulate a drag/swipe gesture in the Solar2D simulator. Specify start and end bounding boxes using percentages. The gesture moves from the center of the start box to the center of the end box over the given duration.",
inputSchema={
"type": "object",
"properties": {
"project_path": {
"type": "string",
"description": "Path to the project directory or main.lua file"
},
"start_left": {
"type": "number",
"description": "Left edge of start area as percentage (0=left edge of screen)"
},
"start_right": {
"type": "number",
"description": "Right edge of start area as percentage (100=right edge of screen)"
},
"start_top": {
"type": "number",
"description": "Top edge of start area as percentage (0=top of screen)"
},
"start_bottom": {
"type": "number",
"description": "Bottom edge of start area as percentage (100=bottom of screen)"
},
"end_left": {
"type": "number",
"description": "Left edge of end area as percentage (0=left edge of screen)"
},
"end_right": {
"type": "number",
"description": "Right edge of end area as percentage (100=right edge of screen)"
},
"end_top": {
"type": "number",
"description": "Top edge of end area as percentage (0=top of screen)"
},
"end_bottom": {
"type": "number",
"description": "Bottom edge of end area as percentage (100=bottom of screen)"
},
"duration": {
"type": "number",
"description": "Duration of the drag in milliseconds (default: 300)",
"default": 300
}
},
"required": ["project_path", "start_left", "start_right", "start_top", "start_bottom",
"end_left", "end_right", "end_top", "end_bottom"]
}
)
# Export all tools
TOOLS = [SIMULATE_TAP_TOOL, SIMULATE_DRAG_TOOL, GET_DISPLAY_INFO_TOOL]
def _get_project_name(project_path: str) -> str:
"""Get the project name from the path."""
main_lua_path = find_main_lua(project_path)
project_dir = str(Path(main_lua_path).parent)
return Path(project_dir).name
def _get_control_file(project_name: str) -> str:
"""Get the touch control file path."""
return os.path.join(tempfile.gettempdir(), f"solar2d_touch_{project_name}.control")
def _get_info_file(project_name: str) -> str:
"""Get the display info output file path."""
return os.path.join(tempfile.gettempdir(), f"solar2d_display_{project_name}.json")
async def handle_simulate_tap(arguments: dict) -> list[TextContent]:
"""Handle simulate_tap tool call."""
project_path = arguments.get("project_path")
left = arguments.get("left")
right = arguments.get("right")
top = arguments.get("top")
bottom = arguments.get("bottom")
if not project_path:
return [TextContent(type="text", text="Error: project_path is required")]
if None in (left, right, top, bottom):
return [TextContent(type="text", text="Error: left, right, top, bottom are all required")]
project_name = _get_project_name(project_path)
control_file = _get_control_file(project_name)
info_file = _get_info_file(project_name)
# Read display info to convert percentages to coordinates
if not os.path.exists(info_file):
return [TextContent(
type="text",
text="Display info not found. Make sure the simulator is running."
)]
try:
with open(info_file, 'r') as f:
info = json.load(f)
content_width = info.get('contentWidth')
content_height = info.get('contentHeight')
if not content_width or not content_height:
return [TextContent(type="text", text="Error: Invalid display info")]
# Calculate center of bounding box and convert to pixels
x_percent = (left + right) / 2
y_percent = (top + bottom) / 2
x = int(content_width * x_percent / 100)
y = int(content_height * y_percent / 100)
except Exception as e:
return [TextContent(type="text", text=f"Error reading display info: {str(e)}")]
# Write tap command to control file
command = f"tap,{x},{y}"
with open(control_file, 'w') as f:
f.write(command)
return [TextContent(
type="text",
text=f"Tap sent to center of box ({left}-{right}%, {top}-{bottom}%)"
)]
async def handle_simulate_drag(arguments: dict) -> list[TextContent]:
"""Handle simulate_drag tool call."""
project_path = arguments.get("project_path")
start_left = arguments.get("start_left")
start_right = arguments.get("start_right")
start_top = arguments.get("start_top")
start_bottom = arguments.get("start_bottom")
end_left = arguments.get("end_left")
end_right = arguments.get("end_right")
end_top = arguments.get("end_top")
end_bottom = arguments.get("end_bottom")
duration = arguments.get("duration", 300)
if not project_path:
return [TextContent(type="text", text="Error: project_path is required")]
if None in (start_left, start_right, start_top, start_bottom,
end_left, end_right, end_top, end_bottom):
return [TextContent(type="text", text="Error: all start and end bounding box parameters are required")]
project_name = _get_project_name(project_path)
control_file = _get_control_file(project_name)
info_file = _get_info_file(project_name)
# Read display info to convert percentages to coordinates
if not os.path.exists(info_file):
return [TextContent(
type="text",
text="Display info not found. Make sure the simulator is running."
)]
try:
with open(info_file, 'r') as f:
info = json.load(f)
content_width = info.get('contentWidth')
content_height = info.get('contentHeight')
if not content_width or not content_height:
return [TextContent(type="text", text="Error: Invalid display info")]
# Calculate center of start bounding box
sx_percent = (start_left + start_right) / 2
sy_percent = (start_top + start_bottom) / 2
x1 = int(content_width * sx_percent / 100)
y1 = int(content_height * sy_percent / 100)
# Calculate center of end bounding box
ex_percent = (end_left + end_right) / 2
ey_percent = (end_top + end_bottom) / 2
x2 = int(content_width * ex_percent / 100)
y2 = int(content_height * ey_percent / 100)
except Exception as e:
return [TextContent(type="text", text=f"Error reading display info: {str(e)}")]
# Write drag command to control file
command = f"drag,{x1},{y1},{x2},{y2},{int(duration)}"
with open(control_file, 'w') as f:
f.write(command)
return [TextContent(
type="text",
text=f"Drag sent from ({start_left}-{start_right}%, {start_top}-{start_bottom}%) to ({end_left}-{end_right}%, {end_top}-{end_bottom}%) over {int(duration)}ms"
)]
async def handle_get_display_info(arguments: dict) -> list[TextContent]:
"""Handle get_display_info tool call."""
project_path = arguments.get("project_path")
if not project_path:
return [TextContent(type="text", text="Error: project_path is required")]
project_name = _get_project_name(project_path)
info_file = _get_info_file(project_name)
if not os.path.exists(info_file):
return [TextContent(
type="text",
text="Display info not found. Make sure the simulator is running (info is written on startup)."
)]
try:
with open(info_file, 'r') as f:
info = json.load(f)
lines = [
"Solar2D Display Info:",
"",
f"Content Size: {info.get('contentWidth', '?')} x {info.get('contentHeight', '?')}",
f"Actual Content Size: {info.get('actualContentWidth', '?')} x {info.get('actualContentHeight', '?')}",
f"Screen Origin: ({info.get('screenOriginX', '?')}, {info.get('screenOriginY', '?')})",
"",
"Note: Screenshots are captured at content size.",
"Tap coordinates should be in content space (0,0 is top-left of content area)."
]
return [TextContent(type="text", text="\n".join(lines))]
except Exception as e:
return [TextContent(type="text", text=f"Error reading display info: {str(e)}")]
