easy-mcp-nest 0.3.0 → 0.4.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (80) hide show
  1. package/README.md +176 -0
  2. package/dist/EasyMCP.js +0 -3
  3. package/dist/EasyMCP.js.map +1 -1
  4. package/dist/adapters/express/express-adapter.js +10 -0
  5. package/dist/adapters/express/express-adapter.js.map +1 -1
  6. package/dist/core/batch/batch-executor.service.d.ts +21 -0
  7. package/dist/core/batch/batch-executor.service.js +101 -0
  8. package/dist/core/batch/batch-executor.service.js.map +1 -0
  9. package/dist/core/documentation/openapi-generator.service.d.ts +13 -0
  10. package/dist/core/documentation/openapi-generator.service.js +116 -0
  11. package/dist/core/documentation/openapi-generator.service.js.map +1 -0
  12. package/dist/core/errors/error-handler.interface.d.ts +11 -0
  13. package/dist/core/errors/error-handler.interface.js +3 -0
  14. package/dist/core/errors/error-handler.interface.js.map +1 -0
  15. package/dist/core/health/health-check.service.d.ts +33 -0
  16. package/dist/core/health/health-check.service.js +71 -0
  17. package/dist/core/health/health-check.service.js.map +1 -0
  18. package/dist/core/mcp-server/mcp-server.service.d.ts +14 -1
  19. package/dist/core/mcp-server/mcp-server.service.js +196 -3
  20. package/dist/core/mcp-server/mcp-server.service.js.map +1 -1
  21. package/dist/core/middleware/middleware-executor.service.d.ts +8 -0
  22. package/dist/core/middleware/middleware-executor.service.js +46 -0
  23. package/dist/core/middleware/middleware-executor.service.js.map +1 -0
  24. package/dist/core/middleware/middleware.interface.d.ts +6 -0
  25. package/dist/core/middleware/middleware.interface.js +3 -0
  26. package/dist/core/middleware/middleware.interface.js.map +1 -0
  27. package/dist/core/observability/metrics.service.d.ts +29 -0
  28. package/dist/core/observability/metrics.service.js +124 -0
  29. package/dist/core/observability/metrics.service.js.map +1 -0
  30. package/dist/core/observability/tracing.service.d.ts +12 -0
  31. package/dist/core/observability/tracing.service.js +88 -0
  32. package/dist/core/observability/tracing.service.js.map +1 -0
  33. package/dist/core/progress/progress-notifier.service.d.ts +16 -0
  34. package/dist/core/progress/progress-notifier.service.js +96 -0
  35. package/dist/core/progress/progress-notifier.service.js.map +1 -0
  36. package/dist/core/rate-limiting/rate-limit.interface.d.ts +11 -0
  37. package/dist/core/rate-limiting/rate-limit.interface.js +3 -0
  38. package/dist/core/rate-limiting/rate-limit.interface.js.map +1 -0
  39. package/dist/core/rate-limiting/rate-limiter.service.d.ts +13 -0
  40. package/dist/core/rate-limiting/rate-limiter.service.js +147 -0
  41. package/dist/core/rate-limiting/rate-limiter.service.js.map +1 -0
  42. package/dist/core/resilience/circuit-breaker.interface.d.ts +15 -0
  43. package/dist/core/resilience/circuit-breaker.interface.js +3 -0
  44. package/dist/core/resilience/circuit-breaker.interface.js.map +1 -0
  45. package/dist/core/resilience/circuit-breaker.service.d.ts +11 -0
  46. package/dist/core/resilience/circuit-breaker.service.js +117 -0
  47. package/dist/core/resilience/circuit-breaker.service.js.map +1 -0
  48. package/dist/core/resilience/retry.service.d.ts +6 -0
  49. package/dist/core/resilience/retry.service.js +66 -0
  50. package/dist/core/resilience/retry.service.js.map +1 -0
  51. package/dist/core/utils/decorator-scanner.d.ts +4 -0
  52. package/dist/core/utils/decorator-scanner.js +112 -0
  53. package/dist/core/utils/decorator-scanner.js.map +1 -0
  54. package/dist/core/utils/sanitize.util.d.ts +1 -0
  55. package/dist/core/utils/sanitize.util.js +19 -0
  56. package/dist/core/utils/sanitize.util.js.map +1 -1
  57. package/dist/decorators/index.d.ts +3 -0
  58. package/dist/decorators/index.js +12 -1
  59. package/dist/decorators/index.js.map +1 -1
  60. package/dist/decorators/mcp-error-handler.decorator.d.ts +4 -0
  61. package/dist/decorators/mcp-error-handler.decorator.js +15 -0
  62. package/dist/decorators/mcp-error-handler.decorator.js.map +1 -0
  63. package/dist/decorators/mcp-middleware.decorator.d.ts +4 -0
  64. package/dist/decorators/mcp-middleware.decorator.js +17 -0
  65. package/dist/decorators/mcp-middleware.decorator.js.map +1 -0
  66. package/dist/decorators/mcp-tool.decorator.d.ts +28 -0
  67. package/dist/decorators/mcp-tool.decorator.js +36 -0
  68. package/dist/decorators/mcp-tool.decorator.js.map +1 -0
  69. package/dist/index.d.ts +2 -2
  70. package/dist/index.js +4 -1
  71. package/dist/index.js.map +1 -1
  72. package/dist/interface/interface.interface.d.ts +2 -0
  73. package/dist/interface/mcp-protocol.interface.d.ts +14 -0
  74. package/dist/interface/mcp-protocol.interface.js.map +1 -1
  75. package/dist/tooling/tool-registry/tool-registry.service.d.ts +7 -1
  76. package/dist/tooling/tool-registry/tool-registry.service.js +40 -3
  77. package/dist/tooling/tool-registry/tool-registry.service.js.map +1 -1
  78. package/dist/tooling/tool.interface.d.ts +7 -1
  79. package/dist/tsconfig.build.tsbuildinfo +1 -1
  80. package/package.json +1 -1
