observability-toolkit 1.8.2 → 1.8.5

package/dist/server.js

@@ -11,112 +11,15 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
 import { zodToJsonSchema } from 'zod-to-json-schema';
+import { createRequire } from 'module';
 import { queryTracesTool, queryMetricsTool, queryLogsTool, queryLLMEventsTool, queryEvaluationsTool, healthCheckTool, contextStatsTool, getTraceUrlTool, setupClaudeignoreTool, } from './tools/index.js';
 import { sanitizeErrorForResponse } from './lib/error-sanitizer.js';
-// Rate limiting configuration
-const RATE_LIMIT = {
-    windowMs: 60_000, // 1 minute window
-    maxRequests: 100, // max requests per window
-    bucketCount: 60, // 60 buckets for 1-second granularity
-};
-/**
- * Sliding window counter rate limiter with O(1) operations and bounded memory.
- * Uses fixed-size bucket array to track request counts per time slice.
- *
- * Memory usage: O(bucketCount) - fixed regardless of request volume
- * Time complexity: O(bucketCount) for sum, but bucketCount is constant (60)
- *
- * @exported for testing
- */
-export class RateLimiter {
-    buckets;
-    bucketTimestamps;
-    bucketMs;
-    windowMs;
-    maxRequests;
-    bucketCount;
-    constructor(windowMs = RATE_LIMIT.windowMs, maxRequests = RATE_LIMIT.maxRequests, bucketCount = RATE_LIMIT.bucketCount) {
-        this.windowMs = windowMs;
-        this.maxRequests = maxRequests;
-        this.bucketCount = bucketCount;
-        this.bucketMs = Math.floor(windowMs / bucketCount);
-        // Pre-allocate fixed-size arrays
-        this.buckets = new Array(bucketCount).fill(0);
-        this.bucketTimestamps = new Array(bucketCount).fill(0);
-    }
-    /**
-     * Get the bucket index for a given timestamp.
-     * Uses modulo to create circular buffer behavior.
-     */
-    getBucketIndex(timestamp) {
-        return Math.floor(timestamp / this.bucketMs) % this.bucketCount;
-    }
-    /**
-     * Clear expired buckets and return the current valid count.
-     * A bucket is expired if its timestamp is older than windowMs ago.
-     */
-    cleanAndCount(now) {
-        const windowStart = now - this.windowMs;
-        let count = 0;
-        for (let i = 0; i < this.bucketCount; i++) {
-            if (this.bucketTimestamps[i] < windowStart) {
-                // Bucket is expired, reset it
-                this.buckets[i] = 0;
-                this.bucketTimestamps[i] = 0;
-            }
-            else {
-                count += this.buckets[i];
-            }
-        }
-        return count;
-    }
-    /**
-     * Check if a request is allowed and record it if so.
-     * O(bucketCount) time complexity where bucketCount is constant.
-     */
-    isAllowed() {
-        const now = Date.now();
-        const currentCount = this.cleanAndCount(now);
-        if (currentCount >= this.maxRequests) {
-            return false;
-        }
-        // Record the request in the current bucket
-        const bucketIndex = this.getBucketIndex(now);
-        // If this bucket is from a different time period, reset it
-        const bucketWindowStart = now - this.windowMs;
-        if (this.bucketTimestamps[bucketIndex] < bucketWindowStart) {
-            this.buckets[bucketIndex] = 0;
-        }
-        this.buckets[bucketIndex]++;
-        this.bucketTimestamps[bucketIndex] = now;
-        return true;
-    }
-    /**
-     * Get the number of remaining allowed requests in the current window.
-     */
-    getRemainingRequests() {
-        const now = Date.now();
-        const currentCount = this.cleanAndCount(now);
-        return Math.max(0, this.maxRequests - currentCount);
-    }
-    /**
-     * Get the current memory footprint (for testing).
-     * Returns the fixed size of internal arrays.
-     */
-    getMemoryFootprint() {
-        return {
-            bucketCount: this.bucketCount,
-            arraySize: this.buckets.length + this.bucketTimestamps.length,
-        };
-    }
-    /**
-     * Reset the rate limiter state (for testing).
-     */
-    reset() {
-        this.buckets.fill(0);
-        this.bucketTimestamps.fill(0);
-    }
-}
+import { RateLimiter, ServerInitError } from './lib/server-utils.js';
+// Import version from package.json (L1 fix)
+const require = createRequire(import.meta.url);
+const { version: packageVersion } = require('../package.json');
+// Re-export for backwards compatibility
+export { RateLimiter, ServerInitError } from './lib/server-utils.js';
 const rateLimiter = new RateLimiter();
 // Tool definitions
 const tools = [
@@ -142,71 +45,160 @@ function convertZodToJsonSchema(zodSchema) {
     const { $schema, ...rest } = schema;
     return rest;
 }
-async function main() {
-    const server = new Server({
-        name: 'observability-toolkit',
-        version: '1.0.0',
-    }, {
-        capabilities: {
-            tools: {},
-        },
-    });
-    // List available tools
-    server.setRequestHandler(ListToolsRequestSchema, async () => {
-        return {
-            tools: tools.map(tool => ({
-                name: tool.name,
-                description: tool.description,
-                inputSchema: convertZodToJsonSchema(tool.inputSchema),
-            })),
-        };
-    });
-    // Handle tool calls
-    server.setRequestHandler(CallToolRequestSchema, async (request) => {
-        const { name, arguments: args } = request.params;
-        // Rate limiting check
-        if (!rateLimiter.isAllowed()) {
-            return {
-                content: [{ type: 'text', text: `Rate limit exceeded. Please wait before making more requests. Remaining: ${rateLimiter.getRemainingRequests()}` }],
-                isError: true,
-            };
+/**
+ * Validate that all tools are properly configured before starting the server.
+ * @throws ServerInitError if any tool is misconfigured
+ */
+function validateTools() {
+    if (tools.length === 0) {
+        throw new ServerInitError('tool-validation', null, 'No tools registered. Server cannot start without at least one tool.');
+    }
+    for (const tool of tools) {
+        if (!tool.name || typeof tool.name !== 'string') {
+            throw new ServerInitError('tool-validation', null, `Tool missing required 'name' property`);
         }
-        const tool = tools.find(t => t.name === name);
-        if (!tool) {
-            return {
-                content: [{ type: 'text', text: `Unknown tool: ${name}` }],
-                isError: true,
-            };
+        if (!tool.handler || typeof tool.handler !== 'function') {
+            throw new ServerInitError('tool-validation', null, `Tool '${tool.name}' missing required 'handler' function`);
         }
-        try {
-            // Parse and validate input
-            const input = tool.inputSchema.parse(args || {});
-            // Execute handler - use type assertion since we've validated input
-            const result = await tool.handler(input);
-            return {
-                content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
-            };
+        if (!tool.inputSchema) {
+            throw new ServerInitError('tool-validation', null, `Tool '${tool.name}' missing required 'inputSchema'`);
+        }
+        if (!tool.description || typeof tool.description !== 'string') {
+            throw new ServerInitError('tool-validation', null, `Tool '${tool.name}' missing required 'description'`);
         }
-        catch (error) {
-            // Security: sanitize error messages to prevent information disclosure
-            // - Removes absolute file paths
-            // - Removes stack traces in production
-            // - Returns generic messages for internal errors
-            const errorText = sanitizeErrorForResponse(error);
+    }
+}
+async function main() {
+    // Step 1: Validate tool configuration
+    console.error('[init] Validating tool configuration...');
+    try {
+        validateTools();
+        console.error(`[init] Validated ${tools.length} tools successfully`);
+    }
+    catch (error) {
+        if (error instanceof ServerInitError) {
+            throw error;
+        }
+        throw new ServerInitError('tool-validation', error, 'Failed to validate tool configuration');
+    }
+    // Step 2: Create MCP server instance
+    console.error('[init] Creating MCP server instance...');
+    let server;
+    try {
+        server = new Server({
+            name: 'observability-toolkit',
+            version: packageVersion,
+        }, {
+            capabilities: {
+                tools: {},
+            },
+        });
+    }
+    catch (error) {
+        throw new ServerInitError('server-creation', error, 'Failed to create MCP server instance. Check @modelcontextprotocol/sdk installation.');
+    }
+    // Step 3: Register request handlers
+    console.error('[init] Registering request handlers...');
+    try {
+        // List available tools
+        server.setRequestHandler(ListToolsRequestSchema, async () => {
             return {
-                content: [{ type: 'text', text: errorText }],
-                isError: true,
+                tools: tools.map(tool => ({
+                    name: tool.name,
+                    description: tool.description,
+                    inputSchema: convertZodToJsonSchema(tool.inputSchema),
+                })),
             };
-        }
-    });
-    // Connect via stdio
-    const transport = new StdioServerTransport();
-    await server.connect(transport);
-    // Log startup to stderr (stdout is for MCP protocol)
-    console.error('Observability Toolkit MCP server started');
+        });
+        // Handle tool calls
+        server.setRequestHandler(CallToolRequestSchema, async (request) => {
+            const { name, arguments: args } = request.params;
+            // Rate limiting check
+            if (!rateLimiter.isAllowed()) {
+                const limitedCount = rateLimiter.getRateLimitedCount();
+                console.warn(`[rate-limit] Request rejected for tool '${name}' (total rejected: ${limitedCount})`);
+                return {
+                    content: [{ type: 'text', text: `Rate limit exceeded. Please wait before making more requests. Remaining: ${rateLimiter.getRemainingRequests()}` }],
+                    isError: true,
+                };
+            }
+            const tool = tools.find(t => t.name === name);
+            if (!tool) {
+                return {
+                    content: [{ type: 'text', text: `Unknown tool: ${name}` }],
+                    isError: true,
+                };
+            }
+            try {
+                // Parse and validate input
+                const input = tool.inputSchema.parse(args || {});
+                // Execute handler - use type assertion since we've validated input
+                const result = await tool.handler(input);
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
+                };
+            }
+            catch (error) {
+                // Security: sanitize error messages to prevent information disclosure
+                // - Removes absolute file paths
+                // - Removes stack traces in production
+                // - Returns generic messages for internal errors
+                const errorText = sanitizeErrorForResponse(error);
+                return {
+                    content: [{ type: 'text', text: errorText }],
+                    isError: true,
+                };
+            }
+        });
+    }
+    catch (error) {
+        throw new ServerInitError('handler-registration', error, 'Failed to register request handlers');
+    }
+    // Step 4: Create and connect transport
+    console.error('[init] Connecting stdio transport...');
+    try {
+        const transport = new StdioServerTransport();
+        await server.connect(transport);
+    }
+    catch (error) {
+        throw new ServerInitError('transport-connection', error, 'Failed to connect stdio transport. Ensure stdin/stdout are available.');
+    }
+    // Log startup success to stderr (stdout is for MCP protocol)
+    console.error('[init] Observability Toolkit MCP server started successfully');
+    console.error(`[init] Available tools: ${tools.map(t => t.name).join(', ')}`);
 }
 main().catch((error) => {
-    console.error('Fatal error:', error);
+    if (error instanceof ServerInitError) {
+        console.error(`[FATAL] Server initialization failed at step '${error.step}'`);
+        console.error(`[FATAL] Error: ${error.message}`);
+        if (error.cause) {
+            console.error(`[FATAL] Cause: ${error.cause instanceof Error ? error.cause.message : String(error.cause)}`);
+        }
+        console.error('[FATAL] Please check:');
+        switch (error.step) {
+            case 'tool-validation':
+                console.error('  - All tool modules are properly imported');
+                console.error('  - Each tool has name, description, inputSchema, and handler');
+                break;
+            case 'server-creation':
+                console.error('  - @modelcontextprotocol/sdk is installed correctly');
+                console.error('  - Run: npm install @modelcontextprotocol/sdk');
+                break;
+            case 'handler-registration':
+                console.error('  - Request schema imports are correct');
+                console.error('  - zod-to-json-schema is installed');
+                break;
+            case 'transport-connection':
+                console.error('  - Server is being run with proper stdin/stdout');
+                console.error('  - Not running in a detached process');
+                break;
+            default:
+                console.error('  - Check the error message above for details');
+        }
+    }
+    else {
+        console.error('[FATAL] Unexpected error during server startup:', error);
+    }
     process.exit(1);
 });
 //# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -1 +1 @@
-{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":";AACA;;;;;;;GAOG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,aAAa,EACb,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,qBAAqB,GACtB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AAEpE,8BAA8B;AAC9B,MAAM,UAAU,GAAG;IACjB,QAAQ,EAAE,MAAM,EAAE,kBAAkB;IACpC,WAAW,EAAE,GAAG,EAAE,0BAA0B;IAC5C,WAAW,EAAE,EAAE,EAAG,sCAAsC;CACzD,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,OAAO,WAAW;IACd,OAAO,CAAW;IAClB,gBAAgB,CAAW;IAClB,QAAQ,CAAS;IACjB,QAAQ,CAAS;IACjB,WAAW,CAAS;IACpB,WAAW,CAAS;IAErC,YACE,WAAmB,UAAU,CAAC,QAAQ,EACtC,cAAsB,UAAU,CAAC,WAAW,EAC5C,cAAsB,UAAU,CAAC,WAAW;QAE5C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAC,CAAC;QAEnD,iCAAiC;QACjC,IAAI,CAAC,OAAO,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAI,CAAC,gBAAgB,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IAED;;;OAGG;IACK,cAAc,CAAC,SAAiB;QACtC,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;IAClE,CAAC;IAED;;;OAGG;IACK,aAAa,CAAC,GAAW;QAC/B,MAAM,WAAW,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC;QACxC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,IAAI,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,WAAW,EAAE,CAAC;gBAC3C,8BAA8B;gBAC9B,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACpB,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;OAGG;IACH,SAAS;QACP,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAE7C,IAAI,YAAY,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,2CAA2C;QAC3C,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QAE7C,2DAA2D;QAC3D,MAAM,iBAAiB,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9C,IAAI,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,GAAG,iBAAiB,EAAE,CAAC;YAC3D,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5B,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;QAEzC,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,GAAG,YAAY,CAAC,CAAC;IACtD,CAAC;IAED;;;OAGG;IACH,kBAAkB;QAChB,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM;SAC9D,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC;CACF;AAED,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AAEtC,mBAAmB;AACnB,MAAM,KAAK,GAAG;IACZ,eAAe;IACf,gBAAgB;IAChB,aAAa;IACb,kBAAkB;IAClB,oBAAoB;IACpB,eAAe;IACf,gBAAgB;IAChB,eAAe;IACf,qBAAqB;CACtB,CAAC;AAEF;;;GAGG;AACH,SAAS,sBAAsB,CAAC,SAAkB;IAChD,MAAM,MAAM,GAAG,eAAe,CAAC,SAAkD,EAAE;QACjF,YAAY,EAAE,MAAM,EAAE,wCAAwC;KAC/D,CAAC,CAAC;IACH,8CAA8C;IAC9C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,MAAiC,CAAC;IAC/D,OAAO,IAAI,CAAC;AACd,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;QACE,IAAI,EAAE,uBAAuB;QAC7B,OAAO,EAAE,OAAO;KACjB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,uBAAuB;IACvB,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QAC1D,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACxB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC;aACtD,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,sBAAsB;QACtB,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC;YAC7B,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4EAA4E,WAAW,CAAC,oBAAoB,EAAE,EAAE,EAAE,CAAC;gBACnJ,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;gBAC1D,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,2BAA2B;YAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;YAEjD,mEAAmE;YACnE,MAAM,MAAM,GAAG,MAAO,IAAI,CAAC,OAAgD,CAAC,KAAK,CAAC,CAAC;YAEnF,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;aACnE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sEAAsE;YACtE,gCAAgC;YAChC,uCAAuC;YACvC,iDAAiD;YACjD,MAAM,SAAS,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;YAClD,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;gBAC5C,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAEhC,qDAAqD;IACrD,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;AAC5D,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":";AACA;;;;;;;GAOG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACvC,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,aAAa,EACb,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,qBAAqB,GACtB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACpE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAErE,4CAA4C;AAC5C,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC/C,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAwB,CAAC;AAEtF,wCAAwC;AACxC,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAErE,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AAEtC,mBAAmB;AACnB,MAAM,KAAK,GAAG;IACZ,eAAe;IACf,gBAAgB;IAChB,aAAa;IACb,kBAAkB;IAClB,oBAAoB;IACpB,eAAe;IACf,gBAAgB;IAChB,eAAe;IACf,qBAAqB;CACtB,CAAC;AAEF;;;GAGG;AACH,SAAS,sBAAsB,CAAC,SAAkB;IAChD,MAAM,MAAM,GAAG,eAAe,CAAC,SAAkD,EAAE;QACjF,YAAY,EAAE,MAAM,EAAE,wCAAwC;KAC/D,CAAC,CAAC;IACH,8CAA8C;IAC9C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,MAAiC,CAAC;IAC/D,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,SAAS,aAAa;IACpB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,IAAI,EAAE,qEAAqE,CAAC,CAAC;IAC5H,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAChD,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,IAAI,EAAE,uCAAuC,CAAC,CAAC;QAC9F,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;YACxD,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,IAAI,EAAE,SAAS,IAAI,CAAC,IAAI,uCAAuC,CAAC,CAAC;QAChH,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,IAAI,EAAE,SAAS,IAAI,CAAC,IAAI,kCAAkC,CAAC,CAAC;QAC3G,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;YAC9D,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,IAAI,EAAE,SAAS,IAAI,CAAC,IAAI,kCAAkC,CAAC,CAAC;QAC3G,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,sCAAsC;IACtC,OAAO,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;IACzD,IAAI,CAAC;QACH,aAAa,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,oBAAoB,KAAK,CAAC,MAAM,qBAAqB,CAAC,CAAC;IACvE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;YACrC,MAAM,KAAK,CAAC;QACd,CAAC;QACD,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,KAAK,EAAE,uCAAuC,CAAC,CAAC;IAC/F,CAAC;IAED,qCAAqC;IACrC,OAAO,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;IACxD,IAAI,MAAc,CAAC;IACnB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,MAAM,CACjB;YACE,IAAI,EAAE,uBAAuB;YAC7B,OAAO,EAAE,cAAc;SACxB,EACD;YACE,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE;aACV;SACF,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,KAAK,EAAE,qFAAqF,CAAC,CAAC;IAC7I,CAAC;IAED,oCAAoC;IACpC,OAAO,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;IACxD,IAAI,CAAC;QACH,uBAAuB;QACvB,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YAC1D,OAAO;gBACL,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACxB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,WAAW,EAAE,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC;iBACtD,CAAC,CAAC;aACJ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,oBAAoB;QACpB,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;YAEjD,sBAAsB;YACtB,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC;gBAC7B,MAAM,YAAY,GAAG,WAAW,CAAC,mBAAmB,EAAE,CAAC;gBACvD,OAAO,CAAC,IAAI,CAAC,2CAA2C,IAAI,sBAAsB,YAAY,GAAG,CAAC,CAAC;gBACnG,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4EAA4E,WAAW,CAAC,oBAAoB,EAAE,EAAE,EAAE,CAAC;oBACnJ,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;oBAC1D,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,2BAA2B;gBAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBAEjD,mEAAmE;gBACnE,MAAM,MAAM,GAAG,MAAO,IAAI,CAAC,OAAgD,CAAC,KAAK,CAAC,CAAC;gBAEnF,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;iBACnE,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,sEAAsE;gBACtE,gCAAgC;gBAChC,uCAAuC;gBACvC,iDAAiD;gBACjD,MAAM,SAAS,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;gBAClD,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;oBAC5C,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,eAAe,CAAC,sBAAsB,EAAE,KAAK,EAAE,qCAAqC,CAAC,CAAC;IAClG,CAAC;IAED,uCAAuC;IACvC,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;IACtD,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,eAAe,CAAC,sBAAsB,EAAE,KAAK,EAAE,uEAAuE,CAAC,CAAC;IACpI,CAAC;IAED,6DAA6D;IAC7D,OAAO,CAAC,KAAK,CAAC,8DAA8D,CAAC,CAAC;IAC9E,OAAO,CAAC,KAAK,CAAC,2BAA2B,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAChF,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;QACrC,OAAO,CAAC,KAAK,CAAC,iDAAiD,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC;QAC9E,OAAO,CAAC,KAAK,CAAC,kBAAkB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACjD,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,kBAAkB,KAAK,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9G,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;QACvC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,iBAAiB;gBACpB,OAAO,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBAC5D,OAAO,CAAC,KAAK,CAAC,+DAA+D,CAAC,CAAC;gBAC/E,MAAM;YACR,KAAK,iBAAiB;gBACpB,OAAO,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;gBACtE,OAAO,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;gBAChE,MAAM;YACR,KAAK,sBAAsB;gBACzB,OAAO,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;gBACxD,OAAO,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC;gBACrD,MAAM;YACR,KAAK,sBAAsB;gBACzB,OAAO,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;gBAClE,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBACvD,MAAM;YACR;gBACE,OAAO,CAAC,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,KAAK,CAAC,iDAAiD,EAAE,KAAK,CAAC,CAAC;IAC1E,CAAC;IACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/server.test.js

@@ -545,9 +545,10 @@ describe('CallToolRequestSchema handler', () => {
     });
 });
 /**
- * Import the RateLimiter for testing
+ * Import the RateLimiter and ServerInitError for testing
+ * Imported from server-utils to avoid starting the MCP server
  */
-import { RateLimiter } from './server.js';
+import { RateLimiter, ServerInitError } from './lib/server-utils.js';
 describe('RateLimiter', () => {
     describe('basic functionality', () => {
         it('should allow requests under the limit', () => {
@@ -704,11 +705,6 @@ describe('RateLimiter', () => {
         });
     });
     describe('edge cases', () => {
-        it('should handle zero requests allowed', () => {
-            const limiter = new RateLimiter(1000, 0, 10);
-            assert.strictEqual(limiter.isAllowed(), false);
-            assert.strictEqual(limiter.getRemainingRequests(), 0);
-        });
         it('should handle single bucket', () => {
             const limiter = new RateLimiter(1000, 5, 1);
             for (let i = 0; i < 5; i++) {
@@ -735,6 +731,34 @@ describe('RateLimiter', () => {
             assert.strictEqual(blocked, 50);
         });
     });
+    describe('bounds validation', () => {
+        it('should throw error for bucketCount < 1', () => {
+            assert.throws(() => new RateLimiter(1000, 10, 0), /bucketCount must be between 1 and/);
+        });
+        it('should throw error for bucketCount > maxBucketCount', () => {
+            assert.throws(() => new RateLimiter(1000, 10, 3601), /bucketCount must be between 1 and/);
+        });
+        it('should throw error for maxRequests < 1', () => {
+            assert.throws(() => new RateLimiter(1000, 0, 10), /maxRequests must be >= 1/);
+        });
+        it('should throw error for negative maxRequests', () => {
+            assert.throws(() => new RateLimiter(1000, -1, 10), /maxRequests must be >= 1/);
+        });
+        it('should throw error for windowMs < 1', () => {
+            assert.throws(() => new RateLimiter(0, 10, 10), /windowMs must be between 1ms and 86400000ms/);
+        });
+        it('should throw error for windowMs > 86400000 (1 day)', () => {
+            assert.throws(() => new RateLimiter(86400001, 10, 10), /windowMs must be between 1ms and 86400000ms/);
+        });
+        it('should accept valid maximum bucketCount (3600)', () => {
+            const limiter = new RateLimiter(3600000, 100, 3600);
+            assert.strictEqual(limiter.getMemoryFootprint().bucketCount, 3600);
+        });
+        it('should accept valid minimum bucketCount (1)', () => {
+            const limiter = new RateLimiter(1000, 10, 1);
+            assert.strictEqual(limiter.getMemoryFootprint().bucketCount, 1);
+        });
+    });
     describe('default configuration', () => {
         it('should use default values when not specified', () => {
             const limiter = new RateLimiter();
@@ -745,5 +769,172 @@ describe('RateLimiter', () => {
             assert.strictEqual(limiter.getRemainingRequests(), 100);
         });
     });
+    describe('observability', () => {
+        it('should track rate-limited requests count', () => {
+            const limiter = new RateLimiter(1000, 3, 10);
+            // Initially zero
+            assert.strictEqual(limiter.getRateLimitedCount(), 0);
+            // Use up all requests
+            for (let i = 0; i < 3; i++) {
+                limiter.isAllowed();
+            }
+            // Still zero - no rejections yet
+            assert.strictEqual(limiter.getRateLimitedCount(), 0);
+            // These should be rejected
+            limiter.isAllowed();
+            assert.strictEqual(limiter.getRateLimitedCount(), 1);
+            limiter.isAllowed();
+            assert.strictEqual(limiter.getRateLimitedCount(), 2);
+        });
+        it('should reset rate-limited count on reset', () => {
+            const limiter = new RateLimiter(1000, 2, 10);
+            // Exhaust limit and trigger rejections
+            limiter.isAllowed();
+            limiter.isAllowed();
+            limiter.isAllowed(); // rejected
+            limiter.isAllowed(); // rejected
+            assert.strictEqual(limiter.getRateLimitedCount(), 2);
+            // Reset should clear counter
+            limiter.reset();
+            assert.strictEqual(limiter.getRateLimitedCount(), 0);
+        });
+    });
+});
+describe('ServerInitError', () => {
+    describe('constructor', () => {
+        it('should create error with step and cause', () => {
+            const cause = new Error('Underlying error');
+            const error = new ServerInitError('tool-validation', cause);
+            assert.strictEqual(error.step, 'tool-validation');
+            assert.strictEqual(error.cause, cause);
+            assert.strictEqual(error.name, 'ServerInitError');
+            assert.ok(error.message.includes('tool-validation'));
+        });
+        it('should create error with custom message', () => {
+            const error = new ServerInitError('server-creation', null, 'Custom error message');
+            assert.strictEqual(error.message, 'Custom error message');
+            assert.strictEqual(error.step, 'server-creation');
+            assert.strictEqual(error.cause, null);
+        });
+        it('should create error without cause', () => {
+            const error = new ServerInitError('tool-validation', undefined);
+            assert.strictEqual(error.step, 'tool-validation');
+            assert.strictEqual(error.cause, undefined);
+        });
+        it('should generate default message from step name', () => {
+            const error = new ServerInitError('server-creation', null);
+            assert.ok(error.message.includes('server-creation'));
+            assert.ok(error.message.includes('initialization failed'));
+        });
+    });
+    describe('inheritance', () => {
+        it('should be instanceof Error', () => {
+            const error = new ServerInitError('tool-validation', null);
+            assert.ok(error instanceof Error);
+        });
+        it('should be instanceof ServerInitError', () => {
+            const error = new ServerInitError('tool-validation', null);
+            assert.ok(error instanceof ServerInitError);
+        });
+        it('should have correct name property', () => {
+            const error = new ServerInitError('tool-validation', null);
+            assert.strictEqual(error.name, 'ServerInitError');
+        });
+    });
+    describe('initialization steps', () => {
+        it('should accept tool-validation step', () => {
+            const error = new ServerInitError('tool-validation', null, 'Tool validation failed');
+            assert.strictEqual(error.step, 'tool-validation');
+        });
+        it('should accept server-creation step', () => {
+            const error = new ServerInitError('server-creation', null, 'Server creation failed');
+            assert.strictEqual(error.step, 'server-creation');
+        });
+        it('should accept handler-registration step', () => {
+            const error = new ServerInitError('handler-registration', null, 'Handler registration failed');
+            assert.strictEqual(error.step, 'handler-registration');
+        });
+        it('should accept transport-connection step', () => {
+            const error = new ServerInitError('transport-connection', null, 'Transport connection failed');
+            assert.strictEqual(error.step, 'transport-connection');
+        });
+    });
+    describe('cause handling', () => {
+        it('should preserve Error cause', () => {
+            const originalError = new Error('Original problem');
+            const error = new ServerInitError('handler-registration', originalError);
+            assert.strictEqual(error.cause, originalError);
+            assert.strictEqual(error.cause.message, 'Original problem');
+        });
+        it('should handle string cause', () => {
+            const error = new ServerInitError('handler-registration', 'string cause');
+            assert.strictEqual(error.cause, 'string cause');
+        });
+        it('should handle object cause', () => {
+            const cause = { code: 'ENOENT', path: '/missing/file' };
+            const error = new ServerInitError('handler-registration', cause);
+            assert.deepStrictEqual(error.cause, cause);
+        });
+        it('should handle null cause', () => {
+            const error = new ServerInitError('handler-registration', null);
+            assert.strictEqual(error.cause, null);
+        });
+        it('should handle undefined cause', () => {
+            const error = new ServerInitError('handler-registration', undefined);
+            assert.strictEqual(error.cause, undefined);
+        });
+    });
+    describe('error messaging', () => {
+        it('should include step in default message', () => {
+            const error = new ServerInitError('transport-connection', null);
+            assert.ok(error.message.includes('transport-connection'));
+        });
+        it('should use custom message when provided', () => {
+            const error = new ServerInitError('transport-connection', null, 'Custom: Failed to initialize the widget');
+            assert.strictEqual(error.message, 'Custom: Failed to initialize the widget');
+        });
+        it('should support detailed error messages', () => {
+            const error = new ServerInitError('tool-validation', null, "Tool 'obs_query_traces' missing required 'handler' function");
+            assert.ok(error.message.includes('obs_query_traces'));
+            assert.ok(error.message.includes('handler'));
+        });
+    });
+    describe('error handling patterns', () => {
+        it('should work with try-catch', () => {
+            let caught = null;
+            try {
+                throw new ServerInitError('server-creation', null, 'Test error');
+            }
+            catch (e) {
+                if (e instanceof ServerInitError) {
+                    caught = e;
+                }
+            }
+            assert.ok(caught !== null);
+            assert.strictEqual(caught?.step, 'server-creation');
+        });
+        it('should work with instanceof check', () => {
+            const error = new ServerInitError('tool-validation', null);
+            if (error instanceof ServerInitError) {
+                assert.strictEqual(error.step, 'tool-validation');
+            }
+            else {
+                assert.fail('Should be instanceof ServerInitError');
+            }
+        });
+        it('should serialize to JSON with step and cause', () => {
+            const error = new ServerInitError('transport-connection', { code: 123 }, 'Test message');
+            const json = JSON.stringify({
+                name: error.name,
+                message: error.message,
+                step: error.step,
+                cause: error.cause,
+            });
+            const parsed = JSON.parse(json);
+            assert.strictEqual(parsed.name, 'ServerInitError');
+            assert.strictEqual(parsed.step, 'transport-connection');
+            assert.deepStrictEqual(parsed.cause, { code: 123 });
+        });
+    });
 });
 //# sourceMappingURL=server.test.js.map