From 6cb0f90db28f22712780c30d43145b60d1a69578 Mon Sep 17 00:00:00 2001
From: Mcat12 <newtoncat12@yahoo.com>
Date: Tue, 12 Sep 2017 22:18:09 -0400
Subject: [PATCH] Add whitelist/blacklist button to query log rows

Signed-off-by: Mcat12 <newtoncat12@yahoo.com>
---
 src/components/QueryLog.js | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/src/components/QueryLog.js b/src/components/QueryLog.js
index 2b18459..ab47108 100644
--- a/src/components/QueryLog.js
+++ b/src/components/QueryLog.js
@@ -129,6 +129,20 @@ const columns = [
   {
     Header: "Action",
     minWidth: 80,
-    filterable: false
+    filterable: false,
+    Cell: data => {
+      if([1, 4, 5].includes(data.row.status))
+        return (
+          <button type="button" className="btn btn-success" onClick={() => api.addWhitelist(data.row.domain)}>
+            Whitelist
+          </button>
+        );
+      if([2, 3].includes(data.row.status))
+        return (
+          <button type="button" className="btn btn-danger" onClick={() => api.addBlacklist(data.row.domain)}>
+            Blacklist
+          </button>
+        );
+    }
   }
 ];