@burdenoff/mcp-libs 2026.1026.1 → 2026.1028.1

package/dist/index.cjs

@@ -7,9 +7,9 @@ var graphqlRequest = require('graphql-request');
 var fetch = require('cross-fetch');
 var mcp_js = require('@modelcontextprotocol/sdk/server/mcp.js');
 var types_js = require('@modelcontextprotocol/sdk/types.js');
+var crypto = require('crypto');
 var zod = require('zod');
 var stdio_js = require('@modelcontextprotocol/sdk/server/stdio.js');
-var crypto = require('crypto');
 var streamableHttp_js = require('@modelcontextprotocol/sdk/server/streamableHttp.js');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
@@ -731,6 +731,56 @@ function inferReadOnlyHint(toolName) {
 }
 
 // src/core/tool-governance.ts
+function canonicalJson(value, seen = /* @__PURE__ */ new Set()) {
+  if (value === null || value === void 0) return "null";
+  if (typeof value === "number") return Number.isFinite(value) ? JSON.stringify(value) : "null";
+  if (typeof value === "string" || typeof value === "boolean") return JSON.stringify(value);
+  if (typeof value === "bigint") return JSON.stringify(value.toString());
+  if (typeof value !== "object") return "null";
+  if (seen.has(value)) return "null";
+  seen.add(value);
+  try {
+    if (Array.isArray(value))
+      return `[${value.map((item) => canonicalJson(item, seen)).join(",")}]`;
+    const record = value;
+    const members = Object.keys(record).filter((key) => record[key] !== void 0).sort().map((key) => `${JSON.stringify(key)}:${canonicalJson(record[key], seen)}`);
+    return `{${members.join(",")}}`;
+  } finally {
+    seen.delete(value);
+  }
+}
+function toolArgsSha256(args) {
+  return crypto.createHash("sha256").update(canonicalJson(args), "utf8").digest("hex");
+}
+function isErrorResult(result) {
+  return typeof result === "object" && result !== null && result.isError === true;
+}
+function wrapAuditedHandler(name, handler, onToolCall) {
+  return async (...callbackArgs) => {
+    const started = Date.now();
+    const args = callbackArgs.length === 2 ? callbackArgs[0] : {};
+    const report = (status, error) => {
+      try {
+        onToolCall({
+          name,
+          argsSha256: toolArgsSha256(args),
+          status,
+          durationMs: Date.now() - started,
+          ...error ? { error } : {}
+        });
+      } catch {
+      }
+    };
+    try {
+      const result = await handler(...callbackArgs);
+      report(isErrorResult(result) ? "error" : "ok");
+      return result;
+    } catch (error) {
+      report("error", error instanceof Error ? error.message : String(error));
+      throw error;
+    }
+  };
+}
 var CONFIRM_FIELD_DESCRIPTION = "Must be true to execute this destructive tool. The call is rejected (no changes made) if omitted or false.";
 function resolveAnnotations(name, declared) {
   const destructiveHint = declared?.destructiveHint ?? inferDestructiveHint(name);
@@ -765,10 +815,11 @@ function wrapServerWithToolGovernance(server, options = {}) {
   const enforceConfirmation = options.enforceDestructiveConfirmation ?? false;
   const target = server;
   const originalRegisterTool = target.registerTool.bind(target);
-  target.registerTool = (name, config, handler) => {
+  const register = (name, config, handler) => {
     const annotations = resolveAnnotations(name, config.annotations);
+    const audited = options.onToolCall ? wrapAuditedHandler(name, handler, options.onToolCall) : handler;
     if (!enforceConfirmation || !annotations.destructiveHint) {
-      return originalRegisterTool(name, { ...config, annotations }, handler);
+      return originalRegisterTool(name, { ...config, annotations }, audited);
     }
     const gatedInputSchema = {
       ...config.inputSchema ?? {},
@@ -777,9 +828,16 @@ function wrapServerWithToolGovernance(server, options = {}) {
     return originalRegisterTool(
       name,
       { ...config, inputSchema: gatedInputSchema, annotations },
-      wrapDestructiveHandler(name, handler)
+      wrapDestructiveHandler(name, audited)
     );
   };
+  target.registerTool = (name, config, handler) => {
+    const registered = register(name, config, handler);
+    if (options.toolFilter && !options.toolFilter(name)) {
+      registered?.disable?.();
+    }
+    return registered;
+  };
   return server;
 }
 
@@ -807,7 +865,9 @@ function createMcpServer(ctx, modules, opts) {
     { capabilities: toServerCapabilities(capabilities) }
   );
   wrapServerWithToolGovernance(server, {
-    enforceDestructiveConfirmation: opts.enforceDestructiveConfirmation
+    enforceDestructiveConfirmation: opts.enforceDestructiveConfirmation,
+    toolFilter: opts.toolFilter,
+    onToolCall: opts.onToolCall
   });
   ctx._internal.serverRef.current = server;
   const inner = server.server ?? server;