@mcp-shark/mcp-shark 1.5.4 → 1.5.6

package/core/repositories/AuditRepository.js

@@ -0,0 +1,233 @@
+/**
+ * Repository for audit logging operations
+ * Handles logging of request and response packets
+ */
+export class AuditRepository {
+  constructor(db) {
+    this.db = db;
+  }
+
+  _getTimestampNs() {
+    return Number(process.hrtime.bigint());
+  }
+
+  _getTimestampISO() {
+    return new Date().toISOString();
+  }
+
+  _calculateDurationMs(startNs, endNs) {
+    return (endNs - startNs) / 1_000_000;
+  }
+
+  _normalizeSessionId(headers) {
+    if (!headers || typeof headers !== 'object') {
+      return null;
+    }
+
+    const sessionHeaderKeys = [
+      'mcp-session-id',
+      'Mcp-Session-Id',
+      'X-MCP-Session-Id',
+      'x-mcp-session-id',
+      'MCP-Session-Id',
+    ];
+
+    for (const key of sessionHeaderKeys) {
+      if (headers[key]) {
+        return headers[key];
+      }
+    }
+
+    return null;
+  }
+
+  _extractJsonRpcMetadata(bodyJson) {
+    if (!bodyJson) {
+      return { id: null, method: null, result: null, error: null };
+    }
+
+    try {
+      const parsed = typeof bodyJson === 'string' ? JSON.parse(bodyJson) : bodyJson;
+      return {
+        id: parsed.id !== undefined ? String(parsed.id) : null,
+        method: parsed.method || null,
+        result: parsed.result ? JSON.stringify(parsed.result) : null,
+        error: parsed.error ? JSON.stringify(parsed.error) : null,
+      };
+    } catch {
+      return { id: null, method: null, result: null, error: null };
+    }
+  }
+
+  _generateInfo(direction, method, url, statusCode, jsonrpcMethod) {
+    if (direction === 'request') {
+      const rpcInfo = jsonrpcMethod ? ` ${jsonrpcMethod}` : '';
+      return `${method} ${url}${rpcInfo}`;
+    }
+
+    const rpcInfo = jsonrpcMethod ? ` ${jsonrpcMethod}` : '';
+    return `${statusCode}${rpcInfo}`;
+  }
+
+  _normalizeBody(body) {
+    if (!body) {
+      return { bodyRaw: '', bodyJson: null };
+    }
+    if (typeof body === 'string') {
+      return { bodyRaw: body, bodyJson: body };
+    }
+    if (typeof body === 'object') {
+      const raw = JSON.stringify(body);
+      return { bodyRaw: raw, bodyJson: raw };
+    }
+    return { bodyRaw: '', bodyJson: null };
+  }
+
+  logRequestPacket(options) {
+    const {
+      method,
+      url,
+      headers = {},
+      body,
+      userAgent = null,
+      remoteAddress = null,
+      sessionId: providedSessionId = null,
+    } = options;
+
+    const timestampNs = this._getTimestampNs();
+    const timestampISO = this._getTimestampISO();
+    const sessionId = providedSessionId || this._normalizeSessionId(headers);
+    const host = headers.host || headers.Host || null;
+
+    const { bodyRaw, bodyJson } = this._normalizeBody(body);
+    const headersJson = JSON.stringify(headers);
+
+    const jsonrpc = this._extractJsonRpcMetadata(bodyJson || bodyRaw);
+    const jsonrpcId = jsonrpc.id;
+    const jsonrpcMethod = jsonrpc.method;
+
+    const length = Buffer.byteLength(headersJson, 'utf8') + Buffer.byteLength(bodyRaw, 'utf8');
+    const info = this._generateInfo('request', method, url, null, jsonrpcMethod);
+
+    const stmt = this.db.prepare(`
+      INSERT INTO packets (
+        timestamp_ns,
+        timestamp_iso,
+        direction,
+        protocol,
+        session_id,
+        method,
+        url,
+        headers_json,
+        body_raw,
+        body_json,
+        jsonrpc_id,
+        jsonrpc_method,
+        length,
+        info,
+        user_agent,
+        remote_address,
+        host
+      ) VALUES (?, ?, 'request', 'HTTP', ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `);
+
+    const result = stmt.run(
+      timestampNs,
+      timestampISO,
+      sessionId,
+      method,
+      url,
+      headersJson,
+      bodyRaw,
+      bodyJson,
+      jsonrpcId,
+      jsonrpcMethod,
+      length,
+      info,
+      userAgent,
+      remoteAddress,
+      host
+    );
+
+    const frameNumber = result.lastInsertRowid;
+
+    return { frameNumber, timestampNs, jsonrpcId, sessionId };
+  }
+
+  logResponsePacket(options) {
+    const {
+      statusCode,
+      headers = {},
+      body,
+      requestFrameNumber: _requestFrameNumber = null,
+      requestTimestampNs: _requestTimestampNs = null,
+      jsonrpcId = null,
+      userAgent = null,
+      remoteAddress = null,
+      sessionId: providedSessionId = null,
+    } = options;
+
+    const timestampNs = this._getTimestampNs();
+    const timestampISO = this._getTimestampISO();
+    const sessionId = providedSessionId || this._normalizeSessionId(headers);
+    const host = headers.host || headers.Host || null;
+
+    const { bodyRaw, bodyJson } = this._normalizeBody(body);
+    const headersJson = JSON.stringify(headers);
+
+    const jsonrpc = this._extractJsonRpcMetadata(bodyJson || bodyRaw);
+    const jsonrpcIdFromBody = jsonrpc.id || jsonrpcId;
+    const jsonrpcMethod = jsonrpc.method;
+    const jsonrpcResult = jsonrpc.result;
+    const jsonrpcError = jsonrpc.error;
+
+    const length = Buffer.byteLength(headersJson, 'utf8') + Buffer.byteLength(bodyRaw, 'utf8');
+    const info = this._generateInfo('response', null, null, statusCode, jsonrpcMethod);
+
+    const stmt = this.db.prepare(`
+      INSERT INTO packets (
+        timestamp_ns,
+        timestamp_iso,
+        direction,
+        protocol,
+        session_id,
+        status_code,
+        headers_json,
+        body_raw,
+        body_json,
+        jsonrpc_id,
+        jsonrpc_method,
+        jsonrpc_result,
+        jsonrpc_error,
+        length,
+        info,
+        user_agent,
+        remote_address,
+        host
+      ) VALUES (?, ?, 'response', 'HTTP', ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `);
+
+    const result = stmt.run(
+      timestampNs,
+      timestampISO,
+      sessionId,
+      statusCode,
+      headersJson,
+      bodyRaw,
+      bodyJson,
+      jsonrpcIdFromBody,
+      jsonrpcMethod,
+      jsonrpcResult,
+      jsonrpcError,
+      length,
+      info,
+      userAgent,
+      remoteAddress,
+      host
+    );
+
+    const frameNumber = result.lastInsertRowid;
+
+    return { frameNumber, timestampNs, jsonrpcId: jsonrpcIdFromBody, sessionId };
+  }
+}

