@loopstack/quota 0.20.7

package/dist/services/__tests__/quota-client.service.spec.js

@@ -0,0 +1,83 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const quota_client_service_1 = require("../quota-client.service");
+describe('QuotaClientService', () => {
+    describe('without Redis (null)', () => {
+        let service;
+        beforeEach(() => {
+            service = new quota_client_service_1.QuotaClientService(null);
+        });
+        it('checkQuota should return not exceeded with unlimited', async () => {
+            const result = await service.checkQuota('user-1', 'default-token');
+            expect(result).toEqual({ exceeded: false, used: 0, limit: -1 });
+        });
+        it('report should complete without error', async () => {
+            await expect(service.report('user-1', 'default-token', 100)).resolves.toBeUndefined();
+        });
+    });
+    describe('with Redis mock', () => {
+        let service;
+        let redisMock;
+        beforeEach(() => {
+            redisMock = {
+                get: jest.fn(),
+                incrby: jest.fn(),
+            };
+            service = new quota_client_service_1.QuotaClientService(redisMock);
+        });
+        describe('checkQuota', () => {
+            it('should return not exceeded when under limit', async () => {
+                redisMock.get.mockResolvedValueOnce('50').mockResolvedValueOnce('1000');
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: false, used: 50, limit: 1000 });
+            });
+            it('should return exceeded when at limit', async () => {
+                redisMock.get.mockResolvedValueOnce('1000').mockResolvedValueOnce('1000');
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: true, used: 1000, limit: 1000 });
+            });
+            it('should return exceeded when over limit', async () => {
+                redisMock.get.mockResolvedValueOnce('1500').mockResolvedValueOnce('1000');
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: true, used: 1500, limit: 1000 });
+            });
+            it('should block when no limit key exists (no quota assigned)', async () => {
+                redisMock.get.mockResolvedValueOnce('500').mockResolvedValueOnce(null);
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: true, used: 500, limit: 0 });
+            });
+            it('should treat missing used as 0', async () => {
+                redisMock.get.mockResolvedValueOnce(null).mockResolvedValueOnce('1000');
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: false, used: 0, limit: 1000 });
+            });
+            it('should treat explicit limit of -1 as unlimited', async () => {
+                redisMock.get.mockResolvedValueOnce('999999').mockResolvedValueOnce('-1');
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: false, used: 999999, limit: -1 });
+            });
+            it('should block when no limit key and no usage exist', async () => {
+                redisMock.get.mockResolvedValueOnce(null).mockResolvedValueOnce(null);
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: true, used: 0, limit: 0 });
+            });
+            it('should fail-open on Redis error', async () => {
+                redisMock.get.mockRejectedValue(new Error('Connection refused'));
+                const result = await service.checkQuota('user-1', 'default-token');
+                expect(result).toEqual({ exceeded: false, used: 0, limit: -1 });
+            });
+        });
+        describe('report', () => {
+            it('should increment the used key', async () => {
+                redisMock.incrby.mockResolvedValue(150);
+                await service.report('user-1', 'default-token', 150);
+                expect(redisMock.incrby).toHaveBeenCalledWith('user:user-1:quota:default-token:used', 150);
+            });
+            it('should not throw on Redis error', async () => {
+                redisMock.incrby.mockRejectedValue(new Error('Connection refused'));
+                await expect(service.report('user-1', 'default-token', 100)).resolves.toBeUndefined();
+            });
+        });
+    });
+});
+//# sourceMappingURL=quota-client.service.spec.js.map

