@graphql-mesh/plugin-rate-limit 0.0.1-alpha-d8211f7b2.0

package/index.d.ts

@@ -0,0 +1,3 @@
+import { MeshPluginOptions, YamlConfig } from '@graphql-mesh/types';
+import { Plugin } from '@envelop/core';
+export default function useMeshRateLimit(options: MeshPluginOptions<YamlConfig.RateLimitPluginConfig>): Plugin;

package/index.js

@@ -0,0 +1,61 @@
+'use strict';
+
+const stringInterpolation = require('@graphql-mesh/string-interpolation');
+const graphql = require('graphql');
+const crossHelpers = require('@graphql-mesh/cross-helpers');
+
+function useMeshRateLimit(options) {
+    const pathRateLimitDef = new Map();
+    const tokenMap = new Map();
+    const timeouts = new Set();
+    if (options.config) {
+        options.config.forEach(config => {
+            pathRateLimitDef.set(`${config.type}.${config.field}`, config);
+        });
+    }
+    if (options.pubsub) {
+        const id = options.pubsub.subscribe('destroy', () => {
+            options.pubsub.unsubscribe(id);
+            timeouts.forEach(timeout => clearTimeout(timeout));
+        });
+    }
+    return {
+        onValidate(onValidateParams) {
+            onValidateParams.addValidationRule(validationContext => ({
+                Field: () => {
+                    const parentType = validationContext.getParentType();
+                    const fieldDef = validationContext.getFieldDef();
+                    const path = `${parentType.name}.${fieldDef.name}`;
+                    const rateLimitConfig = pathRateLimitDef.get(path);
+                    if (rateLimitConfig) {
+                        const identifier = stringInterpolation.stringInterpolator.parse(rateLimitConfig.identifier, {
+                            env: crossHelpers.process.env,
+                            context: onValidateParams.context,
+                        });
+                        const mapKey = `${identifier}-${path}`;
+                        let remainingTokens = tokenMap.get(mapKey);
+                        if (remainingTokens == null) {
+                            remainingTokens = rateLimitConfig.max;
+                            const timeout = setTimeout(() => {
+                                tokenMap.delete(mapKey);
+                                timeouts.delete(timeout);
+                            }, rateLimitConfig.ttl);
+                            timeouts.add(timeout);
+                        }
+                        if (remainingTokens === 0) {
+                            validationContext.reportError(new graphql.GraphQLError(`Rate limit of "${path}" exceeded for "${identifier}"`));
+                            // Remove this field from the selection set
+                            return null;
+                        }
+                        else {
+                            tokenMap.set(mapKey, remainingTokens - 1);
+                        }
+                    }
+                    return false;
+                },
+            }));
+        },
+    };
+}
+
+module.exports = useMeshRateLimit;

package/index.mjs

@@ -0,0 +1,59 @@
+import { stringInterpolator } from '@graphql-mesh/string-interpolation';
+import { GraphQLError } from 'graphql';
+import { process } from '@graphql-mesh/cross-helpers';
+
+function useMeshRateLimit(options) {
+    const pathRateLimitDef = new Map();
+    const tokenMap = new Map();
+    const timeouts = new Set();
+    if (options.config) {
+        options.config.forEach(config => {
+            pathRateLimitDef.set(`${config.type}.${config.field}`, config);
+        });
+    }
+    if (options.pubsub) {
+        const id = options.pubsub.subscribe('destroy', () => {
+            options.pubsub.unsubscribe(id);
+            timeouts.forEach(timeout => clearTimeout(timeout));
+        });
+    }
+    return {
+        onValidate(onValidateParams) {
+            onValidateParams.addValidationRule(validationContext => ({
+                Field: () => {
+                    const parentType = validationContext.getParentType();
+                    const fieldDef = validationContext.getFieldDef();
+                    const path = `${parentType.name}.${fieldDef.name}`;
+                    const rateLimitConfig = pathRateLimitDef.get(path);
+                    if (rateLimitConfig) {
+                        const identifier = stringInterpolator.parse(rateLimitConfig.identifier, {
+                            env: process.env,
+                            context: onValidateParams.context,
+                        });
+                        const mapKey = `${identifier}-${path}`;
+                        let remainingTokens = tokenMap.get(mapKey);
+                        if (remainingTokens == null) {
+                            remainingTokens = rateLimitConfig.max;
+                            const timeout = setTimeout(() => {
+                                tokenMap.delete(mapKey);
+                                timeouts.delete(timeout);
+                            }, rateLimitConfig.ttl);
+                            timeouts.add(timeout);
+                        }
+                        if (remainingTokens === 0) {
+                            validationContext.reportError(new GraphQLError(`Rate limit of "${path}" exceeded for "${identifier}"`));
+                            // Remove this field from the selection set
+                            return null;
+                        }
+                        else {
+                            tokenMap.set(mapKey, remainingTokens - 1);
+                        }
+                    }
+                    return false;
+                },
+            }));
+        },
+    };
+}
+
+export default useMeshRateLimit;

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@graphql-mesh/plugin-rate-limit",
+  "version": "0.0.1-alpha-d8211f7b2.0",
+  "sideEffects": false,
+  "peerDependencies": {
+    "graphql": "*"
+  },
+  "dependencies": {
+    "@envelop/core": "^2.3.2",
+    "@graphql-mesh/cross-helpers": "0.1.6",
+    "@graphql-mesh/string-interpolation": "0.3.0",
+    "@graphql-mesh/types": "0.77.0-alpha-d8211f7b2.0",
+    "tslib": "^2.4.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "Urigo/graphql-mesh",
+    "directory": "packages/plugins/response-cache"
+  },
+  "license": "MIT",
+  "main": "index.js",
+  "module": "index.mjs",
+  "typings": "index.d.ts",
+  "typescript": {
+    "definition": "index.d.ts"
+  },
+  "exports": {
+    ".": {
+      "require": "./index.js",
+      "import": "./index.mjs"
+    },
+    "./*": {
+      "require": "./*.js",
+      "import": "./*.mjs"
+    },
+    "./package.json": "./package.json"
+  }
+}