A11yWatch/a11ywatch-core

import { makeExecutableSchema } from "@graphql-tools/schema";
import gql from "graphql-tag";
import { applyMiddleware } from "graphql-middleware";
import { resolvers } from "./resolvers";
import {
  user,
  pageInsights,
  feature,
  website,
  issue,
  page,
  query,
  mutation,
  subscription,
  history,
  analytic,
  payments,
  meta,
  input,
  invoice,
} from "./graph/gql_types";
import { getGqlRateLimitDirective } from "../web/limiters";

const typeDefs = gql`
  directive @rateLimit(
    max: Int
    window: String
    message: String
    identityArgs: [String]
    arrayLengthField: String
  ) on FIELD_DEFINITION

  ${invoice}
  ${pageInsights}
  ${meta}
  ${payments}
  ${user}
  ${feature}
  ${website}
  ${page}
  ${issue}
  ${analytic}
  ${history}
  ${query}
  ${input}
  ${mutation}
  ${subscription}
`;

const createScheme = () => {
  const rateLimit = getGqlRateLimitDirective();

  const scheme = {
    typeDefs,
    resolvers,
    schemaDirectives: {
      rateLimit,
    },
  };

  return applyMiddleware(makeExecutableSchema(scheme as any));
};

export { createScheme, typeDefs };