ayiin 2.0.39 → 2.0.41

package/dist/index.d.ts

@@ -1,6 +1,8 @@
+import Redis from "ioredis";
 import Methods from "./methods";
 export type { DurationInput, RateLimitStore, RateLimiterOptions, } from "./types";
 export default class Ayiin extends Methods {
     version: string;
-    constructor();
+    cache: Redis | undefined;
+    constructor(databaseUrl?: string);
 }

package/dist/index.js

@@ -3,12 +3,20 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+const ioredis_1 = __importDefault(require("ioredis"));
 const package_json_1 = __importDefault(require("../package.json"));
 const methods_1 = __importDefault(require("./methods"));
 class Ayiin extends methods_1.default {
     version = package_json_1.default.version;
-    constructor() {
+    cache;
+    constructor(databaseUrl) {
         super();
+        if (databaseUrl) {
+            this.cache = new ioredis_1.default(databaseUrl);
+        }
+        else {
+            this.cache = undefined;
+        }
     }
 }
 exports.default = Ayiin;

package/dist/methods/limiter.js

@@ -57,40 +57,53 @@ class Limiter extends tools_1.default {
                 const blockedUntil = new Date(blockedUntilStr).getTime();
                 const remainingMs = blockedUntil - now;
                 if (remainingMs > 0) {
-                    return res.status(429).json({
-                        success: false,
-                        error: `Floodwait aktif untuk ${options.keyPrefix}. Tunggu ${this.formatRemainingTime(remainingMs)}.`,
-                    });
+                    const msg = typeof options.messageFormat === "function"
+                        ? options.messageFormat({
+                            api: options.keyPrefix,
+                            remainMs: remainingMs,
+                            remain: this.formatRemainingTime(remainingMs),
+                            limit: options.limit,
+                        })
+                        : (options.messageFormat ??
+                            `[FloodWait] - Silakan coba lagi dalam ${this.formatRemainingTime(remainingMs)}.`);
+                    return res.status(429).json({ success: false, error: msg });
                 }
             }
             const rateKey = `${options.keyPrefix}:rate:${userId}`;
             const hits = await store.incr(rateKey);
             const windowMs = this.toMilliseconds(options.windowMs);
-            const baseFloodwaitMs = this.toMilliseconds(options.floodwaitMs);
-            const levelTtlMs = this.toMilliseconds(options.levelTtl ?? { hours: 24 });
+            const refreshAtMs = this.toMilliseconds(options.refreshAt ?? { hours: 24 });
             if (hits === 1) {
                 await store.expire(rateKey, windowMs);
             }
             if (hits > options.limit) {
-                let floodwaitMs = baseFloodwaitMs;
-                if (options.levels && options.levels.length > 0) {
-                    let level = parseInt((await store.get(levelKey)) ?? "0", 10);
-                    level += 1;
-                    // simpan level dengan TTL (reset otomatis setelah levelTtlMs)
-                    await store.set(levelKey, level.toString(), levelTtlMs);
-                    // pilih durasi floodwait dari config levels
-                    const idx = Math.min(level - 1, options.levels.length - 1);
-                    const levelDuration = options.levels[idx];
-                    if (levelDuration) {
-                        floodwaitMs = this.toMilliseconds(levelDuration);
-                    }
+                let floodwaitMs;
+                let level;
+                if (Array.isArray(options.floodwaitMs)) {
+                    // escalation mode
+                    level = parseInt((await store.get(levelKey)) ?? "0", 10) + 1;
+                    await store.set(levelKey, level.toString(), refreshAtMs);
+                    const idx = Math.min(level - 1, options.floodwaitMs.length - 1);
+                    const floodwait = options.floodwaitMs[idx];
+                    floodwaitMs = this.toMilliseconds(floodwait ?? options.floodwaitMs[0]);
+                }
+                else {
+                    // fixed mode
+                    floodwaitMs = this.toMilliseconds(options.floodwaitMs);
                 }
                 const untilDate = new Date(now + floodwaitMs).toISOString();
                 await store.set(floodKey, untilDate, floodwaitMs);
-                return res.status(429).json({
-                    success: false,
-                    error: `Terlalu banyak request ${options.keyPrefix}. Floodwait aktif selama ${this.formatRemainingTime(floodwaitMs)}.`,
-                });
+                const msg = typeof options.messageFormat === "function"
+                    ? options.messageFormat({
+                        api: options.keyPrefix,
+                        remainMs: floodwaitMs,
+                        remain: this.formatRemainingTime(floodwaitMs),
+                        level,
+                        limit: options.limit,
+                    })
+                    : (options.messageFormat ??
+                        `[FloodWait] - Terlalu banyak permintaan. Silakan coba lagi dalam ${this.formatRemainingTime(floodwaitMs)}.`);
+                return res.status(429).json({ success: false, error: msg });
             }
             next();
         };

package/dist/types.d.ts

@@ -1,4 +1,10 @@
 import { Request, Response } from "express";
+export interface RateLimitStore {
+    get(key: string): Promise<string | undefined>;
+    set(key: string, value: string, ttlMs?: number): Promise<void>;
+    incr(key: string): Promise<number>;
+    expire(key: string, ttlMs: number): Promise<void>;
+}
 export interface DurationInput {
     hours?: number;
     minutes?: number;
@@ -8,15 +14,15 @@ export interface RateLimiterOptions {
     keyPrefix: string;
     limit: number;
     windowMs: DurationInput;
-    floodwaitMs: DurationInput;
+    floodwaitMs: DurationInput | DurationInput[];
     identifyUser?: (req: Request, res: Response) => string | Response<any, Record<string, any>>;
     store?: RateLimitStore;
-    levels?: DurationInput[];
-    levelTtl?: DurationInput;
-}
-export interface RateLimitStore {
-    get(key: string): Promise<string | undefined>;
-    set(key: string, value: string, ttlMs?: number): Promise<void>;
-    incr(key: string): Promise<number>;
-    expire(key: string, ttlMs: number): Promise<void>;
+    refreshAt?: DurationInput;
+    messageFormat?: string | ((ctx: {
+        api: string;
+        remainMs: number;
+        remain: string;
+        level?: number;
+        limit?: number;
+    }) => string);
 }

package/package.json

@@ -1,30 +1,33 @@
 {
   "name": "ayiin",
-  "version": "2.0.39",
-  "description": "Library Pribadi Yang Gak Ada Isinya",
-  "main": "./dist/index.js",
-  "types": "./dist/index.d.ts",
-  "bin": {
-    "axd": "./dist/bin/cli.js"
-  },
-  "scripts": {
-    "build": "tsc"
+  "version": "2.0.41",
+  "author": {
+    "name": "AyiinXd",
+    "url": "https://github.com/AyiinXd"
   },
   "repository": {
     "type": "git",
     "url": "git+https://github.com/AyiinXd/ayiin.git"
   },
-  "author": {
-    "name": "AyiinXd",
-    "url": "https://github.com/AyiinXd"
+  "main": "./dist/index.js",
+  "devDependencies": {
+    "@types/express": "^5.0.6",
+    "@types/node": "^26.3.0",
+    "typescript": "^7"
+  },
+  "bin": {
+    "axd": "./dist/bin/cli.js"
   },
+  "description": "Library Pribadi Yang Gak Ada Isinya",
   "keywords": [
     "ayiin"
   ],
   "license": "GPL-3.0-only",
-  "devDependencies": {
-    "@types/express": "^5.0.6",
-    "@types/node": "^26.3.0",
-    "typescript": "^7"
+  "scripts": {
+    "build": "tsc"
+  },
+  "types": "./dist/index.d.ts",
+  "dependencies": {
+    "ioredis": "^6.0.0"
   }
 }