From 7b3082fb2f7532ce2c754e24dcfa06ce7a633098 Mon Sep 17 00:00:00 2001
From: dilan-dio4 <54545871+dilan-dio4@users.noreply.github.com>
Date: Mon, 15 Nov 2021 12:21:12 -0500
Subject: [PATCH] Fixes #7

---
 src/cache.ts        | 66 ++++++++++++++++++++++++---------------------
 src/types/types.tsx |  6 ++++-
 2 files changed, 41 insertions(+), 31 deletions(-)

diff --git a/src/cache.ts b/src/cache.ts
index 802506e..ac4c3c8 100644
--- a/src/cache.ts
+++ b/src/cache.ts
@@ -1,6 +1,6 @@
 import Storage from './storage';
 
-const storage = new Storage({ storageBackend: window.localStorage });
+const storage = window !== undefined ? new Storage({ storageBackend: window.localStorage }) : undefined;
 
 // https://github.com/sunnylqm/react-native-storage
 export async function getCacheTokens(cookieName: string): Promise<Record<string, any>> {
@@ -8,17 +8,19 @@ export async function getCacheTokens(cookieName: string): Promise<Record<string,
     let cacheRefreshToken = false;
     let cacheSession = false;
 
-    try {
-        cacheToken = await storage.load({ key: cookieName + "token" });
-    } catch (_) {}
+    if (storage) {
+        try {
+            cacheToken = await storage.load({ key: cookieName + "token" });
+        } catch (_) { }
 
-    try {
-        cacheRefreshToken = await storage.load({ key: cookieName + "refreshToken" });
-    } catch (_) {}
+        try {
+            cacheRefreshToken = await storage.load({ key: cookieName + "refreshToken" });
+        } catch (_) { }
 
-    try {
-        cacheSession = await storage.load({ key: cookieName + "session" });
-    } catch (_) {}
+        try {
+            cacheSession = await storage.load({ key: cookieName + "session" });
+        } catch (_) { }
+    }
 
     return {
         cacheToken,
@@ -28,27 +30,31 @@ export async function getCacheTokens(cookieName: string): Promise<Record<string,
 }
 
 export async function clearCacheTokens(cookieName: string) {
-    await storage.remove({ key: cookieName + "token" });
-    await storage.remove({ key: cookieName + "refreshToken" });
-    await storage.remove({ key: cookieName + "session" });
+    if (storage) {
+        await storage.remove({ key: cookieName + "token" });
+        await storage.remove({ key: cookieName + "refreshToken" });
+        await storage.remove({ key: cookieName + "session" });
+    }
 }
 
 export async function setCacheTokens(g: any, cookieName: string) {
-    await storage.save({
-        key: cookieName + "token",
-        data: g.token,
-        expires: 3600 * 1000 * 24
-    });
-
-    await storage.save({
-        key: cookieName + "refreshToken",
-        data: g.refreshToken,
-        expires: 3600 * 1000 * 24
-    });
-
-    await storage.save({
-        key: cookieName + "session",
-        data: g.session,
-        expires: null
-    });
+    if (storage) {
+        await storage.save({
+            key: cookieName + "token",
+            data: g.token,
+            expires: 3600 * 1000 * 24
+        });
+
+        await storage.save({
+            key: cookieName + "refreshToken",
+            data: g.refreshToken,
+            expires: 3600 * 1000 * 24
+        });
+
+        await storage.save({
+            key: cookieName + "session",
+            data: g.session,
+            expires: null
+        });
+    }
 }
diff --git a/src/types/types.tsx b/src/types/types.tsx
index dcc47cb..65250ed 100644
--- a/src/types/types.tsx
+++ b/src/types/types.tsx
@@ -291,7 +291,9 @@ export interface ContextValue {
     /**
      * Custom stateful hook to an instance of `db().return`. Other local changes will automatically re-fetch the query 
      * as detailed in the passed-in db.
-     * ```jsx
+     * 
+     * ```
+     * 
      * const { frame } = useReturn(() => db('MYTABLE').return().where(e.gt('rating', ratingState)).limit(10), [ratingState])
      * 
      * const onButtonClick = (_key) => {
@@ -300,7 +302,9 @@ export interface ContextValue {
      * 
      * // Stays fresh after call to `.delete()`
      * return (<div>{ frame.map(ele => <Card {...ele} />) }</div>)
+     * 
      * ```
+     * 
      * @param {function():SQW} dbInstance Function returning an instance of `db().return` without having called `.all` or `.one`
      * @param {React.DependencyList} deps If present, instance will be reloaded if the values in the list change.
      * @return {UseReturnValue} Object with the required values to statefully access an array that is subscribed to local executions to the corresponding db instance.