Allow rate limiting to be disabled

DopplerUniversity · Oct 2, 2020 · cfddc83 · cfddc83
1 parent b360096
commit cfddc83
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/sample.env b/sample.env
@@ -6,3 +6,4 @@ PORT="3000"
 TRANSLATION_SUGGESTION="Secrets must not be stored in git repositories"
 YODA_TRANSLATE_API_ENDPOINT="https://api.funtranslations.com/translate/yoda.json"
 YODA_TRANSLATE_API_KEY=""
+RATE_LIMITING_ENABLED="false"
diff --git a/src/config.js b/src/config.js
@@ -22,6 +22,7 @@ const config = Object.freeze({
     TRANSLATION_SUGGESTION: process.env.TRANSLATION_SUGGESTION,
     YODA_TRANSLATE_API_ENDPOINT: process.env.YODA_TRANSLATE_API_ENDPOINT,
     YODA_TRANSLATE_API_KEY: process.env.YODA_TRANSLATE_API_KEY,
+    RATE_LIMITING_ENABLED: process.env.RATE_LIMITING_ENABLED === 'true' ? true : false,
 })
 
 export default config
diff --git a/src/router.js b/src/router.js
@@ -14,9 +14,13 @@ const PREDEFINED_TRANSLATIONS = {
 
 const app = express()
 const router = express.Router()
+
+if (config.RATE_LIMITING_ENABLED) {
+    console.log(colors.green('[info]: rate limiting enabled'))
+}
 const translationLimiter = rateLimit({
     windowMs: 60 * 60 * 1000,
-    max: 60,
+    max: config.RATE_LIMITING_ENABLED ? 60 : 0,
     handler: (req, res) => {
         res.json({
             text: req.body.text,