import { rateLimiter as rateLimit } from "hono-rate-limiter";
import sha256 from "../utils/sha256";

const getIp = (req: Request): string => {
  const ips = (
    req.headers.get("cf-connecting-ip") ||
    req.headers.get("x-real-ip") ||
    req.headers.get("x-forwarded-for") ||
    "unknown"
  ).split(",");

  return ips[0].trim();
};

const hashedIp = (req: Request): string => sha256(getIp(req)).slice(0, 25);

interface RateLimiterOptions {
  requests: number;
}

const rateLimiter = ({ requests }: RateLimiterOptions) => {
  return rateLimit({
    keyGenerator: (c) => {
      const pathHash = sha256(c.req.path).slice(0, 25);
      const key = `rate-limit:${pathHash}:${hashedIp(c.req.raw)}`;
      return key;
    }, // 1 minute
    limit: requests,
    standardHeaders: "draft-6",
    windowMs: 1 * 60 * 1000
  });
};

export default rateLimiter;