package/dist/services/__tests__/quota-client.service.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-client.service.spec.js","sourceRoot":"","sources":["../../../src/services/__tests__/quota-client.service.spec.ts"],"names":[],"mappings":";;AAAA,kEAA6D;AAE7D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,IAAI,OAA2B,CAAC;QAEhC,UAAU,CAAC,GAAG,EAAE;YACd,OAAO,GAAG,IAAI,yCAAkB,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,IAAI,OAA2B,CAAC;QAChC,IAAI,SAAoC,CAAC;QAEzC,UAAU,CAAC,GAAG,EAAE;YACd,SAAS,GAAG;gBACV,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;aAClB,CAAC;YAEF,OAAO,GAAG,IAAI,yCAAkB,CAAC,SAAgB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;YAC1B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;gBAC3D,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBAExE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;gBACpD,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBAE1E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;gBACtD,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBAE1E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;gBACzE,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBAEvE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;gBAC9C,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBAExE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;gBAC9D,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBAE1E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;gBACjE,SAAS,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBAEtE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;gBAC/C,SAAS,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAEjE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;YACtB,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;gBAC7C,SAAS,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;gBAExC,MAAM,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,CAAC,CAAC;gBAErD,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,sCAAsC,EAAE,GAAG,CAAC,CAAC;YAC7F,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;gBAC/C,SAAS,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAEpE,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YACxF,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/__tests__/quota.interceptor.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=quota.interceptor.spec.d.ts.map

package/dist/services/__tests__/quota.interceptor.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota.interceptor.spec.d.ts","sourceRoot":"","sources":["../../../src/services/__tests__/quota.interceptor.spec.ts"],"names":[],"mappings":""}

package/dist/services/__tests__/quota.interceptor.spec.js

