@mcp-shark/mcp-shark 1.5.3 → 1.5.5

package/lib/common/db/query.js

@@ -1,403 +0,0 @@
-/**
- * Query functions for forensic analysis of HTTP packets
- * Can be used independently in other projects
- */
-
-/**
- * Query packets with forensic filters
- */
-export function queryPackets(db, filters = {}) {
-  const {
-    sessionId = null,
-    direction = null,
-    method = null,
-    jsonrpcMethod = null,
-    statusCode = null,
-    startTime = null,
-    endTime = null,
-    jsonrpcId = null,
-    limit = 1000,
-    offset = 0,
-  } = filters;
-
-  const queryParts = ['SELECT * FROM packets WHERE 1=1'];
-  const params = [];
-
-  if (sessionId) {
-    queryParts.push('AND session_id = ?');
-    params.push(sessionId);
-  }
-
-  if (direction) {
-    queryParts.push('AND direction = ?');
-    params.push(direction);
-  }
-
-  if (method) {
-    queryParts.push('AND method = ?');
-    params.push(method);
-  }
-
-  if (jsonrpcMethod) {
-    queryParts.push('AND jsonrpc_method = ?');
-    params.push(jsonrpcMethod);
-  }
-
-  if (statusCode !== null && statusCode !== undefined) {
-    queryParts.push('AND status_code = ?');
-    params.push(statusCode);
-  }
-
-  if (startTime) {
-    queryParts.push('AND timestamp_ns >= ?');
-    params.push(startTime);
-  }
-
-  if (endTime) {
-    queryParts.push('AND timestamp_ns <= ?');
-    params.push(endTime);
-  }
-
-  if (jsonrpcId) {
-    queryParts.push('AND jsonrpc_id = ?');
-    params.push(jsonrpcId);
-  }
-
-  queryParts.push('ORDER BY timestamp_ns ASC LIMIT ? OFFSET ?');
-  params.push(limit, offset);
-
-  const query = queryParts.join(' ');
-  const stmt = db.prepare(query);
-  return stmt.all(...params);
-}
-
-/**
- * Enhanced query function for requests/responses with search and server name filtering
- * Supports partial matching and general search across multiple fields
- */
-export function queryRequests(db, filters = {}) {
-  const {
-    sessionId = null,
-    direction = null,
-    method = null,
-    jsonrpcMethod = null,
-    statusCode = null,
-    startTime = null,
-    endTime = null,
-    jsonrpcId = null,
-    search = null, // General search across multiple fields
-    serverName = null, // MCP server name filter
-    limit = 1000,
-    offset = 0,
-  } = filters;
-
-  const queryParts = ['SELECT * FROM packets WHERE 1=1'];
-  const params = [];
-
-  // General search - searches across multiple fields with partial matching
-  // Also searches for server names in JSON-RPC params (e.g., "params":{"name":"server-name.tool")
-  if (search) {
-    const searchPattern = `%${search}%`;
-    // Also create a pattern to search for server name in params (e.g., "name":"server-name")
-    const serverNamePattern = `%"name":"${search}%`;
-    queryParts.push(`AND (
-      session_id LIKE ? ESCAPE '\\' OR
-      method LIKE ? ESCAPE '\\' OR
-      url LIKE ? ESCAPE '\\' OR
-      jsonrpc_method LIKE ? ESCAPE '\\' OR
-      jsonrpc_id LIKE ? ESCAPE '\\' OR
-      info LIKE ? ESCAPE '\\' OR
-      body_raw LIKE ? ESCAPE '\\' OR
-      body_json LIKE ? ESCAPE '\\' OR
-      headers_json LIKE ? ESCAPE '\\' OR
-      host LIKE ? ESCAPE '\\' OR
-      remote_address LIKE ? ESCAPE '\\' OR
-      -- Search for server name in JSON-RPC params (e.g., "params":{"name":"server-name.tool")
-      body_json LIKE ? ESCAPE '\\' OR
-      body_raw LIKE ? ESCAPE '\\'
-    )`);
-    // Add the pattern for each field (11 regular fields + 2 server name fields = 13 total)
-    for (let i = 0; i < 11; i++) {
-      params.push(searchPattern);
-    }
-    // Add server name specific patterns
-    params.push(serverNamePattern);
-    params.push(serverNamePattern);
-  }
-
-  // Specific field filters with partial matching
-  if (sessionId) {
-    queryParts.push("AND session_id LIKE ? ESCAPE '\\'");
-    params.push(`%${sessionId}%`);
-  }
-  if (direction) {
-    queryParts.push('AND direction = ?');
-    params.push(direction);
-  }
-  if (method) {
-    queryParts.push("AND method LIKE ? ESCAPE '\\'");
-    params.push(`%${method}%`);
-  }
-  if (jsonrpcMethod) {
-    queryParts.push("AND jsonrpc_method LIKE ? ESCAPE '\\'");
-    params.push(`%${jsonrpcMethod}%`);
-  }
-  if (statusCode !== null && statusCode !== undefined) {
-    queryParts.push('AND status_code = ?');
-    params.push(statusCode);
-  }
-  if (startTime) {
-    queryParts.push('AND timestamp_ns >= ?');
-    params.push(startTime);
-  }
-  if (endTime) {
-    queryParts.push('AND timestamp_ns <= ?');
-    params.push(endTime);
-  }
-  if (jsonrpcId) {
-    queryParts.push("AND jsonrpc_id LIKE ? ESCAPE '\\'");
-    params.push(`%${jsonrpcId}%`);
-  }
-
-  // Filter by MCP server name - search in JSON-RPC params
-  // Server names appear as "params":{"name":"server-name.tool-name" or "name":"server-name.tool-name"
-  if (serverName) {
-    const serverPattern = `%"name":"${serverName}.%`;
-    const serverPattern2 = `%"name":"${serverName}"%`;
-    queryParts.push(`AND (
-      body_json LIKE ? ESCAPE '\\' OR
-      body_raw LIKE ? ESCAPE '\\' OR
-      body_json LIKE ? ESCAPE '\\' OR
-      body_raw LIKE ? ESCAPE '\\'
-    )`);
-    params.push(serverPattern);
-    params.push(serverPattern);
-    params.push(serverPattern2);
-    params.push(serverPattern2);
-  }
-
-  queryParts.push('ORDER BY timestamp_ns DESC LIMIT ? OFFSET ?');
-  params.push(limit, offset);
-
-  const query = queryParts.join(' ');
-  const stmt = db.prepare(query);
-  return stmt.all(...params);
-}
-
-/**
- * Get conversation flow (request/response pairs)
- */
-export function queryConversations(db, filters = {}) {
-  const {
-    sessionId = null,
-    method = null,
-    status = null,
-    startTime = null,
-    endTime = null,
-    jsonrpcId = null,
-    limit = 1000,
-    offset = 0,
-  } = filters;
-
-  const queryParts = [
-    'SELECT',
-    '  c.*,',
-    '  req.frame_number as req_frame,',
-    '  req.timestamp_iso as req_timestamp_iso,',
-    '  req.method as req_method,',
-    '  req.url as req_url,',
-    '  req.jsonrpc_method as req_jsonrpc_method,',
-    '  req.body_json as req_body_json,',
-    '  req.headers_json as req_headers_json,',
-    '  resp.frame_number as resp_frame,',
-    '  resp.timestamp_iso as resp_timestamp_iso,',
-    '  resp.status_code as resp_status_code,',
-    '  resp.jsonrpc_method as resp_jsonrpc_method,',
-    '  resp.body_json as resp_body_json,',
-    '  resp.headers_json as resp_headers_json,',
-    '  resp.jsonrpc_result as resp_jsonrpc_result,',
-    '  resp.jsonrpc_error as resp_jsonrpc_error',
-    'FROM conversations c',
-    'LEFT JOIN packets req ON c.request_frame_number = req.frame_number',
-    'LEFT JOIN packets resp ON c.response_frame_number = resp.frame_number',
-    'WHERE 1=1',
-  ];
-
-  const params = [];
-
-  if (sessionId) {
-    queryParts.push('AND c.session_id = ?');
-    params.push(sessionId);
-  }
-
-  if (method) {
-    queryParts.push('AND c.method = ?');
-    params.push(method);
-  }
-
-  if (status) {
-    queryParts.push('AND c.status = ?');
-    params.push(status);
-  }
-
-  if (startTime) {
-    queryParts.push('AND c.request_timestamp_ns >= ?');
-    params.push(startTime);
-  }
-
-  if (endTime) {
-    queryParts.push('AND c.request_timestamp_ns <= ?');
-    params.push(endTime);
-  }
-
-  if (jsonrpcId) {
-    queryParts.push('AND c.jsonrpc_id = ?');
-    params.push(jsonrpcId);
-  }
-
-  queryParts.push('ORDER BY c.request_timestamp_ns ASC LIMIT ? OFFSET ?');
-  params.push(limit, offset);
-
-  const query = queryParts.join(' ');
-  const stmt = db.prepare(query);
-  return stmt.all(...params);
-}
-
-/**
- * Get all packets for a specific session (for forensic analysis)
- */
-export function getSessionPackets(db, sessionId, limit = 10000) {
-  const stmt = db.prepare(`
-    SELECT * FROM packets
-    WHERE session_id = ?
-    ORDER BY timestamp_ns ASC
-    LIMIT ?
-  `);
-  return stmt.all(sessionId, limit);
-}
-
-/**
- * Get all requests for a specific session (ordered by most recent first)
- */
-export function getSessionRequests(db, sessionId, limit = 10000) {
-  const stmt = db.prepare(`
-    SELECT * FROM packets
-    WHERE session_id = ?
-    ORDER BY timestamp_ns DESC
-    LIMIT ?
-  `);
-  return stmt.all(sessionId, limit);
-}
-
-/**
- * Get session metadata
- */
-export function getSessions(db, filters = {}) {
-  const { startTime = null, endTime = null, limit = 1000, offset = 0 } = filters;
-
-  const queryParts = ['SELECT * FROM sessions WHERE 1=1'];
-  const params = [];
-
-  if (startTime) {
-    queryParts.push('AND first_seen_ns >= ?');
-    params.push(startTime);
-  }
-
-  if (endTime) {
-    queryParts.push('AND last_seen_ns <= ?');
-    params.push(endTime);
-  }
-
-  queryParts.push('ORDER BY first_seen_ns DESC LIMIT ? OFFSET ?');
-  params.push(limit, offset);
-
-  const query = queryParts.join(' ');
-  const stmt = db.prepare(query);
-  return stmt.all(...params);
-}
-
-/**
- * Get statistics for forensic analysis
- */
-export function getStatistics(db, filters = {}) {
-  const { sessionId = null, startTime = null, endTime = null } = filters;
-
-  const whereParts = ['WHERE 1=1'];
-  const params = [];
-
-  if (sessionId) {
-    whereParts.push('AND session_id = ?');
-    params.push(sessionId);
-  }
-
-  if (startTime) {
-    whereParts.push('AND timestamp_ns >= ?');
-    params.push(startTime);
-  }
-
-  if (endTime) {
-    whereParts.push('AND timestamp_ns <= ?');
-    params.push(endTime);
-  }
-
-  const whereClause = whereParts.join(' ');
-  const statsQuery = `
-    SELECT 
-      COUNT(*) as total_packets,
-      COUNT(CASE WHEN direction = 'request' THEN 1 END) as total_requests,
-      COUNT(CASE WHEN direction = 'response' THEN 1 END) as total_responses,
-      COUNT(CASE WHEN status_code >= 400 THEN 1 END) as total_errors,
-      COUNT(DISTINCT session_id) as unique_sessions,
-      AVG(length) as avg_packet_size,
-      SUM(length) as total_bytes,
-      MIN(timestamp_ns) as first_packet_ns,
-      MAX(timestamp_ns) as last_packet_ns
-    FROM packets
-    ${whereClause}
-  `;
-
-  const stmt = db.prepare(statsQuery);
-  return stmt.get(...params);
-}
-
-/**
- * Get conversation statistics
- */
-export function getConversationStatistics(db, filters = {}) {
-  const { sessionId = null, startTime = null, endTime = null } = filters;
-
-  const whereParts = ['WHERE 1=1'];
-  const params = [];
-
-  if (sessionId) {
-    whereParts.push('AND session_id = ?');
-    params.push(sessionId);
-  }
-
-  if (startTime) {
-    whereParts.push('AND request_timestamp_ns >= ?');
-    params.push(startTime);
-  }
-
-  if (endTime) {
-    whereParts.push('AND request_timestamp_ns <= ?');
-    params.push(endTime);
-  }
-
-  const whereClause = whereParts.join(' ');
-  const statsQuery = `
-    SELECT 
-      COUNT(*) as total_conversations,
-      COUNT(CASE WHEN status = 'completed' THEN 1 END) as completed,
-      COUNT(CASE WHEN status = 'pending' THEN 1 END) as pending,
-      COUNT(CASE WHEN status = 'error' THEN 1 END) as errors,
-      AVG(duration_ms) as avg_duration_ms,
-      MIN(duration_ms) as min_duration_ms,
-      MAX(duration_ms) as max_duration_ms
-    FROM conversations
-    ${whereClause}
-  `;
-
-  const stmt = db.prepare(statsQuery);
-  return stmt.get(...params);
-}

