Parse and cast objects from browser storage to class object

Author: Stigmatoz

src/entity/tab.ts

@@ -1,13 +1,17 @@
 import { todayLocalDate } from "../utils/common";
 
-export class Tab {
-    url: string;
-    favicon: string;
+interface Serializable<T> {
+    deserialize(input: Object): T;
+}
+
+export class Tab implements Serializable<Tab> {
+    url: string = '';
+    favicon: string = '';
     summaryTime: number = 0;
     counter: number = 0
     days: TabDay[] = [];
 
-    constructor(url: string, favicon: string){
+    init(url: string, favicon: string){
         this.url = url;
         this.favicon = favicon;
     }
@@ -30,17 +34,29 @@ export class Tab {
     }
 
     addNewDay() :void {
-        const newTabDay = new TabDay(todayLocalDate());
+        const newTabDay = new TabDay();
+        newTabDay.init(todayLocalDate());
         this.days.push(newTabDay);
     }
+
+    deserialize(input: Tab) {
+        this.url = input.url;
+        this.counter = input.counter;
+        this.favicon = input.favicon;
+        this.summaryTime = input.summaryTime;
+        if (input.days?.length > 0)
+            this.days = input.days.map(x =>  new TabDay().deserialize(x));
+
+        return this;
+    }
 }
 
-export class TabDay {
-    counter: number;
-    date: string;
-    summary: number;
+export class TabDay implements Serializable<TabDay> {
+    counter: number = 0;
+    date: string = '';
+    summary: number = 0;
 
-    constructor(date: string){
+    init(date: string){
         this.date = date;
         this.counter = 1;
         this.summary = 1;
@@ -53,4 +69,12 @@ export class TabDay {
     incCounter() :void {
         this.counter += 1;
     }
-}
+
+    deserialize(input: TabDay): TabDay {
+        this.counter = input.counter;
+        this.date = input.date;
+        this.summary = input.summary;
+
+        return this;
+    }
+}

src/repository/tabs-repository.ts

@@ -41,7 +41,9 @@ export class TabsRepository implements ITabsRepository {
                         if (!favicon) {
                             favicon = 'chrome://favicon/' + domain;
                         }
-                        this.tabs.push(new Tab(domain, favicon));
+                        const newTab = new Tab();
+                        newTab.init(domain, favicon);
+                        this.tabs.push(newTab);
                     }
                     else {
                         tabFromStorage.incCounter();

src/storage/local-storage.ts

@@ -5,21 +5,26 @@ import Browser from 'webextension-polyfill';
 import { isEmpty } from "../common/utility";
 
 export class LocalStorage implements IStorage {
-    getTabs(): Promise<Tab[]> {
+    async getTabs(): Promise<Tab[]> {
         return new Promise(async resolve => {
             const { tabs } = await Browser.storage.local.get(StorageParams.TABS);
-            if (tabs != undefined)
-                return resolve(tabs);
+            if (tabs != undefined){
+                let tempTabs: Tab[] = [];
+                for (let i = 0; i < tabs.length; i++) {
+                    tempTabs.push(new Tab().deserialize(tabs[i]));
+                }
+                return resolve(tempTabs);
+            }
             else resolve([]);
         });
     }
 
-    saveTabs(value: Tab[]): Promise<void> {
-        return Browser.storage.local.set({ tabs: value });
+    async saveTabs(value: Tab[]): Promise<void> {
+        return await Browser.storage.local.set({ tabs: value });
     }
 
-    saveValue(name: StorageParams, value: any): Promise<void> {
-        return Browser.storage.local.set({
+    async saveValue(name: StorageParams, value: any): Promise<void> {
+        return await Browser.storage.local.set({
             [name]: value
         });
     }