@@ -0,0 +1,117 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const quota_calculator_registry_service_1 = require("../quota-calculator-registry.service");
+const quota_interceptor_1 = require("../quota.interceptor");
+class FakeTool {
+    name;
+    constructor(name) {
+        this.name = name;
+        Object.defineProperty(this.constructor, 'name', { value: name });
+    }
+}
+function createContext(toolName, overrides) {
+    return {
+        tool: new FakeTool(toolName),
+        args: undefined,
+        runContext: { userId: 'user-1' },
+        ...overrides,
+    };
+}
+describe('QuotaInterceptor', () => {
+    let interceptor;
+    let quotaClientService;
+    let registry;
+    beforeEach(() => {
+        quotaClientService = {
+            checkQuota: jest.fn().mockResolvedValue({ exceeded: false, used: 0, limit: -1 }),
+            report: jest.fn().mockResolvedValue(undefined),
+        };
+        registry = new quota_calculator_registry_service_1.QuotaCalculatorRegistry();
+        interceptor = new quota_interceptor_1.QuotaInterceptor(quotaClientService, registry);
+    });
+    describe('beforeExecute', () => {
+        it('should check processing time quota even when no calculator is registered', async () => {
+            const context = createContext('UnknownTool');
+            await interceptor.beforeExecute(context);
+            expect(quotaClientService.checkQuota).toHaveBeenCalledWith('user-1', 'processing-time-ms');
+            expect(quotaClientService.checkQuota).toHaveBeenCalledTimes(1);
+        });
+        it('should throw when processing time quota is exceeded for any tool', async () => {
+            quotaClientService.checkQuota.mockResolvedValue({ exceeded: true, used: 60000, limit: 60000 });
+            const context = createContext('UnknownTool');
+            await expect(interceptor.beforeExecute(context)).rejects.toThrow('Quota exceeded for "processing-time-ms": 60000/60000');
+        });
+        it('should check both processing time and tool-specific quota', async () => {
+            registry.register('AiGenerateText', {
+                quotaType: 'default-token',
+                calculateQuotaUsage: jest.fn(),
+            });
+            const context = createContext('AiGenerateText');
+            await interceptor.beforeExecute(context);
+            expect(quotaClientService.checkQuota).toHaveBeenCalledWith('user-1', 'processing-time-ms');
+            expect(quotaClientService.checkQuota).toHaveBeenCalledWith('user-1', 'default-token');
+            expect(quotaClientService.checkQuota).toHaveBeenCalledTimes(2);
+        });
+        it('should throw when tool-specific quota is exceeded', async () => {
+            registry.register('AiGenerateText', {
+                quotaType: 'default-token',
+                calculateQuotaUsage: jest.fn(),
+            });
+            quotaClientService.checkQuota
+                .mockResolvedValueOnce({ exceeded: false, used: 0, limit: -1 })
+                .mockResolvedValueOnce({ exceeded: true, used: 1000, limit: 1000 });
+            const context = createContext('AiGenerateText');
+            await expect(interceptor.beforeExecute(context)).rejects.toThrow('Quota exceeded for "default-token": 1000/1000');
+        });
+        it('should not throw when all quotas are within limits', async () => {
+            registry.register('AiGenerateText', {
+                quotaType: 'default-token',
+                calculateQuotaUsage: jest.fn(),
+            });
+            quotaClientService.checkQuota.mockResolvedValue({ exceeded: false, used: 500, limit: 1000 });
+            const context = createContext('AiGenerateText');
+            await expect(interceptor.beforeExecute(context)).resolves.toBeUndefined();
+        });
+    });
+    describe('afterExecute', () => {
+        const result = { data: {} };
+        it('should report processing time for every tool', async () => {
+            const context = createContext('UnknownTool', { metrics: { durationMs: 250 } });
+            await interceptor.afterExecute(context, result);
+            expect(quotaClientService.report).toHaveBeenCalledWith('user-1', 'processing-time-ms', 250);
+        });
+        it('should not report processing time when metrics are missing', async () => {
+            const context = createContext('UnknownTool');
+            await interceptor.afterExecute(context, result);
+            expect(quotaClientService.report).not.toHaveBeenCalled();
+        });
+        it('should report tool-specific usage when calculator returns usage', async () => {
+            registry.register('AiGenerateText', {
+                quotaType: 'default-token',
+                calculateQuotaUsage: jest.fn().mockReturnValue({ quotaType: 'default-token', actualAmount: 500 }),
+            });
+            const context = createContext('AiGenerateText', { metrics: { durationMs: 1000 } });
+            await interceptor.afterExecute(context, result);
+            expect(quotaClientService.report).toHaveBeenCalledWith('user-1', 'processing-time-ms', 1000);
+            expect(quotaClientService.report).toHaveBeenCalledWith('user-1', 'default-token', 500);
+            expect(quotaClientService.report).toHaveBeenCalledTimes(2);
+        });
+        it('should not report tool-specific usage when calculator returns null', async () => {
+            registry.register('AiGenerateText', {
+                quotaType: 'default-token',
+                calculateQuotaUsage: jest.fn().mockReturnValue(null),
+            });
+            const context = createContext('AiGenerateText', { metrics: { durationMs: 100 } });
+            await interceptor.afterExecute(context, result);
+            expect(quotaClientService.report).toHaveBeenCalledTimes(1);
+            expect(quotaClientService.report).toHaveBeenCalledWith('user-1', 'processing-time-ms', 100);
+        });
+    });
+    describe('onError', () => {
+        it('should not throw', async () => {
+            const context = createContext('AiGenerateText');
+            await expect(interceptor.onError(context, new Error('fail'))).resolves.toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=quota.interceptor.spec.js.map

package/dist/services/__tests__/quota.interceptor.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota.interceptor.spec.js","sourceRoot":"","sources":["../../../src/services/__tests__/quota.interceptor.spec.ts"],"names":[],"mappings":";;AAEA,4FAA+E;AAE/E,4DAAwD;AAExD,MAAM,QAAQ;IACgB;IAA5B,YAA4B,IAAY;QAAZ,SAAI,GAAJ,IAAI,CAAQ;QACtC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;CACF;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,SAAyC;IAChF,OAAO;QACL,IAAI,EAAE,IAAI,QAAQ,CAAC,QAAQ,CAAC;QAC5B,IAAI,EAAE,SAAS;QACf,UAAU,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAgB;QAC9C,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,WAA6B,CAAC;IAClC,IAAI,kBAAmD,CAAC;IACxD,IAAI,QAAiC,CAAC;IAEtC,UAAU,CAAC,GAAG,EAAE;QAEd,kBAAkB,GAAG;YACnB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC;YAChF,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;SACxC,CAAC;QAET,QAAQ,GAAG,IAAI,2DAAuB,EAAE,CAAC;QACzC,WAAW,GAAG,IAAI,oCAAgB,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,OAAO,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;YAE7C,MAAM,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAEzC,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;YAC3F,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,kBAAkB,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YAC/F,MAAM,OAAO,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;YAE7C,MAAM,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9D,sDAAsD,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAClC,SAAS,EAAE,eAAe;gBAC1B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;aAC/B,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAAC;YAEhD,MAAM,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAEzC,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;YAC3F,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;YACtF,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAClC,SAAS,EAAE,eAAe;gBAC1B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;aAC/B,CAAC,CAAC;YACH,kBAAkB,CAAC,UAAU;iBAC1B,qBAAqB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC;iBAC9D,qBAAqB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACtE,MAAM,OAAO,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAAC;YAEhD,MAAM,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,+CAA+C,CAAC,CAAC;QACpH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAClC,SAAS,EAAE,eAAe;gBAC1B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;aAC/B,CAAC,CAAC;YACH,kBAAkB,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC7F,MAAM,OAAO,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAAC;YAEhD,MAAM,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,MAAM,MAAM,GAAe,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAExC,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,OAAO,GAAG,aAAa,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YAE/E,MAAM,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAEhD,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC;QAC9F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,OAAO,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;YAE7C,MAAM,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAEhD,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAClC,SAAS,EAAE,eAAe;gBAC1B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,SAAS,EAAE,eAAe,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;aAClG,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,aAAa,CAAC,gBAAgB,EAAE,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAEnF,MAAM,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAEhD,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,oBAAoB,EAAE,IAAI,CAAC,CAAC;YAC7F,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,CAAC,CAAC;YACvF,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAClC,SAAS,EAAE,eAAe;gBAC1B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;aACrD,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,aAAa,CAAC,gBAAgB,EAAE,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YAElF,MAAM,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAEhD,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC;QAC9F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;YAChC,MAAM,OAAO,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAAC;YAEhD,MAAM,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/index.d.ts

@@ -0,0 +1,4 @@
+export * from './quota-calculator-registry.service';
+export * from './quota-client.service';
+export * from './quota.interceptor';
+//# sourceMappingURL=index.d.ts.map

package/dist/services/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":"AAAA,cAAc,qCAAqC,CAAC;AACpD,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC"}

package/dist/services/index.js

@@ -0,0 +1,20 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./quota-calculator-registry.service"), exports);
+__exportStar(require("./quota-client.service"), exports);
+__exportStar(require("./quota.interceptor"), exports);
+//# sourceMappingURL=index.js.map

package/dist/services/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sEAAoD;AACpD,yDAAuC;AACvC,sDAAoC"}

package/dist/services/quota-calculator-registry.service.d.ts

@@ -0,0 +1,9 @@
+import { ToolQuotaCalculator } from '../interfaces/tool-quota-calculator.interface';
+export declare class QuotaCalculatorRegistry {
+    private readonly logger;
+    private readonly calculators;
+    register(toolClassName: string, calculator: ToolQuotaCalculator): void;
+    get(toolClassName: string): ToolQuotaCalculator | undefined;
+    has(toolClassName: string): boolean;
+}
+//# sourceMappingURL=quota-calculator-registry.service.d.ts.map

package/dist/services/quota-calculator-registry.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-calculator-registry.service.d.ts","sourceRoot":"","sources":["../../src/services/quota-calculator-registry.service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAC;AAEpF,qBACa,uBAAuB;IAClC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA4C;IACnE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA0C;IAEtE,QAAQ,CAAC,aAAa,EAAE,MAAM,EAAE,UAAU,EAAE,mBAAmB,GAAG,IAAI;IAQtE,GAAG,CAAC,aAAa,EAAE,MAAM,GAAG,mBAAmB,GAAG,SAAS;IAI3D,GAAG,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO;CAGpC"}

package/dist/services/quota-calculator-registry.service.js

@@ -0,0 +1,33 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var QuotaCalculatorRegistry_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QuotaCalculatorRegistry = void 0;
+const common_1 = require("@nestjs/common");
+let QuotaCalculatorRegistry = QuotaCalculatorRegistry_1 = class QuotaCalculatorRegistry {
+    logger = new common_1.Logger(QuotaCalculatorRegistry_1.name);
+    calculators = new Map();
+    register(toolClassName, calculator) {
+        if (this.calculators.has(toolClassName)) {
+            this.logger.warn(`Quota calculator for "${toolClassName}" already registered, overriding`);
+        }
+        this.calculators.set(toolClassName, calculator);
+        this.logger.log(`Registered quota calculator for tool: ${toolClassName}`);
+    }
+    get(toolClassName) {
+        return this.calculators.get(toolClassName);
+    }
+    has(toolClassName) {
+        return this.calculators.has(toolClassName);
+    }
+};
+exports.QuotaCalculatorRegistry = QuotaCalculatorRegistry;
+exports.QuotaCalculatorRegistry = QuotaCalculatorRegistry = QuotaCalculatorRegistry_1 = __decorate([
+    (0, common_1.Injectable)()
+], QuotaCalculatorRegistry);
+//# sourceMappingURL=quota-calculator-registry.service.js.map

package/dist/services/quota-calculator-registry.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-calculator-registry.service.js","sourceRoot":"","sources":["../../src/services/quota-calculator-registry.service.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAAoD;AAI7C,IAAM,uBAAuB,+BAA7B,MAAM,uBAAuB;IACjB,MAAM,GAAG,IAAI,eAAM,CAAC,yBAAuB,CAAC,IAAI,CAAC,CAAC;IAClD,WAAW,GAAG,IAAI,GAAG,EAA+B,CAAC;IAEtE,QAAQ,CAAC,aAAqB,EAAE,UAA+B;QAC7D,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,aAAa,kCAAkC,CAAC,CAAC;QAC7F,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yCAAyC,aAAa,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED,GAAG,CAAC,aAAqB;QACvB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC7C,CAAC;IAED,GAAG,CAAC,aAAqB;QACvB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC7C,CAAC;CACF,CAAA;AAnBY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,mBAAU,GAAE;GACA,uBAAuB,CAmBnC"}

package/dist/services/quota-client.service.d.ts

@@ -0,0 +1,12 @@
+import Redis from 'ioredis';
+import { QuotaCheckResult, QuotaClientServiceInterface } from '../interfaces';
+export declare const QUOTA_REDIS = "QUOTA_REDIS";
+export declare class QuotaClientService implements QuotaClientServiceInterface {
+    private readonly redis;
+    private readonly logger;
+    constructor(redis: Redis | null);
+    checkQuota(userId: string, quotaType: string): Promise<QuotaCheckResult>;
+    report(userId: string, quotaType: string, amount: number): Promise<void>;
+    private key;
+}
+//# sourceMappingURL=quota-client.service.d.ts.map

package/dist/services/quota-client.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-client.service.d.ts","sourceRoot":"","sources":["../../src/services/quota-client.service.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,MAAM,SAAS,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,MAAM,eAAe,CAAC;AAE9E,eAAO,MAAM,WAAW,gBAAgB,CAAC;AAEzC,qBACa,kBAAmB,YAAW,2BAA2B;IAMlE,OAAO,CAAC,QAAQ,CAAC,KAAK;IALxB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuC;gBAK3C,KAAK,EAAE,KAAK,GAAG,IAAI;IAGhC,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAsCxE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB9E,OAAO,CAAC,GAAG;CAGZ"}

package/dist/services/quota-client.service.js

@@ -0,0 +1,74 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var QuotaClientService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QuotaClientService = exports.QUOTA_REDIS = void 0;
+const common_1 = require("@nestjs/common");
+exports.QUOTA_REDIS = 'QUOTA_REDIS';
+let QuotaClientService = QuotaClientService_1 = class QuotaClientService {
+    redis;
+    logger = new common_1.Logger(QuotaClientService_1.name);
+    constructor(redis) {
+        this.redis = redis;
+    }
+    async checkQuota(userId, quotaType) {
+        if (!this.redis) {
+            return { exceeded: false, used: 0, limit: -1 };
+        }
+        const usedKey = this.key(userId, quotaType, 'used');
+        const limitKey = this.key(userId, quotaType, 'limit');
+        try {
+            const [usedStr, limitStr] = await Promise.all([this.redis.get(usedKey), this.redis.get(limitKey)]);
+            const used = usedStr ? parseInt(usedStr, 10) : 0;
+            if (limitStr == null) {
+                return { exceeded: true, used, limit: 0 };
+            }
+            const limit = parseInt(limitStr, 10);
+            if (limit === -1) {
+                return { exceeded: false, used, limit };
+            }
+            if (used >= limit) {
+                return { exceeded: true, used, limit };
+            }
+            return { exceeded: false, used, limit };
+        }
+        catch (error) {
+            this.logger.warn(`Quota check failed for user ${userId}, quota ${quotaType}: ${String(error)}. Allowing (fail-open).`);
+            return { exceeded: false, used: 0, limit: -1 };
+        }
+    }
+    async report(userId, quotaType, amount) {
+        if (!this.redis) {
+            return;
+        }
+        const usedKey = this.key(userId, quotaType, 'used');
+        try {
+            await this.redis.incrby(usedKey, amount);
+        }
+        catch (error) {
+            this.logger.warn(`Quota report failed for user ${userId}, quota ${quotaType}: ${String(error)}. Skipping (fail-open).`);
+        }
+    }
+    key(userId, quotaType, suffix) {
+        return `user:${userId}:quota:${quotaType}:${suffix}`;
+    }
+};
+exports.QuotaClientService = QuotaClientService;
+exports.QuotaClientService = QuotaClientService = QuotaClientService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Optional)()),
+    __param(0, (0, common_1.Inject)(exports.QUOTA_REDIS)),
+    __metadata("design:paramtypes", [Object])
+], QuotaClientService);
+//# sourceMappingURL=quota-client.service.js.map

package/dist/services/quota-client.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-client.service.js","sourceRoot":"","sources":["../../src/services/quota-client.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAsE;AAIzD,QAAA,WAAW,GAAG,aAAa,CAAC;AAGlC,IAAM,kBAAkB,0BAAxB,MAAM,kBAAkB;IAMV;IALF,MAAM,GAAG,IAAI,eAAM,CAAC,oBAAkB,CAAC,IAAI,CAAC,CAAC;IAE9D,YAGmB,KAAmB;QAAnB,UAAK,GAAL,KAAK,CAAc;IACnC,CAAC;IAEJ,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,SAAiB;QAChD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC;QACjD,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAEtD,IAAI,CAAC;YACH,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAEnG,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAGjD,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YAC5C,CAAC;YAED,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAGrC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjB,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;YAC1C,CAAC;YAED,IAAI,IAAI,IAAI,KAAK,EAAE,CAAC;gBAClB,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;YACzC,CAAC;YAED,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,+BAA+B,MAAM,WAAW,SAAS,KAAK,MAAM,CAAC,KAAK,CAAC,yBAAyB,CACrG,CAAC;YACF,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC;QACjD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,SAAiB,EAAE,MAAc;QAC5D,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAEpD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gCAAgC,MAAM,WAAW,SAAS,KAAK,MAAM,CAAC,KAAK,CAAC,yBAAyB,CACtG,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,MAAc,EAAE,SAAiB,EAAE,MAAc;QAC3D,OAAO,QAAQ,MAAM,UAAU,SAAS,IAAI,MAAM,EAAE,CAAC;IACvD,CAAC;CACF,CAAA;AAlEY,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,mBAAU,GAAE;IAKR,WAAA,IAAA,iBAAQ,GAAE,CAAA;IACV,WAAA,IAAA,eAAM,EAAC,mBAAW,CAAC,CAAA;;GALX,kBAAkB,CAkE9B"}

package/dist/services/quota.interceptor.d.ts

@@ -0,0 +1,14 @@
+import { ToolExecutionContext, ToolExecutionInterceptor, ToolResult } from '@loopstack/common';
+import { QuotaCalculatorRegistry } from './quota-calculator-registry.service';
+import { QuotaClientService } from './quota-client.service';
+export declare class QuotaInterceptor implements ToolExecutionInterceptor {
+    private readonly quotaClientService;
+    private readonly calculatorRegistry;
+    private readonly logger;
+    private readonly processingTimeCalculator;
+    constructor(quotaClientService: QuotaClientService, calculatorRegistry: QuotaCalculatorRegistry);
+    beforeExecute(context: ToolExecutionContext): Promise<void>;
+    afterExecute(context: ToolExecutionContext, result: ToolResult): Promise<void>;
+    onError(_context: ToolExecutionContext, _error: unknown): Promise<void>;
+}
+//# sourceMappingURL=quota.interceptor.d.ts.map

package/dist/services/quota.interceptor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota.interceptor.d.ts","sourceRoot":"","sources":["../../src/services/quota.interceptor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,wBAAwB,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/F,OAAO,EAAE,uBAAuB,EAAE,MAAM,qCAAqC,CAAC;AAC9E,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,qBACa,gBAAiB,YAAW,wBAAwB;IAK7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IALrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqC;IAC5D,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAAuC;gBAG7D,kBAAkB,EAAE,kBAAkB,EACtC,kBAAkB,EAAE,uBAAuB;IAGxD,aAAa,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAsB3D,YAAY,CAAC,OAAO,EAAE,oBAAoB,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAoB9E,OAAO,CAAC,QAAQ,EAAE,oBAAoB,EAAE,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;CAG9E"}

package/dist/services/quota.interceptor.js

@@ -0,0 +1,66 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var QuotaInterceptor_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QuotaInterceptor = void 0;
+const common_1 = require("@nestjs/common");
+const calculators_1 = require("../calculators");
+const quota_calculator_registry_service_1 = require("./quota-calculator-registry.service");
+const quota_client_service_1 = require("./quota-client.service");
+let QuotaInterceptor = QuotaInterceptor_1 = class QuotaInterceptor {
+    quotaClientService;
+    calculatorRegistry;
+    logger = new common_1.Logger(QuotaInterceptor_1.name);
+    processingTimeCalculator = new calculators_1.ProcessingTimeQuotaCalculator();
+    constructor(quotaClientService, calculatorRegistry) {
+        this.quotaClientService = quotaClientService;
+        this.calculatorRegistry = calculatorRegistry;
+    }
+    async beforeExecute(context) {
+        const userId = context.runContext.userId;
+        const timeCheck = await this.quotaClientService.checkQuota(userId, this.processingTimeCalculator.quotaType);
+        if (timeCheck.exceeded) {
+            throw new Error(`Quota exceeded for "${this.processingTimeCalculator.quotaType}": ${timeCheck.used}/${timeCheck.limit}`);
+        }
+        const toolClassName = context.tool.constructor.name;
+        const calculator = this.calculatorRegistry.get(toolClassName);
+        if (!calculator)
+            return;
+        const checkResult = await this.quotaClientService.checkQuota(userId, calculator.quotaType);
+        if (checkResult.exceeded) {
+            throw new Error(`Quota exceeded for "${calculator.quotaType}": ${checkResult.used}/${checkResult.limit}`);
+        }
+    }
+    async afterExecute(context, result) {
+        const userId = context.runContext.userId;
+        const timeUsage = this.processingTimeCalculator.calculateQuotaUsage(context, result);
+        if (timeUsage) {
+            await this.quotaClientService.report(userId, timeUsage.quotaType, timeUsage.actualAmount);
+        }
+        const toolClassName = context.tool.constructor.name;
+        const calculator = this.calculatorRegistry.get(toolClassName);
+        if (!calculator)
+            return;
+        const usage = calculator.calculateQuotaUsage(context, result);
+        if (!usage)
+            return;
+        await this.quotaClientService.report(userId, usage.quotaType, usage.actualAmount);
+    }
+    async onError(_context, _error) {
+    }
+};
+exports.QuotaInterceptor = QuotaInterceptor;
+exports.QuotaInterceptor = QuotaInterceptor = QuotaInterceptor_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [quota_client_service_1.QuotaClientService,
+        quota_calculator_registry_service_1.QuotaCalculatorRegistry])
+], QuotaInterceptor);
+//# sourceMappingURL=quota.interceptor.js.map

