easy-mcp-nest 0.3.0 → 0.5.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 (96) hide show
  1. package/README.md +280 -29
  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/config/config-validator.js +36 -14
  7. package/dist/config/config-validator.js.map +1 -1
  8. package/dist/config/mcp-config.interface.d.ts +9 -4
  9. package/dist/core/batch/batch-executor.service.d.ts +21 -0
  10. package/dist/core/batch/batch-executor.service.js +101 -0
  11. package/dist/core/batch/batch-executor.service.js.map +1 -0
  12. package/dist/core/documentation/openapi-generator.service.d.ts +13 -0
  13. package/dist/core/documentation/openapi-generator.service.js +116 -0
  14. package/dist/core/documentation/openapi-generator.service.js.map +1 -0
  15. package/dist/core/errors/error-handler.interface.d.ts +11 -0
  16. package/dist/core/errors/error-handler.interface.js +3 -0
  17. package/dist/core/errors/error-handler.interface.js.map +1 -0
  18. package/dist/core/health/health-check.service.d.ts +33 -0
  19. package/dist/core/health/health-check.service.js +71 -0
  20. package/dist/core/health/health-check.service.js.map +1 -0
  21. package/dist/core/mcp-server/mcp-server.service.d.ts +14 -1
  22. package/dist/core/mcp-server/mcp-server.service.js +198 -4
  23. package/dist/core/mcp-server/mcp-server.service.js.map +1 -1
  24. package/dist/core/middleware/middleware-executor.service.d.ts +8 -0
  25. package/dist/core/middleware/middleware-executor.service.js +46 -0
  26. package/dist/core/middleware/middleware-executor.service.js.map +1 -0
  27. package/dist/core/middleware/middleware.interface.d.ts +6 -0
  28. package/dist/core/middleware/middleware.interface.js +3 -0
  29. package/dist/core/middleware/middleware.interface.js.map +1 -0
  30. package/dist/core/observability/metrics.service.d.ts +29 -0
  31. package/dist/core/observability/metrics.service.js +124 -0
  32. package/dist/core/observability/metrics.service.js.map +1 -0
  33. package/dist/core/observability/tracing.service.d.ts +12 -0
  34. package/dist/core/observability/tracing.service.js +88 -0
  35. package/dist/core/observability/tracing.service.js.map +1 -0
  36. package/dist/core/progress/progress-notifier.service.d.ts +16 -0
  37. package/dist/core/progress/progress-notifier.service.js +96 -0
  38. package/dist/core/progress/progress-notifier.service.js.map +1 -0
  39. package/dist/core/rate-limiting/rate-limit.interface.d.ts +11 -0
  40. package/dist/core/rate-limiting/rate-limit.interface.js +3 -0
  41. package/dist/core/rate-limiting/rate-limit.interface.js.map +1 -0
  42. package/dist/core/rate-limiting/rate-limiter.service.d.ts +13 -0
  43. package/dist/core/rate-limiting/rate-limiter.service.js +147 -0
  44. package/dist/core/rate-limiting/rate-limiter.service.js.map +1 -0
  45. package/dist/core/resilience/circuit-breaker.interface.d.ts +15 -0
  46. package/dist/core/resilience/circuit-breaker.interface.js +3 -0
  47. package/dist/core/resilience/circuit-breaker.interface.js.map +1 -0
  48. package/dist/core/resilience/circuit-breaker.service.d.ts +11 -0
  49. package/dist/core/resilience/circuit-breaker.service.js +117 -0
  50. package/dist/core/resilience/circuit-breaker.service.js.map +1 -0
  51. package/dist/core/resilience/retry.service.d.ts +6 -0
  52. package/dist/core/resilience/retry.service.js +66 -0
  53. package/dist/core/resilience/retry.service.js.map +1 -0
  54. package/dist/core/utils/decorator-scanner.d.ts +4 -0
  55. package/dist/core/utils/decorator-scanner.js +112 -0
  56. package/dist/core/utils/decorator-scanner.js.map +1 -0
  57. package/dist/core/utils/sanitize.util.d.ts +1 -0
  58. package/dist/core/utils/sanitize.util.js +19 -0
  59. package/dist/core/utils/sanitize.util.js.map +1 -1
  60. package/dist/decorators/index.d.ts +3 -0
  61. package/dist/decorators/index.js +12 -1
  62. package/dist/decorators/index.js.map +1 -1
  63. package/dist/decorators/mcp-error-handler.decorator.d.ts +4 -0
  64. package/dist/decorators/mcp-error-handler.decorator.js +15 -0
  65. package/dist/decorators/mcp-error-handler.decorator.js.map +1 -0
  66. package/dist/decorators/mcp-middleware.decorator.d.ts +4 -0
  67. package/dist/decorators/mcp-middleware.decorator.js +17 -0
  68. package/dist/decorators/mcp-middleware.decorator.js.map +1 -0
  69. package/dist/decorators/mcp-tool.decorator.d.ts +28 -0
  70. package/dist/decorators/mcp-tool.decorator.js +36 -0
  71. package/dist/decorators/mcp-tool.decorator.js.map +1 -0
  72. package/dist/index.d.ts +4 -4
  73. package/dist/index.js +5 -1
  74. package/dist/index.js.map +1 -1
  75. package/dist/interface/interface.interface.d.ts +2 -0
  76. package/dist/interface/mcp-protocol.interface.d.ts +14 -0
  77. package/dist/interface/mcp-protocol.interface.js.map +1 -1
  78. package/dist/resources/resource-registry.service.d.ts +3 -2
  79. package/dist/resources/resource-registry.service.js +7 -3
  80. package/dist/resources/resource-registry.service.js.map +1 -1
  81. package/dist/resources/resource.interface.d.ts +2 -1
  82. package/dist/standalone/core-services.d.ts +5 -0
  83. package/dist/standalone/core-services.js +18 -2
  84. package/dist/standalone/core-services.js.map +1 -1
  85. package/dist/tooling/tool-registry/tool-registry.service.d.ts +7 -1
  86. package/dist/tooling/tool-registry/tool-registry.service.js +54 -7
  87. package/dist/tooling/tool-registry/tool-registry.service.js.map +1 -1
  88. package/dist/tooling/tool.interface.d.ts +7 -1
  89. package/dist/tsconfig.build.tsbuildinfo +1 -1
  90. package/dist/validation/index.d.ts +1 -1
  91. package/dist/validation/index.js +2 -1
  92. package/dist/validation/index.js.map +1 -1
  93. package/dist/validation/zod-integration.d.ts +1 -0
  94. package/dist/validation/zod-integration.js +14 -0
  95. package/dist/validation/zod-integration.js.map +1 -1
  96. package/package.json +1 -1
