From 045eef73eed666f45664b05604658a48157afa8f Mon Sep 17 00:00:00 2001
From: Mcat12 <newtoncat12@yahoo.com>
Date: Mon, 31 Jul 2017 21:28:38 -0400
Subject: [PATCH] Add remove functionality to whitelist

Signed-off-by: Mcat12 <newtoncat12@yahoo.com>
---
 src/views/Whitelist/Whitelist.js | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/views/Whitelist/Whitelist.js b/src/views/Whitelist/Whitelist.js
index d35e91c..f3e94c7 100644
--- a/src/views/Whitelist/Whitelist.js
+++ b/src/views/Whitelist/Whitelist.js
@@ -18,18 +18,29 @@ export default class Whitelist extends Component {
   onAdd(domain) {
     if(domain.length > 0) {
       if(this.state.domains.includes(domain))
-        this.setState({ errorMsg: domain + " is already added", successMsg: "", infoMsg: "" });
+        this.setState({
+          infoMsg: "",
+          successMsg: "",
+          errorMsg: domain + " is already added"
+        });
       else {
         this.setState(prevState => ({
           domains: [...prevState.domains, domain],
-          successMsg: "Successfully added " + domain,
           infoMsg: "",
+          successMsg: "Successfully added " + domain,
           errorMsg: ""
         }));
       }
     }
   }
 
+  onRemove(domain) {
+    if(this.state.domains.includes(domain)) {
+      //TODO: Run API call to remove domain
+      this.setState(prevState => ({ domains: prevState.domains.filter(item => item !== domain) }));
+    }
+  }
+
   render() {
     return (
       <div>
@@ -60,7 +71,7 @@ export default class Whitelist extends Component {
           {
             this.state.domains.map(item => (
               <li key={item} className="list-group-item">
-                <button className="btn btn-danger btn-sm pull-right" type="button">
+                <button className="btn btn-danger btn-sm pull-right" type="button" onClick={() => this.onRemove(item)}>
                   <span className="fa fa-trash-o"/>
                 </button>
                 <span style={{ display: "table-cell", verticalAlign: "middle", height: "32px" }}>