@teikyo/rate-limit 0.2.0 → 0.3.0

package/lib/index.d.ts

@@ -1,4 +1,5 @@
 import { type HTTPPlugin } from '@sozai/http-server';
+import type { Context } from 'hono';
 import { type Store } from 'hono-rate-limiter';
 export declare const RATE_LIMIT: import("@sozai/http-server").PluginName<"http:rate-limit", void>;
 export type RateLimitPluginParams = {
@@ -6,6 +7,8 @@ export type RateLimitPluginParams = {
     windowMs: number;
     /** Maximum number of requests per client within a window. */
     limit: number;
+    /** Bucket by IP (default), principal attribution, or a custom key with nullish IP fallback. */
+    key?: 'ip' | 'principal' | ((c: Context) => string | null | undefined | Promise<string | null | undefined>);
     /**
      * Custom store, defaults to an in-memory store. The plugin calls `shutdown()` on it when
      * the server closes, so do not pass a store shared with or owned by something else.
@@ -13,7 +16,7 @@ export type RateLimitPluginParams = {
     store?: Store;
 };
 /**
- * Rate limit every request by resolved client IP (IPv6 grouped by /64), answering `429` with a
+ * Rate limit requests by IP, principal attribution, or a custom key, answering `429` with a
  * `Retry-After` header past the limit. Health routes are never limited.
  */
 export declare function rateLimitPlugin(params: RateLimitPluginParams): HTTPPlugin<'http:rate-limit', void, []>;

package/lib/index.js

@@ -1,4 +1,5 @@
 import { pluginName } from '@sozai/http-server';
+import { attributionOf, getPrincipal } from '@teikyo/principal';
 import { MemoryStore, rateLimiter } from 'hono-rate-limiter';
 import { clientKey } from './key.js';
 export const RATE_LIMIT = pluginName()('http:rate-limit');
@@ -8,12 +9,12 @@ function assertPositiveInteger(name, value) {
     }
 }
 /**
- * Rate limit every request by resolved client IP (IPv6 grouped by /64), answering `429` with a
+ * Rate limit requests by IP, principal attribution, or a custom key, answering `429` with a
  * `Retry-After` header past the limit. Health routes are never limited.
  */ export function rateLimitPlugin(params) {
     assertPositiveInteger('windowMs', params.windowMs);
     assertPositiveInteger('limit', params.limit);
-    const { windowMs, limit } = params;
+    const { windowMs, limit, key = 'ip' } = params;
     return {
         name: RATE_LIMIT,
         dependsOn: [],
@@ -25,7 +26,16 @@ function assertPositiveInteger(name, value) {
                 standardHeaders: 'draft-7',
                 // IPv6 clients share one bucket per /64. A request with no resolvable client IP falls
                 // into one shared bucket.
-                keyGenerator: (c)=>clientKey(ctx.clientIP(c)),
+                keyGenerator: async (c)=>{
+                    if (typeof key === 'function') {
+                        const customKey = await key(c);
+                        if (customKey != null) return customKey;
+                    } else if (key === 'principal') {
+                        const principal = getPrincipal(c);
+                        if (principal != null) return `p:${attributionOf(principal).onBehalfOf}`;
+                    }
+                    return `ip:${clientKey(ctx.clientIP(c))}`;
+                },
                 store
             }));
             ctx.onClose(()=>store.shutdown?.());

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teikyo/rate-limit",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Rate limiting plugin for Sozai HTTP servers",
   "repository": {
     "type": "git",
@@ -19,17 +19,18 @@
     "lib/*"
   ],
   "dependencies": {
-    "@sozai/http-server": "^0.2.0",
+    "@sozai/http-server": "^0.2.1",
+    "@teikyo/principal": "^0.1.0",
     "hono-rate-limiter": "^0.5.4"
   },
   "devDependencies": {
-    "@types/node": "^26.6.5",
+    "@types/node": "^26.6.6",
     "del-cli": "^8.0.1",
-    "hono": "^4.13.13",
-    "vitest": "5.0.5"
+    "hono": "^4.13.17",
+    "vitest": "5.1.0"
   },
   "peerDependencies": {
-    "hono": "^4.13.13"
+    "hono": "^4.13.17"
   },
   "engines": {
     "node": ">=24"