@@ -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
+ const 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,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC;QAGnE,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
+ }
@@ -0,0 +1,117 @@
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.CircuitBreakerService = void 0;
10
+ const common_1 = require("@nestjs/common");
11
+ const logger_util_1 = require("../utils/logger.util");
12
+ const DEFAULT_CONFIG = {
13
+ errorThreshold: 0.5,
14
+ timeWindow: 60000,
15
+ minRequests: 10,
16
+ halfOpenTimeout: 30000,
17
+ };
18
+ let CircuitBreakerService = class CircuitBreakerService {
19
+ circuits = new Map();
20
+ configs = new Map();
21
+ getCircuit(toolName, config) {
22
+ if (!this.circuits.has(toolName)) {
23
+ this.circuits.set(toolName, {
24
+ state: "closed",
25
+ failures: 0,
26
+ successes: 0,
27
+ });
28
+ this.configs.set(toolName, config || DEFAULT_CONFIG);
29
+ }
30
+ return this.circuits.get(toolName);
31
+ }
32
+ isOpen(toolName, config) {
33
+ const circuit = this.getCircuit(toolName, config);
34
+ const now = Date.now();
35
+ if (circuit.state === "half-open" && circuit.nextAttemptTime && now < circuit.nextAttemptTime) {
36
+ return false;
37
+ }
38
+ if (circuit.state === "open") {
39
+ const cfg = this.configs.get(toolName) || config || DEFAULT_CONFIG;
40
+ if (circuit.nextAttemptTime && now >= circuit.nextAttemptTime) {
41
+ circuit.state = "half-open";
42
+ circuit.failures = 0;
43
+ circuit.successes = 0;
44
+ logger_util_1.logger.info("CircuitBreakerService", `Circuit half-open for ${toolName}`, {
45
+ component: "Resilience",
46
+ toolName,
47
+ });
48
+ return false;
49
+ }
50
+ return true;
51
+ }
52
+ return false;
53
+ }
54
+ recordSuccess(toolName) {
55
+ const circuit = this.getCircuit(toolName);
56
+ circuit.successes++;
57
+ circuit.lastSuccessTime = Date.now();
58
+ if (circuit.state === "half-open") {
59
+ circuit.state = "closed";
60
+ circuit.failures = 0;
61
+ circuit.successes = 0;
62
+ circuit.nextAttemptTime = undefined;
63
+ logger_util_1.logger.info("CircuitBreakerService", `Circuit closed for ${toolName}`, {
64
+ component: "Resilience",
65
+ toolName,
66
+ });
67
+ }
68
+ }
69
+ recordFailure(toolName) {
70
+ const circuit = this.getCircuit(toolName);
71
+ const config = this.configs.get(toolName) || DEFAULT_CONFIG;
72
+ circuit.failures++;
73
+ circuit.lastFailureTime = Date.now();
74
+ const total = circuit.failures + circuit.successes;
75
+ const errorRate = circuit.failures / total;
76
+ if (circuit.state === "closed" &&
77
+ total >= config.minRequests &&
78
+ errorRate >= config.errorThreshold) {
79
+ circuit.state = "open";
80
+ circuit.nextAttemptTime = Date.now() + config.halfOpenTimeout;
81
+ logger_util_1.logger.warn("CircuitBreakerService", `Circuit opened for ${toolName}`, {
82
+ component: "Resilience",
83
+ toolName,
84
+ errorRate,
85
+ failures: circuit.failures,
86
+ total,
87
+ });
88
+ }
89
+ else if (circuit.state === "half-open") {
90
+ circuit.state = "open";
91
+ circuit.nextAttemptTime = Date.now() + config.halfOpenTimeout;
92
+ logger_util_1.logger.warn("CircuitBreakerService", `Circuit re-opened for ${toolName}`, {
93
+ component: "Resilience",
94
+ toolName,
95
+ });
96
+ }
97
+ }
98
+ getState(toolName) {
99
+ const circuit = this.getCircuit(toolName);
100
+ return circuit.state;
101
+ }
102
+ reset(toolName) {
103
+ if (toolName) {
104
+ this.circuits.delete(toolName);
105
+ this.configs.delete(toolName);
106
+ }
107
+ else {
108
+ this.circuits.clear();
109
+ this.configs.clear();
110
+ }
111
+ }
112
+ };
113
+ exports.CircuitBreakerService = CircuitBreakerService;
114
+ exports.CircuitBreakerService = CircuitBreakerService = __decorate([
115
+ (0, common_1.Injectable)()
116
+ ], CircuitBreakerService);
117
+ //# sourceMappingURL=circuit-breaker.service.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"circuit-breaker.service.js","sourceRoot":"","sources":["../../../src/core/resilience/circuit-breaker.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAM5C,sDAA8C;AAK9C,MAAM,cAAc,GAAyB;IAC3C,cAAc,EAAE,GAAG;IACnB,UAAU,EAAE,KAAK;IACjB,WAAW,EAAE,EAAE;IACf,eAAe,EAAE,KAAK;CACvB,CAAC;AAMK,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IACxB,QAAQ,GAAG,IAAI,GAAG,EAA+B,CAAC;IAClD,OAAO,GAAG,IAAI,GAAG,EAAgC,CAAC;IAKlD,UAAU,CAAC,QAAgB,EAAE,MAA6B;QAChE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE;gBAC1B,KAAK,EAAE,QAAQ;gBACf,QAAQ,EAAE,CAAC;gBACX,SAAS,EAAE,CAAC;aACb,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,IAAI,cAAc,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;IACtC,CAAC;IAKD,MAAM,CAAC,QAAgB,EAAE,MAA6B;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAGvB,IAAI,OAAO,CAAC,KAAK,KAAK,WAAW,IAAI,OAAO,CAAC,eAAe,IAAI,GAAG,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;YAC9F,OAAO,KAAK,CAAC;QACf,CAAC;QAGD,IAAI,OAAO,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;YAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,MAAM,IAAI,cAAc,CAAC;YACnE,IAAI,OAAO,CAAC,eAAe,IAAI,GAAG,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAC9D,OAAO,CAAC,KAAK,GAAG,WAAW,CAAC;gBAC5B,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;gBACrB,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;gBACtB,oBAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,yBAAyB,QAAQ,EAAE,EAAE;oBACxE,SAAS,EAAE,YAAY;oBACvB,QAAQ;iBACT,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAKD,aAAa,CAAC,QAAgB;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1C,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAGrC,IAAI,OAAO,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;YAClC,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC;YACzB,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;YACrB,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;YACtB,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;YACpC,oBAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,sBAAsB,QAAQ,EAAE,EAAE;gBACrE,SAAS,EAAE,YAAY;gBACvB,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAKD,aAAa,CAAC,QAAgB;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;QAC5D,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAErC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;QACnD,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,GAAG,KAAK,CAAC;QAG3C,IACE,OAAO,CAAC,KAAK,KAAK,QAAQ;YAC1B,KAAK,IAAI,MAAM,CAAC,WAAW;YAC3B,SAAS,IAAI,MAAM,CAAC,cAAc,EAClC,CAAC;YACD,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;YACvB,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,eAAe,CAAC;YAC9D,oBAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,sBAAsB,QAAQ,EAAE,EAAE;gBACrE,SAAS,EAAE,YAAY;gBACvB,QAAQ;gBACR,SAAS;gBACT,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,KAAK;aACN,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,OAAO,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;YAEzC,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;YACvB,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,eAAe,CAAC;YAC9D,oBAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,yBAAyB,QAAQ,EAAE,EAAE;gBACxE,SAAS,EAAE,YAAY;gBACvB,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAKD,QAAQ,CAAC,QAAgB;QACvB,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1C,OAAO,OAAO,CAAC,KAAK,CAAC;IACvB,CAAC;IAKD,KAAK,CAAC,QAAiB;QACrB,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;CACF,CAAA;AAjIY,sDAAqB;gCAArB,qBAAqB;IADjC,IAAA,mBAAU,GAAE;GACA,qBAAqB,CAiIjC"}
@@ -0,0 +1,6 @@
1
+ import { RetryConfig } from "../../decorators/mcp-tool.decorator";
2
+ export declare class RetryService {
3
+ executeWithRetry<T>(fn: () => Promise<T>, config: RetryConfig): Promise<T>;
4
+ private calculateDelay;
5
+ private sleep;
6
+ }