GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-07-10 15:02:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add uixt tool select_device

Browse Source
This commit is contained in:
lilong.129
2025-05-20 23:11:22 +08:00
parent 1fa87819ae
commit 044eb07a35
2 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/version/VERSION
View File

@@ -1 +1 @@
v5.0.0-beta-2505202258 v5.0.0-beta-2505202311

25
mcphost/mcp_server.go
View File

@@ -91,6 +91,16 @@ func (ums *MCPServer4XTDriver) addTools() {
ums.tools = append(ums.tools, listDevicesTool) ums.tools = append(ums.tools, listDevicesTool)
ums.handlerMap[listDevicesTool.Name] = ums.handleListAvailableDevices ums.handlerMap[listDevicesTool.Name] = ums.handleListAvailableDevices
// SelectDevice Tool
selectDeviceTool := mcp.NewTool("select_device",
mcp.WithDescription("Select a device to use from the list of available devices. Use the list_available_devices tool to get a list of available devices."),
mcp.WithString("platform", mcp.Enum("android", "ios"), mcp.Description("The type of device to select")),
mcp.WithString("serial", mcp.Description("The device serial/udid to select")),
)
ums.mcpServer.AddTool(selectDeviceTool, ums.handleSelectDevice)
ums.tools = append(ums.tools, selectDeviceTool)
ums.handlerMap[selectDeviceTool.Name] = ums.handleSelectDevice
// TapXY Tool // TapXY Tool
tapParams := append( tapParams := append(
[]mcp.ToolOption{mcp.WithDescription("Taps on the device screen at the given coordinates.")}, []mcp.ToolOption{mcp.WithDescription("Taps on the device screen at the given coordinates.")},
@@ -160,6 +170,17 @@ func (ums *MCPServer4XTDriver) handleListAvailableDevices(ctx context.Context, r
return mcp.NewToolResultText(string(jsonResult)), nil return mcp.NewToolResultText(string(jsonResult)), nil
} }
// handleSelectDevice handles the select_device tool call.
func (ums *MCPServer4XTDriver) handleSelectDevice(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
driverExt, err := ums.setupXTDriver(ctx, request.Params.Arguments)
if err != nil {
return nil, err
}
uuid := driverExt.IDriver.GetDevice().UUID()
return mcp.NewToolResultText(fmt.Sprintf("Selected device: %s", uuid)), nil
}
// handleTapXY handles the tap_xy tool call. // handleTapXY handles the tap_xy tool call.
func (ums *MCPServer4XTDriver) handleTapXY(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { func (ums *MCPServer4XTDriver) handleTapXY(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
driverExt, err := ums.setupXTDriver(ctx, request.Params.Arguments) driverExt, err := ums.setupXTDriver(ctx, request.Params.Arguments)
@@ -237,8 +258,8 @@ func (ums *MCPServer4XTDriver) handleScreenShot(ctx context.Context, request mcp
func (ums *MCPServer4XTDriver) setupXTDriver(_ context.Context, args map[string]interface{}) (*uixt.XTDriver, error) { func (ums *MCPServer4XTDriver) setupXTDriver(_ context.Context, args map[string]interface{}) (*uixt.XTDriver, error) {
platform, _ := args["platform"].(string) platform, _ := args["platform"].(string)
serial, _ := args["serial"].(string) serial, _ := args["serial"].(string)
if platform == "" || serial == "" { if platform == "" {
return nil, fmt.Errorf("platform and serial are required") platform = "android"
} }
// Check if driver exists in cache // Check if driver exists in cache