rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add tasks tools

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-11-24 16:23:08 +01:00
parent 0d985b40cc
commit 66484113d3
5 changed files with 698 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
pkg/server/api/mcp/v1/server/server.go
View File

@@ -65,6 +65,12 @@ type ResolverInterface interface {
UnlinkControlAuditTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UnlinkControlAuditInput) (*mcp.CallToolResult, types.UnlinkControlAuditOutput, error)
LinkControlSnapshotTool(ctx context.Context, req *mcp.CallToolRequest, input *types.LinkControlSnapshotInput) (*mcp.CallToolResult, types.LinkControlSnapshotOutput, error)
UnlinkControlSnapshotTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UnlinkControlSnapshotInput) (*mcp.CallToolResult, types.UnlinkControlSnapshotOutput, error)
ListTasksTool(ctx context.Context, req *mcp.CallToolRequest, input *types.ListTasksInput) (*mcp.CallToolResult, types.ListTasksOutput, error)
GetTaskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetTaskInput) (*mcp.CallToolResult, types.GetTaskOutput, error)
AddTaskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddTaskInput) (*mcp.CallToolResult, types.AddTaskOutput, error)
UpdateTaskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateTaskInput) (*mcp.CallToolResult, types.UpdateTaskOutput, error)
AssignTaskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AssignTaskInput) (*mcp.CallToolResult, types.AssignTaskOutput, error)
UnassignTaskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UnassignTaskInput) (*mcp.CallToolResult, types.UnassignTaskOutput, error)
}
// New creates a new MCP server instance with all handlers registered.
@@ -634,4 +640,64 @@ func registerToolHandlers(server *mcp.Server, resolver ResolverInterface) {
},
resolver.UnlinkControlSnapshotTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "listTasks",
Description: "List all tasks for the organization or measure",
InputSchema: types.ListTasksToolInputSchema,
OutputSchema: types.ListTasksToolOutputSchema,
},
resolver.ListTasksTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "getTask",
Description: "Get a task by ID",
InputSchema: types.GetTaskToolInputSchema,
OutputSchema: types.GetTaskToolOutputSchema,
},
resolver.GetTaskTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "addTask",
Description: "Add a new task to the organization",
InputSchema: types.AddTaskToolInputSchema,
OutputSchema: types.AddTaskToolOutputSchema,
},
resolver.AddTaskTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "updateTask",
Description: "Update an existing task",
InputSchema: types.UpdateTaskToolInputSchema,
OutputSchema: types.UpdateTaskToolOutputSchema,
},
resolver.UpdateTaskTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "assignTask",
Description: "Assign a task to a person",
InputSchema: types.AssignTaskToolInputSchema,
OutputSchema: types.AssignTaskToolOutputSchema,
},
resolver.AssignTaskTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "unassignTask",
Description: "Unassign a task from a person",
InputSchema: types.UnassignTaskToolInputSchema,
OutputSchema: types.UnassignTaskToolOutputSchema,
},
resolver.UnassignTaskTool,
)
}