chore(klesia): adjust rate limiter

palladians · Sep 3, 2024 · c9eda78 · c9eda78
1 parent 41f8605
commit c9eda78
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/apps/klesia/src/index.ts b/apps/klesia/src/index.ts
@@ -19,7 +19,11 @@ const api = new OpenAPIHono();
 api.use(logger());
 api.use(
 	rateLimiter({
-		keyGenerator: (c) => getConnInfo(c).remote.address ?? nanoid(),
+		skip: (c) => c.req.path !== "/api" || c.req.method !== "POST",
+		keyGenerator: (c) =>
+			c.req.header("x-forwarded-for") ??
+			getConnInfo(c).remote.address ??
+			nanoid(),
 		limit: 10,
 	}),
 );

diff --git a/packages/klesia-sdk/src/client.ts b/packages/klesia-sdk/src/client.ts
@@ -5,13 +5,15 @@ import { z } from "zod";
 
 const NetworkMatcher = z.enum(["mainnet", "devnet"]);
 
-export const createClient = ({ network }: { network: 'mainnet' | 'devnet' }) => {
+export const createClient = ({
+	network,
+}: { network: "mainnet" | "devnet" }) => {
 	return match(NetworkMatcher.parse(network))
 		.with("devnet", () =>
 			hc<KlesiaRpc>("https://devnet.klesia.palladians.xyz/api"),
 		)
 		.with("mainnet", () =>
 			hc<KlesiaRpc>("https://mainnet.klesia.palladians.xyz/api"),
 		)
-	  .exhaustive();
+		.exhaustive();
 };