From d682eea9ac81d0c418c02b744554a1268f6a0a6c Mon Sep 17 00:00:00 2001
From: Stanislaw Adaszewski <s.adaszewski@gmail.com>
Date: Tue, 4 Feb 2020 15:41:00 +0100
Subject: [PATCH] Better sign out.

---
 frontend/src/js/page/wb-app.js      | 2 +-
 frontend/src/js/page/wb-sign-out.js | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/frontend/src/js/page/wb-app.js b/frontend/src/js/page/wb-app.js
index 91e5c5c..c2993fc 100644
--- a/frontend/src/js/page/wb-app.js
+++ b/frontend/src/js/page/wb-app.js
@@ -48,7 +48,7 @@ class WBApp extends Component {
 
         <WBSignIn path="/sign-in" appState={ this.state } />
 
-        <WBSignOut path='/sign-out' />
+        <WBSignOut path='/sign-out' app={ this } />
 
         <WBBrowse path="/browse/:ownerUuid?/:activePage?/:objTypeTab?/:collectionPage?/:processPage?/:workflowPage?"
           appState={ this.state }
diff --git a/frontend/src/js/page/wb-sign-out.js b/frontend/src/js/page/wb-sign-out.js
index 04e98c8..0ecae35 100644
--- a/frontend/src/js/page/wb-sign-out.js
+++ b/frontend/src/js/page/wb-sign-out.js
@@ -3,9 +3,12 @@ import { route } from 'preact-router';
 
 class WBSignOut extends Component {
   componentDidMount() {
-    delete window.localStorage['arvHost'];
-    delete window.localStorage['arvToken'];
-    delete window.localStorage['currentUser'];
+    const { app } = this.props;
+    const keys = ['arvHost', 'arvToken', 'currentUser'];
+    for (let i = 0; i < keys.length; i++) {
+      delete app.state[keys[i]];
+      delete window.localStorage[keys[i]];
+    }
     route('/sign-in');
   }