package/dist/services/quota.interceptor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota.interceptor.js","sourceRoot":"","sources":["../../src/services/quota.interceptor.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AAEpD,gDAA+D;AAC/D,2FAA8E;AAC9E,iEAA4D;AAGrD,IAAM,gBAAgB,wBAAtB,MAAM,gBAAgB;IAKR;IACA;IALF,MAAM,GAAG,IAAI,eAAM,CAAC,kBAAgB,CAAC,IAAI,CAAC,CAAC;IAC3C,wBAAwB,GAAG,IAAI,2CAA6B,EAAE,CAAC;IAEhF,YACmB,kBAAsC,EACtC,kBAA2C;QAD3C,uBAAkB,GAAlB,kBAAkB,CAAoB;QACtC,uBAAkB,GAAlB,kBAAkB,CAAyB;IAC3D,CAAC;IAEJ,KAAK,CAAC,aAAa,CAAC,OAA6B;QAC/C,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;QAGzC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;QAC5G,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CACb,uBAAuB,IAAI,CAAC,wBAAwB,CAAC,SAAS,MAAM,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,KAAK,EAAE,CACxG,CAAC;QACJ,CAAC;QAGD,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;QACpD,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC9D,IAAI,CAAC,UAAU;YAAE,OAAO;QAExB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC;QAC3F,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,uBAAuB,UAAU,CAAC,SAAS,MAAM,WAAW,CAAC,IAAI,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC;QAC5G,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAA6B,EAAE,MAAkB;QAClE,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;QAGzC,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACrF,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,SAAS,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAC5F,CAAC;QAGD,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;QACpD,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC9D,IAAI,CAAC,UAAU;YAAE,OAAO;QAExB,MAAM,KAAK,GAAG,UAAU,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC9D,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;IACpF,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAA8B,EAAE,MAAe;IAE7D,CAAC;CACF,CAAA;AAtDY,4CAAgB;2BAAhB,gBAAgB;IAD5B,IAAA,mBAAU,GAAE;qCAM4B,yCAAkB;QAClB,2DAAuB;GANnD,gBAAgB,CAsD5B"}