package/core/repositories/ConversationRepository.js

@@ -0,0 +1,182 @@
+import { Defaults } from '#core/constants/Defaults.js';
+
+/**
+ * Repository for conversation-related database operations
+ */
+export class ConversationRepository {
+  constructor(db) {
+    this.db = db;
+  }
+
+  /**
+   * Get conversation flow (request/response pairs)
+   */
+  queryConversations(filters = {}) {
+    const {
+      sessionId = null,
+      method = null,
+      status = null,
+      startTime = null,
+      endTime = null,
+      jsonrpcId = null,
+      limit = Defaults.DEFAULT_LIMIT,
+      offset = Defaults.DEFAULT_OFFSET,
+    } = 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 = this.db.prepare(query);
+    return stmt.all(...params);
+  }
+
+  /**
+   * Create conversation entry for request
+   */
+  createConversation(frameNumber, sessionId, jsonrpcId, method, timestampNs) {
+    const stmt = this.db.prepare(`
+      INSERT INTO conversations (
+        request_frame_number,
+        session_id,
+        jsonrpc_id,
+        method,
+        request_timestamp_ns,
+        status
+      ) VALUES (?, ?, ?, ?, ?, 'pending')
+    `);
+    stmt.run(frameNumber, sessionId, jsonrpcId, method, timestampNs);
+  }
+
+  /**
+   * Update conversation entry with response
+   */
+  updateConversationWithResponse(
+    requestFrameNumber,
+    responseFrameNumber,
+    timestampNs,
+    durationMs,
+    status
+  ) {
+    const stmt = this.db.prepare(`
+      UPDATE conversations
+      SET response_frame_number = ?,
+          response_timestamp_ns = ?,
+          duration_ms = ?,
+          status = ?
+      WHERE request_frame_number = ?
+    `);
+    stmt.run(responseFrameNumber, timestampNs, durationMs, status, requestFrameNumber);
+  }
+
+  /**
+   * Find conversation by JSON-RPC ID
+   */
+  findConversationByJsonRpcId(jsonrpcId) {
+    const stmt = this.db.prepare(`
+      SELECT request_frame_number FROM conversations
+      WHERE jsonrpc_id = ? AND response_frame_number IS NULL
+      ORDER BY request_timestamp_ns DESC
+      LIMIT 1
+    `);
+    return stmt.get(jsonrpcId);
+  }
+
+  /**
+   * Get conversation statistics
+   */
+  getConversationStatistics(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 = this.db.prepare(statsQuery);
+    return stmt.get(...params);
+  }
+}

