import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
// Tool definitions — shared by the STDIO (npx) and HTTP (web-URL) entrypoints.
export const TOOLS = [
    {
        name: "list_couriers",
        annotations: { title: "List couriers", readOnlyHint: true, destructiveHint: false, openWorldHint: true },
        description: "Show supported Bangladeshi couriers and check which credentials are active.",
        inputSchema: { type: "object", properties: {}, required: [] },
    },
    {
        name: "create_parcel",
        annotations: { title: "Create parcel (book delivery)", readOnlyHint: false, destructiveHint: false, openWorldHint: true },
        description: "Book a new parcel delivery across Bangladesh (Steadfast, Pathao, RedX or Paperfly) with normalized response.",
        inputSchema: {
            type: "object",
            properties: {
                courier: { type: "string", enum: ["steadfast", "pathao", "redx", "paperfly", "auto"], description: "Target courier or 'auto' for AI smart routing (default: auto)" },
                invoice: { type: "string", description: "Unique order invoice number (e.g. INV-1002)" },
                recipient_name: { type: "string", description: "Customer full name" },
                recipient_phone: { type: "string", description: "11-digit Bangladeshi mobile number (e.g. 017XXXXXXXX)" },
                recipient_address: { type: "string", description: "Delivery address (Thana, District, Street)" },
                cod_amount: { type: "number", description: "Cash on delivery amount in BDT (0 if prepaid)" },
                note: { type: "string", description: "Special instructions for delivery rider" },
                item_weight: { type: "number", description: "Parcel weight in KG (default: 0.5)" },
                item_type: { type: "string", description: "What's inside the parcel (used by RedX)" },
                value: { type: "number", description: "Declared parcel value in BDT (used by RedX; defaults to the COD amount)" },
                delivery_area_id: { type: "number", description: "RedX only: numeric delivery-area id (auto-resolved from the address if omitted)" },
                pickup_store_id: { type: "number", description: "RedX only: your pickup store id (optional)" },
            },
            required: ["invoice", "recipient_name", "recipient_phone", "recipient_address", "cod_amount"],
        },
    },
    {
        name: "track_parcel",
        annotations: { title: "Track parcel", readOnlyHint: true, destructiveHint: false, openWorldHint: true },
        description: "Track shipment delivery status across Steadfast or Pathao using Tracking Code / Consignment ID.",
        inputSchema: {
            type: "object",
            properties: {
                tracking_code: { type: "string", description: "Consignment ID or tracking code" },
                courier: { type: "string", enum: ["steadfast", "pathao", "redx", "paperfly"], description: "Optional courier name if known" },
            },
            required: ["tracking_code"],
        },
    },
    {
        name: "get_balance",
        annotations: { title: "Get merchant balance", readOnlyHint: true, destructiveHint: false, openWorldHint: true },
        description: "Retrieve current merchant account balance and payout details from a courier.",
        inputSchema: {
            type: "object",
            properties: {
                courier: { type: "string", enum: ["steadfast", "pathao"], description: "Courier provider to check balance for" },
            },
            required: ["courier"],
        },
    },
    {
        name: "check_fraud_risk",
        annotations: { title: "Check COD fraud risk", readOnlyHint: true, destructiveHint: false, openWorldHint: true },
        description: "Analyze Bangladeshi customer phone number delivery history and return/fraud risk score before dispatching.",
        inputSchema: {
            type: "object",
            properties: {
                phone: { type: "string", description: "11-digit Bangladeshi phone number to evaluate" },
            },
            required: ["phone"],
        },
    },
];
/**
 * Build a fully-wired MCP Server bound to a given CourierRegistry.
 * The registry carries the credentials — for STDIO it comes from env,
 * for HTTP it is built per-request from that client's keys.
 */
export function buildMcpServer(registry) {
    const server = new Server({ name: "aura-courier-mcp", version: "2.3.2" }, { capabilities: { tools: {} } });
    server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOLS }));
    server.setRequestHandler(CallToolRequestSchema, async (request) => {
        const { name, arguments: args } = request.params;
        try {
            switch (name) {
                case "list_couriers":
                    return { content: [{ type: "text", text: JSON.stringify(registry.listCouriers(), null, 2) }] };
                case "create_parcel": {
                    const result = await registry.createParcel({
                        courier: args?.courier,
                        invoice: String(args?.invoice),
                        recipient_name: String(args?.recipient_name),
                        recipient_phone: String(args?.recipient_phone),
                        recipient_address: String(args?.recipient_address),
                        cod_amount: Number(args?.cod_amount || 0),
                        note: args?.note ? String(args?.note) : undefined,
                        item_weight: args?.item_weight ? Number(args?.item_weight) : 0.5,
                    });
                    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
                }
                case "track_parcel": {
                    const result = await registry.trackParcel(String(args?.tracking_code), args?.courier);
                    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
                }
                case "get_balance": {
                    const result = await registry.getBalance(String(args?.courier));
                    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
                }
                case "check_fraud_risk": {
                    const result = registry.checkFraudRisk(String(args?.phone));
                    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
                }
                default:
                    throw new Error(`Unknown tool: ${name}`);
            }
        }
        catch (error) {
            return {
                isError: true,
                content: [{ type: "text", text: `Aura Courier MCP Error: ${error?.message || String(error)}` }],
            };
        }
    });
    return server;
}