GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-11 18:11:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add uixt tool swipe

Browse Source
This commit is contained in:
lilong.129
2025-05-21 17:35:30 +08:00
parent 7724cf0062
commit 60e608f101
3 changed files with 46 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
mcphost/mcp_server.go
View File

@@ -167,10 +167,10 @@ func (ums *MCPServer4XTDriver) addTools() {
// Swipe Tool
swipeParams := append(
[]mcp.ToolOption{mcp.WithDescription("Swipes on the device screen from one point to another.")},
[]mcp.ToolOption{mcp.WithDescription("Swipe on the screen")},
commonToolOptions...,
)
swipeParams = append(swipeParams, generateMCPOptions(types.DragRequest{})...)
swipeParams = append(swipeParams, generateMCPOptions(types.SwipeRequest{})...)
swipeTool := mcp.NewTool("swipe", swipeParams...)
ums.mcpServer.AddTool(swipeTool, ums.handleSwipe)
ums.tools = append(ums.tools, swipeTool)
@@ -354,22 +354,54 @@ func (ums *MCPServer4XTDriver) handleSwipe(ctx context.Context, request mcp.Call
if err != nil {
return nil, err
}
var swipeReq types.DragRequest
var swipeReq types.SwipeRequest
if err := mapToStruct(request.Params.Arguments, &swipeReq); err != nil {
return mcp.NewToolResultError("parse parameters error: " + err.Error()), nil
}
actionOptions := []option.ActionOption{}
if swipeReq.Duration > 0 {
actionOptions = append(actionOptions, option.WithDuration(swipeReq.Duration/1000.0))
// enum direction: up, down, left, right
switch swipeReq.Direction {
case "up":
err = driverExt.Swipe(0.5, 0.5, 0.5, 0.1)
case "down":
err = driverExt.Swipe(0.5, 0.5, 0.5, 0.9)
case "left":
err = driverExt.Swipe(0.5, 0.5, 0.1, 0.5)
case "right":
err = driverExt.Swipe(0.5, 0.5, 0.9, 0.5)
default:
return mcp.NewToolResultError(fmt.Sprintf("get unexpected swipe direction: %s", swipeReq.Direction)), nil
}
err = driverExt.Swipe(swipeReq.FromX, swipeReq.FromY,
swipeReq.ToX, swipeReq.ToY, actionOptions...)
if err != nil {
return mcp.NewToolResultError("Swipe failed: " + err.Error()), nil
}
return mcp.NewToolResultText(
fmt.Sprintf("swipe %s success", swipeReq.Direction),
), nil
}
// handleDrag handles the drag tool call.
func (ums *MCPServer4XTDriver) handleDrag(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
driverExt, err := ums.setupXTDriver(ctx, request.Params.Arguments)
if err != nil {
return nil, err
}
var dragReq types.DragRequest
if err := mapToStruct(request.Params.Arguments, &dragReq); err != nil {
return mcp.NewToolResultError("parse parameters error: " + err.Error()), nil
}
actionOptions := []option.ActionOption{}
if dragReq.Duration > 0 {
actionOptions = append(actionOptions, option.WithDuration(dragReq.Duration/1000.0))
}
err = driverExt.Swipe(dragReq.FromX, dragReq.FromY,
dragReq.ToX, dragReq.ToY, actionOptions...)
if err != nil {
return mcp.NewToolResultError("Swipe failed: " + err.Error()), nil
}
return mcp.NewToolResultText(
fmt.Sprintf("swipe (%f,%f)->(%f,%f) success",
swipeReq.FromX, swipeReq.FromY, swipeReq.ToX, swipeReq.ToY),
dragReq.FromX, dragReq.FromY, dragReq.ToX, dragReq.ToY),
), nil
}