From aa33c86af09afc524833e4ff5f5764c2c3be553c Mon Sep 17 00:00:00 2001
From: marzavec <admin@marzavec.com>
Date: Tue, 13 Mar 2018 23:22:23 -0700
Subject: Added unban by hash

---
 server/src/core/rateLimiter.js | 22 ++++++++++------------
 1 file changed, 10 insertions(+), 12 deletions(-)

(limited to 'server/src/core')

diff --git a/server/src/core/rateLimiter.js b/server/src/core/rateLimiter.js
index f5c739b..0d94ef2 100644
--- a/server/src/core/rateLimiter.js
+++ b/server/src/core/rateLimiter.js
@@ -18,6 +18,7 @@ class Police {
     this._records = {};
     this._halflife = 30000; // ms
     this._threshold = 25;
+    this._hashes = [];
   }
 
   /**
@@ -76,12 +77,11 @@ class Police {
     *
     * @memberof Police
     */
-  arrest (id) {
-    var record = this.search(id);
+  arrest (id, hash) {
+    let record = this.search(id);
 
-    if (record) {
-      record.arrested = true;
-    }
+    record.arrested = true;
+    this._hashes[hash] = id;
   }
 
   /**
@@ -93,14 +93,12 @@ class Police {
     * @memberof Police
     */
   pardon (id) {
-    var record = this.search(id);
-
-    if (record) {
-      record.arrested = false;
-      return true;
+    if (typeof this._hashes[id] !== 'undefined') {
+      id = this._hashes[id];
     }
-
-    return false;
+    
+    let record = this.search(id);
+    record.arrested = false;
   }
 }
 
-- 
cgit v1.2.1