@kewacode/guard 1.0.2 → 1.0.4

package/kewa-guard.module.js

@@ -0,0 +1,44 @@
+"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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var KewaGuardModule_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KewaGuardModule = void 0;
+const common_1 = require("@nestjs/common");
+const ioredis_1 = __importDefault(require("ioredis"));
+const constants_1 = require("./constants");
+const kewa_guard_service_1 = require("./kewa-guard.service");
+let KewaGuardModule = KewaGuardModule_1 = class KewaGuardModule {
+    static register(options) {
+        return {
+            module: KewaGuardModule_1,
+            providers: [
+                {
+                    provide: constants_1.KEWA_REDIS_CLIENT,
+                    useFactory: () => {
+                        return new ioredis_1.default({
+                            host: options.host,
+                            port: options.port,
+                            password: options.password,
+                        });
+                    },
+                },
+                kewa_guard_service_1.KewaGuardService,
+            ],
+            exports: [kewa_guard_service_1.KewaGuardService],
+        };
+    }
+};
+exports.KewaGuardModule = KewaGuardModule;
+exports.KewaGuardModule = KewaGuardModule = KewaGuardModule_1 = __decorate([
+    (0, common_1.Global)(),
+    (0, common_1.Module)({})
+], KewaGuardModule);
+//# sourceMappingURL=kewa-guard.module.js.map

package/kewa-guard.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kewa-guard.module.js","sourceRoot":"","sources":["../../../libs/kewa-guard/src/kewa-guard.module.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAA+D;AAC/D,sDAA4B;AAC5B,2CAAgD;AAChD,6DAAwD;AAKjD,IAAM,eAAe,uBAArB,MAAM,eAAe;IAC1B,MAAM,CAAC,QAAQ,CAAC,OAAyB;QACvC,OAAO;YACL,MAAM,EAAE,iBAAe;YACvB,SAAS,EAAE;gBACT;oBACE,OAAO,EAAE,6BAAiB;oBAC1B,UAAU,EAAE,GAAG,EAAE;wBACf,OAAO,IAAI,iBAAK,CAAC;4BACf,IAAI,EAAE,OAAO,CAAC,IAAI;4BAClB,IAAI,EAAE,OAAO,CAAC,IAAI;4BAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;yBAC3B,CAAC,CAAC;oBACL,CAAC;iBACF;gBACD,qCAAgB;aACjB;YACD,OAAO,EAAE,CAAC,qCAAgB,CAAC;SAC5B,CAAC;IACJ,CAAC;CACF,CAAA;AApBY,0CAAe;0BAAf,eAAe;IAF3B,IAAA,eAAM,GAAE;IACR,IAAA,eAAM,EAAC,EAAE,CAAC;GACE,eAAe,CAoB3B"}

package/kewa-guard.service.d.ts

@@ -0,0 +1,17 @@
+import { OnModuleDestroy } from '@nestjs/common';
+import Redis from 'ioredis';
+interface RateLimitResult {
+    allowed: boolean;
+    remaining: number;
+    resetAt: number;
+}
+export declare class KewaGuardService implements OnModuleDestroy {
+    private readonly redis;
+    constructor(redis: Redis);
+    onModuleDestroy(): void;
+    checkLimit(key: string, limit: number, ttlSeconds: number): Promise<RateLimitResult>;
+    ping(): Promise<"PONG">;
+    getClient(): Redis;
+}
+export {};
+//# sourceMappingURL=kewa-guard.service.d.ts.map

package/kewa-guard.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"kewa-guard.service.d.ts","sourceRoot":"","sources":["../../../libs/kewa-guard/src/kewa-guard.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACrE,OAAO,KAAK,MAAM,SAAS,CAAC;AAG5B,UAAU,eAAe;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,qBACa,gBAAiB,YAAW,eAAe;IACf,OAAO,CAAC,QAAQ,CAAC,KAAK;gBAAL,KAAK,EAAE,KAAK;IAEpE,eAAe;IAIT,UAAU,CACd,GAAG,EAAE,MAAM,EACX,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,eAAe,CAAC;IAkDrB,IAAI;IAIV,SAAS;CAGV"}

