All files / src server.ts

100% Statements 25/25
90% Branches 9/10
100% Functions 3/3
100% Lines 23/23

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67                1x 1x       1x   1x 7x   11x 5x   5x 11x 8x       5x         1x             1x           9x         9x 9x   9x 24x   24x 24x 24x 24x         9x    
import { createRequire } from "node:module";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import type { PostNLClient } from "./postnl-client.js";
import { registerShippingTools } from "./tools/shipping.js";
import { registerTrackingTools } from "./tools/tracking.js";
import { registerDeliveryTools } from "./tools/delivery.js";
import { registerLocationTools } from "./tools/locations.js";
 
const require = createRequire(import.meta.url);
const { version } = require("../package.json") as { version: string };
 
export type Toolset = "shipping" | "tracking" | "delivery" | "locations";
 
const ALL_TOOLSETS: Toolset[] = ["shipping", "tracking", "delivery", "locations"];
 
export const parseToolsets = (env?: string): Set<Toolset> => {
  if (!env) return new Set(ALL_TOOLSETS);
 
  const requested = env.split(",").map((s) => s.trim().toLowerCase());
  const valid = new Set<Toolset>();
 
  for (const name of requested) {
    if (ALL_TOOLSETS.includes(name as Toolset)) {
      valid.add(name as Toolset);
    }
  }
 
  return valid.size > 0 ? valid : new Set(ALL_TOOLSETS);
};
 
type ToolRegisterer = (server: McpServer, client: PostNLClient, customerCode: string, customerNumber: string) => void;
 
const toolsetRegistry: Record<Toolset, ToolRegisterer[]> = {
  shipping: [registerShippingTools],
  tracking: [registerTrackingTools],
  delivery: [registerDeliveryTools],
  locations: [registerLocationTools],
};
 
export const createServer = (
  client: PostNLClient,
  customerCode: string,
  customerNumber: string,
  toolsets?: Set<Toolset>,
): McpServer => {
  const server = new McpServer({
    name: "postnl-mcp",
    version,
  });
 
  const enabled = toolsets ?? new Set(ALL_TOOLSETS);
  const registered = new Set<ToolRegisterer>();
 
  for (const toolset of enabled) {
    const registerers = toolsetRegistry[toolset];
 
    for (const register of registerers) {
      Eif (!registered.has(register)) {
        registered.add(register);
        register(server, client, customerCode, customerNumber);
      }
    }
  }
 
  return server;
};