import context
import mcp
import x.async as xasync

struct EchoArgs {
pub:
	text string
}

fn dispatch_tool(raw string) !mcp.Response {
	req := mcp.decode_request(raw)!
	if req.method != 'tools/call' {
		return error('unsupported MCP method')
	}
	args := req.decode_params[EchoArgs]()!
	result := mcp.tool_text_result('echo: ${args.text}')
	return mcp.new_response(1, result, mcp.ResponseError{})
}

fn main() {
	request := mcp.new_request(1, 'tools/call', EchoArgs{
		text: 'hello'
	})
	mut task := xasync.run[mcp.Response](fn [request] (mut ctx context.Context) !mcp.Response {
		done := ctx.done()
		select {
			_ := <-done {
				return ctx.err()
			}
			else {}
		}
		return dispatch_tool(request.encode())!
	})!

	response := task.wait()!
	println(response.encode())
}