@@ -0,0 +1,124 @@
1
+ "use strict";
2
+ var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
3
+ var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
4
+ if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
5
+ else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
6
+ return c > 3 && r && Object.defineProperty(target, key, r), r;
7
+ };
8
+ Object.defineProperty(exports, "__esModule", { value: true });
9
+ exports.MetricsService = void 0;
10
+ const common_1 = require("@nestjs/common");
11
+ const logger_util_1 = require("../utils/logger.util");
12
+ let MetricsService = class MetricsService {
13
+ toolMetrics = new Map();
14
+ requestCount = 0;
15
+ activeRequests = 0;
16
+ startTime = Date.now();
17
+ onModuleInit() {
18
+ logger_util_1.logger.info("MetricsService", "Metrics collection initialized", {
19
+ component: "Observability",
20
+ });
21
+ }
22
+ recordToolStart(toolName) {
23
+ const startTime = Date.now();
24
+ this.activeRequests++;
25
+ this.requestCount++;
26
+ return () => {
27
+ const executionTime = Date.now() - startTime;
28
+ this.activeRequests--;
29
+ const metrics = this.toolMetrics.get(toolName) || {
30
+ toolName,
31
+ executionCount: 0,
32
+ totalExecutionTime: 0,
33
+ errorCount: 0,
34
+ };
35
+ metrics.executionCount++;
36
+ metrics.totalExecutionTime += executionTime;
37
+ metrics.lastExecutionTime = Date.now();
38
+ this.toolMetrics.set(toolName, metrics);
39
+ };
40
+ }
41
+ recordToolError(toolName) {
42
+ const metrics = this.toolMetrics.get(toolName) || {
43
+ toolName,
44
+ executionCount: 0,
45
+ totalExecutionTime: 0,
46
+ errorCount: 0,
47
+ };
48
+ metrics.errorCount++;
49
+ this.toolMetrics.set(toolName, metrics);
50
+ }
51
+ getToolMetrics(toolName) {
52
+ return this.toolMetrics.get(toolName);
53
+ }
54
+ getAllToolMetrics() {
55
+ return Array.from(this.toolMetrics.values());
56
+ }
57
+ getAverageExecutionTime(toolName) {
58
+ const metrics = this.toolMetrics.get(toolName);
59
+ if (!metrics || metrics.executionCount === 0) {
60
+ return 0;
61
+ }
62
+ return metrics.totalExecutionTime / metrics.executionCount;
63
+ }
64
+ getErrorRate(toolName) {
65
+ const metrics = this.toolMetrics.get(toolName);
66
+ if (!metrics || metrics.executionCount === 0) {
67
+ return 0;
68
+ }
69
+ return metrics.errorCount / metrics.executionCount;
70
+ }
71
+ getServerMetrics() {
72
+ return {
73
+ totalRequests: this.requestCount,
74
+ activeRequests: this.activeRequests,
75
+ uptime: Date.now() - this.startTime,
76
+ toolCount: this.toolMetrics.size,
77
+ };
78
+ }
79
+ getPrometheusMetrics() {
80
+ const lines = [];
81
+ const serverMetrics = this.getServerMetrics();
82
+ lines.push(`# HELP mcp_server_total_requests Total number of requests`);
83
+ lines.push(`# TYPE mcp_server_total_requests counter`);
84
+ lines.push(`mcp_server_total_requests ${serverMetrics.totalRequests}`);
85
+ lines.push(`# HELP mcp_server_active_requests Current number of active requests`);
86
+ lines.push(`# TYPE mcp_server_active_requests gauge`);
87
+ lines.push(`mcp_server_active_requests ${serverMetrics.activeRequests}`);
88
+ lines.push(`# HELP mcp_server_uptime_seconds Server uptime in seconds`);
89
+ lines.push(`# TYPE mcp_server_uptime_seconds gauge`);
90
+ lines.push(`mcp_server_uptime_seconds ${Math.floor(serverMetrics.uptime / 1000)}`);
91
+ for (const metrics of this.toolMetrics.values()) {
92
+ const toolName = metrics.toolName.replace(/[^a-zA-Z0-9_]/g, "_");
93
+ lines.push(`# HELP mcp_tool_executions_total Total number of tool executions`);
94
+ lines.push(`# TYPE mcp_tool_executions_total counter`);
95
+ lines.push(`mcp_tool_executions_total{tool="${toolName}"} ${metrics.executionCount}`);
96
+ lines.push(`# HELP mcp_tool_execution_time_seconds Total execution time in seconds`);
97
+ lines.push(`# TYPE mcp_tool_execution_time_seconds counter`);
98
+ lines.push(`mcp_tool_execution_time_seconds{tool="${toolName}"} ${metrics.totalExecutionTime / 1000}`);
99
+ lines.push(`# HELP mcp_tool_errors_total Total number of tool errors`);
100
+ lines.push(`# TYPE mcp_tool_errors_total counter`);
101
+ lines.push(`mcp_tool_errors_total{tool="${toolName}"} ${metrics.errorCount}`);
102
+ const avgTime = this.getAverageExecutionTime(metrics.toolName);
103
+ lines.push(`# HELP mcp_tool_avg_execution_time_seconds Average execution time in seconds`);
104
+ lines.push(`# TYPE mcp_tool_avg_execution_time_seconds gauge`);
105
+ lines.push(`mcp_tool_avg_execution_time_seconds{tool="${toolName}"} ${avgTime / 1000}`);
106
+ const errorRate = this.getErrorRate(metrics.toolName);
107
+ lines.push(`# HELP mcp_tool_error_rate Error rate (0.0 to 1.0)`);
108
+ lines.push(`# TYPE mcp_tool_error_rate gauge`);
109
+ lines.push(`mcp_tool_error_rate{tool="${toolName}"} ${errorRate}`);
110
+ }
111
+ return lines.join("\n");
112
+ }
113
+ reset() {
114
+ this.toolMetrics.clear();
115
+ this.requestCount = 0;
116
+ this.activeRequests = 0;
117
+ this.startTime = Date.now();
118
+ }
119
+ };
120
+ exports.MetricsService = MetricsService;
121
+ exports.MetricsService = MetricsService = __decorate([
122
+ (0, common_1.Injectable)()
123
+ ], MetricsService);
124
+ //# sourceMappingURL=metrics.service.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"metrics.service.js","sourceRoot":"","sources":["../../../src/core/observability/metrics.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA0D;AAC1D,sDAA8C;AAsBvC,IAAM,cAAc,GAApB,MAAM,cAAc;IACjB,WAAW,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC7C,YAAY,GAAG,CAAC,CAAC;IACjB,cAAc,GAAG,CAAC,CAAC;IACnB,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE/B,YAAY;QAEV,oBAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,gCAAgC,EAAE;YAC9D,SAAS,EAAE,eAAe;SAC3B,CAAC,CAAC;IACL,CAAC;IAKD,eAAe,CAAC,QAAgB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,OAAO,GAAG,EAAE;YAEV,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,IAAI,CAAC,cAAc,EAAE,CAAC;YAEtB,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI;gBAChD,QAAQ;gBACR,cAAc,EAAE,CAAC;gBACjB,kBAAkB,EAAE,CAAC;gBACrB,UAAU,EAAE,CAAC;aACd,CAAC;YAEF,OAAO,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,CAAC,kBAAkB,IAAI,aAAa,CAAC;YAC5C,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEvC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC,CAAC;IACJ,CAAC;IAKD,eAAe,CAAC,QAAgB;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI;YAChD,QAAQ;YACR,cAAc,EAAE,CAAC;YACjB,kBAAkB,EAAE,CAAC;YACrB,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;IAKD,cAAc,CAAC,QAAgB;QAC7B,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAKD,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,CAAC;IAKD,uBAAuB,CAAC,QAAgB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,cAAc,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,OAAO,CAAC,kBAAkB,GAAG,OAAO,CAAC,cAAc,CAAC;IAC7D,CAAC;IAKD,YAAY,CAAC,QAAgB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,cAAc,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC,cAAc,CAAC;IACrD,CAAC;IAKD,gBAAgB;QAMd,OAAO;YACL,aAAa,EAAE,IAAI,CAAC,YAAY;YAChC,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS;YACnC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;SACjC,CAAC;IACJ,CAAC;IAKD,oBAAoB;QAClB,MAAM,KAAK,GAAa,EAAE,CAAC;QAG3B,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9C,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;QACxE,KAAK,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QACvD,KAAK,CAAC,IAAI,CAAC,6BAA6B,aAAa,CAAC,aAAa,EAAE,CAAC,CAAC;QAEvE,KAAK,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;QAClF,KAAK,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;QACtD,KAAK,CAAC,IAAI,CAAC,8BAA8B,aAAa,CAAC,cAAc,EAAE,CAAC,CAAC;QAEzE,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;QACxE,KAAK,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QACrD,KAAK,CAAC,IAAI,CAAC,6BAA6B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC;QAGnF,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;YAEjE,KAAK,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;YAC/E,KAAK,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YACvD,KAAK,CAAC,IAAI,CAAC,mCAAmC,QAAQ,MAAM,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;YAEtF,KAAK,CAAC,IAAI,CAAC,wEAAwE,CAAC,CAAC;YACrF,KAAK,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC7D,KAAK,CAAC,IAAI,CACR,yCAAyC,QAAQ,MAAM,OAAO,CAAC,kBAAkB,GAAG,IAAI,EAAE,CAC3F,CAAC;YAEF,KAAK,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;YACvE,KAAK,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;YACnD,KAAK,CAAC,IAAI,CAAC,+BAA+B,QAAQ,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;YAE9E,MAAM,OAAO,GAAG,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC/D,KAAK,CAAC,IAAI,CAAC,8EAA8E,CAAC,CAAC;YAC3F,KAAK,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YAC/D,KAAK,CAAC,IAAI,CAAC,6CAA6C,QAAQ,MAAM,OAAO,GAAG,IAAI,EAAE,CAAC,CAAC;YAExF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtD,KAAK,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;YACjE,KAAK,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC/C,KAAK,CAAC,IAAI,CAAC,6BAA6B,QAAQ,MAAM,SAAS,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAKD,KAAK;QACH,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;CACF,CAAA;AA1KY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;GACA,cAAc,CA0K1B"}
@@ -0,0 +1,12 @@
1
+ export interface TraceContext {
2
+ traceId: string;
3
+ spanId: string;
4
+ parentSpanId?: string;
5
+ startTime: number;
6
+ }
7
+ export declare class TracingService {
8
+ createTraceContext(parentTraceId?: string, parentSpanId?: string): TraceContext;
9
+ private validateTraceId;
10
+ extractTraceContext(headers?: Record<string, any>): TraceContext | undefined;
11
+ createChildSpan(parent: TraceContext): TraceContext;
12
+ }
@@ -0,0 +1,88 @@
1
+ "use strict";
2
+ var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
3
+ var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
4
+ if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
5
+ else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
6
+ return c > 3 && r && Object.defineProperty(target, key, r), r;
7
+ };
8
+ Object.defineProperty(exports, "__esModule", { value: true });
9
+ exports.TracingService = void 0;
10
+ const common_1 = require("@nestjs/common");
11
+ const crypto_1 = require("crypto");
12
+ function generateUUID() {
13
+ if (typeof crypto_1.randomUUID === "function") {
14
+ return (0, crypto_1.randomUUID)();
15
+ }
16
+ if (typeof crypto !== "undefined" && crypto.getRandomValues) {
17
+ const bytes = new Uint8Array(16);
18
+ crypto.getRandomValues(bytes);
19
+ bytes[6] = (bytes[6] & 0x0f) | 0x40;
20
+ bytes[8] = (bytes[8] & 0x3f) | 0x80;
21
+ const hex = Array.from(bytes)
22
+ .map((b) => b.toString(16).padStart(2, "0"))
23
+ .join("");
24
+ return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20, 32)}`;
25
+ }
26
+ return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
27
+ const r = (Math.random() * 16) | 0;
28
+ const v = c === "x" ? r : (r & 0x3) | 0x8;
29
+ return v.toString(16);
30
+ });
31
+ }
32
+ let TracingService = class TracingService {
33
+ createTraceContext(parentTraceId, parentSpanId) {
34
+ return {
35
+ traceId: parentTraceId || generateUUID(),
36
+ spanId: generateUUID(),
37
+ parentSpanId,
38
+ startTime: Date.now(),
39
+ };
40
+ }
41
+ validateTraceId(traceId) {
42
+ if (!traceId || typeof traceId !== "string") {
43
+ return null;
44
+ }
45
+ let sanitized = traceId.replace(/[\x00-\x1F\x7F-\x9F\n\r]/g, "");
46
+ const MAX_TRACE_ID_LENGTH = 128;
47
+ if (sanitized.length > MAX_TRACE_ID_LENGTH) {
48
+ return null;
49
+ }
50
+ if (!/^[a-zA-Z0-9_-]{1,128}$/.test(sanitized)) {
51
+ return null;
52
+ }
53
+ return sanitized;
54
+ }
55
+ extractTraceContext(headers) {
56
+ if (!headers) {
57
+ return undefined;
58
+ }
59
+ const traceId = headers["x-trace-id"] || headers["trace-id"];
60
+ const spanId = headers["x-span-id"] || headers["span-id"];
61
+ const parentSpanId = headers["x-parent-span-id"] || headers["parent-span-id"];
62
+ const validatedTraceId = traceId ? this.validateTraceId(traceId) : null;
63
+ if (!validatedTraceId) {
64
+ return undefined;
65
+ }
66
+ const validatedSpanId = spanId ? this.validateTraceId(spanId) : null;
67
+ const validatedParentSpanId = parentSpanId ? this.validateTraceId(parentSpanId) : null;
68
+ return {
69
+ traceId: validatedTraceId,
70
+ spanId: validatedSpanId || generateUUID(),
71
+ parentSpanId: validatedParentSpanId || undefined,
72
+ startTime: Date.now(),
73
+ };
74
+ }
75
+ createChildSpan(parent) {
76
+ return {
77
+ traceId: parent.traceId,
78
+ spanId: generateUUID(),
79
+ parentSpanId: parent.spanId,
80
+ startTime: Date.now(),
81
+ };
82
+ }
83
+ };
84
+ exports.TracingService = TracingService;
85
+ exports.TracingService = TracingService = __decorate([
86
+ (0, common_1.Injectable)()
87
+ ], TracingService);
88
+ //# sourceMappingURL=tracing.service.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tracing.service.js","sourceRoot":"","sources":["../../../src/core/observability/tracing.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAC5C,mCAAoC;AAMpC,SAAS,YAAY;IAEnB,IAAI,OAAO,mBAAU,KAAK,UAAU,EAAE,CAAC;QACrC,OAAO,IAAA,mBAAU,GAAE,CAAC;IACtB,CAAC;IAGD,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,CAAC;QACjC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAE9B,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;QACpC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;QAGpC,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;aAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;aAC3C,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;IACjH,CAAC;IAID,OAAO,sCAAsC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;QACnE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;QAC1C,OAAO,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;AACL,CAAC;AAgBM,IAAM,cAAc,GAApB,MAAM,cAAc;IAIzB,kBAAkB,CAAC,aAAsB,EAAE,YAAqB;QAC9D,OAAO;YACL,OAAO,EAAE,aAAa,IAAI,YAAY,EAAE;YACxC,MAAM,EAAE,YAAY,EAAE;YACtB,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;IAOO,eAAe,CAAC,OAAY;QAClC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC;QAGjE,MAAM,mBAAmB,GAAG,GAAG,CAAC;QAChC,IAAI,SAAS,CAAC,MAAM,GAAG,mBAAmB,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC;QACd,CAAC;QAID,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAMD,mBAAmB,CAAC,OAA6B;QAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,YAAY,GAAG,OAAO,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAG9E,MAAM,gBAAgB,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACxE,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QAGD,MAAM,eAAe,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,MAAM,qBAAqB,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAEvF,OAAO;YACL,OAAO,EAAE,gBAAgB;YACzB,MAAM,EAAE,eAAe,IAAI,YAAY,EAAE;YACzC,YAAY,EAAE,qBAAqB,IAAI,SAAS;YAChD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;IAKD,eAAe,CAAC,MAAoB;QAClC,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,YAAY,EAAE;YACtB,YAAY,EAAE,MAAM,CAAC,MAAM;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;CACF,CAAA;AAnFY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;GACA,cAAc,CAmF1B"}
@@ -0,0 +1,16 @@
1
+ import type { IInterfaceLayer } from "../../interface/interface.interface";
2
+ export type ProgressCallback = (progress: {
3
+ progress: number;
4
+ total?: number;
5
+ message?: string;
6
+ }) => void;
7
+ export declare class ProgressNotifierService {
8
+ private interfaceLayer?;
9
+ private progressTokens;
10
+ private nextTokenId;
11
+ constructor(interfaceLayer?: IInterfaceLayer | undefined);
12
+ createProgressCallback(requestId: string | number | null): ProgressCallback;
13
+ private getOrCreateProgressToken;
14
+ private sendProgress;
15
+ cleanup(requestId: string | number | null): void;
16
+ }
@@ -0,0 +1,96 @@
1
+ "use strict";
2
+ var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
3
+ var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
4
+ if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
5
+ else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
6
+ return c > 3 && r && Object.defineProperty(target, key, r), r;
7
+ };
8
+ var __metadata = (this && this.__metadata) || function (k, v) {
9
+ if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
10
+ };
11
+ var __param = (this && this.__param) || function (paramIndex, decorator) {
12
+ return function (target, key) { decorator(target, key, paramIndex); }
13
+ };
14
+ Object.defineProperty(exports, "__esModule", { value: true });
15
+ exports.ProgressNotifierService = void 0;
16
+ const common_1 = require("@nestjs/common");
17
+ const constants_1 = require("../../config/constants");
18
+ const logger_util_1 = require("../utils/logger.util");
19
+ let ProgressNotifierService = class ProgressNotifierService {
20
+ interfaceLayer;
21
+ progressTokens = new Map();
22
+ nextTokenId = 0;
23
+ constructor(interfaceLayer) {
24
+ this.interfaceLayer = interfaceLayer;
25
+ }
26
+ createProgressCallback(requestId) {
27
+ if (requestId === null || requestId === undefined) {
28
+ return () => { };
29
+ }
30
+ const progressToken = this.getOrCreateProgressToken(requestId);
31
+ return (progress) => {
32
+ this.sendProgress(requestId, progressToken, progress);
33
+ };
34
+ }
35
+ getOrCreateProgressToken(requestId) {
36
+ if (!this.progressTokens.has(requestId)) {
37
+ const token = `progress_${this.nextTokenId++}_${Date.now()}`;
38
+ this.progressTokens.set(requestId, token);
39
+ }
40
+ return this.progressTokens.get(requestId);
41
+ }
42
+ sendProgress(requestId, progressToken, progress) {
43
+ if (!this.interfaceLayer) {
44
+ logger_util_1.logger.debug("ProgressNotifierService", "Progress update", {
45
+ component: "ProgressNotifier",
46
+ requestId,
47
+ progress: progress.progress,
48
+ message: progress.message,
49
+ });
50
+ return;
51
+ }
52
+ const clampedProgress = Math.max(0, Math.min(1, progress.progress));
53
+ const notification = {
54
+ jsonrpc: "2.0",
55
+ method: "notifications/progress",
56
+ params: {
57
+ progressToken,
58
+ progress: clampedProgress,
59
+ ...(progress.total !== undefined && { total: progress.total }),
60
+ ...(progress.message && { message: progress.message }),
61
+ },
62
+ };
63
+ try {
64
+ if (this.interfaceLayer.sendNotification) {
65
+ this.interfaceLayer.sendNotification(notification);
66
+ }
67
+ else {
68
+ logger_util_1.logger.debug("ProgressNotifierService", "Progress notification", {
69
+ component: "ProgressNotifier",
70
+ requestId,
71
+ notification,
72
+ });
73
+ }
74
+ }
75
+ catch (error) {
76
+ logger_util_1.logger.error("ProgressNotifierService", "Failed to send progress notification", {
77
+ component: "ProgressNotifier",
78
+ requestId,
79
+ error: error instanceof Error ? error.message : String(error),
80
+ });
81
+ }
82
+ }
83
+ cleanup(requestId) {
84
+ if (requestId !== null && requestId !== undefined) {
85
+ this.progressTokens.delete(requestId);
86
+ }
87
+ }
88
+ };
89
+ exports.ProgressNotifierService = ProgressNotifierService;
90
+ exports.ProgressNotifierService = ProgressNotifierService = __decorate([
91
+ (0, common_1.Injectable)(),
92
+ __param(0, (0, common_1.Optional)()),
93
+ __param(0, (0, common_1.Inject)(constants_1.INTERFACE_LAYER_TOKEN)),
94
+ __metadata("design:paramtypes", [Object])
95
+ ], ProgressNotifierService);
96
+ //# sourceMappingURL=progress-notifier.service.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"progress-notifier.service.js","sourceRoot":"","sources":["../../../src/core/progress/progress-notifier.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAA8D;AAG9D,sDAA+D;AAC/D,sDAA8C;AAevC,IAAM,uBAAuB,GAA7B,MAAM,uBAAuB;IAKmB;IAJ7C,cAAc,GAAG,IAAI,GAAG,EAA2B,CAAC;IACpD,WAAW,GAAG,CAAC,CAAC;IAExB,YACqD,cAAgC;QAAhC,mBAAc,GAAd,cAAc,CAAkB;IAClF,CAAC;IASJ,sBAAsB,CAAC,SAAiC;QACtD,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAElD,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;QAE/D,OAAO,CAAC,QAAgE,EAAE,EAAE;YAC1E,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;QACxD,CAAC,CAAC;IACJ,CAAC;IAKO,wBAAwB,CAAC,SAA0B;QACzD,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,YAAY,IAAI,CAAC,WAAW,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC7D,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;IAC7C,CAAC;IAKO,YAAY,CAClB,SAA0B,EAC1B,aAAqB,EACrB,QAAgE;QAEhE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAEzB,oBAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,iBAAiB,EAAE;gBACzD,SAAS,EAAE,kBAAkB;gBAC7B,SAAS;gBACT,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAGD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEpE,MAAM,YAAY,GAAyB;YACzC,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,wBAAwB;YAChC,MAAM,EAAE;gBACN,aAAa;gBACb,QAAQ,EAAE,eAAe;gBACzB,GAAG,CAAC,QAAQ,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC9D,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC;aACvD;SACF,CAAC;QAKF,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,cAAc,CAAC,gBAAgB,EAAE,CAAC;gBACzC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;YACrD,CAAC;iBAAM,CAAC;gBAEN,oBAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,uBAAuB,EAAE;oBAC/D,SAAS,EAAE,kBAAkB;oBAC7B,SAAS;oBACT,YAAY;iBACb,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,oBAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,sCAAsC,EAAE;gBAC9E,SAAS,EAAE,kBAAkB;gBAC7B,SAAS;gBACT,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAKD,OAAO,CAAC,SAAiC;QACvC,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAClD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;CACF,CAAA;AAvGY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,mBAAU,GAAE;IAMR,WAAA,IAAA,iBAAQ,GAAE,CAAA;IAAE,WAAA,IAAA,eAAM,EAAC,iCAAqB,CAAC,CAAA;;GALjC,uBAAuB,CAuGnC"}
@@ -0,0 +1,11 @@
1
+ import { RateLimitConfig } from "../../decorators/mcp-tool.decorator";
2
+ export type { RateLimitConfig };
3
+ export interface RateLimitEntry {
4
+ count: number;
5
+ resetTime: number;
6
+ }
7
+ export interface RateLimitResult {
8
+ allowed: boolean;
9
+ remaining: number;
10
+ resetTime: number;
11
+ }
@@ -0,0 +1,3 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ //# sourceMappingURL=rate-limit.interface.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"rate-limit.interface.js","sourceRoot":"","sources":["../../../src/core/rate-limiting/rate-limit.interface.ts"],"names":[],"mappings":""}
@@ -0,0 +1,13 @@
1
+ import { OnModuleInit, OnModuleDestroy } from "@nestjs/common";
2
+ import { RateLimitConfig, RateLimitResult } from "./rate-limit.interface";
3
+ export declare class RateLimiterService implements OnModuleInit, OnModuleDestroy {
4
+ private rateLimiters;
5
+ private cleanupInterval?;
6
+ private static readonly MAX_ENTRIES_PER_TOOL;
7
+ private static readonly CLEANUP_INTERVAL_MS;
8
+ onModuleInit(): void;
9
+ onModuleDestroy(): void;
10
+ checkRateLimit(toolName: string, identifier: string, config: RateLimitConfig): RateLimitResult;
11
+ cleanup(): void;
12
+ reset(toolName?: string): void;
13
+ }
@@ -0,0 +1,147 @@
1
+ "use strict";
2
+ var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
3
+ var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
4
+ if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
5
+ else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
6
+ return c > 3 && r && Object.defineProperty(target, key, r), r;
7
+ };
8
+ var RateLimiterService_1;
9
+ Object.defineProperty(exports, "__esModule", { value: true });
10
+ exports.RateLimiterService = void 0;
11
+ const common_1 = require("@nestjs/common");
12
+ const logger_util_1 = require("../utils/logger.util");
13
+ const sanitize_util_1 = require("../utils/sanitize.util");
14
+ function parseTimeWindow(window) {
15
+ const match = window.match(/^(\d+)([smhd])$/);
16
+ if (!match) {
17
+ logger_util_1.logger.warn("RateLimiterService", `Invalid time window format: ${window}. Using default 1 minute`, {
18
+ component: "RateLimiter",
19
+ invalidWindow: window,
20
+ });
21
+ return 60 * 1000;
22
+ }
23
+ const value = parseInt(match[1], 10);
24
+ const unit = match[2];
25
+ if (isNaN(value) || value <= 0) {
26
+ logger_util_1.logger.warn("RateLimiterService", `Invalid time window value: ${value}. Using default 1 minute`, {
27
+ component: "RateLimiter",
28
+ invalidValue: value,
29
+ });
30
+ return 60 * 1000;
31
+ }
32
+ const multipliers = {
33
+ s: 1000,
34
+ m: 60 * 1000,
35
+ h: 60 * 60 * 1000,
36
+ d: 24 * 60 * 60 * 1000,
37
+ };
38
+ const multiplier = multipliers[unit];
39
+ if (!multiplier) {
40
+ logger_util_1.logger.warn("RateLimiterService", `Invalid time window unit: ${unit}. Using default 1 minute`, {
41
+ component: "RateLimiter",
42
+ invalidUnit: unit,
43
+ });
44
+ return 60 * 1000;
45
+ }
46
+ return value * multiplier;
47
+ }
48
+ let RateLimiterService = class RateLimiterService {
49
+ static { RateLimiterService_1 = this; }
50
+ rateLimiters = new Map();
51
+ cleanupInterval;
52
+ static MAX_ENTRIES_PER_TOOL = 10000;
53
+ static CLEANUP_INTERVAL_MS = 5 * 60 * 1000;
54
+ onModuleInit() {
55
+ this.cleanupInterval = setInterval(() => {
56
+ this.cleanup();
57
+ }, RateLimiterService_1.CLEANUP_INTERVAL_MS);
58
+ }
59
+ onModuleDestroy() {
60
+ if (this.cleanupInterval) {
61
+ clearInterval(this.cleanupInterval);
62
+ this.cleanupInterval = undefined;
63
+ }
64
+ }
65
+ checkRateLimit(toolName, identifier, config) {
66
+ try {
67
+ const windowMs = parseTimeWindow(config.window);
68
+ const now = Date.now();
69
+ if (!this.rateLimiters.has(toolName)) {
70
+ this.rateLimiters.set(toolName, new Map());
71
+ }
72
+ const toolLimiter = this.rateLimiters.get(toolName);
73
+ if (toolLimiter.size >= RateLimiterService_1.MAX_ENTRIES_PER_TOOL) {
74
+ const entries = Array.from(toolLimiter.entries());
75
+ const expired = entries.filter(([, entry]) => entry.resetTime < now);
76
+ if (expired.length > 0) {
77
+ expired.forEach(([id]) => toolLimiter.delete(id));
78
+ }
79
+ else {
80
+ const sorted = entries.sort(([, a], [, b]) => a.resetTime - b.resetTime);
81
+ const toRemove = Math.floor(sorted.length * 0.1);
82
+ sorted.slice(0, toRemove).forEach(([id]) => toolLimiter.delete(id));
83
+ }
84
+ }
85
+ let entry = toolLimiter.get(identifier);
86
+ if (!entry || entry.resetTime < now) {
87
+ entry = {
88
+ count: 0,
89
+ resetTime: now + windowMs,
90
+ };
91
+ toolLimiter.set(identifier, entry);
92
+ }
93
+ if (entry.count >= config.max) {
94
+ return {
95
+ allowed: false,
96
+ remaining: 0,
97
+ resetTime: entry.resetTime,
98
+ };
99
+ }
100
+ entry.count++;
101
+ return {
102
+ allowed: true,
103
+ remaining: config.max - entry.count,
104
+ resetTime: entry.resetTime,
105
+ };
106
+ }
107
+ catch (error) {
108
+ logger_util_1.logger.error("RateLimiterService", "Rate limit check failed", {
109
+ component: "RateLimiter",
110
+ toolName: (0, sanitize_util_1.sanitizeActorId)(toolName),
111
+ identifier: (0, sanitize_util_1.sanitizeActorId)(identifier),
112
+ error: error instanceof Error ? error.message : String(error),
113
+ });
114
+ return {
115
+ allowed: false,
116
+ remaining: 0,
117
+ resetTime: Date.now() + 60000,
118
+ };
119
+ }
120
+ }
121
+ cleanup() {
122
+ const now = Date.now();
123
+ for (const [toolName, toolLimiter] of this.rateLimiters.entries()) {
124
+ for (const [identifier, entry] of toolLimiter.entries()) {
125
+ if (entry.resetTime < now) {
126
+ toolLimiter.delete(identifier);
127
+ }
128
+ }
129
+ if (toolLimiter.size === 0) {
130
+ this.rateLimiters.delete(toolName);
131
+ }
132
+ }
133
+ }
134
+ reset(toolName) {
135
+ if (toolName) {
136
+ this.rateLimiters.delete(toolName);
137
+ }
138
+ else {
139
+ this.rateLimiters.clear();
140
+ }
141
+ }
142
+ };
143
+ exports.RateLimiterService = RateLimiterService;
144
+ exports.RateLimiterService = RateLimiterService = RateLimiterService_1 = __decorate([
145
+ (0, common_1.Injectable)()
146
+ ], RateLimiterService);
147
+ //# sourceMappingURL=rate-limiter.service.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"rate-limiter.service.js","sourceRoot":"","sources":["../../../src/core/rate-limiting/rate-limiter.service.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAA2E;AAE3E,sDAA8C;AAC9C,0DAAyD;AAMzD,SAAS,eAAe,CAAC,MAAc;IACrC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,oBAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,+BAA+B,MAAM,0BAA0B,EAAE;YACjG,SAAS,EAAE,aAAa;YACxB,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,OAAO,EAAE,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAEtB,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QAC/B,oBAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,8BAA8B,KAAK,0BAA0B,EAAE;YAC/F,SAAS,EAAE,aAAa;YACxB,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAA2B;QAC1C,CAAC,EAAE,IAAI;QACP,CAAC,EAAE,EAAE,GAAG,IAAI;QACZ,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI;QACjB,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI;KACvB,CAAC;IAEF,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,oBAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,6BAA6B,IAAI,0BAA0B,EAAE;YAC7F,SAAS,EAAE,aAAa;YACxB,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,OAAO,KAAK,GAAG,UAAU,CAAC;AAC5B,CAAC;AAMM,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;;IACrB,YAAY,GAAG,IAAI,GAAG,EAAuC,CAAC;IAC9D,eAAe,CAAkB;IAEjC,MAAM,CAAU,oBAAoB,GAAG,KAAK,CAAC;IAE7C,MAAM,CAAU,mBAAmB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IAM5D,YAAY;QAEV,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,EAAE,oBAAkB,CAAC,mBAAmB,CAAC,CAAC;IAC7C,CAAC;IAMD,eAAe;QACb,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACnC,CAAC;IACH,CAAC;IAKD,cAAc,CACZ,QAAgB,EAChB,UAAkB,EAClB,MAAuB;QAEvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAGvB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAC7C,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;YAGrD,IAAI,WAAW,CAAC,IAAI,IAAI,oBAAkB,CAAC,oBAAoB,EAAE,CAAC;gBAEhE,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC;gBAClD,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;gBACrE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAEvB,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBAEN,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;oBACzE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;oBACjD,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAGD,IAAI,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;gBAEpC,KAAK,GAAG;oBACN,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,GAAG,GAAG,QAAQ;iBAC1B,CAAC;gBACF,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrC,CAAC;YAGD,IAAI,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;gBAC9B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,SAAS,EAAE,CAAC;oBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B,CAAC;YACJ,CAAC;YAGD,KAAK,CAAC,KAAK,EAAE,CAAC;YAEd,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,MAAM,CAAC,GAAG,GAAG,KAAK,CAAC,KAAK;gBACnC,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAIf,oBAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,yBAAyB,EAAE;gBAC5D,SAAS,EAAE,aAAa;gBACxB,QAAQ,EAAE,IAAA,+BAAe,EAAC,QAAQ,CAAC;gBACnC,UAAU,EAAE,IAAA,+BAAe,EAAC,UAAU,CAAC;gBACvC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;YAGH,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;aAC9B,CAAC;QACJ,CAAC;IACH,CAAC;IAKD,OAAO;QACL,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,IAAI,KAAK,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;oBAC1B,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;YACD,IAAI,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,QAAiB;QACrB,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;;AA1IU,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,mBAAU,GAAE;GACA,kBAAkB,CA2I9B"}
@@ -0,0 +1,15 @@
1
+ export type CircuitState = "closed" | "open" | "half-open";
2
+ export interface CircuitBreakerConfig {
3
+ errorThreshold: number;
4
+ timeWindow: number;
5
+ minRequests: number;
6
+ halfOpenTimeout: number;
7
+ }
8
+ export interface CircuitBreakerState {
9
+ state: CircuitState;
10
+ failures: number;
11
+ successes: number;
12
+ lastFailureTime?: number;
13
+ lastSuccessTime?: number;
14
+ nextAttemptTime?: number;
15
+ }
@@ -0,0 +1,3 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ //# sourceMappingURL=circuit-breaker.interface.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"circuit-breaker.interface.js","sourceRoot":"","sources":["../../../src/core/resilience/circuit-breaker.interface.ts"],"names":[],"mappings":""}
@@ -0,0 +1,11 @@
1
+ import { CircuitBreakerConfig, CircuitState } from "./circuit-breaker.interface";
2
+ export declare class CircuitBreakerService {
3
+ private circuits;
4
+ private configs;
5
+ private getCircuit;
6
+ isOpen(toolName: string, config?: CircuitBreakerConfig): boolean;
7
+ recordSuccess(toolName: string): void;
8
+ recordFailure(toolName: string): void;
9
+ getState(toolName: string): CircuitState;
10
+ reset(toolName?: string): void;
11
+ }