@lenne.tech/nest-server 11.32.3 → 11.33.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.
- package/.claude/rules/architecture.md +16 -0
- package/.claude/rules/configurable-features.md +11 -3
- package/.claude/rules/testing.md +101 -1
- package/CLAUDE.md +5 -0
- package/FRAMEWORK-API.md +58 -3
- package/bin/migrate.js +13 -3
- package/dist/core/common/helpers/content-disposition.helper.d.ts +1 -0
- package/dist/core/common/helpers/content-disposition.helper.js +14 -0
- package/dist/core/common/helpers/content-disposition.helper.js.map +1 -0
- package/dist/core/common/helpers/file.helper.d.ts +21 -2
- package/dist/core/common/helpers/file.helper.js +69 -13
- package/dist/core/common/helpers/file.helper.js.map +1 -1
- package/dist/core/common/helpers/graceful-shutdown.helper.d.ts +2 -0
- package/dist/core/common/helpers/graceful-shutdown.helper.js +55 -0
- package/dist/core/common/helpers/graceful-shutdown.helper.js.map +1 -0
- package/dist/core/common/helpers/project-name.helper.d.ts +2 -0
- package/dist/core/common/helpers/project-name.helper.js +38 -0
- package/dist/core/common/helpers/project-name.helper.js.map +1 -0
- package/dist/core/common/interfaces/cron-job-config.interface.d.ts +1 -0
- package/dist/core/common/interfaces/server-options.interface.d.ts +39 -0
- package/dist/core/common/services/core-cron-jobs.initializer.d.ts +10 -0
- package/dist/core/common/services/core-cron-jobs.initializer.js +46 -0
- package/dist/core/common/services/core-cron-jobs.initializer.js.map +1 -0
- package/dist/core/common/services/core-cron-jobs.registry.d.ts +8 -0
- package/dist/core/common/services/core-cron-jobs.registry.js +12 -0
- package/dist/core/common/services/core-cron-jobs.registry.js.map +1 -0
- package/dist/core/common/services/core-cron-jobs.service.d.ts +37 -7
- package/dist/core/common/services/core-cron-jobs.service.js +278 -27
- package/dist/core/common/services/core-cron-jobs.service.js.map +1 -1
- package/dist/core/common/services/core-redis-pubsub.d.ts +20 -0
- package/dist/core/common/services/core-redis-pubsub.js +112 -0
- package/dist/core/common/services/core-redis-pubsub.js.map +1 -0
- package/dist/core/common/services/core-redis.service.d.ts +25 -0
- package/dist/core/common/services/core-redis.service.js +154 -0
- package/dist/core/common/services/core-redis.service.js.map +1 -0
- package/dist/core/common/services/core-s3.service.d.ts +51 -0
- package/dist/core/common/services/core-s3.service.js +261 -0
- package/dist/core/common/services/core-s3.service.js.map +1 -0
- package/dist/core/common/services/core-trust-proxy.initializer.d.ts +16 -0
- package/dist/core/common/services/core-trust-proxy.initializer.js +77 -0
- package/dist/core/common/services/core-trust-proxy.initializer.js.map +1 -0
- package/dist/core/common/services/rate-limit-store.d.ts +61 -0
- package/dist/core/common/services/rate-limit-store.js +235 -0
- package/dist/core/common/services/rate-limit-store.js.map +1 -0
- package/dist/core/modules/ai/core-ai-mcp.controller.d.ts +19 -3
- package/dist/core/modules/ai/core-ai-mcp.controller.js +148 -18
- package/dist/core/modules/ai/core-ai-mcp.controller.js.map +1 -1
- package/dist/core/modules/ai/services/core-ai.service.d.ts +7 -2
- package/dist/core/modules/ai/services/core-ai.service.js +24 -20
- package/dist/core/modules/ai/services/core-ai.service.js.map +1 -1
- package/dist/core/modules/auth/core-auth.module.js +4 -1
- package/dist/core/modules/auth/core-auth.module.js.map +1 -1
- package/dist/core/modules/auth/guards/legacy-auth-rate-limit.guard.d.ts +1 -1
- package/dist/core/modules/auth/guards/legacy-auth-rate-limit.guard.js +3 -14
- package/dist/core/modules/auth/guards/legacy-auth-rate-limit.guard.js.map +1 -1
- package/dist/core/modules/auth/services/legacy-auth-rate-limiter.service.d.ts +12 -10
- package/dist/core/modules/auth/services/legacy-auth-rate-limiter.service.js +30 -59
- package/dist/core/modules/auth/services/legacy-auth-rate-limiter.service.js.map +1 -1
- package/dist/core/modules/better-auth/core-better-auth-email-verification.service.d.ts +8 -2
- package/dist/core/modules/better-auth/core-better-auth-email-verification.service.js +118 -45
- package/dist/core/modules/better-auth/core-better-auth-email-verification.service.js.map +1 -1
- package/dist/core/modules/better-auth/core-better-auth-rate-limit.middleware.d.ts +1 -1
- package/dist/core/modules/better-auth/core-better-auth-rate-limit.middleware.js +2 -11
- package/dist/core/modules/better-auth/core-better-auth-rate-limit.middleware.js.map +1 -1
- package/dist/core/modules/better-auth/core-better-auth-rate-limiter.service.d.ts +12 -10
- package/dist/core/modules/better-auth/core-better-auth-rate-limiter.service.js +30 -85
- package/dist/core/modules/better-auth/core-better-auth-rate-limiter.service.js.map +1 -1
- package/dist/core/modules/better-auth/core-better-auth-user.mapper.js +0 -1
- package/dist/core/modules/better-auth/core-better-auth-user.mapper.js.map +1 -1
- package/dist/core/modules/file/core-file.controller.d.ts +15 -3
- package/dist/core/modules/file/core-file.controller.js +90 -20
- package/dist/core/modules/file/core-file.controller.js.map +1 -1
- package/dist/core/modules/file/core-file.resolver.d.ts +4 -4
- package/dist/core/modules/file/core-file.resolver.js +23 -16
- package/dist/core/modules/file/core-file.resolver.js.map +1 -1
- package/dist/core/modules/file/core-file.service.d.ts +41 -7
- package/dist/core/modules/file/core-file.service.js +227 -7
- package/dist/core/modules/file/core-file.service.js.map +1 -1
- package/dist/core/modules/file/file-metadata.helper.d.ts +16 -0
- package/dist/core/modules/file/file-metadata.helper.js +33 -0
- package/dist/core/modules/file/file-metadata.helper.js.map +1 -0
- package/dist/core/modules/file/file-roles.helper.d.ts +4 -0
- package/dist/core/modules/file/file-roles.helper.js +53 -0
- package/dist/core/modules/file/file-roles.helper.js.map +1 -0
- package/dist/core/modules/file/file-storage.helper.d.ts +13 -0
- package/dist/core/modules/file/file-storage.helper.js +65 -0
- package/dist/core/modules/file/file-storage.helper.js.map +1 -0
- package/dist/core/modules/file/filesystem-file.helper.d.ts +22 -0
- package/dist/core/modules/file/filesystem-file.helper.js +77 -0
- package/dist/core/modules/file/filesystem-file.helper.js.map +1 -0
- package/dist/core/modules/file/interfaces/file-service-options.interface.d.ts +1 -0
- package/dist/core/modules/file/interfaces/file-upload.interface.d.ts +7 -1
- package/dist/core/modules/file/s3-file.helper.d.ts +37 -0
- package/dist/core/modules/file/s3-file.helper.js +81 -0
- package/dist/core/modules/file/s3-file.helper.js.map +1 -0
- package/dist/core/modules/hub/core-hub.controller.d.ts +8 -8
- package/dist/core/modules/hub/core-hub.controller.js +24 -22
- package/dist/core/modules/hub/core-hub.controller.js.map +1 -1
- package/dist/core/modules/hub/helpers/hub-mask.helper.js +2 -2
- package/dist/core/modules/hub/helpers/hub-mask.helper.js.map +1 -1
- package/dist/core/modules/hub/hub-buffer.d.ts +23 -0
- package/dist/core/modules/hub/hub-buffer.js +101 -0
- package/dist/core/modules/hub/hub-buffer.js.map +1 -0
- package/dist/core/modules/hub/services/core-hub-mailbox.service.d.ts +7 -5
- package/dist/core/modules/hub/services/core-hub-mailbox.service.js +14 -10
- package/dist/core/modules/hub/services/core-hub-mailbox.service.js.map +1 -1
- package/dist/core/modules/hub/services/hub-log-buffer.service.d.ts +5 -2
- package/dist/core/modules/hub/services/hub-log-buffer.service.js +17 -8
- package/dist/core/modules/hub/services/hub-log-buffer.service.js.map +1 -1
- package/dist/core/modules/hub/services/hub-query-profiler.service.d.ts +4 -2
- package/dist/core/modules/hub/services/hub-query-profiler.service.js +13 -7
- package/dist/core/modules/hub/services/hub-query-profiler.service.js.map +1 -1
- package/dist/core/modules/hub/services/hub-trace-buffer.service.d.ts +4 -2
- package/dist/core/modules/hub/services/hub-trace-buffer.service.js +13 -8
- package/dist/core/modules/hub/services/hub-trace-buffer.service.js.map +1 -1
- package/dist/core/modules/migrate/cli/migrate-cli.d.ts +3 -1
- package/dist/core/modules/migrate/cli/migrate-cli.js +29 -4
- package/dist/core/modules/migrate/cli/migrate-cli.js.map +1 -1
- package/dist/core/modules/migrate/helpers/migration.helper.d.ts +3 -6
- package/dist/core/modules/migrate/helpers/migration.helper.js +55 -8
- package/dist/core/modules/migrate/helpers/migration.helper.js.map +1 -1
- package/dist/core/modules/migrate/migration-runner.d.ts +1 -0
- package/dist/core/modules/migrate/migration-runner.js +4 -0
- package/dist/core/modules/migrate/migration-runner.js.map +1 -1
- package/dist/core/modules/migrate/mongo-state-store.d.ts +1 -0
- package/dist/core/modules/migrate/mongo-state-store.js +98 -9
- package/dist/core/modules/migrate/mongo-state-store.js.map +1 -1
- package/dist/core/modules/system-setup/core-system-setup.service.d.ts +2 -0
- package/dist/core/modules/system-setup/core-system-setup.service.js +31 -1
- package/dist/core/modules/system-setup/core-system-setup.service.js.map +1 -1
- package/dist/core/modules/tenant/core-tenant.guard.d.ts +17 -3
- package/dist/core/modules/tenant/core-tenant.guard.js +92 -15
- package/dist/core/modules/tenant/core-tenant.guard.js.map +1 -1
- package/dist/core/modules/tenant/core-tenant.module.js +3 -4
- package/dist/core/modules/tenant/core-tenant.module.js.map +1 -1
- package/dist/core/modules/tus/core-tus.controller.d.ts +2 -0
- package/dist/core/modules/tus/core-tus.controller.js +29 -3
- package/dist/core/modules/tus/core-tus.controller.js.map +1 -1
- package/dist/core/modules/tus/core-tus.service.d.ts +24 -1
- package/dist/core/modules/tus/core-tus.service.js +182 -28
- package/dist/core/modules/tus/core-tus.service.js.map +1 -1
- package/dist/core/modules/tus/interfaces/tus-config.interface.js +3 -0
- package/dist/core/modules/tus/interfaces/tus-config.interface.js.map +1 -1
- package/dist/core/modules/tus/tus-redis-locker.d.ts +25 -0
- package/dist/core/modules/tus/tus-redis-locker.js +87 -0
- package/dist/core/modules/tus/tus-redis-locker.js.map +1 -0
- package/dist/core/modules/tus/tus.module.d.ts +1 -0
- package/dist/core/modules/tus/tus.module.js +30 -4
- package/dist/core/modules/tus/tus.module.js.map +1 -1
- package/dist/core.module.js +18 -1
- package/dist/core.module.js.map +1 -1
- package/dist/index.d.ts +10 -0
- package/dist/index.js +10 -0
- package/dist/index.js.map +1 -1
- package/dist/main.js +2 -1
- package/dist/main.js.map +1 -1
- package/dist/server/modules/file/file.controller.js.map +1 -1
- package/dist/server/modules/file/file.resolver.js +1 -12
- package/dist/server/modules/file/file.resolver.js.map +1 -1
- package/dist/server/modules/file/file.service.d.ts +5 -1
- package/dist/server/modules/file/file.service.js +12 -3
- package/dist/server/modules/file/file.service.js.map +1 -1
- package/dist/server/modules/user/avatar.controller.d.ts +5 -1
- package/dist/server/modules/user/avatar.controller.js +26 -10
- package/dist/server/modules/user/avatar.controller.js.map +1 -1
- package/dist/server/modules/user/user.model.js +3 -2
- package/dist/server/modules/user/user.model.js.map +1 -1
- package/dist/server/modules/user/user.module.js +6 -2
- package/dist/server/modules/user/user.module.js.map +1 -1
- package/dist/server/modules/user/user.service.d.ts +1 -1
- package/dist/server/modules/user/user.service.js +6 -13
- package/dist/server/modules/user/user.service.js.map +1 -1
- package/dist/tsconfig.build.tsbuildinfo +1 -1
- package/docs/REQUEST-LIFECYCLE.md +57 -8
- package/docs/security-overrides.md +30 -5
- package/migration-guides/11.32.3-to-11.32.4.md +335 -0
- package/migration-guides/11.32.x-to-11.33.x.md +1551 -0
- package/package.json +74 -42
- package/src/core/common/helpers/content-disposition.helper.ts +90 -0
- package/src/core/common/helpers/file.helper.ts +201 -20
- package/src/core/common/helpers/graceful-shutdown.helper.ts +116 -0
- package/src/core/common/helpers/project-name.helper.ts +71 -0
- package/src/core/common/interfaces/cron-job-config.interface.ts +16 -0
- package/src/core/common/interfaces/server-options.interface.ts +439 -2
- package/src/core/common/services/core-cron-jobs.initializer.ts +39 -0
- package/src/core/common/services/core-cron-jobs.registry.ts +45 -0
- package/src/core/common/services/core-cron-jobs.service.ts +589 -47
- package/src/core/common/services/core-redis-pubsub.ts +183 -0
- package/src/core/common/services/core-redis.service.ts +249 -0
- package/src/core/common/services/core-s3.service.ts +465 -0
- package/src/core/common/services/core-trust-proxy.initializer.ts +112 -0
- package/src/core/common/services/rate-limit-store.ts +492 -0
- package/src/core/modules/ai/README.md +25 -0
- package/src/core/modules/ai/core-ai-mcp.controller.ts +253 -22
- package/src/core/modules/ai/services/core-ai.service.ts +33 -24
- package/src/core/modules/auth/core-auth.module.ts +7 -1
- package/src/core/modules/auth/guards/legacy-auth-rate-limit.guard.ts +12 -21
- package/src/core/modules/auth/services/legacy-auth-rate-limiter.service.ts +49 -93
- package/src/core/modules/better-auth/core-better-auth-email-verification.service.ts +202 -66
- package/src/core/modules/better-auth/core-better-auth-rate-limit.middleware.ts +11 -18
- package/src/core/modules/better-auth/core-better-auth-rate-limiter.service.ts +51 -132
- package/src/core/modules/better-auth/core-better-auth-user.mapper.ts +7 -1
- package/src/core/modules/file/INTEGRATION-CHECKLIST.md +108 -0
- package/src/core/modules/file/README.md +324 -41
- package/src/core/modules/file/core-file.controller.ts +267 -19
- package/src/core/modules/file/core-file.resolver.ts +46 -12
- package/src/core/modules/file/core-file.service.ts +482 -16
- package/src/core/modules/file/file-metadata.helper.ts +142 -0
- package/src/core/modules/file/file-roles.helper.ts +110 -0
- package/src/core/modules/file/file-storage.helper.ts +163 -0
- package/src/core/modules/file/filesystem-file.helper.ts +184 -0
- package/src/core/modules/file/interfaces/file-service-options.interface.ts +14 -0
- package/src/core/modules/file/interfaces/file-upload.interface.ts +17 -1
- package/src/core/modules/file/s3-file.helper.ts +209 -0
- package/src/core/modules/hub/README.md +48 -0
- package/src/core/modules/hub/core-hub.controller.ts +18 -16
- package/src/core/modules/hub/helpers/hub-mask.helper.ts +21 -4
- package/src/core/modules/hub/hub-buffer.ts +198 -0
- package/src/core/modules/hub/services/core-hub-mailbox.service.ts +22 -12
- package/src/core/modules/hub/services/hub-log-buffer.service.ts +24 -8
- package/src/core/modules/hub/services/hub-query-profiler.service.ts +9 -6
- package/src/core/modules/hub/services/hub-trace-buffer.service.ts +14 -9
- package/src/core/modules/migrate/README.md +68 -6
- package/src/core/modules/migrate/cli/migrate-cli.ts +69 -6
- package/src/core/modules/migrate/helpers/migration.helper.ts +144 -10
- package/src/core/modules/migrate/migration-runner.ts +10 -1
- package/src/core/modules/migrate/mongo-state-store.ts +179 -11
- package/src/core/modules/system-setup/README.md +7 -2
- package/src/core/modules/system-setup/core-system-setup.service.ts +95 -1
- package/src/core/modules/tenant/README.md +21 -1
- package/src/core/modules/tenant/core-tenant.guard.ts +135 -14
- package/src/core/modules/tenant/core-tenant.module.ts +8 -4
- package/src/core/modules/tus/INTEGRATION-CHECKLIST.md +93 -21
- package/src/core/modules/tus/README.md +35 -2
- package/src/core/modules/tus/core-tus.controller.ts +64 -12
- package/src/core/modules/tus/core-tus.service.ts +358 -34
- package/src/core/modules/tus/interfaces/tus-config.interface.ts +9 -0
- package/src/core/modules/tus/tus-redis-locker.ts +139 -0
- package/src/core/modules/tus/tus.module.ts +78 -6
- package/src/core.module.ts +26 -1
- package/src/index.ts +10 -0
- package/src/main.ts +7 -1
- package/src/server/modules/file/file.controller.ts +7 -3
- package/src/server/modules/file/file.resolver.ts +5 -17
- package/src/server/modules/file/file.service.ts +42 -3
- package/src/server/modules/user/avatar.controller.ts +39 -9
- package/src/server/modules/user/user.model.ts +9 -3
- package/src/server/modules/user/user.module.ts +12 -3
- package/src/server/modules/user/user.service.ts +20 -17
- package/src/test/README.md +7 -2
- package/src/core/common/helpers/logging.helper.spec.ts +0 -61
- package/src/core/common/helpers/process-diagnostics.helper.spec.ts +0 -310
- package/src/core/common/services/brevo.service.spec.ts +0 -266
- package/src/core/modules/hub/core-hub.module.spec.ts +0 -108
- package/src/core/modules/hub/helpers/hub-command-shape.helper.spec.ts +0 -48
- package/src/core/modules/hub/helpers/hub-mask.helper.spec.ts +0 -67
- package/src/core/modules/hub/helpers/hub-mermaid.helper.spec.ts +0 -61
- package/src/core/modules/hub/helpers/hub-shell.helper.spec.ts +0 -106
- package/src/core/modules/hub/hub-config.helper.spec.ts +0 -108
- package/src/core/modules/hub/hub-ring-buffer.spec.ts +0 -95
- package/src/core/modules/hub/services/core-hub-mailbox.service.spec.ts +0 -116
- package/src/core/modules/hub/services/hub-trace-buffer.service.spec.ts +0 -112
|
@@ -0,0 +1,235 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RedisRateLimitStore = exports.InMemoryRateLimitStore = void 0;
|
|
4
|
+
exports.rateLimitKey = rateLimitKey;
|
|
5
|
+
exports.rateLimitKeyPrefix = rateLimitKeyPrefix;
|
|
6
|
+
const common_1 = require("@nestjs/common");
|
|
7
|
+
function escapeGlob(value) {
|
|
8
|
+
return value.replace(/[?[\]*\\^]/g, (char) => `\\${char}`);
|
|
9
|
+
}
|
|
10
|
+
function rateLimitKey(...parts) {
|
|
11
|
+
return parts.map((part) => part.replace(/\\/g, '\\\\').replace(/:/g, '\\:')).join(':');
|
|
12
|
+
}
|
|
13
|
+
function rateLimitKeyPrefix(...parts) {
|
|
14
|
+
return `${rateLimitKey(...parts)}:`;
|
|
15
|
+
}
|
|
16
|
+
function namespaceLabel(namespace) {
|
|
17
|
+
return `namespace ${namespace}`;
|
|
18
|
+
}
|
|
19
|
+
const OVERFLOW_BUCKETS = 64;
|
|
20
|
+
function overflowSlot(key) {
|
|
21
|
+
let hash = 0;
|
|
22
|
+
for (let i = 0; i < key.length; i++) {
|
|
23
|
+
hash = (hash * 31 + key.charCodeAt(i)) | 0;
|
|
24
|
+
}
|
|
25
|
+
return Math.abs(hash) % OVERFLOW_BUCKETS;
|
|
26
|
+
}
|
|
27
|
+
class InMemoryRateLimitStore {
|
|
28
|
+
maxEntries;
|
|
29
|
+
store = new Map();
|
|
30
|
+
cleanupInterval = null;
|
|
31
|
+
atCapacity = false;
|
|
32
|
+
capacityWarned = false;
|
|
33
|
+
overflow = new Map();
|
|
34
|
+
constructor(maxEntries = 10000) {
|
|
35
|
+
this.maxEntries = maxEntries;
|
|
36
|
+
this.cleanupInterval = setInterval(() => this.removeExpired(), 5 * 60 * 1000);
|
|
37
|
+
this.cleanupInterval.unref?.();
|
|
38
|
+
}
|
|
39
|
+
async clear() {
|
|
40
|
+
this.store.clear();
|
|
41
|
+
this.overflow.clear();
|
|
42
|
+
this.atCapacity = false;
|
|
43
|
+
this.capacityWarned = false;
|
|
44
|
+
}
|
|
45
|
+
async hit(key, windowSeconds) {
|
|
46
|
+
const now = Date.now();
|
|
47
|
+
let entry = this.store.get(key);
|
|
48
|
+
if (!entry || now >= entry.resetTime) {
|
|
49
|
+
if (!entry && this.store.size >= this.maxEntries) {
|
|
50
|
+
this.atCapacity = false;
|
|
51
|
+
this.evictOldest();
|
|
52
|
+
if (this.atCapacity) {
|
|
53
|
+
if (!this.capacityWarned) {
|
|
54
|
+
this.capacityWarned = true;
|
|
55
|
+
console.warn(`Rate limit store is at capacity (${this.maxEntries} live counters). New clients now share ` +
|
|
56
|
+
`${OVERFLOW_BUCKETS} coarse counters until entries expire, so they may be limited earlier than ` +
|
|
57
|
+
'their own traffic warrants. Existing limits are unaffected.');
|
|
58
|
+
}
|
|
59
|
+
return this.hitOverflow(key, windowSeconds, now);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
entry = { count: 1, resetTime: now + windowSeconds * 1000 };
|
|
63
|
+
this.store.set(key, entry);
|
|
64
|
+
return { count: 1, resetIn: windowSeconds };
|
|
65
|
+
}
|
|
66
|
+
entry.count++;
|
|
67
|
+
return { count: entry.count, resetIn: Math.ceil((entry.resetTime - now) / 1000) };
|
|
68
|
+
}
|
|
69
|
+
async resetByPrefix(prefix) {
|
|
70
|
+
for (const key of this.store.keys()) {
|
|
71
|
+
if (key.startsWith(prefix)) {
|
|
72
|
+
this.store.delete(key);
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
hitOverflow(key, windowSeconds, now) {
|
|
77
|
+
const slot = overflowSlot(key);
|
|
78
|
+
const entry = this.overflow.get(slot);
|
|
79
|
+
if (!entry || now >= entry.resetTime) {
|
|
80
|
+
this.overflow.set(slot, { count: 1, resetTime: now + windowSeconds * 1000 });
|
|
81
|
+
return { count: 1, resetIn: windowSeconds };
|
|
82
|
+
}
|
|
83
|
+
entry.count++;
|
|
84
|
+
return { count: entry.count, resetIn: Math.ceil((entry.resetTime - now) / 1000) };
|
|
85
|
+
}
|
|
86
|
+
size() {
|
|
87
|
+
return this.store.size;
|
|
88
|
+
}
|
|
89
|
+
destroy() {
|
|
90
|
+
if (this.cleanupInterval) {
|
|
91
|
+
clearInterval(this.cleanupInterval);
|
|
92
|
+
this.cleanupInterval = null;
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
removeExpired() {
|
|
96
|
+
const now = Date.now();
|
|
97
|
+
let removed = 0;
|
|
98
|
+
for (const [key, entry] of this.store.entries()) {
|
|
99
|
+
if (now >= entry.resetTime) {
|
|
100
|
+
this.store.delete(key);
|
|
101
|
+
removed++;
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
return removed;
|
|
105
|
+
}
|
|
106
|
+
evictOldest() {
|
|
107
|
+
const removed = this.removeExpired();
|
|
108
|
+
if (removed > 0 || this.store.size < this.maxEntries) {
|
|
109
|
+
return;
|
|
110
|
+
}
|
|
111
|
+
this.atCapacity = true;
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
exports.InMemoryRateLimitStore = InMemoryRateLimitStore;
|
|
115
|
+
const HIT_SCRIPT = `
|
|
116
|
+
local window = tonumber(ARGV[1])
|
|
117
|
+
local target = KEYS[1]
|
|
118
|
+
local overflow = 0
|
|
119
|
+
if redis.call('EXISTS', KEYS[1]) == 0 then
|
|
120
|
+
local seen = redis.call('INCR', KEYS[2])
|
|
121
|
+
if seen == 1 then
|
|
122
|
+
redis.call('EXPIRE', KEYS[2], window)
|
|
123
|
+
end
|
|
124
|
+
if seen > tonumber(ARGV[2]) then
|
|
125
|
+
target = KEYS[3]
|
|
126
|
+
overflow = 1
|
|
127
|
+
end
|
|
128
|
+
end
|
|
129
|
+
local c = redis.call('INCR', target)
|
|
130
|
+
if c == 1 then
|
|
131
|
+
redis.call('EXPIRE', target, window)
|
|
132
|
+
end
|
|
133
|
+
local ttl = redis.call('TTL', target)
|
|
134
|
+
if ttl < 0 then
|
|
135
|
+
redis.call('EXPIRE', target, window)
|
|
136
|
+
ttl = window
|
|
137
|
+
end
|
|
138
|
+
return {c, ttl, overflow}
|
|
139
|
+
`;
|
|
140
|
+
const HIT_COMMAND = 'ltRateLimitHit';
|
|
141
|
+
class RedisRateLimitStore {
|
|
142
|
+
redisService;
|
|
143
|
+
namespace;
|
|
144
|
+
maxKeys;
|
|
145
|
+
fallback = new InMemoryRateLimitStore();
|
|
146
|
+
degraded = false;
|
|
147
|
+
logger = new common_1.Logger(RedisRateLimitStore.name);
|
|
148
|
+
capacityWarned = false;
|
|
149
|
+
prepared = new WeakSet();
|
|
150
|
+
constructor(redisService, namespace, maxKeys = 10000) {
|
|
151
|
+
this.redisService = redisService;
|
|
152
|
+
this.namespace = namespace;
|
|
153
|
+
this.maxKeys = maxKeys;
|
|
154
|
+
}
|
|
155
|
+
async clear() {
|
|
156
|
+
await this.fallback.clear();
|
|
157
|
+
this.capacityWarned = false;
|
|
158
|
+
await this.guard(() => this.deleteByPattern(this.redisKeyPattern('')), undefined);
|
|
159
|
+
}
|
|
160
|
+
async hit(key, windowSeconds) {
|
|
161
|
+
return this.guard(async () => {
|
|
162
|
+
const [count, ttl, overflow] = await this.runHit(key, windowSeconds);
|
|
163
|
+
if (this.degraded) {
|
|
164
|
+
this.degraded = false;
|
|
165
|
+
this.logger.log('Redis rate limiting recovered — limits are shared across replicas again');
|
|
166
|
+
}
|
|
167
|
+
if (overflow && !this.capacityWarned) {
|
|
168
|
+
this.capacityWarned = true;
|
|
169
|
+
this.logger.warn(`Rate limit keyspace for ${namespaceLabel(this.namespace)} reached ${this.maxKeys} distinct counters ` +
|
|
170
|
+
`within one window. New clients now share ${OVERFLOW_BUCKETS} coarse counters until the window rolls ` +
|
|
171
|
+
'over, so they may be limited earlier than their own traffic warrants. Existing limits are unaffected.');
|
|
172
|
+
}
|
|
173
|
+
return { count, resetIn: ttl };
|
|
174
|
+
}, () => this.fallback.hit(key, windowSeconds));
|
|
175
|
+
}
|
|
176
|
+
async runHit(key, windowSeconds) {
|
|
177
|
+
const client = this.redisService.getClient();
|
|
178
|
+
const args = [this.redisKey(key), this.cardinalityKey(), this.overflowKey(key), windowSeconds, this.maxKeys];
|
|
179
|
+
if (typeof client.defineCommand === 'function') {
|
|
180
|
+
if (!this.prepared.has(client)) {
|
|
181
|
+
client.defineCommand(HIT_COMMAND, { lua: HIT_SCRIPT, numberOfKeys: 3 });
|
|
182
|
+
this.prepared.add(client);
|
|
183
|
+
}
|
|
184
|
+
return (await client[HIT_COMMAND](...args));
|
|
185
|
+
}
|
|
186
|
+
return (await client.eval(HIT_SCRIPT, 3, ...args));
|
|
187
|
+
}
|
|
188
|
+
async resetByPrefix(prefix) {
|
|
189
|
+
await this.fallback.resetByPrefix(prefix);
|
|
190
|
+
await this.guard(() => this.deleteByPattern(this.redisKeyPattern(prefix)), undefined);
|
|
191
|
+
}
|
|
192
|
+
size() {
|
|
193
|
+
return -1;
|
|
194
|
+
}
|
|
195
|
+
destroy() {
|
|
196
|
+
this.fallback.destroy();
|
|
197
|
+
}
|
|
198
|
+
async guard(operation, onFailure) {
|
|
199
|
+
try {
|
|
200
|
+
return await operation();
|
|
201
|
+
}
|
|
202
|
+
catch (error) {
|
|
203
|
+
if (!this.degraded) {
|
|
204
|
+
this.degraded = true;
|
|
205
|
+
this.logger.error(`Redis rate limiting unavailable (${namespaceLabel(this.namespace)}): ${error instanceof Error ? error.message : 'Unknown error'}. Falling back to per-replica in-memory limits until Redis recovers.`);
|
|
206
|
+
}
|
|
207
|
+
return onFailure ? await onFailure() : undefined;
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
redisKey(key) {
|
|
211
|
+
return this.redisService.key('rate-limit', this.namespace, key);
|
|
212
|
+
}
|
|
213
|
+
redisKeyPattern(prefix) {
|
|
214
|
+
return `${escapeGlob(this.redisKey(prefix))}*`;
|
|
215
|
+
}
|
|
216
|
+
cardinalityKey() {
|
|
217
|
+
return this.redisService.key('rate-limit', this.namespace, '#meta', 'cardinality');
|
|
218
|
+
}
|
|
219
|
+
overflowKey(key) {
|
|
220
|
+
return this.redisService.key('rate-limit', this.namespace, '#overflow', String(overflowSlot(key)));
|
|
221
|
+
}
|
|
222
|
+
async deleteByPattern(pattern) {
|
|
223
|
+
const client = this.redisService.getClient();
|
|
224
|
+
let cursor = '0';
|
|
225
|
+
do {
|
|
226
|
+
const [nextCursor, keys] = await client.scan(cursor, 'MATCH', pattern, 'COUNT', 200);
|
|
227
|
+
if (keys.length) {
|
|
228
|
+
await client.del(...keys);
|
|
229
|
+
}
|
|
230
|
+
cursor = nextCursor;
|
|
231
|
+
} while (cursor !== '0');
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
exports.RedisRateLimitStore = RedisRateLimitStore;
|
|
235
|
+
//# sourceMappingURL=rate-limit-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limit-store.js","sourceRoot":"","sources":["../../../../src/core/common/services/rate-limit-store.ts"],"names":[],"mappings":";;;AAiCA,oCAEC;AAOD,gDAEC;AA5CD,2CAAwC;AAgBxC,SAAS,UAAU,CAAC,KAAa;IAC/B,OAAO,KAAK,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;AAC7D,CAAC;AAeD,SAAgB,YAAY,CAAC,GAAG,KAAe;IAC7C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzF,CAAC;AAOD,SAAgB,kBAAkB,CAAC,GAAG,KAAe;IACnD,OAAO,GAAG,YAAY,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC;AACtC,CAAC;AAKD,SAAS,cAAc,CAAC,SAAiB;IACvC,OAAO,aAAa,SAAS,EAAE,CAAC;AAClC,CAAC;AA2CD,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAG5B,SAAS,YAAY,CAAC,GAAW;IAC/B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,IAAI,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAC7C,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC;AAC3C,CAAC;AAMD,MAAa,sBAAsB;IA0BF;IAzBZ,KAAK,GAAG,IAAI,GAAG,EAAgD,CAAC;IACzE,eAAe,GAA0B,IAAI,CAAC;IAG9C,UAAU,GAAG,KAAK,CAAC;IAGnB,cAAc,GAAG,KAAK,CAAC;IAgBd,QAAQ,GAAG,IAAI,GAAG,EAAgD,CAAC;IAEtF,YAA+B,aAAa,KAAK;QAAlB,eAAU,GAAV,UAAU,CAAQ;QAE/C,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC9E,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QAInB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,aAAqB;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,IAAI,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACrC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACjD,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACpB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;wBACzB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;wBAC3B,OAAO,CAAC,IAAI,CACV,oCAAoC,IAAI,CAAC,UAAU,yCAAyC;4BAC1F,GAAG,gBAAgB,6EAA6E;4BAChG,6DAA6D,CAChE,CAAC;oBACJ,CAAC;oBACD,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,aAAa,EAAE,GAAG,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;YACD,KAAK,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,GAAG,aAAa,GAAG,IAAI,EAAE,CAAC;YAC5D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAC3B,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QAC9C,CAAC;QACD,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC;IACpF,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YACpC,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;IACH,CAAC;IAOS,WAAW,CAAC,GAAW,EAAE,aAAqB,EAAE,GAAW;QACnE,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEtC,IAAI,CAAC,KAAK,IAAI,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACrC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,GAAG,aAAa,GAAG,IAAI,EAAE,CAAC,CAAC;YAC7E,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QAC9C,CAAC;QAED,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC;IACpF,CAAC;IAED,IAAI;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;IAGD,OAAO;QACL,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;IAES,aAAa;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YAChD,IAAI,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACvB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAMS,WAAW;QACnB,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACrC,IAAI,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACrD,OAAO;QACT,CAAC;QAMD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;IACzB,CAAC;CACF;AAvID,wDAuIC;AAkBD,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;;;;;;;;CAwBlB,CAAC;AAGF,MAAM,WAAW,GAAG,gBAAgB,CAAC;AAWrC,MAAa,mBAAmB;IAmCT;IACA;IACA;IA1BF,QAAQ,GAAG,IAAI,sBAAsB,EAAE,CAAC;IAGjD,QAAQ,GAAG,KAAK,CAAC;IAER,MAAM,GAAG,IAAI,eAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAGvD,cAAc,GAAG,KAAK,CAAC;IASd,QAAQ,GAAG,IAAI,OAAO,EAAU,CAAC;IAMpD,YACqB,YAA8B,EAC9B,SAAiB,EACjB,UAAU,KAAK;QAFf,iBAAY,GAAZ,YAAY,CAAkB;QAC9B,cAAS,GAAT,SAAS,CAAQ;QACjB,YAAO,GAAP,OAAO,CAAQ;IACjC,CAAC;IAEJ,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAI5B,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACpF,CAAC;IAkBD,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,aAAqB;QAC1C,OAAO,IAAI,CAAC,KAAK,CACf,KAAK,IAAI,EAAE;YACT,MAAM,CAAC,KAAK,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;YACrE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC;YAC7F,CAAC;YACD,IAAI,QAAQ,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;gBACrC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;gBAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,2BAA2B,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,IAAI,CAAC,OAAO,qBAAqB;oBACpG,4CAA4C,gBAAgB,0CAA0C;oBACtG,uGAAuG,CAC1G,CAAC;YACJ,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;QACjC,CAAC,EACD,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,aAAa,CAAC,CAC5C,CAAC;IACJ,CAAC;IAUS,KAAK,CAAC,MAAM,CAAC,GAAW,EAAE,aAAqB;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAS,CAAC;QACpD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAE7G,IAAI,OAAO,MAAM,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;YAC/C,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/B,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;gBACxE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;YACD,OAAO,CAAC,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,IAAI,CAAC,CAA6B,CAAC;QAC1E,CAAC;QACD,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAA6B,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACxF,CAAC;IAED,IAAI;QACF,OAAO,CAAC,CAAC,CAAC;IACZ,CAAC;IAGD,OAAO;QACL,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAKS,KAAK,CAAC,KAAK,CAAI,SAA2B,EAAE,SAA6C;QACjG,IAAI,CAAC;YACH,OAAO,MAAM,SAAS,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;gBACrB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,oCAAoC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,MAChE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAC3C,sEAAsE,CACvE,CAAC;YACJ,CAAC;YACD,OAAO,SAAS,CAAC,CAAC,CAAC,MAAM,SAAS,EAAE,CAAC,CAAC,CAAE,SAAe,CAAC;QAC1D,CAAC;IACH,CAAC;IAGS,QAAQ,CAAC,GAAW;QAC5B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAClE,CAAC;IASS,eAAe,CAAC,MAAc;QACtC,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC;IACjD,CAAC;IAeS,cAAc;QACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;IACrF,CAAC;IAGS,WAAW,CAAC,GAAW;QAC/B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,EAAE,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACrG,CAAC;IAES,KAAK,CAAC,eAAe,CAAC,OAAe;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;QAC7C,IAAI,MAAM,GAAG,GAAG,CAAC;QACjB,GAAG,CAAC;YACF,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;YACrF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,MAAM,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC5B,CAAC;YACD,MAAM,GAAG,UAAU,CAAC;QACtB,CAAC,QAAQ,MAAM,KAAK,GAAG,EAAE;IAC3B,CAAC;CACF;AA/LD,kDA+LC"}
|
|
@@ -1,21 +1,37 @@
|
|
|
1
|
-
import { Logger } from '@nestjs/common';
|
|
1
|
+
import { Logger, OnModuleDestroy } from '@nestjs/common';
|
|
2
2
|
import { Request, Response } from 'express';
|
|
3
|
+
import { CoreRedisService } from '../../common/services/core-redis.service';
|
|
3
4
|
import { CoreAiMcpOAuthService } from './services/core-ai-mcp-oauth.service';
|
|
4
5
|
import { CoreAiMcpService } from './services/core-ai-mcp.service';
|
|
5
|
-
export declare class CoreAiMcpController {
|
|
6
|
+
export declare class CoreAiMcpController implements OnModuleDestroy {
|
|
6
7
|
private readonly mcpService;
|
|
7
8
|
private readonly oauthService;
|
|
9
|
+
protected readonly redisService?: CoreRedisService;
|
|
8
10
|
protected readonly logger: Logger;
|
|
9
11
|
private readonly transports;
|
|
12
|
+
private readonly maxSessionsPerUser;
|
|
10
13
|
private readonly maxSessions;
|
|
11
|
-
|
|
14
|
+
protected readonly instanceId: string;
|
|
15
|
+
protected readonly sessionTtlSeconds = 3600;
|
|
16
|
+
constructor(mcpService: CoreAiMcpService, oauthService: CoreAiMcpOAuthService, redisService?: CoreRedisService);
|
|
12
17
|
handlePost(req: Request, res: Response): Promise<void>;
|
|
13
18
|
handleGet(req: Request, res: Response): Promise<void>;
|
|
14
19
|
handleDelete(req: Request, res: Response): Promise<void>;
|
|
15
20
|
protected resolveUser(req: Request): Promise<any | null>;
|
|
16
21
|
protected oauthEnabled(): boolean;
|
|
17
22
|
private handleSessionRequest;
|
|
23
|
+
protected foreignSessionOwner(sessionId: string, userId: string): Promise<string | undefined>;
|
|
24
|
+
protected parseSessionEntry(raw: null | string): undefined | {
|
|
25
|
+
instanceId: string;
|
|
26
|
+
userId: string;
|
|
27
|
+
};
|
|
28
|
+
protected registerSession(sessionId: string, userId: string): void;
|
|
29
|
+
protected releaseSession(sessionId: string): Promise<void>;
|
|
30
|
+
onModuleDestroy(): Promise<void>;
|
|
31
|
+
protected sessionKey(sessionId: string): string;
|
|
32
|
+
protected foreignSession(res: Response, owner: string): void;
|
|
18
33
|
private mcpUnavailable;
|
|
19
34
|
private unauthorized;
|
|
20
35
|
private evictIfNeeded;
|
|
36
|
+
private evictOldestOf;
|
|
21
37
|
}
|
|
@@ -16,9 +16,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
16
16
|
exports.CoreAiMcpController = void 0;
|
|
17
17
|
const common_1 = require("@nestjs/common");
|
|
18
18
|
const swagger_1 = require("@nestjs/swagger");
|
|
19
|
+
const node_os_1 = require("node:os");
|
|
19
20
|
const roles_decorator_1 = require("../../common/decorators/roles.decorator");
|
|
20
21
|
const role_enum_1 = require("../../common/enums/role.enum");
|
|
21
22
|
const config_service_1 = require("../../common/services/config.service");
|
|
23
|
+
const core_redis_service_1 = require("../../common/services/core-redis.service");
|
|
22
24
|
const core_better_auth_module_1 = require("../better-auth/core-better-auth.module");
|
|
23
25
|
const error_codes_1 = require("../error-code/error-codes");
|
|
24
26
|
const core_ai_mcp_oauth_service_1 = require("./services/core-ai-mcp-oauth.service");
|
|
@@ -26,12 +28,17 @@ const core_ai_mcp_service_1 = require("./services/core-ai-mcp.service");
|
|
|
26
28
|
let CoreAiMcpController = CoreAiMcpController_1 = class CoreAiMcpController {
|
|
27
29
|
mcpService;
|
|
28
30
|
oauthService;
|
|
31
|
+
redisService;
|
|
29
32
|
logger = new common_1.Logger(CoreAiMcpController_1.name);
|
|
30
33
|
transports = new Map();
|
|
34
|
+
maxSessionsPerUser = 25;
|
|
31
35
|
maxSessions = 500;
|
|
32
|
-
|
|
36
|
+
instanceId = `${(0, node_os_1.hostname)()}:${process.pid}`;
|
|
37
|
+
sessionTtlSeconds = 3600;
|
|
38
|
+
constructor(mcpService, oauthService, redisService) {
|
|
33
39
|
this.mcpService = mcpService;
|
|
34
40
|
this.oauthService = oauthService;
|
|
41
|
+
this.redisService = redisService;
|
|
35
42
|
}
|
|
36
43
|
async handlePost(req, res) {
|
|
37
44
|
const user = await this.resolveUser(req);
|
|
@@ -41,6 +48,18 @@ let CoreAiMcpController = CoreAiMcpController_1 = class CoreAiMcpController {
|
|
|
41
48
|
}
|
|
42
49
|
const sessionId = req.headers['mcp-session-id'];
|
|
43
50
|
let entry = sessionId ? this.transports.get(sessionId) : undefined;
|
|
51
|
+
if (entry && entry.ownerId !== user.id) {
|
|
52
|
+
entry = undefined;
|
|
53
|
+
res.status(404).json({ error: 'Unknown or expired MCP session' });
|
|
54
|
+
return;
|
|
55
|
+
}
|
|
56
|
+
if (!entry && sessionId) {
|
|
57
|
+
const owner = await this.foreignSessionOwner(sessionId, user.id);
|
|
58
|
+
if (owner) {
|
|
59
|
+
this.foreignSession(res, owner);
|
|
60
|
+
return;
|
|
61
|
+
}
|
|
62
|
+
}
|
|
44
63
|
if (!entry) {
|
|
45
64
|
let StreamableHTTPServerTransport;
|
|
46
65
|
try {
|
|
@@ -56,15 +75,19 @@ let CoreAiMcpController = CoreAiMcpController_1 = class CoreAiMcpController {
|
|
|
56
75
|
transport.onclose = () => {
|
|
57
76
|
if (transport.sessionId) {
|
|
58
77
|
this.transports.delete(transport.sessionId);
|
|
78
|
+
this.releaseSession(transport.sessionId);
|
|
59
79
|
}
|
|
60
80
|
};
|
|
61
|
-
entry = { lastUsed: Date.now(), transport };
|
|
81
|
+
entry = { lastUsed: Date.now(), ownerId: user.id, transport };
|
|
62
82
|
}
|
|
63
83
|
entry.lastUsed = Date.now();
|
|
64
84
|
await entry.transport.handleRequest(req, res, req.body);
|
|
65
|
-
if (entry.transport.sessionId
|
|
66
|
-
this.
|
|
67
|
-
|
|
85
|
+
if (entry.transport.sessionId) {
|
|
86
|
+
if (!this.transports.has(entry.transport.sessionId)) {
|
|
87
|
+
this.evictIfNeeded(user.id);
|
|
88
|
+
this.transports.set(entry.transport.sessionId, entry);
|
|
89
|
+
}
|
|
90
|
+
this.registerSession(entry.transport.sessionId, user.id);
|
|
68
91
|
}
|
|
69
92
|
}
|
|
70
93
|
async handleGet(req, res) {
|
|
@@ -110,14 +133,93 @@ let CoreAiMcpController = CoreAiMcpController_1 = class CoreAiMcpController {
|
|
|
110
133
|
return;
|
|
111
134
|
}
|
|
112
135
|
const sessionId = req.headers['mcp-session-id'];
|
|
113
|
-
const
|
|
136
|
+
const held = sessionId ? this.transports.get(sessionId) : undefined;
|
|
137
|
+
const entry = held && held.ownerId === user.id ? held : undefined;
|
|
114
138
|
if (!entry) {
|
|
139
|
+
const owner = sessionId ? await this.foreignSessionOwner(sessionId, user.id) : undefined;
|
|
140
|
+
if (owner) {
|
|
141
|
+
this.foreignSession(res, owner);
|
|
142
|
+
return;
|
|
143
|
+
}
|
|
115
144
|
res.status(404).json({ error: 'Unknown or expired MCP session' });
|
|
116
145
|
return;
|
|
117
146
|
}
|
|
118
147
|
entry.lastUsed = Date.now();
|
|
148
|
+
this.registerSession(sessionId, user.id);
|
|
119
149
|
await entry.transport.handleRequest(req, res, req.body);
|
|
120
150
|
}
|
|
151
|
+
async foreignSessionOwner(sessionId, userId) {
|
|
152
|
+
if (!this.redisService?.enabled) {
|
|
153
|
+
return undefined;
|
|
154
|
+
}
|
|
155
|
+
try {
|
|
156
|
+
const raw = await this.redisService.getClient().get(this.sessionKey(sessionId));
|
|
157
|
+
const entry = this.parseSessionEntry(raw);
|
|
158
|
+
if (!entry || entry.userId !== userId || entry.instanceId === this.instanceId) {
|
|
159
|
+
return undefined;
|
|
160
|
+
}
|
|
161
|
+
return entry.instanceId;
|
|
162
|
+
}
|
|
163
|
+
catch (err) {
|
|
164
|
+
this.logger.debug(`MCP session registry lookup failed: ${err.message}`);
|
|
165
|
+
return undefined;
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
parseSessionEntry(raw) {
|
|
169
|
+
if (!raw) {
|
|
170
|
+
return undefined;
|
|
171
|
+
}
|
|
172
|
+
try {
|
|
173
|
+
const parsed = JSON.parse(raw);
|
|
174
|
+
return parsed?.instanceId && parsed?.userId ? parsed : undefined;
|
|
175
|
+
}
|
|
176
|
+
catch {
|
|
177
|
+
return undefined;
|
|
178
|
+
}
|
|
179
|
+
}
|
|
180
|
+
registerSession(sessionId, userId) {
|
|
181
|
+
if (!this.redisService?.enabled) {
|
|
182
|
+
return;
|
|
183
|
+
}
|
|
184
|
+
this.redisService
|
|
185
|
+
.getClient()
|
|
186
|
+
.set(this.sessionKey(sessionId), JSON.stringify({ instanceId: this.instanceId, userId }), 'EX', this.sessionTtlSeconds)
|
|
187
|
+
.catch((err) => this.logger.debug(`MCP session registry write failed: ${err.message}`));
|
|
188
|
+
}
|
|
189
|
+
releaseSession(sessionId) {
|
|
190
|
+
if (!this.redisService?.enabled) {
|
|
191
|
+
return Promise.resolve();
|
|
192
|
+
}
|
|
193
|
+
return this.redisService
|
|
194
|
+
.getClient()
|
|
195
|
+
.del(this.sessionKey(sessionId))
|
|
196
|
+
.then(() => undefined)
|
|
197
|
+
.catch((err) => this.logger.debug(`MCP session registry delete failed: ${err.message}`));
|
|
198
|
+
}
|
|
199
|
+
async onModuleDestroy() {
|
|
200
|
+
const sessions = [...this.transports.keys()];
|
|
201
|
+
for (const [, entry] of this.transports) {
|
|
202
|
+
try {
|
|
203
|
+
await entry.transport?.close?.();
|
|
204
|
+
}
|
|
205
|
+
catch (error) {
|
|
206
|
+
this.logger.debug(`MCP transport close failed: ${error instanceof Error ? error.message : 'unknown'}`);
|
|
207
|
+
}
|
|
208
|
+
}
|
|
209
|
+
this.transports.clear();
|
|
210
|
+
await Promise.allSettled(sessions.map((sessionId) => this.releaseSession(sessionId)));
|
|
211
|
+
}
|
|
212
|
+
sessionKey(sessionId) {
|
|
213
|
+
return this.redisService.key('ai-mcp-session', sessionId);
|
|
214
|
+
}
|
|
215
|
+
foreignSession(res, owner) {
|
|
216
|
+
res.status(409).json({
|
|
217
|
+
error: `MCP session belongs to another server instance (${owner}); this instance is ${this.instanceId}. ` +
|
|
218
|
+
'The MCP transport is held in process memory and cannot be shared, so /ai/mcp requires sticky ' +
|
|
219
|
+
'sessions — route every request of one MCP session to the same replica.',
|
|
220
|
+
statusCode: 409,
|
|
221
|
+
});
|
|
222
|
+
}
|
|
121
223
|
mcpUnavailable(res, err) {
|
|
122
224
|
this.logger.error(`MCP SDK not available: ${err.message}`);
|
|
123
225
|
res.status(503).json({
|
|
@@ -135,26 +237,52 @@ let CoreAiMcpController = CoreAiMcpController_1 = class CoreAiMcpController {
|
|
|
135
237
|
.set({ 'WWW-Authenticate': `Bearer resource_metadata="${baseUrl}/ai/mcp"` })
|
|
136
238
|
.json({ error: 'Unauthorized: a valid Bearer token is required' });
|
|
137
239
|
}
|
|
138
|
-
evictIfNeeded() {
|
|
240
|
+
evictIfNeeded(ownerId) {
|
|
241
|
+
const perUser = new Map();
|
|
242
|
+
for (const [key, value] of this.transports) {
|
|
243
|
+
const owned = perUser.get(value.ownerId);
|
|
244
|
+
if (owned) {
|
|
245
|
+
owned.push(key);
|
|
246
|
+
}
|
|
247
|
+
else {
|
|
248
|
+
perUser.set(value.ownerId, [key]);
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
if ((perUser.get(ownerId)?.length ?? 0) >= this.maxSessionsPerUser) {
|
|
252
|
+
this.evictOldestOf(perUser.get(ownerId));
|
|
253
|
+
return;
|
|
254
|
+
}
|
|
139
255
|
if (this.transports.size < this.maxSessions) {
|
|
140
256
|
return;
|
|
141
257
|
}
|
|
258
|
+
let largest;
|
|
259
|
+
for (const owned of perUser.values()) {
|
|
260
|
+
if (!largest || owned.length > largest.length) {
|
|
261
|
+
largest = owned;
|
|
262
|
+
}
|
|
263
|
+
}
|
|
264
|
+
this.evictOldestOf(largest);
|
|
265
|
+
}
|
|
266
|
+
evictOldestOf(sessionIds) {
|
|
142
267
|
let oldestKey;
|
|
143
268
|
let oldest = Infinity;
|
|
144
|
-
for (const
|
|
145
|
-
|
|
269
|
+
for (const key of sessionIds ?? []) {
|
|
270
|
+
const value = this.transports.get(key);
|
|
271
|
+
if (value && value.lastUsed < oldest) {
|
|
146
272
|
oldest = value.lastUsed;
|
|
147
273
|
oldestKey = key;
|
|
148
274
|
}
|
|
149
275
|
}
|
|
150
|
-
if (oldestKey) {
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
276
|
+
if (!oldestKey) {
|
|
277
|
+
return;
|
|
278
|
+
}
|
|
279
|
+
const evicted = this.transports.get(oldestKey);
|
|
280
|
+
this.transports.delete(oldestKey);
|
|
281
|
+
this.releaseSession(oldestKey);
|
|
282
|
+
try {
|
|
283
|
+
evicted?.transport.close?.();
|
|
284
|
+
}
|
|
285
|
+
catch {
|
|
158
286
|
}
|
|
159
287
|
}
|
|
160
288
|
};
|
|
@@ -187,7 +315,9 @@ exports.CoreAiMcpController = CoreAiMcpController = CoreAiMcpController_1 = __de
|
|
|
187
315
|
(0, swagger_1.ApiExcludeController)(),
|
|
188
316
|
(0, common_1.Controller)('ai/mcp'),
|
|
189
317
|
(0, roles_decorator_1.Roles)(role_enum_1.RoleEnum.S_EVERYONE),
|
|
318
|
+
__param(2, (0, common_1.Optional)()),
|
|
190
319
|
__metadata("design:paramtypes", [core_ai_mcp_service_1.CoreAiMcpService,
|
|
191
|
-
core_ai_mcp_oauth_service_1.CoreAiMcpOAuthService
|
|
320
|
+
core_ai_mcp_oauth_service_1.CoreAiMcpOAuthService,
|
|
321
|
+
core_redis_service_1.CoreRedisService])
|
|
192
322
|
], CoreAiMcpController);
|
|
193
323
|
//# sourceMappingURL=core-ai-mcp.controller.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"core-ai-mcp.controller.js","sourceRoot":"","sources":["../../../../src/core/modules/ai/core-ai-mcp.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAiF;AACjF,6CAAuD;AAGvD,6EAAgE;AAChE,4DAAwD;AACxD,yEAAqE;AACrE,oFAA8E;AAC9E,2DAAsD;AACtD,oFAA6E;AAC7E,wEAAkE;AAsB3D,IAAM,mBAAmB,2BAAzB,MAAM,mBAAmB;IAUX;IACA;IAVA,MAAM,GAAG,IAAI,eAAM,CAAC,qBAAmB,CAAC,IAAI,CAAC,CAAC;IAGhD,UAAU,GAAG,IAAI,GAAG,EAAgD,CAAC;IAGrE,WAAW,GAAG,GAAG,CAAC;IAEnC,YACmB,UAA4B,EAC5B,YAAmC;QADnC,eAAU,GAAV,UAAU,CAAkB;QAC5B,iBAAY,GAAZ,YAAY,CAAuB;IACnD,CAAC;IAGE,AAAN,KAAK,CAAC,UAAU,CAAQ,GAAY,EAAS,GAAa;QACxD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC;YACd,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QACtE,IAAI,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEnE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,6BAAkC,CAAC;YACvC,IAAI,CAAC;gBACH,CAAC,EAAE,6BAA6B,EAAE,GAAG,2CAAa,oDAAoD,EAAC,CAAC,CAAC;YAC3G,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBAQb,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAY,CAAC,CAAC;YAChD,CAAC;YACD,MAAM,EAAE,UAAU,EAAE,GAAG,2CAAa,aAAa,EAAC,CAAC;YACnD,MAAM,SAAS,GAAQ,IAAI,6BAA6B,CAAC,EAAE,kBAAkB,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACrG,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAIhC,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE;gBACvB,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;oBACxB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC,CAAC;YACF,KAAK,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC;QAC9C,CAAC;QAED,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,MAAM,KAAK,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QAGxD,IAAI,KAAK,CAAC,SAAS,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;YACjF,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAGK,AAAN,KAAK,CAAC,SAAS,CAAQ,GAAY,EAAS,GAAa;QACvD,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC5C,CAAC;IAGK,AAAN,KAAK,CAAC,YAAY,CAAQ,GAAY,EAAS,GAAa;QAC1D,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC5C,CAAC;IAQS,KAAK,CAAC,WAAW,CAAC,GAAY;QACtC,MAAM,WAAW,GAAI,GAAW,CAAC,IAAI,CAAC;QACtC,IAAI,WAAW,EAAE,EAAE,EAAE,CAAC;YACpB,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAGpF,MAAM,YAAY,GAAG,8CAAoB,CAAC,uBAAuB,EAAE,CAAC;QACpE,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC;gBACH,MAAM,EAAE,KAAK,EAAE,GAAG,YAAY,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;gBAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,MAAM,YAAY,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBACxE,IAAI,IAAI,EAAE,EAAE,EAAE,CAAC;oBACb,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;QACH,CAAC;QAGD,IAAI,MAAM,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC5D,IAAI,OAAO,EAAE,GAAG,EAAE,CAAC;gBACjB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAKS,YAAY;QACpB,MAAM,GAAG,GAAG,8BAAa,CAAC,GAAG,CAAsB,QAAQ,CAAC,CAAC;QAC7D,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC;IACxD,CAAC;IAKO,KAAK,CAAC,oBAAoB,CAAC,GAAY,EAAE,GAAa;QAC5D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC;YACd,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QACtE,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QACD,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,MAAM,KAAK,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAG,GAAW,CAAC,IAAI,CAAC,CAAC;IACnE,CAAC;IAoBO,cAAc,CAAC,GAAa,EAAE,GAAU;QAC9C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3D,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACnB,KAAK,EACH,GAAG,uBAAS,CAAC,mBAAmB,6BAA6B;gBAC7D,6EAA6E;gBAC7E,kFAAkF;gBAClF,8EAA8E;YAChF,UAAU,EAAE,GAAG;SAChB,CAAC,CAAC;IACL,CAAC;IAKO,YAAY,CAAC,GAAY,EAAE,GAAa;QAC9C,MAAM,OAAO,GAAG,GAAG,GAAG,CAAC,QAAQ,MAAM,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;QACvD,GAAG;aACA,MAAM,CAAC,GAAG,CAAC;aACX,GAAG,CAAC,EAAE,kBAAkB,EAAE,6BAA6B,OAAO,UAAU,EAAE,CAAC;aAC3E,IAAI,CAAC,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC,CAAC;IACvE,CAAC;IAKO,aAAa;QACnB,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC5C,OAAO;QACT,CAAC;QACD,IAAI,SAA6B,CAAC;QAClC,IAAI,MAAM,GAAG,QAAQ,CAAC;QACtB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,QAAQ,GAAG,MAAM,EAAE,CAAC;gBAC5B,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC;gBACxB,SAAS,GAAG,GAAG,CAAC;YAClB,CAAC;QACH,CAAC;QACD,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC/C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC;gBACH,OAAO,EAAE,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC;YAC/B,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;QACH,CAAC;IACH,CAAC;CACF,CAAA;AA9MY,kDAAmB;AAexB;IADL,IAAA,aAAI,GAAE;IACW,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;qDA+C3C;AAGK;IADL,IAAA,YAAG,GAAE;IACW,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;oDAE1C;AAGK;IADL,IAAA,eAAM,GAAE;IACW,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;uDAE7C;8BAxEU,mBAAmB;IAH/B,IAAA,8BAAoB,GAAE;IACtB,IAAA,mBAAU,EAAC,QAAQ,CAAC;IACpB,IAAA,uBAAK,EAAC,oBAAQ,CAAC,UAAU,CAAC;qCAWM,sCAAgB;QACd,iDAAqB;GAX3C,mBAAmB,CA8M/B"}
|
|
1
|
+
{"version":3,"file":"core-ai-mcp.controller.js","sourceRoot":"","sources":["../../../../src/core/modules/ai/core-ai-mcp.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAA4G;AAC5G,6CAAuD;AAEvD,qCAAmC;AAEnC,6EAAgE;AAChE,4DAAwD;AACxD,yEAAqE;AACrE,iFAA4E;AAC5E,oFAA8E;AAC9E,2DAAsD;AACtD,oFAA6E;AAC7E,wEAAkE;AAoC3D,IAAM,mBAAmB,2BAAzB,MAAM,mBAAmB;IAsCX;IACA;IACc;IAvCd,MAAM,GAAG,IAAI,eAAM,CAAC,qBAAmB,CAAC,IAAI,CAAC,CAAC;IAYhD,UAAU,GAAG,IAAI,GAAG,EAAiE,CAAC;IAStF,kBAAkB,GAAG,EAAE,CAAC;IAGxB,WAAW,GAAG,GAAG,CAAC;IAGhB,UAAU,GAAG,GAAG,IAAA,kBAAQ,GAAE,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAO5C,iBAAiB,GAAG,IAAI,CAAC;IAE5C,YACmB,UAA4B,EAC5B,YAAmC,EACrB,YAA+B;QAF7C,eAAU,GAAV,UAAU,CAAkB;QAC5B,iBAAY,GAAZ,YAAY,CAAuB;QACrB,iBAAY,GAAZ,YAAY,CAAmB;IAC7D,CAAC;IAGE,AAAN,KAAK,CAAC,UAAU,CAAQ,GAAY,EAAS,GAAa;QACxD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC;YACd,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QACtE,IAAI,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAKnE,IAAI,KAAK,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,EAAE,EAAE,CAAC;YACvC,KAAK,GAAG,SAAS,CAAC;YAClB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,IAAI,SAAS,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YACjE,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAChC,OAAO;YACT,CAAC;QACH,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,6BAAkC,CAAC;YACvC,IAAI,CAAC;gBACH,CAAC,EAAE,6BAA6B,EAAE,GAAG,2CAAa,oDAAoD,EAAC,CAAC,CAAC;YAC3G,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBAQb,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAY,CAAC,CAAC;YAChD,CAAC;YACD,MAAM,EAAE,UAAU,EAAE,GAAG,2CAAa,aAAa,EAAC,CAAC;YACnD,MAAM,SAAS,GAAQ,IAAI,6BAA6B,CAAC,EAAE,kBAAkB,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACrG,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAIhC,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE;gBACvB,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;oBACxB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC,CAAC;YACF,KAAK,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,CAAC;QAChE,CAAC;QAED,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,MAAM,KAAK,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QAGxD,IAAI,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACxD,CAAC;YAED,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAGK,AAAN,KAAK,CAAC,SAAS,CAAQ,GAAY,EAAS,GAAa;QACvD,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC5C,CAAC;IAGK,AAAN,KAAK,CAAC,YAAY,CAAQ,GAAY,EAAS,GAAa;QAC1D,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC5C,CAAC;IAQS,KAAK,CAAC,WAAW,CAAC,GAAY;QACtC,MAAM,WAAW,GAAI,GAAW,CAAC,IAAI,CAAC;QACtC,IAAI,WAAW,EAAE,EAAE,EAAE,CAAC;YACpB,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAGpF,MAAM,YAAY,GAAG,8CAAoB,CAAC,uBAAuB,EAAE,CAAC;QACpE,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC;gBACH,MAAM,EAAE,KAAK,EAAE,GAAG,YAAY,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;gBAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,MAAM,YAAY,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBACxE,IAAI,IAAI,EAAE,EAAE,EAAE,CAAC;oBACb,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;QACH,CAAC;QAGD,IAAI,MAAM,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC5D,IAAI,OAAO,EAAE,GAAG,EAAE,CAAC;gBACjB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAKS,YAAY;QACpB,MAAM,GAAG,GAAG,8BAAa,CAAC,GAAG,CAAsB,QAAQ,CAAC,CAAC;QAC7D,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC;IACxD,CAAC;IAKO,KAAK,CAAC,oBAAoB,CAAC,GAAY,EAAE,GAAa;QAC5D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC;YACd,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QACtE,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEpE,MAAM,KAAK,GAAG,IAAI,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QAClE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACzF,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAChC,OAAO;YACT,CAAC;YACD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QACD,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,eAAe,CAAC,SAAmB,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACnD,MAAM,KAAK,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAG,GAAW,CAAC,IAAI,CAAC,CAAC;IACnE,CAAC;IAeS,KAAK,CAAC,mBAAmB,CAAC,SAAiB,EAAE,MAAc;QACnE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;YAChF,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAC1C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9E,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,OAAO,KAAK,CAAC,UAAU,CAAC;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAwC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YACnF,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IASS,iBAAiB,CAAC,GAAkB;QAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC/B,OAAO,MAAM,EAAE,UAAU,IAAI,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;QACnE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAGS,eAAe,CAAC,SAAiB,EAAE,MAAc;QACzD,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,YAAY;aACd,SAAS,EAAE;aACX,GAAG,CACF,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAC1B,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC,EACvD,IAAI,EACJ,IAAI,CAAC,iBAAiB,CACvB;aACA,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACnG,CAAC;IAGS,cAAc,CAAC,SAAiB;QACxC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QACD,OAAO,IAAI,CAAC,YAAY;aACrB,SAAS,EAAE;aACX,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;aAC/B,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACrB,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACpG,CAAC;IAWD,KAAK,CAAC,eAAe;QACnB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7C,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;YACnC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;YACzG,CAAC;QACH,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACxF,CAAC;IAES,UAAU,CAAC,SAAiB;QACpC,OAAO,IAAI,CAAC,YAAa,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;IAC7D,CAAC;IAOS,cAAc,CAAC,GAAa,EAAE,KAAa;QACnD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACnB,KAAK,EACH,mDAAmD,KAAK,uBAAuB,IAAI,CAAC,UAAU,IAAI;gBAClG,+FAA+F;gBAC/F,wEAAwE;YAC1E,UAAU,EAAE,GAAG;SAChB,CAAC,CAAC;IACL,CAAC;IAoBO,cAAc,CAAC,GAAa,EAAE,GAAU;QAC9C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3D,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACnB,KAAK,EACH,GAAG,uBAAS,CAAC,mBAAmB,6BAA6B;gBAC7D,6EAA6E;gBAC7E,kFAAkF;gBAClF,8EAA8E;YAChF,UAAU,EAAE,GAAG;SAChB,CAAC,CAAC;IACL,CAAC;IAKO,YAAY,CAAC,GAAY,EAAE,GAAa;QAC9C,MAAM,OAAO,GAAG,GAAG,GAAG,CAAC,QAAQ,MAAM,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;QACvD,GAAG;aACA,MAAM,CAAC,GAAG,CAAC;aACX,GAAG,CAAC,EAAE,kBAAkB,EAAE,6BAA6B,OAAO,UAAU,EAAE,CAAC;aAC3E,IAAI,CAAC,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC,CAAC;IACvE,CAAC;IAWO,aAAa,CAAC,OAAe;QACnC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAoB,CAAC;QAC5C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACzC,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACnE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,IAAI,OAA6B,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC9C,OAAO,GAAG,KAAK,CAAC;YAClB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAKO,aAAa,CAAC,UAAgC;QACpD,IAAI,SAA6B,CAAC;QAClC,IAAI,MAAM,GAAG,QAAQ,CAAC;QACtB,KAAK,MAAM,GAAG,IAAI,UAAU,IAAI,EAAE,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACvC,IAAI,KAAK,IAAI,KAAK,CAAC,QAAQ,GAAG,MAAM,EAAE,CAAC;gBACrC,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC;gBACxB,SAAS,GAAG,GAAG,CAAC;YAClB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC/C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC/B,IAAI,CAAC;YACH,OAAO,EAAE,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;QAET,CAAC;IACH,CAAC;CACF,CAAA;AAraY,kDAAmB;AA4CxB;IADL,IAAA,aAAI,GAAE;IACW,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;qDAqE3C;AAGK;IADL,IAAA,YAAG,GAAE;IACW,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;oDAE1C;AAGK;IADL,IAAA,eAAM,GAAE;IACW,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;uDAE7C;8BA3HU,mBAAmB;IAH/B,IAAA,8BAAoB,GAAE;IACtB,IAAA,mBAAU,EAAC,QAAQ,CAAC;IACpB,IAAA,uBAAK,EAAC,oBAAQ,CAAC,UAAU,CAAC;IAyCtB,WAAA,IAAA,iBAAQ,GAAE,CAAA;qCAFkB,sCAAgB;QACd,iDAAqB;QACN,qCAAgB;GAxCrD,mBAAmB,CAqa/B"}
|
|
@@ -1,5 +1,7 @@
|
|
|
1
1
|
import { Logger } from '@nestjs/common';
|
|
2
2
|
import { ServiceOptions } from '../../../common/interfaces/service-options.interface';
|
|
3
|
+
import { CoreRedisService } from '../../../common/services/core-redis.service';
|
|
4
|
+
import { RateLimitStore } from '../../../common/services/rate-limit-store';
|
|
3
5
|
import { AiToolAuthorization, AiToolContext, IAiTool } from '../interfaces/ai-tool.interface';
|
|
4
6
|
import { LlmMessage, LlmResponse, LlmToolCall } from '../interfaces/llm-provider.interface';
|
|
5
7
|
import { ResolvedAiConnection } from '../interfaces/resolved-ai-connection.interface';
|
|
@@ -63,9 +65,10 @@ export declare class CoreAiService {
|
|
|
63
65
|
protected readonly toolPolicyService?: CoreAiToolPolicyService;
|
|
64
66
|
protected readonly modeService?: CoreAiModeService;
|
|
65
67
|
protected readonly placeholderRegistry?: CoreAiPlaceholderRegistry;
|
|
68
|
+
protected readonly coreRedisService?: CoreRedisService;
|
|
66
69
|
protected readonly logger: Logger;
|
|
67
|
-
|
|
68
|
-
constructor(connectionService: CoreAiConnectionService, providerFactory: LlmProviderFactory, toolRegistry: AiToolRegistry, promptBuilder: CoreAiPromptBuilderService, interactionService?: CoreAiInteractionService, conversationService?: CoreAiConversationService, budgetService?: CoreAiBudgetService, connectionResolver?: CoreAiConnectionResolverService, hintService?: CoreAiPromptHintService, toolGrantService?: CoreAiToolGrantService, hookRegistry?: AiHookRegistry, toolPolicyService?: CoreAiToolPolicyService, modeService?: CoreAiModeService, placeholderRegistry?: CoreAiPlaceholderRegistry);
|
|
70
|
+
protected rateLimitStore?: RateLimitStore;
|
|
71
|
+
constructor(connectionService: CoreAiConnectionService, providerFactory: LlmProviderFactory, toolRegistry: AiToolRegistry, promptBuilder: CoreAiPromptBuilderService, interactionService?: CoreAiInteractionService, conversationService?: CoreAiConversationService, budgetService?: CoreAiBudgetService, connectionResolver?: CoreAiConnectionResolverService, hintService?: CoreAiPromptHintService, toolGrantService?: CoreAiToolGrantService, hookRegistry?: AiHookRegistry, toolPolicyService?: CoreAiToolPolicyService, modeService?: CoreAiModeService, placeholderRegistry?: CoreAiPlaceholderRegistry, coreRedisService?: CoreRedisService);
|
|
69
72
|
prompt(input: CoreAiPromptInput, serviceOptions: ServiceOptions): Promise<CoreAiResponse>;
|
|
70
73
|
protected resolvePromptPlaceholders(input: CoreAiPromptInput, serviceOptions: ServiceOptions): Promise<CoreAiPromptInput>;
|
|
71
74
|
protected unavailableResponse(input: CoreAiPromptInput, language?: string): CoreAiResponse;
|
|
@@ -100,6 +103,8 @@ export declare class CoreAiService {
|
|
|
100
103
|
}): AiInteractionRecord;
|
|
101
104
|
protected audit(record: AiInteractionRecord): Promise<void>;
|
|
102
105
|
protected checkRateLimit(userId?: string): Promise<void>;
|
|
106
|
+
protected getRateLimitStore(): RateLimitStore;
|
|
107
|
+
onModuleDestroy(): void;
|
|
103
108
|
protected executeToolCall(call: LlmToolCall, availableTools: IAiTool[], context: AiToolContext, input?: CoreAiPromptInput): Promise<CoreAiAction>;
|
|
104
109
|
protected extractToolCalls(text: string): LlmToolCall[] | undefined;
|
|
105
110
|
protected extractJsonObject(text: string): any | null;
|