package/lib/common/logger.js

@@ -1,90 +0,0 @@
-import { consola } from 'consola';
-
-/**
- * Unified logger for the entire codebase
- * Uses consola (already installed) with a consistent API
- * Supports both Pino-style API (object, message) and consola-style (multiple args)
- */
-const logger = {
-  info: (...args) => {
-    if (args.length === 0) {
-      return;
-    }
-    // If first arg is an object and second is a string, use Pino-style
-    if (
-      args.length === 2 &&
-      typeof args[0] === 'object' &&
-      args[0] !== null &&
-      typeof args[1] === 'string'
-    ) {
-      consola.info(args[1] || '', args[0]);
-    } else {
-      // Otherwise, pass all args to consola (supports multiple strings/values)
-      consola.info(...args);
-    }
-  },
-  error: (...args) => {
-    if (args.length === 0) {
-      return;
-    }
-    if (
-      args.length === 2 &&
-      typeof args[0] === 'object' &&
-      args[0] !== null &&
-      typeof args[1] === 'string'
-    ) {
-      consola.error(args[1] || '', args[0]);
-    } else {
-      consola.error(...args);
-    }
-  },
-  warn: (...args) => {
-    if (args.length === 0) {
-      return;
-    }
-    if (
-      args.length === 2 &&
-      typeof args[0] === 'object' &&
-      args[0] !== null &&
-      typeof args[1] === 'string'
-    ) {
-      consola.warn(args[1] || '', args[0]);
-    } else {
-      consola.warn(...args);
-    }
-  },
-  debug: (...args) => {
-    if (args.length === 0) {
-      return;
-    }
-    if (
-      args.length === 2 &&
-      typeof args[0] === 'object' &&
-      args[0] !== null &&
-      typeof args[1] === 'string'
-    ) {
-      consola.debug(args[1] || '', args[0]);
-    } else {
-      consola.debug(...args);
-    }
-  },
-  log: (...args) => {
-    if (args.length === 0) {
-      return;
-    }
-    if (
-      args.length === 2 &&
-      typeof args[0] === 'object' &&
-      args[0] !== null &&
-      typeof args[1] === 'string'
-    ) {
-      consola.log(args[1] || '', args[0]);
-    } else {
-      consola.log(...args);
-    }
-  },
-  // Expose consola directly for advanced usage
-  consola,
-};
-
-export default logger;

