From 691e9ca8f28a8430e4765ab1a9ae13687289a53b Mon Sep 17 00:00:00 2001
From: Shin'ya Ueoka <ueokande@i-beam.org>
Date: Sat, 28 Jul 2018 17:06:20 +0900
Subject: Fix last tab is undefined

---
 src/background/infrastructures/memory-storage.js | 6 +++++-
 src/background/usecases/operation.js             | 4 +++-
 2 files changed, 8 insertions(+), 2 deletions(-)

(limited to 'src/background')

diff --git a/src/background/infrastructures/memory-storage.js b/src/background/infrastructures/memory-storage.js
index 234a9b8..3a7e4f2 100644
--- a/src/background/infrastructures/memory-storage.js
+++ b/src/background/infrastructures/memory-storage.js
@@ -10,6 +10,10 @@ export default class MemoryStorage {
   }
 
   get(name) {
-    return JSON.parse(db[name]);
+    let data = db[name];
+    if (!data) {
+      return undefined;
+    }
+    return JSON.parse(data);
   }
 }
diff --git a/src/background/usecases/operation.js b/src/background/usecases/operation.js
index f19c632..86f39ca 100644
--- a/src/background/usecases/operation.js
+++ b/src/background/usecases/operation.js
@@ -183,7 +183,9 @@ export default class OperationInteractor {
 
   onTabSelected(tabId) {
     let lastId = this.cache.get(CURRENT_SELECTED_KEY);
-    this.cache.set(LAST_SELECTED_KEY, lastId);
+    if (lastId) {
+      this.cache.set(LAST_SELECTED_KEY, lastId);
+    }
     this.cache.set(CURRENT_SELECTED_KEY, tabId);
   }
 }
-- 
cgit v1.2.3