1 files changed, 7 insertions, 5 deletions
diff --git a/app.js b/app.js
index 9a466d8..a834f26 100644
--- a/app.js
+++ b/app.js
@@ -1,6 +1,6 @@
 const express = require('express')
 const app = express()
-const limit = require('express-limit').limit
+const rateLimit = require('express-rate-limit')
 const helmet = require('helmet')
 const bp = require('body-parser')
 const cookieParser = require('cookie-parser')
@@ -32,6 +32,11 @@ const assetDir = [
     'img',
     'vid'
 ]
+const limiter = rateLimit({
+    windowMs: 30 * 1000,
+    max: 100,
+    message: '<h1>Rate Limit Exceeded</h1><p>You have exceeded the allowed number of requests. Please try again later.</p>',
+})
 require('dotenv').config()
 const rss = require('./source/rss')({baseUrl: process.env.baseUrl})
 
@@ -52,10 +57,7 @@ class App {
         app.use(cookieParser())
         if (process.env.NODE_ENV === 'prod') {
             app.use(helmet())
-            app.use(limit({
-                max: 100, // 100 requests
-                period: 300 * 1000 // per minute (five minutes)
-            }))
+            app.use(limiter)
         }
 
         app.route('*')