package/mcp-server/index.js

@@ -1,111 +0,0 @@
-import { createServer } from 'node:http';
-
-import serverLogger from '#common/logger';
-import { isError } from './lib/common/error.js';
-import { runAllExternalServers } from './lib/server/external/all.js';
-
-import { getDatabaseFile, getMcpConfigPath, prepareAppDataSpaces } from '#common/configs';
-import { initDb } from '#common/db/init';
-import { getLogger } from '#common/db/logger';
-import { withAuditRequestResponseHandler } from './lib/auditor/audit.js';
-import { getInternalServer } from './lib/server/internal/run.js';
-import { createInternalServerFactory } from './lib/server/internal/server.js';
-
-function initAuditLogger(serverLogger) {
-  serverLogger.info({ path: getDatabaseFile() }, 'Initializing audit logger at');
-  return getLogger(initDb(getDatabaseFile()));
-}
-
-/**
- * Start the MCP Shark server
- * @param {Object} options - Configuration options
- * @param {string} [options.configPath] - Path to MCP config file (defaults to getMcpConfigPath())
- * @param {number} [options.port=9851] - Port to listen on
- * @param {Function} [options.onError] - Error callback
- * @param {Function} [options.onReady] - Ready callback
- * @returns {Promise<{app: Express, server: http.Server, stop: Function}>} Server instance
- */
-export async function startMcpSharkServer(options = {}) {
-  const { configPath = getMcpConfigPath(), port = 9851, onError, onReady } = options;
-
-  prepareAppDataSpaces();
-
-  serverLogger.info('[MCP-Shark] Starting MCP server...');
-  serverLogger.info(`[MCP-Shark] Config path: ${configPath}`);
-  serverLogger.info(`[MCP-Shark] Database path: ${getDatabaseFile()}`);
-  serverLogger.info(`[MCP-Shark] Working directory: ${process.cwd()}`);
-  serverLogger.info(`[MCP-Shark] PATH: ${process.env.PATH}`);
-
-  try {
-    const auditLogger = initAuditLogger(serverLogger);
-    const externalServersResult = await runAllExternalServers(serverLogger, configPath);
-
-    if (isError(externalServersResult)) {
-      const error = new Error(JSON.stringify(externalServersResult));
-      if (onError) onError(error);
-      throw error;
-    }
-
-    const { kv, servers: externalServers } = externalServersResult;
-
-    const internalServerFactory = createInternalServerFactory(serverLogger, kv);
-    const app = getInternalServer(
-      internalServerFactory,
-      auditLogger,
-      withAuditRequestResponseHandler
-    );
-
-    const httpServer = createServer(app);
-
-    const server = await new Promise((resolve, reject) => {
-      httpServer.on('error', (err) => {
-        if (onError) onError(err);
-        reject(err);
-      });
-
-      httpServer.listen(port, '0.0.0.0', () => {
-        serverLogger.info(`MCP proxy HTTP server listening on http://localhost:${port}/mcp`);
-        if (onReady) onReady();
-        resolve(httpServer);
-      });
-    });
-
-    const stop = async () => {
-      // Close all external server clients
-      for (const serverInfo of externalServers) {
-        if (serverInfo?.client && !isError(serverInfo)) {
-          try {
-            await serverInfo.client.close();
-          } catch (err) {
-            serverLogger.warn({ error: err.message }, 'Error closing external server client');
-          }
-        }
-      }
-
-      // Close all connections forcefully
-      if (server.closeAllConnections) {
-        server.closeAllConnections();
-      }
-
-      // Close the server
-      return new Promise((resolve) => {
-        server.close(() => {
-          serverLogger.info('MCP Shark server stopped');
-          resolve();
-        });
-      });
-    };
-
-    return {
-      app,
-      server,
-      stop,
-      auditLogger,
-    };
-  } catch (error) {
-    const errorMsg = 'Error starting MCP server';
-    serverLogger.error({ error: error.message, stack: error.stack }, `[MCP-Shark] ${errorMsg}`);
-    if (onError) onError(error);
-    throw error;
-  }
-}

