From 0255638f0bd1d2609063b0db260792f81cee6c5d Mon Sep 17 00:00:00 2001
From: noahtalerman <47070608+noahtalerman@users.noreply.github.com>
Date: Fri, 18 Jun 2021 10:32:51 -0400
Subject: [PATCH] Add persistence to saved columns in "Hosts" table after user
 logs out (#1131)

- Add `removeItem` method so that we only remove the `auth_token`, and not the saved columns, from `localStorage` when the user logs out
---
 frontend/redux/middlewares/auth.js | 2 +-
 frontend/utilities/local.js        | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/frontend/redux/middlewares/auth.js b/frontend/redux/middlewares/auth.js
index d431a8ed75..c66a154633 100644
--- a/frontend/redux/middlewares/auth.js
+++ b/frontend/redux/middlewares/auth.js
@@ -35,7 +35,7 @@ const authMiddleware = (store) => (next) => (action) => {
   if (type === LOGOUT_SUCCESS || type === LOGIN_FAILURE) {
     const { LOGIN } = PATHS;
 
-    local.clear();
+    local.removeItem("auth_token");
     Fleet.setBearerToken(null);
     store.dispatch(push(LOGIN));
   }
diff --git a/frontend/utilities/local.js b/frontend/utilities/local.js
index 77688147ba..f5e2c074c1 100644
--- a/frontend/utilities/local.js
+++ b/frontend/utilities/local.js
@@ -16,6 +16,11 @@ const local = {
 
     return localStorage.setItem(`FLEET::${itemName}`, value);
   },
+  removeItem: (itemName) => {
+    const { localStorage } = window;
+
+    localStorage.removeItem(`FLEET::${itemName}`);
+  },
 };
 
 export const authToken = () => {