@cardor/heimdall-mcp 0.1.0

package/README.md

@@ -0,0 +1,391 @@
+# heimdall-mcp
+
+Transparent proxy for any MCP server. Intercepts all JSON-RPC messages, measures latency, and stores traces in a configurable database — without touching the original server.
+
+## Table of Contents
+
+- [How it works](#how-it-works)
+- [Installation](#installation)
+- [Usage modes](#usage-modes)
+  - [Mode 1 — CLI wrapping a subprocess (stdio)](#mode-1--cli-wrapping-a-subprocess-stdio)
+  - [Mode 2 — CLI wrapping a remote HTTP server](#mode-2--cli-wrapping-a-remote-http-server)
+  - [Mode 3 — CLI wrapping a remote SSE server](#mode-3--cli-wrapping-a-remote-sse-server)
+  - [Mode 4 — Library for developers](#mode-4--library-for-developers)
+- [Stores](#stores)
+  - [SQLite](#sqlite)
+  - [PostgreSQL](#postgresql)
+  - [MySQL](#mysql)
+- [What gets recorded](#what-gets-recorded)
+- [Custom interceptors](#custom-interceptors)
+- [CLI reference](#cli-reference)
+
+---
+
+## How it works
+
+```mermaid
+flowchart LR
+    A["MCP Client\n(Claude Desktop / OpenCode / Cursor)"]
+
+    subgraph proxy["heimdall-mcp"]
+        B["TelemetryInterceptor"]
+        C["ForwardInterceptor"]
+        D[("SQLite\nPostgres\nMySQL")]
+        B --> C
+        B -->|"saves span"| D
+    end
+
+    S["Real MCP server\n(subprocess / HTTP / SSE)"]
+
+    A -->|"stdio"| B
+    C -->|"stdio · http · sse"| S
+    S -->|"response"| C
+    C -->|"response"| A
+```
+
+The proxy always exposes **stdio** to the MCP client and speaks the correct transport to the real server. Every request/response pair is converted into a span with timing, attributes, and the input/output body.
+
+---
+
+## Installation
+
+```bash
+npm install -g @cardor/heimdall-mcp
+# or as a project dependency
+npm install @cardor/heimdall-mcp
+```
+
+---
+
+## Usage modes
+
+### Mode 1 — CLI wrapping a subprocess (stdio)
+
+The MCP client thinks it is talking to `heimdall-mcp`. The proxy spawns the real server as a child process and forwards all messages.
+
+**`mcp.json` / Claude Desktop configuration:**
+
+```json
+{
+  "mcpServers": {
+    "my-server": {
+      "command": "heimdall-mcp",
+      "args": [
+        "--store", "sqlite://~/.mcp-traces/traces.db",
+        "--", "node", "my-server.js"
+      ]
+    }
+  }
+}
+```
+
+The `--` separator divides heimdall-mcp flags from the real server command. Everything after it is executed as a subprocess.
+
+**With a globally installed server:**
+
+```json
+{
+  "mcpServers": {
+    "filesystem": {
+      "command": "heimdall-mcp",
+      "args": [
+        "--store", "sqlite://~/.mcp-traces/traces.db",
+        "--", "npx", "@modelcontextprotocol/server-filesystem", "/tmp"
+      ]
+    }
+  }
+}
+```
+
+**With Postgres instead of SQLite:**
+
+```json
+{
+  "mcpServers": {
+    "my-server": {
+      "command": "heimdall-mcp",
+      "args": [
+        "--store", "postgres://user:pass@localhost:5432/traces",
+        "--", "node", "my-server.js"
+      ]
+    }
+  }
+}
+```
+
+---
+
+### Mode 2 — CLI wrapping a remote HTTP server
+
+When the MCP server is already running and exposes an HTTP endpoint.
+
+```json
+{
+  "mcpServers": {
+    "remote-server": {
+      "command": "heimdall-mcp",
+      "args": [
+        "--store",  "sqlite://~/.mcp-traces/traces.db",
+        "--out",    "http",
+        "--target", "http://localhost:3001"
+      ]
+    }
+  }
+}
+```
+
+The proxy exposes **stdio** to the client and forwards each message as an HTTP `POST` to the target URL.
+
+---
+
+### Mode 3 — CLI wrapping a remote SSE server
+
+For servers that use Server-Sent Events.
+
+```json
+{
+  "mcpServers": {
+    "sse-server": {
+      "command": "heimdall-mcp",
+      "args": [
+        "--store",  "postgres://user:pass@host/db",
+        "--out",    "sse",
+        "--target", "http://remote.example.com"
+      ]
+    }
+  }
+}
+```
+
+The proxy connects to `{target}/sse` to receive responses and sends requests as `POST` to `{target}`.
+
+---
+
+### Mode 4 — Library for developers
+
+When you have access to the source code and want to integrate the proxy programmatically.
+
+**Minimal setup:**
+
+```ts
+import { ProxyBuilder } from '@cardor/heimdall-mcp'
+
+const proxy = await ProxyBuilder.create()
+  .inbound({ transport: 'stdio' })
+  .outbound({ transport: 'stdio', command: 'node', args: ['my-server.js'] })
+  .store('sqlite://./traces.db')
+  .build()
+
+await proxy.start()
+
+// clean shutdown
+process.on('SIGINT', () => proxy.stop())
+```
+
+**stdio → remote HTTP:**
+
+```ts
+const proxy = await ProxyBuilder.create()
+  .inbound({ transport: 'stdio' })
+  .outbound({ transport: 'http', url: 'http://localhost:3001' })
+  .store('postgres://user:pass@localhost/traces')
+  .build()
+
+await proxy.start()
+```
+
+**HTTP inbound (proxy listens on a port):**
+
+```ts
+const proxy = await ProxyBuilder.create()
+  .inbound({ transport: 'http', port: 8080 })
+  .outbound({ transport: 'stdio', command: 'node', args: ['server.js'] })
+  .store('mysql://user:pass@localhost/traces')
+  .build()
+
+await proxy.start()
+```
+
+**With a custom interceptor:**
+
+```ts
+import type { Interceptor, InterceptorContext, JsonRpcMessage } from '@cardor/heimdall-mcp'
+
+class LogAllInterceptor implements Interceptor {
+  name = 'LogAllInterceptor'
+
+  async intercept(
+    request: JsonRpcMessage,
+    context: InterceptorContext,
+    next: () => Promise<JsonRpcMessage>
+  ): Promise<JsonRpcMessage> {
+    console.log('→', request.method, request.id)
+    const response = await next()
+    console.log('←', response.id, response.error ? 'ERROR' : 'OK')
+    return response
+  }
+}
+
+const proxy = await ProxyBuilder.create()
+  .inbound({ transport: 'stdio' })
+  .outbound({ transport: 'stdio', command: 'node', args: ['server.js'] })
+  .store('sqlite://./traces.db')
+  .build()
+
+proxy.addInterceptor(new LogAllInterceptor())
+await proxy.start()
+```
+
+---
+
+## Stores
+
+### SQLite
+
+No external server required — ideal for local development.
+
+**Valid connection strings:**
+
+```
+sqlite://./traces.db
+sqlite://~/.mcp-traces/traces.db
+sqlite:///absolute/path/traces.db
+```
+
+Driver: [`@libsql/client`](https://github.com/tursodatabase/libsql-client-ts) — pure WASM, no native compilation required.
+
+**Schema:**
+
+```
+mcp_spans
+  id           TEXT  PRIMARY KEY   → "{trace_id}-{span_id}"
+  trace_id     TEXT  NOT NULL
+  span_id      TEXT  NOT NULL
+  parent_id    TEXT
+  name         TEXT  NOT NULL      → "mcp.tool.call", "mcp.initialize", etc.
+  status       TEXT  NOT NULL      → "ok" | "error"
+  started_at   TEXT  NOT NULL      → ISO 8601
+  ended_at     TEXT  NOT NULL
+  duration_ms  INT   NOT NULL
+  attributes   JSON               → tool_name, client_version, etc.
+  events       JSON               → input/output bodies
+
+mcp_metrics
+  id           INT   PRIMARY KEY
+  tool_name    TEXT  NOT NULL
+  call_count   INT
+  error_count  INT
+  avg_duration REAL
+  updated_at   TEXT
+```
+
+---
+
+### PostgreSQL
+
+```
+postgres://user:pass@localhost:5432/my_db
+postgresql://user:pass@localhost:5432/my_db
+```
+
+Driver: [`postgres`](https://github.com/porsager/postgres) — pure JS, no node-gyp.
+
+Same schema as SQLite but with native Postgres types (`TIMESTAMP`, `INTEGER`, `JSON`).
+
+---
+
+### MySQL
+
+```
+mysql://user:pass@localhost:3306/my_db
+```
+
+Driver: [`mysql2`](https://github.com/sidorares/node-mysql2).
+
+Same schema adapted to MySQL types (`FLOAT`, `INT`, `JSON`, `TIMESTAMP`).
+
+---
+
+## What gets recorded
+
+Every JSON-RPC message produces a span in the `mcp_spans` table. Attributes vary by method:
+
+| MCP method       | Span name             | Key attributes                                              | Events                        |
+|------------------|-----------------------|-------------------------------------------------------------|-------------------------------|
+| `initialize`     | `mcp.initialize`      | `mcp.client_version`, `mcp.server_version`, capabilities   | —                             |
+| `tools/list`     | `mcp.tools.list`      | `mcp.tools_count`, `mcp.duration_ms`                       | tools list as JSON            |
+| `tools/call`     | `mcp.tool.call`       | `gen_ai.tool.name`, `gen_ai.tool.call.id`, duration        | `tool.input` + `tool.output`  |
+| `resources/read` | `mcp.resource.read`   | `url.full`, `mcp.duration_ms`                              | —                             |
+| `resources/list` | `mcp.resources.list`  | `mcp.duration_ms`                                          | —                             |
+| `prompts/get`    | `mcp.prompt.get`      | `mcp.prompt_name`, `mcp.duration_ms`                       | rendered prompt body          |
+| `prompts/list`   | `mcp.prompts.list`    | `mcp.duration_ms`                                          | —                             |
+| `shutdown`       | `mcp.shutdown`        | `mcp.duration_ms`                                          | —                             |
+| any other        | `mcp.{method}`        | `gen_ai.operation.name`, `mcp.duration_ms`                 | —                             |
+
+Attributes use the `gen_ai.*` prefix following the [OpenTelemetry semantic conventions](https://opentelemetry.io/docs/specs/semconv/gen-ai/) for generative AI.
+
+---
+
+## Custom interceptors
+
+The `Interceptor` interface is public. You can add your own logic into the pipeline before the telemetry interceptor:
+
+```ts
+interface Interceptor {
+  name: string
+  intercept(
+    request: JsonRpcMessage,
+    context: InterceptorContext,
+    next: () => Promise<JsonRpcMessage>
+  ): Promise<JsonRpcMessage>
+}
+
+interface InterceptorContext {
+  startedAt: Date
+  traceId: string
+  spanId: string
+  metadata: Record<string, unknown>
+}
+```
+
+Calling `next()` passes control to the next interceptor in the chain. `ForwardInterceptor` is always last — it makes the actual call to the real server.
+
+---
+
+## CLI reference
+
+```
+heimdall-mcp [options] [-- command [args...]]
+
+Options:
+  --store <url>         Store connection string (required)
+                          sqlite://./traces.db
+                          postgres://user:pass@host/db
+                          mysql://user:pass@host/db
+
+  --out <transport>     Transport to the real server (default: stdio)
+                          stdio | http | sse
+
+  --target <url>        Server URL when --out is http or sse
+
+  --in <transport>      Inbound transport (default: stdio)
+                          stdio | http | sse
+
+  --in-port <port>      Port for --in http or --in sse
+
+  -V, --version         Print version
+  -h, --help            Print this help
+
+  --                    Separates proxy flags from the subprocess command
+                        (required when --out is stdio)
+
+Examples:
+  # stdio proxy → subprocess
+  heimdall-mcp --store sqlite://./t.db -- node server.js
+
+  # stdio proxy → remote HTTP server
+  heimdall-mcp --store sqlite://./t.db --out http --target http://localhost:3001
+
+  # stdio proxy → remote SSE server with Postgres
+  heimdall-mcp --store postgres://user:pass@host/db --out sse --target http://remote.com
+```

package/bin/heimdall-mcp.js

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+import '../dist/cli.js'

package/dist/MySqlStore-VOXUWDAJ.js

@@ -0,0 +1,111 @@
+// src/store/MySqlStore.ts
+import { and, between, eq, sql } from "drizzle-orm";
+import { drizzle } from "drizzle-orm/mysql2";
+import mysql from "mysql2/promise";
+
+// src/schema/mysql.schema.ts
+import { float, int, json, mysqlTable, serial, timestamp, varchar } from "drizzle-orm/mysql-core";
+var spans = mysqlTable("mcp_spans", {
+  id: varchar("id", { length: 64 }).primaryKey(),
+  traceId: varchar("trace_id", { length: 32 }).notNull(),
+  spanId: varchar("span_id", { length: 16 }).notNull(),
+  parentId: varchar("parent_id", { length: 16 }),
+  name: varchar("name", { length: 128 }).notNull(),
+  status: varchar("status", { length: 16 }).notNull(),
+  startedAt: timestamp("started_at").notNull(),
+  endedAt: timestamp("ended_at").notNull(),
+  durationMs: int("duration_ms").notNull(),
+  attributes: json("attributes"),
+  events: json("events")
+});
+var metrics = mysqlTable("mcp_metrics", {
+  id: serial("id").primaryKey(),
+  toolName: varchar("tool_name", { length: 128 }).notNull(),
+  callCount: int("call_count").default(0),
+  errorCount: int("error_count").default(0),
+  avgDuration: float("avg_duration"),
+  updatedAt: timestamp("updated_at").notNull()
+});
+
+// src/store/MySqlStore.ts
+var MySqlStore = class {
+  pool;
+  db;
+  constructor(connectionString) {
+    this.pool = mysql.createPool(connectionString);
+    this.db = drizzle(this.pool);
+  }
+  async init() {
+    await this.db.execute(sql`
+      CREATE TABLE IF NOT EXISTS mcp_spans (
+        id           VARCHAR(64)  NOT NULL PRIMARY KEY,
+        trace_id     VARCHAR(32)  NOT NULL,
+        span_id      VARCHAR(16)  NOT NULL,
+        parent_id    VARCHAR(16),
+        name         VARCHAR(128) NOT NULL,
+        status       VARCHAR(16)  NOT NULL,
+        started_at   TIMESTAMP(3) NOT NULL,
+        ended_at     TIMESTAMP(3) NOT NULL,
+        duration_ms  INT          NOT NULL,
+        attributes   JSON,
+        events       JSON
+      )
+    `);
+    await this.db.execute(sql`
+      CREATE TABLE IF NOT EXISTS mcp_metrics (
+        id           BIGINT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY,
+        tool_name    VARCHAR(128)    NOT NULL,
+        call_count   INT             DEFAULT 0,
+        error_count  INT             DEFAULT 0,
+        avg_duration FLOAT,
+        updated_at   TIMESTAMP(3)    NOT NULL
+      )
+    `);
+  }
+  async save(span) {
+    await this.db.insert(spans).values({
+      id: span.id,
+      traceId: span.traceId,
+      spanId: span.spanId,
+      parentId: span.parentId,
+      name: span.name,
+      status: span.status,
+      startedAt: span.startedAt,
+      endedAt: span.endedAt,
+      durationMs: span.durationMs,
+      attributes: span.attributes ?? null,
+      events: span.events ?? null
+    }).onDuplicateKeyUpdate({ set: { id: span.id } });
+  }
+  async query(filters) {
+    const conditions = [];
+    if (filters.traceId) conditions.push(eq(spans.traceId, filters.traceId));
+    if (filters.spanId) conditions.push(eq(spans.spanId, filters.spanId));
+    if (filters.name) conditions.push(eq(spans.name, filters.name));
+    if (filters.status) conditions.push(eq(spans.status, filters.status));
+    if (filters.from && filters.to) {
+      conditions.push(between(spans.startedAt, filters.from, filters.to));
+    }
+    const rows = await this.db.select().from(spans).where(conditions.length ? and(...conditions) : void 0).limit(filters.limit ?? 100);
+    return rows.map((r) => ({
+      id: r.id,
+      traceId: r.traceId,
+      spanId: r.spanId,
+      parentId: r.parentId ?? void 0,
+      name: r.name,
+      status: r.status,
+      startedAt: r.startedAt,
+      endedAt: r.endedAt,
+      durationMs: r.durationMs,
+      attributes: r.attributes,
+      events: r.events
+    }));
+  }
+  async close() {
+    await this.pool.end();
+  }
+};
+export {
+  MySqlStore
+};
+//# sourceMappingURL=MySqlStore-VOXUWDAJ.js.map

package/dist/MySqlStore-VOXUWDAJ.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/store/MySqlStore.ts","../src/schema/mysql.schema.ts"],"sourcesContent":["import { and, between, eq, sql } from 'drizzle-orm'\nimport { drizzle } from 'drizzle-orm/mysql2'\nimport mysql from 'mysql2/promise'\n\nimport { spans } from '@/schema/mysql.schema'\n\nimport type { TraceStore } from './TraceStore'\nimport type { McpSpan, SpanFilters } from '@/types'\n\nexport class MySqlStore implements TraceStore {\n  private pool\n  private db\n\n  constructor(connectionString: string) {\n    this.pool = mysql.createPool(connectionString)\n    this.db = drizzle(this.pool)\n  }\n\n  async init(): Promise<void> {\n    await this.db.execute(sql`\n      CREATE TABLE IF NOT EXISTS mcp_spans (\n        id           VARCHAR(64)  NOT NULL PRIMARY KEY,\n        trace_id     VARCHAR(32)  NOT NULL,\n        span_id      VARCHAR(16)  NOT NULL,\n        parent_id    VARCHAR(16),\n        name         VARCHAR(128) NOT NULL,\n        status       VARCHAR(16)  NOT NULL,\n        started_at   TIMESTAMP(3) NOT NULL,\n        ended_at     TIMESTAMP(3) NOT NULL,\n        duration_ms  INT          NOT NULL,\n        attributes   JSON,\n        events       JSON\n      )\n    `)\n    await this.db.execute(sql`\n      CREATE TABLE IF NOT EXISTS mcp_metrics (\n        id           BIGINT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY,\n        tool_name    VARCHAR(128)    NOT NULL,\n        call_count   INT             DEFAULT 0,\n        error_count  INT             DEFAULT 0,\n        avg_duration FLOAT,\n        updated_at   TIMESTAMP(3)    NOT NULL\n      )\n    `)\n  }\n\n  async save(span: McpSpan): Promise<void> {\n    await this.db.insert(spans).values({\n      id:         span.id,\n      traceId:    span.traceId,\n      spanId:     span.spanId,\n      parentId:   span.parentId,\n      name:       span.name,\n      status:     span.status,\n      startedAt:  span.startedAt,\n      endedAt:    span.endedAt,\n      durationMs: span.durationMs,\n      attributes: span.attributes ?? null,\n      events:     span.events ?? null,\n    }).onDuplicateKeyUpdate({ set: { id: span.id } })\n  }\n\n  async query(filters: SpanFilters): Promise<McpSpan[]> {\n    const conditions = []\n    if (filters.traceId) conditions.push(eq(spans.traceId, filters.traceId))\n    if (filters.spanId)  conditions.push(eq(spans.spanId, filters.spanId))\n    if (filters.name)    conditions.push(eq(spans.name, filters.name))\n    if (filters.status)  conditions.push(eq(spans.status, filters.status))\n    if (filters.from && filters.to) {\n      conditions.push(between(spans.startedAt, filters.from, filters.to))\n    }\n\n    const rows = await this.db\n      .select()\n      .from(spans)\n      .where(conditions.length ? and(...conditions) : undefined)\n      .limit(filters.limit ?? 100)\n\n    return rows.map((r) => ({\n      id:         r.id,\n      traceId:    r.traceId,\n      spanId:     r.spanId,\n      parentId:   r.parentId ?? undefined,\n      name:       r.name,\n      status:     r.status as McpSpan['status'],\n      startedAt:  r.startedAt,\n      endedAt:    r.endedAt,\n      durationMs: r.durationMs,\n      attributes: r.attributes as Record<string, unknown> | undefined,\n      events:     r.events as McpSpan['events'],\n    }))\n  }\n\n  async close(): Promise<void> {\n    await this.pool.end()\n  }\n}\n","import { float, int, json, mysqlTable, serial, timestamp, varchar } from 'drizzle-orm/mysql-core'\n\nexport const spans = mysqlTable('mcp_spans', {\n  id:          varchar('id', { length: 64 }).primaryKey(),\n  traceId:     varchar('trace_id', { length: 32 }).notNull(),\n  spanId:      varchar('span_id', { length: 16 }).notNull(),\n  parentId:    varchar('parent_id', { length: 16 }),\n  name:        varchar('name', { length: 128 }).notNull(),\n  status:      varchar('status', { length: 16 }).notNull(),\n  startedAt:   timestamp('started_at').notNull(),\n  endedAt:     timestamp('ended_at').notNull(),\n  durationMs:  int('duration_ms').notNull(),\n  attributes:  json('attributes'),\n  events:      json('events'),\n})\n\nexport const metrics = mysqlTable('mcp_metrics', {\n  id:          serial('id').primaryKey(),\n  toolName:    varchar('tool_name', { length: 128 }).notNull(),\n  callCount:   int('call_count').default(0),\n  errorCount:  int('error_count').default(0),\n  avgDuration: float('avg_duration'),\n  updatedAt:   timestamp('updated_at').notNull(),\n})\n"],"mappings":";AAAA,SAAS,KAAK,SAAS,IAAI,WAAW;AACtC,SAAS,eAAe;AACxB,OAAO,WAAW;;;ACFlB,SAAS,OAAO,KAAK,MAAM,YAAY,QAAQ,WAAW,eAAe;AAElE,IAAM,QAAQ,WAAW,aAAa;AAAA,EAC3C,IAAa,QAAQ,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,WAAW;AAAA,EACtD,SAAa,QAAQ,YAAY,EAAE,QAAQ,GAAG,CAAC,EAAE,QAAQ;AAAA,EACzD,QAAa,QAAQ,WAAW,EAAE,QAAQ,GAAG,CAAC,EAAE,QAAQ;AAAA,EACxD,UAAa,QAAQ,aAAa,EAAE,QAAQ,GAAG,CAAC;AAAA,EAChD,MAAa,QAAQ,QAAQ,EAAE,QAAQ,IAAI,CAAC,EAAE,QAAQ;AAAA,EACtD,QAAa,QAAQ,UAAU,EAAE,QAAQ,GAAG,CAAC,EAAE,QAAQ;AAAA,EACvD,WAAa,UAAU,YAAY,EAAE,QAAQ;AAAA,EAC7C,SAAa,UAAU,UAAU,EAAE,QAAQ;AAAA,EAC3C,YAAa,IAAI,aAAa,EAAE,QAAQ;AAAA,EACxC,YAAa,KAAK,YAAY;AAAA,EAC9B,QAAa,KAAK,QAAQ;AAC5B,CAAC;AAEM,IAAM,UAAU,WAAW,eAAe;AAAA,EAC/C,IAAa,OAAO,IAAI,EAAE,WAAW;AAAA,EACrC,UAAa,QAAQ,aAAa,EAAE,QAAQ,IAAI,CAAC,EAAE,QAAQ;AAAA,EAC3D,WAAa,IAAI,YAAY,EAAE,QAAQ,CAAC;AAAA,EACxC,YAAa,IAAI,aAAa,EAAE,QAAQ,CAAC;AAAA,EACzC,aAAa,MAAM,cAAc;AAAA,EACjC,WAAa,UAAU,YAAY,EAAE,QAAQ;AAC/C,CAAC;;;ADdM,IAAM,aAAN,MAAuC;AAAA,EACpC;AAAA,EACA;AAAA,EAER,YAAY,kBAA0B;AACpC,SAAK,OAAO,MAAM,WAAW,gBAAgB;AAC7C,SAAK,KAAK,QAAQ,KAAK,IAAI;AAAA,EAC7B;AAAA,EAEA,MAAM,OAAsB;AAC1B,UAAM,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAcrB;AACD,UAAM,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KASrB;AAAA,EACH;AAAA,EAEA,MAAM,KAAK,MAA8B;AACvC,UAAM,KAAK,GAAG,OAAO,KAAK,EAAE,OAAO;AAAA,MACjC,IAAY,KAAK;AAAA,MACjB,SAAY,KAAK;AAAA,MACjB,QAAY,KAAK;AAAA,MACjB,UAAY,KAAK;AAAA,MACjB,MAAY,KAAK;AAAA,MACjB,QAAY,KAAK;AAAA,MACjB,WAAY,KAAK;AAAA,MACjB,SAAY,KAAK;AAAA,MACjB,YAAY,KAAK;AAAA,MACjB,YAAY,KAAK,cAAc;AAAA,MAC/B,QAAY,KAAK,UAAU;AAAA,IAC7B,CAAC,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,KAAK,GAAG,EAAE,CAAC;AAAA,EAClD;AAAA,EAEA,MAAM,MAAM,SAA0C;AACpD,UAAM,aAAa,CAAC;AACpB,QAAI,QAAQ,QAAS,YAAW,KAAK,GAAG,MAAM,SAAS,QAAQ,OAAO,CAAC;AACvE,QAAI,QAAQ,OAAS,YAAW,KAAK,GAAG,MAAM,QAAQ,QAAQ,MAAM,CAAC;AACrE,QAAI,QAAQ,KAAS,YAAW,KAAK,GAAG,MAAM,MAAM,QAAQ,IAAI,CAAC;AACjE,QAAI,QAAQ,OAAS,YAAW,KAAK,GAAG,MAAM,QAAQ,QAAQ,MAAM,CAAC;AACrE,QAAI,QAAQ,QAAQ,QAAQ,IAAI;AAC9B,iBAAW,KAAK,QAAQ,MAAM,WAAW,QAAQ,MAAM,QAAQ,EAAE,CAAC;AAAA,IACpE;AAEA,UAAM,OAAO,MAAM,KAAK,GACrB,OAAO,EACP,KAAK,KAAK,EACV,MAAM,WAAW,SAAS,IAAI,GAAG,UAAU,IAAI,MAAS,EACxD,MAAM,QAAQ,SAAS,GAAG;AAE7B,WAAO,KAAK,IAAI,CAAC,OAAO;AAAA,MACtB,IAAY,EAAE;AAAA,MACd,SAAY,EAAE;AAAA,MACd,QAAY,EAAE;AAAA,MACd,UAAY,EAAE,YAAY;AAAA,MAC1B,MAAY,EAAE;AAAA,MACd,QAAY,EAAE;AAAA,MACd,WAAY,EAAE;AAAA,MACd,SAAY,EAAE;AAAA,MACd,YAAY,EAAE;AAAA,MACd,YAAY,EAAE;AAAA,MACd,QAAY,EAAE;AAAA,IAChB,EAAE;AAAA,EACJ;AAAA,EAEA,MAAM,QAAuB;AAC3B,UAAM,KAAK,KAAK,IAAI;AAAA,EACtB;AACF;","names":[]}

package/dist/PostgresStore-QUSLDMPH.js

@@ -0,0 +1,111 @@
+// src/store/PostgresStore.ts
+import { and, between, eq, sql } from "drizzle-orm";
+import { drizzle } from "drizzle-orm/postgres-js";
+import postgres from "postgres";
+
+// src/schema/pg.schema.ts
+import { integer, json, pgTable, real, serial, timestamp, varchar } from "drizzle-orm/pg-core";
+var spans = pgTable("mcp_spans", {
+  id: varchar("id", { length: 64 }).primaryKey(),
+  traceId: varchar("trace_id", { length: 32 }).notNull(),
+  spanId: varchar("span_id", { length: 16 }).notNull(),
+  parentId: varchar("parent_id", { length: 16 }),
+  name: varchar("name", { length: 128 }).notNull(),
+  status: varchar("status", { length: 16 }).notNull(),
+  startedAt: timestamp("started_at").notNull(),
+  endedAt: timestamp("ended_at").notNull(),
+  durationMs: integer("duration_ms").notNull(),
+  attributes: json("attributes"),
+  events: json("events")
+});
+var metrics = pgTable("mcp_metrics", {
+  id: serial("id").primaryKey(),
+  toolName: varchar("tool_name", { length: 128 }).notNull(),
+  callCount: integer("call_count").default(0),
+  errorCount: integer("error_count").default(0),
+  avgDuration: real("avg_duration"),
+  updatedAt: timestamp("updated_at").notNull()
+});
+
+// src/store/PostgresStore.ts
+var PostgresStore = class {
+  sql;
+  db;
+  constructor(connectionString) {
+    this.sql = postgres(connectionString);
+    this.db = drizzle(this.sql);
+  }
+  async init() {
+    await this.db.execute(sql`
+      CREATE TABLE IF NOT EXISTS mcp_spans (
+        id           VARCHAR(64)  NOT NULL PRIMARY KEY,
+        trace_id     VARCHAR(32)  NOT NULL,
+        span_id      VARCHAR(16)  NOT NULL,
+        parent_id    VARCHAR(16),
+        name         VARCHAR(128) NOT NULL,
+        status       VARCHAR(16)  NOT NULL,
+        started_at   TIMESTAMP    NOT NULL,
+        ended_at     TIMESTAMP    NOT NULL,
+        duration_ms  INTEGER      NOT NULL,
+        attributes   JSON,
+        events       JSON
+      )
+    `);
+    await this.db.execute(sql`
+      CREATE TABLE IF NOT EXISTS mcp_metrics (
+        id           SERIAL       PRIMARY KEY,
+        tool_name    VARCHAR(128) NOT NULL,
+        call_count   INTEGER      DEFAULT 0,
+        error_count  INTEGER      DEFAULT 0,
+        avg_duration REAL,
+        updated_at   TIMESTAMP    NOT NULL
+      )
+    `);
+  }
+  async save(span) {
+    await this.db.insert(spans).values({
+      id: span.id,
+      traceId: span.traceId,
+      spanId: span.spanId,
+      parentId: span.parentId,
+      name: span.name,
+      status: span.status,
+      startedAt: span.startedAt,
+      endedAt: span.endedAt,
+      durationMs: span.durationMs,
+      attributes: span.attributes ?? null,
+      events: span.events ?? null
+    }).onConflictDoNothing();
+  }
+  async query(filters) {
+    const conditions = [];
+    if (filters.traceId) conditions.push(eq(spans.traceId, filters.traceId));
+    if (filters.spanId) conditions.push(eq(spans.spanId, filters.spanId));
+    if (filters.name) conditions.push(eq(spans.name, filters.name));
+    if (filters.status) conditions.push(eq(spans.status, filters.status));
+    if (filters.from && filters.to) {
+      conditions.push(between(spans.startedAt, filters.from, filters.to));
+    }
+    const rows = await this.db.select().from(spans).where(conditions.length ? and(...conditions) : void 0).limit(filters.limit ?? 100);
+    return rows.map((r) => ({
+      id: r.id,
+      traceId: r.traceId,
+      spanId: r.spanId,
+      parentId: r.parentId ?? void 0,
+      name: r.name,
+      status: r.status,
+      startedAt: r.startedAt,
+      endedAt: r.endedAt,
+      durationMs: r.durationMs,
+      attributes: r.attributes,
+      events: r.events
+    }));
+  }
+  async close() {
+    await this.sql.end();
+  }
+};
+export {
+  PostgresStore
+};
+//# sourceMappingURL=PostgresStore-QUSLDMPH.js.map

package/dist/PostgresStore-QUSLDMPH.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/store/PostgresStore.ts","../src/schema/pg.schema.ts"],"sourcesContent":["import { and, between, eq, sql } from 'drizzle-orm'\nimport { drizzle } from 'drizzle-orm/postgres-js'\nimport postgres from 'postgres'\n\nimport { spans } from '@/schema/pg.schema'\n\nimport type { TraceStore } from './TraceStore'\nimport type { McpSpan, SpanFilters } from '@/types'\n\nexport class PostgresStore implements TraceStore {\n  private sql\n  private db\n\n  constructor(connectionString: string) {\n    this.sql = postgres(connectionString)\n    this.db = drizzle(this.sql)\n  }\n\n  async init(): Promise<void> {\n    await this.db.execute(sql`\n      CREATE TABLE IF NOT EXISTS mcp_spans (\n        id           VARCHAR(64)  NOT NULL PRIMARY KEY,\n        trace_id     VARCHAR(32)  NOT NULL,\n        span_id      VARCHAR(16)  NOT NULL,\n        parent_id    VARCHAR(16),\n        name         VARCHAR(128) NOT NULL,\n        status       VARCHAR(16)  NOT NULL,\n        started_at   TIMESTAMP    NOT NULL,\n        ended_at     TIMESTAMP    NOT NULL,\n        duration_ms  INTEGER      NOT NULL,\n        attributes   JSON,\n        events       JSON\n      )\n    `)\n    await this.db.execute(sql`\n      CREATE TABLE IF NOT EXISTS mcp_metrics (\n        id           SERIAL       PRIMARY KEY,\n        tool_name    VARCHAR(128) NOT NULL,\n        call_count   INTEGER      DEFAULT 0,\n        error_count  INTEGER      DEFAULT 0,\n        avg_duration REAL,\n        updated_at   TIMESTAMP    NOT NULL\n      )\n    `)\n  }\n\n  async save(span: McpSpan): Promise<void> {\n    await this.db.insert(spans).values({\n      id:         span.id,\n      traceId:    span.traceId,\n      spanId:     span.spanId,\n      parentId:   span.parentId,\n      name:       span.name,\n      status:     span.status,\n      startedAt:  span.startedAt,\n      endedAt:    span.endedAt,\n      durationMs: span.durationMs,\n      attributes: span.attributes ?? null,\n      events:     span.events ?? null,\n    }).onConflictDoNothing()\n  }\n\n  async query(filters: SpanFilters): Promise<McpSpan[]> {\n    const conditions = []\n    if (filters.traceId) conditions.push(eq(spans.traceId, filters.traceId))\n    if (filters.spanId)  conditions.push(eq(spans.spanId, filters.spanId))\n    if (filters.name)    conditions.push(eq(spans.name, filters.name))\n    if (filters.status)  conditions.push(eq(spans.status, filters.status))\n    if (filters.from && filters.to) {\n      conditions.push(between(spans.startedAt, filters.from, filters.to))\n    }\n\n    const rows = await this.db\n      .select()\n      .from(spans)\n      .where(conditions.length ? and(...conditions) : undefined)\n      .limit(filters.limit ?? 100)\n\n    return rows.map((r) => ({\n      id:         r.id,\n      traceId:    r.traceId,\n      spanId:     r.spanId,\n      parentId:   r.parentId ?? undefined,\n      name:       r.name,\n      status:     r.status as McpSpan['status'],\n      startedAt:  r.startedAt,\n      endedAt:    r.endedAt,\n      durationMs: r.durationMs,\n      attributes: r.attributes as Record<string, unknown> | undefined,\n      events:     r.events as McpSpan['events'],\n    }))\n  }\n\n  async close(): Promise<void> {\n    await this.sql.end()\n  }\n}\n","import { integer, json, pgTable, real, serial, timestamp, varchar } from 'drizzle-orm/pg-core'\n\nexport const spans = pgTable('mcp_spans', {\n  id:          varchar('id', { length: 64 }).primaryKey(),\n  traceId:     varchar('trace_id', { length: 32 }).notNull(),\n  spanId:      varchar('span_id', { length: 16 }).notNull(),\n  parentId:    varchar('parent_id', { length: 16 }),\n  name:        varchar('name', { length: 128 }).notNull(),\n  status:      varchar('status', { length: 16 }).notNull(),\n  startedAt:   timestamp('started_at').notNull(),\n  endedAt:     timestamp('ended_at').notNull(),\n  durationMs:  integer('duration_ms').notNull(),\n  attributes:  json('attributes'),\n  events:      json('events'),\n})\n\nexport const metrics = pgTable('mcp_metrics', {\n  id:          serial('id').primaryKey(),\n  toolName:    varchar('tool_name', { length: 128 }).notNull(),\n  callCount:   integer('call_count').default(0),\n  errorCount:  integer('error_count').default(0),\n  avgDuration: real('avg_duration'),\n  updatedAt:   timestamp('updated_at').notNull(),\n})\n"],"mappings":";AAAA,SAAS,KAAK,SAAS,IAAI,WAAW;AACtC,SAAS,eAAe;AACxB,OAAO,cAAc;;;ACFrB,SAAS,SAAS,MAAM,SAAS,MAAM,QAAQ,WAAW,eAAe;AAElE,IAAM,QAAQ,QAAQ,aAAa;AAAA,EACxC,IAAa,QAAQ,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,WAAW;AAAA,EACtD,SAAa,QAAQ,YAAY,EAAE,QAAQ,GAAG,CAAC,EAAE,QAAQ;AAAA,EACzD,QAAa,QAAQ,WAAW,EAAE,QAAQ,GAAG,CAAC,EAAE,QAAQ;AAAA,EACxD,UAAa,QAAQ,aAAa,EAAE,QAAQ,GAAG,CAAC;AAAA,EAChD,MAAa,QAAQ,QAAQ,EAAE,QAAQ,IAAI,CAAC,EAAE,QAAQ;AAAA,EACtD,QAAa,QAAQ,UAAU,EAAE,QAAQ,GAAG,CAAC,EAAE,QAAQ;AAAA,EACvD,WAAa,UAAU,YAAY,EAAE,QAAQ;AAAA,EAC7C,SAAa,UAAU,UAAU,EAAE,QAAQ;AAAA,EAC3C,YAAa,QAAQ,aAAa,EAAE,QAAQ;AAAA,EAC5C,YAAa,KAAK,YAAY;AAAA,EAC9B,QAAa,KAAK,QAAQ;AAC5B,CAAC;AAEM,IAAM,UAAU,QAAQ,eAAe;AAAA,EAC5C,IAAa,OAAO,IAAI,EAAE,WAAW;AAAA,EACrC,UAAa,QAAQ,aAAa,EAAE,QAAQ,IAAI,CAAC,EAAE,QAAQ;AAAA,EAC3D,WAAa,QAAQ,YAAY,EAAE,QAAQ,CAAC;AAAA,EAC5C,YAAa,QAAQ,aAAa,EAAE,QAAQ,CAAC;AAAA,EAC7C,aAAa,KAAK,cAAc;AAAA,EAChC,WAAa,UAAU,YAAY,EAAE,QAAQ;AAC/C,CAAC;;;ADdM,IAAM,gBAAN,MAA0C;AAAA,EACvC;AAAA,EACA;AAAA,EAER,YAAY,kBAA0B;AACpC,SAAK,MAAM,SAAS,gBAAgB;AACpC,SAAK,KAAK,QAAQ,KAAK,GAAG;AAAA,EAC5B;AAAA,EAEA,MAAM,OAAsB;AAC1B,UAAM,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAcrB;AACD,UAAM,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KASrB;AAAA,EACH;AAAA,EAEA,MAAM,KAAK,MAA8B;AACvC,UAAM,KAAK,GAAG,OAAO,KAAK,EAAE,OAAO;AAAA,MACjC,IAAY,KAAK;AAAA,MACjB,SAAY,KAAK;AAAA,MACjB,QAAY,KAAK;AAAA,MACjB,UAAY,KAAK;AAAA,MACjB,MAAY,KAAK;AAAA,MACjB,QAAY,KAAK;AAAA,MACjB,WAAY,KAAK;AAAA,MACjB,SAAY,KAAK;AAAA,MACjB,YAAY,KAAK;AAAA,MACjB,YAAY,KAAK,cAAc;AAAA,MAC/B,QAAY,KAAK,UAAU;AAAA,IAC7B,CAAC,EAAE,oBAAoB;AAAA,EACzB;AAAA,EAEA,MAAM,MAAM,SAA0C;AACpD,UAAM,aAAa,CAAC;AACpB,QAAI,QAAQ,QAAS,YAAW,KAAK,GAAG,MAAM,SAAS,QAAQ,OAAO,CAAC;AACvE,QAAI,QAAQ,OAAS,YAAW,KAAK,GAAG,MAAM,QAAQ,QAAQ,MAAM,CAAC;AACrE,QAAI,QAAQ,KAAS,YAAW,KAAK,GAAG,MAAM,MAAM,QAAQ,IAAI,CAAC;AACjE,QAAI,QAAQ,OAAS,YAAW,KAAK,GAAG,MAAM,QAAQ,QAAQ,MAAM,CAAC;AACrE,QAAI,QAAQ,QAAQ,QAAQ,IAAI;AAC9B,iBAAW,KAAK,QAAQ,MAAM,WAAW,QAAQ,MAAM,QAAQ,EAAE,CAAC;AAAA,IACpE;AAEA,UAAM,OAAO,MAAM,KAAK,GACrB,OAAO,EACP,KAAK,KAAK,EACV,MAAM,WAAW,SAAS,IAAI,GAAG,UAAU,IAAI,MAAS,EACxD,MAAM,QAAQ,SAAS,GAAG;AAE7B,WAAO,KAAK,IAAI,CAAC,OAAO;AAAA,MACtB,IAAY,EAAE;AAAA,MACd,SAAY,EAAE;AAAA,MACd,QAAY,EAAE;AAAA,MACd,UAAY,EAAE,YAAY;AAAA,MAC1B,MAAY,EAAE;AAAA,MACd,QAAY,EAAE;AAAA,MACd,WAAY,EAAE;AAAA,MACd,SAAY,EAAE;AAAA,MACd,YAAY,EAAE;AAAA,MACd,YAAY,EAAE;AAAA,MACd,QAAY,EAAE;AAAA,IAChB,EAAE;AAAA,EACJ;AAAA,EAEA,MAAM,QAAuB;AAC3B,UAAM,KAAK,IAAI,IAAI;AAAA,EACrB;AACF;","names":[]}