mcp-think-tank/dist/core/connection.js

Structured thinking and knowledge management tool for Model Context Protocol

import { serverState, resetInactivityTimer } from './index.js';
/**
 * Start monitoring connections to detect inactivity
 *
 * @param isToolScanMode Whether the server is running in tool scan mode
 */
export function startConnectionCheck(isToolScanMode) {
    // Remove all connection checking logic since it's causing timeouts
    console.error('[INFO] [connection] Connection checking disabled to prevent timeout issues');
    // Just ensure the server is considered active
    resetInactivityTimer();
    // For tool scan mode, provide additional information to improve scanning
    if (isToolScanMode) {
        console.error('[INFO] [connection] Running in tool scan mode with enhanced Smithery 2025 compatibility');
    }
}
/**
 * Set up connection tracking for HTTP server
 *
 * @param httpServer HTTP server instance
 * @returns The HTTP server with connection tracking
 */
export function setupConnectionTracking(httpServer) {
    console.error('[INFO] [connection] Setting up HTTP connection tracking');
    // Track when a new connection is established
    httpServer.on('connection', (socket) => {
        serverState.connectionCount++;
        // Set up event listener for when the connection closes
        socket.on('close', () => {
            serverState.connectionCount = Math.max(0, serverState.connectionCount - 1);
        });
    });
    // Track when a new request comes in
    httpServer.on('request', (req, res) => {
        // Determine if this is a tool list request (typical for Smithery scanning)
        const isToolListRequest = req.url?.includes('/rpc.listTools') ||
            req.url?.includes('list_tools') ||
            req.url?.includes('toolList');
        // For tool scanning, set appropriate headers to improve Smithery compatibility
        if (isToolListRequest) {
            // Set headers to indicate scanning capabilities to Smithery
            res.setHeader('X-Smithery-Scan-Support', 'enhanced');
            res.setHeader('X-Smithery-Protocol', 'mcp-2025');
        }
        // Consider the server active as long as we're receiving requests
        resetInactivityTimer();
        // Track response completion to update connection status if needed
        res.on('finish', () => {
            // No logging needed
        });
    });
    return httpServer;
}
/**
 * Get HTTP server options based on tool scan mode
 *
 * @param isToolScanMode Whether the server is running in tool scan mode
 * @param toolScanTimeout Timeout for tool scanning in milliseconds
 * @returns Server options object
 */
export function getServerOptions(isToolScanMode, toolScanTimeout) {
    const options = isToolScanMode
        ? {
            keepAliveTimeout: toolScanTimeout,
            toolScanTimeout: toolScanTimeout,
            headersTimeout: toolScanTimeout + 5000, // Add buffer for headers
            timeout: toolScanTimeout + 10000 // General timeout with extra buffer
        }
        : {
            keepAliveTimeout: 3600000 // 1 hour for normal mode
        };
    return options;
}