#!/usr/bin/env node
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { RohlikAPI } from "./rohlik-api.js";
import { createSearchProductsTool } from "./tools/search-products.js";
import { createCartManagementTools } from "./tools/cart-management.js";
import { createShoppingListsTool } from "./tools/shopping-lists.js";
import { createAccountDataTool } from "./tools/account-data.js";
import { createOrderHistoryTool } from "./tools/order-history.js";
import { createDeliveryInfoTool } from "./tools/delivery-info.js";
import { createUpcomingOrdersTool } from "./tools/upcoming-orders.js";
import { createPremiumInfoTool } from "./tools/premium-info.js";
import { createDeliverySlotsTool } from "./tools/delivery-slots.js";
import { createAnnouncementsTool } from "./tools/announcements.js";
import { createReusableBagsTool } from "./tools/reusable-bags.js";
import { createOrderDetailTool } from "./tools/order-detail.js";
import { createFrequentItemsTool } from "./tools/frequent-items.js";
import { createMealSuggestionsTool } from "./tools/meal-suggestions.js";
import { createShoppingScenariosTool } from "./tools/shopping-scenarios.js";
const server = new McpServer(
{
name: "rohlik-mcp",
version: "1.0.0",
},
{
capabilities: {
tools: {},
},
}
);
function getCredentials() {
const username = process.env.ROHLIK_USERNAME;
const password = process.env.ROHLIK_PASSWORD;
if (!username || !password) {
throw new Error('ROHLIK_USERNAME and ROHLIK_PASSWORD environment variables are required');
}
return { username, password };
}
function createRohlikAPI() {
const credentials = getCredentials();
return new RohlikAPI(credentials);
}
// Register all tools
const searchProducts = createSearchProductsTool(createRohlikAPI);
const cartTools = createCartManagementTools(createRohlikAPI);
const shoppingLists = createShoppingListsTool(createRohlikAPI);
const accountData = createAccountDataTool(createRohlikAPI);
const orderHistory = createOrderHistoryTool(createRohlikAPI);
const deliveryInfo = createDeliveryInfoTool(createRohlikAPI);
const upcomingOrders = createUpcomingOrdersTool(createRohlikAPI);
const premiumInfo = createPremiumInfoTool(createRohlikAPI);
const deliverySlots = createDeliverySlotsTool(createRohlikAPI);
const announcements = createAnnouncementsTool(createRohlikAPI);
const reusableBags = createReusableBagsTool(createRohlikAPI);
const orderDetail = createOrderDetailTool(createRohlikAPI);
const frequentItems = createFrequentItemsTool(createRohlikAPI);
const mealSuggestions = createMealSuggestionsTool(createRohlikAPI);
const shoppingScenarios = createShoppingScenariosTool();
// Core functionality
server.registerTool(searchProducts.name, searchProducts.definition, searchProducts.handler);
server.registerTool(cartTools.addToCart.name, cartTools.addToCart.definition, cartTools.addToCart.handler);
server.registerTool(cartTools.getCartContent.name, cartTools.getCartContent.definition, cartTools.getCartContent.handler);
server.registerTool(cartTools.removeFromCart.name, cartTools.removeFromCart.definition, cartTools.removeFromCart.handler);
server.registerTool(shoppingLists.name, shoppingLists.definition, shoppingLists.handler);
server.registerTool(accountData.name, accountData.definition, accountData.handler);
// Order management
server.registerTool(orderHistory.name, orderHistory.definition, orderHistory.handler);
server.registerTool(orderDetail.name, orderDetail.definition, orderDetail.handler);
server.registerTool(upcomingOrders.name, upcomingOrders.definition, upcomingOrders.handler);
// Delivery management
server.registerTool(deliveryInfo.name, deliveryInfo.definition, deliveryInfo.handler);
server.registerTool(deliverySlots.name, deliverySlots.definition, deliverySlots.handler);
// Account features
server.registerTool(premiumInfo.name, premiumInfo.definition, premiumInfo.handler);
server.registerTool(announcements.name, announcements.definition, announcements.handler);
server.registerTool(reusableBags.name, reusableBags.definition, reusableBags.handler);
// Smart shopping features
server.registerTool(frequentItems.name, frequentItems.definition, frequentItems.handler);
server.registerTool(mealSuggestions.name, mealSuggestions.definition, mealSuggestions.handler);
server.registerTool(shoppingScenarios.name, shoppingScenarios.definition, shoppingScenarios.handler);
async function main() {
const transport = new StdioServerTransport();
await server.connect(transport);
console.error("Rohlik MCP server running on stdio");
}
main().catch((error) => {
console.error("Server error:", error);
process.exit(1);
});