package/core/repositories/PacketRepository.js

@@ -0,0 +1,237 @@
+import { Defaults } from '#core/constants/Defaults.js';
+
+/**
+ * Repository for packet-related database operations
+ * Encapsulates all database queries for packets
+ */
+export class PacketRepository {
+  constructor(db) {
+    this.db = db;
+  }
+
+  queryPackets(filters = {}) {
+    const {
+      sessionId = null,
+      direction = null,
+      method = null,
+      jsonrpcMethod = null,
+      statusCode = null,
+      startTime = null,
+      endTime = null,
+      jsonrpcId = null,
+      limit = Defaults.DEFAULT_LIMIT,
+      offset = Defaults.DEFAULT_OFFSET,
+    } = 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 = this.db.prepare(query);
+    return stmt.all(...params);
+  }
+
+  queryRequests(filters = {}) {
+    const {
+      sessionId = null,
+      direction = null,
+      method = null,
+      jsonrpcMethod = null,
+      statusCode = null,
+      startTime = null,
+      endTime = null,
+      jsonrpcId = null,
+      search = null,
+      serverName = null,
+      limit = Defaults.DEFAULT_LIMIT,
+      offset = Defaults.DEFAULT_OFFSET,
+    } = filters;
+
+    const queryParts = ['SELECT * FROM packets WHERE 1=1'];
+    const params = [];
+
+    if (search) {
+      const searchPattern = `%${search}%`;
+      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
+        body_json LIKE ? ESCAPE '\\' OR
+        body_raw LIKE ? ESCAPE '\\'
+      )`);
+      const searchParams = Array.from({ length: 11 }, () => searchPattern);
+      params.push(...searchParams);
+      params.push(serverNamePattern);
+      params.push(serverNamePattern);
+    }
+
+    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}%`);
+    }
+
+    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 = this.db.prepare(query);
+    return stmt.all(...params);
+  }
+
+  getByFrameNumber(frameNumber) {
+    const stmt = this.db.prepare('SELECT * FROM packets WHERE frame_number = ?');
+    return stmt.get(frameNumber);
+  }
+
+  getSessionPackets(sessionId, limit = Defaults.DEFAULT_SESSION_LIMIT) {
+    const stmt = this.db.prepare(`
+      SELECT * FROM packets
+      WHERE session_id = ?
+      ORDER BY timestamp_ns ASC
+      LIMIT ?
+    `);
+    return stmt.all(sessionId, limit);
+  }
+
+  getSessionRequests(sessionId, limit = Defaults.DEFAULT_SESSION_LIMIT) {
+    const stmt = this.db.prepare(`
+      SELECT * FROM packets
+      WHERE session_id = ?
+      ORDER BY timestamp_ns DESC
+      LIMIT ?
+    `);
+    return stmt.all(sessionId, limit);
+  }
+
+  clearAll() {
+    this.db.exec('PRAGMA foreign_keys = OFF');
+
+    const tablesResult = this.db
+      .prepare(
+        `
+        SELECT name FROM sqlite_master 
+        WHERE type='table' AND name NOT LIKE 'sqlite_%'
+      `
+      )
+      .all();
+
+    const existingTables = tablesResult.map((row) => row.name);
+    const trafficTables = ['conversations', 'packets', 'sessions'];
+
+    const clearedTables = [];
+    for (const table of trafficTables) {
+      if (existingTables.includes(table)) {
+        try {
+          this.db.exec(`DELETE FROM ${table}`);
+          clearedTables.push(table);
+        } catch (_err) {
+          // Log error but continue
+        }
+      }
+    }
+
+    this.db.exec('PRAGMA foreign_keys = ON');
+
+    return { clearedTables };
+  }
+
+  getMaxTimestamp() {
+    const stmt = this.db.prepare('SELECT MAX(timestamp_ns) as max_ts FROM packets');
+    return stmt.get();
+  }
+}