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 (