A11yWatch/a11ywatch-core

import { redisClient } from "../../database/memory-client";

import {
  createRateLimitDirective,
  RedisStore as GraphQLRedisStore,
  getGraphQLRateLimiter,
} from "graphql-rate-limit";

const limiter = {
  config: {
    rateLimit: {
      max: 4,
      timeWindow: "1 minute",
    },
  },
};

const registerLimiter = {
  config: {
    rateLimit: {
      max: 4,
      timeWindow: "30 minutes",
    },
  },
};

// crawl limiter
const scanLimiter = {
  config: {
    rateLimit: {
      max: 10000,
      timeWindow: "1 minute",
    },
  },
};

let gqlRateLimiter; // graphql rate limit

const getGqlRateLimitDirective = () => {
  try {
    const rateLimitOptions = {
      identifyContext: (ctx) =>
        ctx?.request?.ipAddress || ctx.id || (ctx.user && ctx.user.id),
      formatError: ({ fieldName, window }) =>
        `Rate limited exceeded for ${fieldName}. Please wait ${
          window / 1000
        }s and try again`,
      store: new GraphQLRedisStore(redisClient),
    };

    gqlRateLimiter = getGraphQLRateLimiter(rateLimitOptions);

    return createRateLimitDirective(rateLimitOptions);
  } catch (e) {
    console.error(e);
  }
};

export {
  gqlRateLimiter,
  limiter,
  registerLimiter,
  scanLimiter,
  getGqlRateLimitDirective,
};