whisher · November 7, 2015 20:10 · Nov 7, 2015
diff --git a/gistfile1.txt b/gistfile1.txt
@@ -0,0 +1,61 @@
+'use strict';
+
+/**
+ * Module dependencies.
+ */
+const Hoek = require('hoek');
+const Limiter = require('ratelimiter');
+const RedisClient = require('../utils/redis');
+const ReplyUtil = require('../utils/reply');
+
+const internals = {};
+internals.defaults = {
+    namespace: 'hrl',
+    global: {
+        limit: -1,
+        duration: 1
+    }
+};
+const MILLISECONDS = 1000;
+
+exports.register = (server, options, next) => {
+    const settings = Hoek.applyToDefaults(internals.defaults, options);
+    server.ext('onPreAuth', (request, reply) => {
+        const route = request.route;
+        const currentSettings = route.settings.plugins && route.settings.plugins['hapi-rate-limit'];
+        if (currentSettings) {
+            const routeSettings = Hoek.applyToDefaults(settings.global, currentSettings);
+            const id = settings.namespace + ':' + request.info.remoteAddress + ':' + route.path;
+            const limiter = new Limiter({
+                id: id,
+                db: RedisClient,
+                max: routeSettings.limit,
+                duration: routeSettings.duration * MILLISECONDS
+            });
+            limiter.get((err, limit) => {
+                if (err) {
+                    return reply(err);
+                }
+                request.plugins['hapi-rate-limit'] = {};
+                request.plugins['hapi-rate-limit'].limit = limit.total;
+                request.plugins['hapi-rate-limit'].remaining = limit.remaining - 1;
+                request.plugins['hapi-rate-limit'].reset = limit.reset;
+                if (limit.remaining <= 0) {
+                    const  error = ReplyUtil.tooManyRequests();
+                    error.output.headers['X-RateLimit-Limit'] =  request.plugins['hapi-rate-limit'].limit;
+                    error.output.headers['X-RateLimit-Remaining'] = request.plugins['hapi-rate-limit'].remaining;
+                    error.output.headers['X-RateLimit-Reset'] = request.plugins['hapi-rate-limit'].reset;
+                    return reply(error);
+                }
+                return reply.continue();
+            });
+        } else {
+            return reply.continue();
+        }
+    });
+    return next();
+};
+exports.register.attributes = {
+    name: 'hapi-rate-limit',
+    version: '1.0.0'
+};