package/mcp-server/lib/server/external/config.js

@@ -1,57 +0,0 @@
-import { readFileSync } from 'node:fs';
-import { CompositeError, isError } from '../../common/error.js';
-
-const DEFAULT_TYPE = 'stdio';
-export class ConfigError extends CompositeError {
-  constructor(message, error) {
-    super('ConfigError', message, error);
-  }
-}
-
-function parseConfig(configPath) {
-  try {
-    const conf = JSON.parse(readFileSync(configPath, 'utf-8'));
-    if (conf && typeof conf === 'object') {
-      return conf;
-    }
-    return new ConfigError(
-      'Invalid config file',
-      new Error(`Invalid config file: ${configPath} - ${JSON.stringify(conf, null, 2)}`)
-    );
-  } catch (error) {
-    return new ConfigError(
-      'Error parsing config',
-      new Error(`Error parsing config: ${configPath} - ${error.message}`)
-    );
-  }
-}
-
-export function normalizeConfig(configPath) {
-  const parsedConfigResult = parseConfig(configPath);
-  if (isError(parsedConfigResult)) {
-    return parsedConfigResult;
-  }
-  const out = new Map();
-  const { servers, mcpServers } = parsedConfigResult;
-  // Servers are the old format
-  if (servers) {
-    Object.entries(servers).forEach(([name, cfg]) => {
-      const type = cfg.type ?? DEFAULT_TYPE;
-      out.set(name, { type, ...cfg });
-    });
-  }
-  // MCP Servers are the new format
-  if (mcpServers) {
-    Object.entries(mcpServers).forEach(([name, cfg]) => {
-      // Cursor/Claude usually omit type; assume stdio if command is given
-      const type = cfg.type ?? DEFAULT_TYPE;
-      out.set(name, { type, ...cfg });
-    });
-  }
-
-  if (out.size === 0) {
-    return new ConfigError('No servers found in config');
-  }
-
-  return Object.fromEntries(out);
-}

