import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { IBClient } from "./ib-client.js";
import { IBGatewayManager } from "./gateway-manager.js";
import { ToolHandlers, ToolHandlerContext } from "./tool-handlers.js";
import {
AuthenticateZodShape,
GetAccountInfoZodShape,
GetPositionsZodShape,
GetMarketDataZodShape,
PlaceOrderZodShape,
GetOrderStatusZodShape,
ConfirmOrderZodShape
} from "./tool-definitions.js";
export function registerTools(
server: McpServer,
ibClient: IBClient,
gatewayManager?: IBGatewayManager,
userConfig?: any
) {
// Create handler context
const context: ToolHandlerContext = {
ibClient,
gatewayManager,
config: userConfig,
};
// Create handlers instance
const handlers = new ToolHandlers(context);
// Register authenticate tool (skip if in headless mode)
if (!userConfig?.IB_HEADLESS_MODE) {
server.tool(
"authenticate",
"Authenticate with Interactive Brokers. Usage: `{ \"confirm\": true }`.",
AuthenticateZodShape,
async (args) => await handlers.authenticate(args)
);
}
// Register get_account_info tool
server.tool(
"get_account_info",
"Get account information and balances. Usage: `{ \"confirm\": true }`.",
GetAccountInfoZodShape,
async (args) => await handlers.getAccountInfo(args)
);
// Register get_positions tool
server.tool(
"get_positions",
"Get current positions. Usage: `{}` or `{ \"accountId\": \"<id>\" }`.",
GetPositionsZodShape,
async (args) => await handlers.getPositions(args)
);
// Register get_market_data tool
server.tool(
"get_market_data",
"Get real-time market data. Usage: `{ \"symbol\": \"AAPL\" }` or `{ \"symbol\": \"AAPL\", \"exchange\": \"NASDAQ\" }`.",
GetMarketDataZodShape,
async (args) => await handlers.getMarketData(args)
);
// Register place_order tool
server.tool(
"place_order",
"Place a trading order. Examples:\n" +
"- Market buy: `{ \"accountId\":\"abc\",\"symbol\":\"AAPL\",\"action\":\"BUY\",\"orderType\":\"MKT\",\"quantity\":1 }`\n" +
"- Limit sell: `{ \"accountId\":\"abc\",\"symbol\":\"AAPL\",\"action\":\"SELL\",\"orderType\":\"LMT\",\"quantity\":1,\"price\":185.5 }`\n" +
"- Stop sell: `{ \"accountId\":\"abc\",\"symbol\":\"AAPL\",\"action\":\"SELL\",\"orderType\":\"STP\",\"quantity\":1,\"stopPrice\":180 }`\n" +
"- Suppress confirmations: `{ \"accountId\":\"abc\",\"symbol\":\"AAPL\",\"action\":\"BUY\",\"orderType\":\"MKT\",\"quantity\":1,\"suppressConfirmations\":true }`",
PlaceOrderZodShape,
async (args) => await handlers.placeOrder(args)
);
// Register get_order_status tool
server.tool(
"get_order_status",
"Get the status of a specific order. Usage: `{ \"orderId\": \"12345\" }`.",
GetOrderStatusZodShape,
async (args) => await handlers.getOrderStatus(args)
);
// Register confirm_order tool
server.tool(
"confirm_order",
"Manually confirm an order that requires confirmation. Usage: `{ \"replyId\": \"742a95a7-55f6-4d67-861b-2fd3e2b61e3c\", \"messageIds\": [\"o10151\", \"o10153\"] }`.",
ConfirmOrderZodShape,
async (args) => await handlers.confirmOrder(args)
);
}
