@envelop/rate-limiter 3.5.0 → 3.5.1-alpha-20220824152235-afd061e6

package/cjs/index.js

@@ -4,6 +4,7 @@ exports.useRateLimiter = exports.DIRECTIVE_SDL = exports.UnauthenticatedError =
 const tslib_1 = require("tslib");
 const utils_js_1 = require("./utils.js");
 const graphql_rate_limit_1 = require("graphql-rate-limit");
+const traced_schema_js_1 = require("./traced-schema.js");
 tslib_1.__exportStar(require("./utils.js"), exports);
 class UnauthenticatedError extends Error {
 }
@@ -14,6 +15,9 @@ exports.DIRECTIVE_SDL = `
 const useRateLimiter = (options) => {
     const rateLimiterFn = (0, graphql_rate_limit_1.getGraphQLRateLimiter)({ identifyContext: options.identifyFn });
     return {
+        onSchemaChange: ({ schema }) => {
+            (0, traced_schema_js_1.prepareTracedSchema)(schema);
+        },
         async onContextBuilding({ extendContext }) {
             extendContext({
                 rateLimiterFn,

package/cjs/traced-schema.js

@@ -0,0 +1,68 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.prepareTracedSchema = void 0;
+const graphql_1 = require("graphql");
+const core_1 = require("@envelop/core");
+const trackedSchemaSymbol = Symbol('TRACKED_SCHEMA');
+function prepareTracedSchema(schema) {
+    if (!schema || schema[trackedSchemaSymbol]) {
+        return;
+    }
+    schema[trackedSchemaSymbol] = true;
+    const entries = Object.values(schema.getTypeMap());
+    for (const type of entries) {
+        if (!(0, graphql_1.isIntrospectionType)(type) && (0, graphql_1.isObjectType)(type)) {
+            const fields = Object.values(type.getFields());
+            for (const field of fields) {
+                let resolverFn = (field.resolve || graphql_1.defaultFieldResolver);
+                field.resolve = async (root, args, context, info) => {
+                    if (context && context[core_1.resolversHooksSymbol]) {
+                        const hooks = context[core_1.resolversHooksSymbol];
+                        const afterCalls = [];
+                        for (const hook of hooks) {
+                            const afterFn = await hook({
+                                root,
+                                args,
+                                context,
+                                info,
+                                resolverFn,
+                                replaceResolverFn: newFn => {
+                                    resolverFn = newFn;
+                                },
+                            });
+                            afterFn && afterCalls.push(afterFn);
+                        }
+                        try {
+                            let result = await resolverFn(root, args, context, info);
+                            for (const afterFn of afterCalls) {
+                                afterFn({
+                                    result,
+                                    setResult: newResult => {
+                                        result = newResult;
+                                    },
+                                });
+                            }
+                            return result;
+                        }
+                        catch (e) {
+                            let resultErr = e;
+                            for (const afterFn of afterCalls) {
+                                afterFn({
+                                    result: resultErr,
+                                    setResult: newResult => {
+                                        resultErr = newResult;
+                                    },
+                                });
+                            }
+                            throw resultErr;
+                        }
+                    }
+                    else {
+                        return resolverFn(root, args, context, info);
+                    }
+                };
+            }
+        }
+    }
+}
+exports.prepareTracedSchema = prepareTracedSchema;

package/esm/index.js

@@ -1,5 +1,6 @@
 import { getDirective } from './utils.js';
 import { getGraphQLRateLimiter } from 'graphql-rate-limit';
+import { prepareTracedSchema } from './traced-schema.js';
 export * from './utils.js';
 export class UnauthenticatedError extends Error {
 }
@@ -9,6 +10,9 @@ export const DIRECTIVE_SDL = /* GraphQL */ `
 export const useRateLimiter = (options) => {
     const rateLimiterFn = getGraphQLRateLimiter({ identifyContext: options.identifyFn });
     return {
+        onSchemaChange: ({ schema }) => {
+            prepareTracedSchema(schema);
+        },
         async onContextBuilding({ extendContext }) {
             extendContext({
                 rateLimiterFn,

package/esm/traced-schema.js

@@ -0,0 +1,64 @@
+import { defaultFieldResolver, isIntrospectionType, isObjectType } from 'graphql';
+import { resolversHooksSymbol } from '@envelop/core';
+const trackedSchemaSymbol = Symbol('TRACKED_SCHEMA');
+export function prepareTracedSchema(schema) {
+    if (!schema || schema[trackedSchemaSymbol]) {
+        return;
+    }
+    schema[trackedSchemaSymbol] = true;
+    const entries = Object.values(schema.getTypeMap());
+    for (const type of entries) {
+        if (!isIntrospectionType(type) && isObjectType(type)) {
+            const fields = Object.values(type.getFields());
+            for (const field of fields) {
+                let resolverFn = (field.resolve || defaultFieldResolver);
+                field.resolve = async (root, args, context, info) => {
+                    if (context && context[resolversHooksSymbol]) {
+                        const hooks = context[resolversHooksSymbol];
+                        const afterCalls = [];
+                        for (const hook of hooks) {
+                            const afterFn = await hook({
+                                root,
+                                args,
+                                context,
+                                info,
+                                resolverFn,
+                                replaceResolverFn: newFn => {
+                                    resolverFn = newFn;
+                                },
+                            });
+                            afterFn && afterCalls.push(afterFn);
+                        }
+                        try {
+                            let result = await resolverFn(root, args, context, info);
+                            for (const afterFn of afterCalls) {
+                                afterFn({
+                                    result,
+                                    setResult: newResult => {
+                                        result = newResult;
+                                    },
+                                });
+                            }
+                            return result;
+                        }
+                        catch (e) {
+                            let resultErr = e;
+                            for (const afterFn of afterCalls) {
+                                afterFn({
+                                    result: resultErr,
+                                    setResult: newResult => {
+                                        resultErr = newResult;
+                                    },
+                                });
+                            }
+                            throw resultErr;
+                        }
+                    }
+                    else {
+                        return resolverFn(root, args, context, info);
+                    }
+                };
+            }
+        }
+    }
+}

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "@envelop/rate-limiter",
-  "version": "3.5.0",
+  "version": "3.5.1-alpha-20220824152235-afd061e6",
   "sideEffects": false,
   "peerDependencies": {
-    "@envelop/core": "^2.5.0",
+    "@envelop/types": "2.3.2-alpha-20220824152235-afd061e6",
+    "@envelop/core": "2.5.1-alpha-20220824152235-afd061e6",
     "graphql": "^14.0.0 || ^15.0.0 || ^16.0.0"
   },
   "dependencies": {

package/typings/traced-schema.d.cts

@@ -0,0 +1,2 @@
+import { GraphQLSchema } from 'graphql';
+export declare function prepareTracedSchema(schema: GraphQLSchema | null | undefined): void;

package/typings/traced-schema.d.ts

@@ -0,0 +1,2 @@
+import { GraphQLSchema } from 'graphql';
+export declare function prepareTracedSchema(schema: GraphQLSchema | null | undefined): void;