package/mcp-server/lib/server/internal/handlers/prompts-get.js

@@ -1,20 +0,0 @@
-import { getBy } from '../../external/kv.js';
-import { InternalServerError } from './error.js';
-
-export function createPromptsGetHandler(logger, mcpServers, requestedMcpServer) {
-  return async (req) => {
-    const name = req.params.name;
-    const promptArgs = req?.params?.arguments || {};
-    logger.debug('Prompt get', name, promptArgs);
-
-    const getPrompt = getBy(mcpServers, requestedMcpServer, name, 'getPrompt', promptArgs);
-    if (!getPrompt) {
-      throw new InternalServerError(`Prompt not found: ${name}`);
-    }
-
-    const result = await getPrompt(name, promptArgs);
-    logger.debug('Prompt get result', result);
-
-    return result;
-  };
-}

package/mcp-server/lib/server/internal/handlers/prompts-list.js

@@ -1,13 +0,0 @@
-import { listAll } from '../../external/kv.js';
-
-export function createPromptsListHandler(logger, mcpServers, requestedMcpServer) {
-  return async (req) => {
-    const path = req.path;
-    logger.debug('Prompts list', path);
-
-    const res = await listAll(mcpServers, requestedMcpServer, 'prompts');
-    const result = Array.isArray(res) ? { prompts: res } : res;
-
-    return result;
-  };
-}

package/mcp-server/lib/server/internal/handlers/resources-list.js

@@ -1,13 +0,0 @@
-import { listAll } from '../../external/kv.js';
-
-export function createResourcesListHandler(logger, mcpServers, requestedMcpServer) {
-  return async (req) => {
-    const path = req.path;
-    logger.debug('Resources list', path);
-
-    const res = await listAll(mcpServers, requestedMcpServer, 'resources');
-    const result = Array.isArray(res) ? { resources: res } : res;
-
-    return result;
-  };
-}