package/kewa-guard.service.js

@@ -0,0 +1,81 @@
+"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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KewaGuardService = void 0;
+const common_1 = require("@nestjs/common");
+const ioredis_1 = __importDefault(require("ioredis"));
+const constants_1 = require("./constants");
+let KewaGuardService = class KewaGuardService {
+    redis;
+    constructor(redis) {
+        this.redis = redis;
+    }
+    onModuleDestroy() {
+        this.redis.disconnect();
+    }
+    async checkLimit(key, limit, ttlSeconds) {
+        const now = Date.now();
+        const windowStart = now - ttlSeconds * 1000;
+        const luaScript = `
+      local key = KEYS[1]
+      local limit = tonumber(ARGV[1])
+      local now = tonumber(ARGV[2])
+      local windowStart = tonumber(ARGV[3])
+      local ttl = tonumber(ARGV[4])
+
+      redis.call('ZREMRANGEBYSCORE', key, 0, windowStart)
+
+      local currentCount = redis.call('ZCARD', key)
+
+      local allowed = 1
+      local remaining = 0
+
+      if currentCount >= limit then
+        allowed = 0
+        remaining = 0
+      else
+        redis.call('ZADD', key, now, now)
+        redis.call('EXPIRE', key, ttl)
+        allowed = 1
+        remaining = limit - (currentCount + 1)
+      end
+
+      local resetAt = math.floor((now + (ttl * 1000)) / 1000)
+      
+      return { allowed, remaining, resetAt }
+    `;
+        const result = (await this.redis.eval(luaScript, 1, key, limit, now, windowStart, ttlSeconds));
+        return {
+            allowed: result[0] === 1,
+            remaining: result[1],
+            resetAt: result[2],
+        };
+    }
+    async ping() {
+        return await this.redis.ping();
+    }
+    getClient() {
+        return this.redis;
+    }
+};
+exports.KewaGuardService = KewaGuardService;
+exports.KewaGuardService = KewaGuardService = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Inject)(constants_1.KEWA_REDIS_CLIENT)),
+    __metadata("design:paramtypes", [ioredis_1.default])
+], KewaGuardService);
+//# sourceMappingURL=kewa-guard.service.js.map

package/kewa-guard.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kewa-guard.service.js","sourceRoot":"","sources":["../../../libs/kewa-guard/src/kewa-guard.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AAAA,2CAAqE;AACrE,sDAA4B;AAC5B,2CAAgD;AASzC,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAC6B;IAAxD,YAAwD,KAAY;QAAZ,UAAK,GAAL,KAAK,CAAO;IAAG,CAAC;IAExE,eAAe;QACb,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,UAAU,CACd,GAAW,EACX,KAAa,EACb,UAAkB;QAElB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,GAAG,GAAG,UAAU,GAAG,IAAI,CAAC;QAE5C,MAAM,SAAS,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;KA2BjB,CAAC;QAEF,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CACnC,SAAS,EACT,CAAC,EACD,GAAG,EACH,KAAK,EACL,GAAG,EACH,WAAW,EACX,UAAU,CACX,CAA6B,CAAC;QAE/B,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;YACxB,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;YACpB,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;SACnB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;CACF,CAAA;AApEY,4CAAgB;2BAAhB,gBAAgB;IAD5B,IAAA,mBAAU,GAAE;IAEE,WAAA,IAAA,eAAM,EAAC,6BAAiB,CAAC,CAAA;qCAAyB,iBAAK;GADzD,gBAAgB,CAoE5B"}

package/package.json

@@ -1,14 +1,15 @@
 {
   "name": "@kewacode/guard",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "Distributed Rate Limiter for NestJS using Redis",
   "author": "João Bertan",
   "license": "MIT",
   "types": "./index.d.ts",
+  "main": "./index.js",
   "peerDependencies": {
     "@nestjs/common": "^10.0.0 || ^11.0.0",
     "@nestjs/core": "^10.0.0 || ^11.0.0",
-    "ioredis": "5.0.0",
+    "ioredis": "^5.0.0",
     "rxjs": "^7.0.0"
   },
   "repository": {