package/package.json

@@ -0,0 +1,53 @@
+{
+  "name": "@loopstack/quota",
+  "displayName": "Loopstack Quota Module",
+  "description": "Opt-in quota tracking and enforcement for Loopstack workflows",
+  "version": "0.20.7",
+  "license": "BSL",
+  "author": {
+    "name": "Jakob Klippel",
+    "url": "https://www.linkedin.com/in/jakob-klippel/"
+  },
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "nest build",
+    "compile": "tsc --noEmit",
+    "format": "prettier --write .",
+    "lint": "eslint .",
+    "test": "jest --passWithNoTests",
+    "watch": "nest build --watch"
+  },
+  "dependencies": {
+    "@loopstack/common": "^0.21.0",
+    "@nestjs/common": "^11.1.14",
+    "ioredis": "^5.6.1"
+  },
+  "devDependencies": {
+    "@nestjs/cli": "^11.0.16",
+    "@types/jest": "^30.0.0",
+    "jest": "^30.2.0",
+    "ts-jest": "^29.4.6"
+  },
+  "files": [
+    "dist"
+  ],
+  "jest": {
+    "moduleFileExtensions": [
+      "js",
+      "json",
+      "ts"
+    ],
+    "rootDir": "src",
+    "testRegex": ".*\\.spec\\.ts$",
+    "transform": {
+      "^.+\\.ts$": "ts-jest"
+    },
+    "collectCoverageFrom": [
+      "**/*.(t|j)s"
+    ],
+    "coverageDirectory": "../coverage",
+    "testEnvironment": "node",
+    "maxWorkers": 1
+  }
+}