Merge pull request #5846 from manucorporat/refactor-storage

refactor(storage): improve code reuse and add clean() method

ionic/platform/storage/local-storage.ts

@@ -79,4 +79,15 @@ export class LocalStorage extends StorageEngine {
       }
     });
   }
+
+  clear(): Promise<any> {
+    return new Promise((resolve, reject) => {
+      try {
+        window.localStorage.clear();
+        resolve();
+      } catch (e) {
+        reject(e);
+      }
+    });
+  }
 }

ionic/platform/storage/sql.ts

@@ -68,7 +68,7 @@ export class SqlStorage extends StorageEngine {
     this._tryInit();
   }
 
-  _getBackupLocation(dbFlag: number) {
+  _getBackupLocation(dbFlag: number): number {
     switch (dbFlag) {
       case SqlStorage.BACKUP_LOCAL:
         return 2;
@@ -83,11 +83,8 @@ export class SqlStorage extends StorageEngine {
 
   // Initialize the DB with our required tables
   _tryInit() {
-    this._db.transaction((tx) => {
-      tx.executeSql('CREATE TABLE IF NOT EXISTS kv (key text primary key, value text)', [], (tx, res) => {
-      }, (tx, err) => {
-        console.error('Storage: Unable to create initial storage tables', tx, err);
-      });
+    this.query('CREATE TABLE IF NOT EXISTS kv (key text primary key, value text)').catch(err => {
+      console.error('Storage: Unable to create initial storage tables', err.tx, err.err);
     });
   }
 
@@ -100,26 +97,17 @@ export class SqlStorage extends StorageEngine {
    * @param {array} params the additional params to use for query placeholders
    * @return {Promise} that resolves or rejects with an object of the form { tx: Transaction, res: Result (or err)}
    */
-  query(query, params=[]): Promise<any> {
+  query(query: string, params = []): Promise<any> {
     return new Promise((resolve, reject) => {
       try {
         this._db.transaction((tx) => {
-          tx.executeSql(query, params, (tx, res) => {
-            resolve({
-              tx: tx,
-              res: res
-            });
-          }, (tx, err) => {
-            reject({
-              tx: tx,
-              err: err
-            });
-          })
-        }, err => {
-          reject(err);
-        });
-      } catch(e) {
-        reject(e);
+          tx.executeSql(query, params,
+            (tx, res) => resolve({ tx: tx, res: res }),
+            (tx, err) => reject({ tx: tx, err: err }));
+        },
+          (err) => reject({ err: err }));
+      } catch (err) {
+        reject({ err: err });
       }
     })
   }
@@ -130,28 +118,9 @@ export class SqlStorage extends StorageEngine {
    * @return {Promise} that resolves or rejects with an object of the form { tx: Transaction, res: Result (or err)}
    */
   get(key: string): Promise<any> {
-    return new Promise((resolve, reject) => {
-      try {
-
-        this._db.transaction(tx => {
-          tx.executeSql("select key, value from kv where key = ? limit 1", [key], (tx, res) => {
-            if (res.rows.length > 0) {
-              let item = res.rows.item(0);
-              resolve(item.value);
-            }
-            resolve(null);
-          }, (tx, err) => {
-            reject({
-              tx: tx,
-              err: err
-            });
-          })
-        }, err => {
-          reject(err);
-        });
-
-      } catch(e) {
-        reject(e);
+    return this.query('select key, value from kv where key = ? limit 1', [key]).then(data => {
+      if (data.res.rows.length > 0) {
+        return data.res.rows.item(0);
       }
     });
   }
@@ -163,24 +132,7 @@ export class SqlStorage extends StorageEngine {
   * @return {Promise} that resolves or rejects with an object of the form { tx: Transaction, res: Result (or err)}
   */
   set(key: string, value: string): Promise<any> {
-    return new Promise((resolve, reject) => {
-      try {
-        this._db.transaction(tx => {
-          tx.executeSql('insert or replace into kv(key, value) values (?, ?)', [key, value], (tx, res) => {
-            resolve();
-          }, (tx, err) => {
-            reject({
-              tx: tx,
-              err: err
-            });
-          })
-        }, err => {
-          reject(err);
-        });
-      } catch(e) {
-        reject(e);
-      }
-    });
+    return this.query('insert or replace into kv(key, value) values (?, ?)', [key, value]);
   }
 
   /**
@@ -189,24 +141,10 @@ export class SqlStorage extends StorageEngine {
   * @return {Promise} that resolves or rejects with an object of the form { tx: Transaction, res: Result (or err)}
   */
   remove(key: string): Promise<any> {
-    return new Promise((resolve, reject) => {
-      try {
-        this._db.transaction(tx => {
-          tx.executeSql('delete from kv where key = ?', [key], (tx, res) => {
-            resolve();
-          }, (tx, err) => {
-            reject({
-              tx: tx,
-              err: err
-            });
-          })
-        }, err => {
-          reject(err);
-        });
-      } catch(e) {
-        reject(e);
+    return this.query('delete from kv where key = ?', [key]);
   }
-    });
 
+  clear(): Promise<any> {
+    return this.query('delete from kv');
   }
 }

ionic/platform/storage/storage.ts

@@ -11,15 +11,17 @@
  * @private
 */
 export class Storage {
-  private _strategy: any;
+  private _strategy: StorageEngine;
 
   constructor(strategyCls: IStorageEngine, options?: any) {
     this._strategy = new strategyCls(options);
   }
-  get(key: string): any {
+
+  get(key: string): Promise<any> {
     return this._strategy.get(key);
   }
-  getJson(key: string): any {
+
+  getJson(key: string): Promise<any> {
     return this.get(key).then(value => {
       try {
         return JSON.parse(value);
@@ -29,6 +31,7 @@ export class Storage {
       }
     });
   }
+
   setJson(key: string, value: any): Promise<any> {
     try {
       return this.set(key, JSON.stringify(value));
@@ -36,15 +39,22 @@ export class Storage {
       return Promise.reject(e);
     }
   }
+
   set(key: string, value: any) {
     return this._strategy.set(key, value);
   }
+
   remove(key: string) {
     return this._strategy.remove(key);
   }
+
   query(query: string, params?: any) {
     return this._strategy.query(query, params);
   }
+
+  clear() {
+    return this._strategy.clear();
+  }
 }
 
 export interface IStorageEngine {
@@ -55,18 +65,21 @@ export interface IStorageEngine {
  * @private
 */
 export class StorageEngine {
-  constructor(options={}) {
-  }
-  get(key, value) {
+  constructor(options = {}) { }
+
+  get(key: string): Promise<any> {
     throw Error("get() not implemented for this storage engine");
   }
-  set(key, value) {
+  set(key: string, value: any): Promise<any> {
     throw Error("set() not implemented for this storage engine");
   }
-  remove(key) {
+  remove(key: string): Promise<any> {
     throw Error("remove() not implemented for this storage engine");
   }
-  query(query, params) {
+  query(query: string, params?: any): Promise<any> {
     throw Error("query() not implemented for this storage engine");
   }
+  clear(): Promise<any> {
+    throw Error("clear() not implemented for this storage engine");
+  }
 }