react-native-nitro-auth 0.1.4 → 0.1.5

package/src/Auth.web.ts

@@ -1,4 +1,11 @@
-import type { Auth, AuthUser, AuthProvider, LoginOptions } from "./Auth.nitro";
+import type {
+  Auth,
+  AuthUser,
+  AuthProvider,
+  LoginOptions,
+  AuthTokens,
+} from "./Auth.nitro";
+import type { AuthStorageAdapter } from "./AuthStorage.nitro";
 import { logger } from "./utils/logger";
 
 const CACHE_KEY = "nitro_auth_user";
@@ -18,26 +25,47 @@ class AuthWeb implements Auth {
   private _currentUser: AuthUser | undefined;
   private _grantedScopes: string[] = [];
   private _listeners: ((user: AuthUser | undefined) => void)[] = [];
+  private _tokenListeners: ((tokens: AuthTokens) => void)[] = [];
+  private _storageAdapter: AuthStorageAdapter | undefined;
 
   constructor() {
-    const cached = localStorage.getItem(CACHE_KEY);
+    this.loadFromCache();
+  }
+
+  private loadFromCache() {
+    const cached = this._storageAdapter
+      ? this._storageAdapter.load(CACHE_KEY)
+      : localStorage.getItem(CACHE_KEY);
+
     if (cached) {
       try {
         this._currentUser = JSON.parse(cached);
       } catch {
-        localStorage.removeItem(CACHE_KEY);
+        this.removeFromCache(CACHE_KEY);
       }
     }
-    const scopes = localStorage.getItem(SCOPES_KEY);
+
+    const scopes = this._storageAdapter
+      ? this._storageAdapter.load(SCOPES_KEY)
+      : localStorage.getItem(SCOPES_KEY);
+
     if (scopes) {
       try {
         this._grantedScopes = JSON.parse(scopes);
       } catch {
-        localStorage.removeItem(SCOPES_KEY);
+        this.removeFromCache(SCOPES_KEY);
       }
     }
   }
 
+  private removeFromCache(key: string) {
+    if (this._storageAdapter) {
+      this._storageAdapter.remove(key);
+    } else {
+      localStorage.removeItem(key);
+    }
+  }
+
   get currentUser(): AuthUser | undefined {
     return this._currentUser;
   }
@@ -60,6 +88,13 @@ class AuthWeb implements Auth {
     };
   }
 
+  onTokensRefreshed(callback: (tokens: AuthTokens) => void): () => void {
+    this._tokenListeners.push(callback);
+    return () => {
+      this._tokenListeners = this._tokenListeners.filter((l) => l !== callback);
+    };
+  }
+
   private notify() {
     this._listeners.forEach((l) => l(this._currentUser));
   }
@@ -103,7 +138,14 @@ class AuthWeb implements Auth {
     this._grantedScopes = this._grantedScopes.filter(
       (s) => !scopes.includes(s)
     );
-    localStorage.setItem(SCOPES_KEY, JSON.stringify(this._grantedScopes));
+    if (this._storageAdapter) {
+      this._storageAdapter.save(
+        SCOPES_KEY,
+        JSON.stringify(this._grantedScopes)
+      );
+    } else {
+      localStorage.setItem(SCOPES_KEY, JSON.stringify(this._grantedScopes));
+    }
     if (this._currentUser) {
       this._currentUser.scopes = this._grantedScopes;
       this.updateUser(this._currentUser);
@@ -132,10 +174,12 @@ class AuthWeb implements Auth {
     await this.loginGoogle(
       this._grantedScopes.length > 0 ? this._grantedScopes : DEFAULT_SCOPES
     );
-    return {
+    const tokens = {
       accessToken: this._currentUser.accessToken,
       idToken: this._currentUser.idToken,
     };
+    this._tokenListeners.forEach((l) => l(tokens));
+    return tokens;
   }
 
   private mapError(error: unknown): Error {
@@ -227,7 +271,11 @@ class AuthWeb implements Auth {
 
             if (idToken) {
               this._grantedScopes = scopes;
-              localStorage.setItem(SCOPES_KEY, JSON.stringify(scopes));
+              if (this._storageAdapter) {
+                this._storageAdapter.save(SCOPES_KEY, JSON.stringify(scopes));
+              } else {
+                localStorage.setItem(SCOPES_KEY, JSON.stringify(scopes));
+              }
 
               const user: AuthUser = {
                 provider: "google",
@@ -315,14 +363,18 @@ class AuthWeb implements Auth {
   logout(): void {
     this._currentUser = undefined;
     this._grantedScopes = [];
-    localStorage.removeItem(CACHE_KEY);
-    localStorage.removeItem(SCOPES_KEY);
+    this.removeFromCache(CACHE_KEY);
+    this.removeFromCache(SCOPES_KEY);
     this.notify();
   }
 
   private updateUser(user: AuthUser) {
     this._currentUser = user;
-    localStorage.setItem(CACHE_KEY, JSON.stringify(user));
+    if (this._storageAdapter) {
+      this._storageAdapter.save(CACHE_KEY, JSON.stringify(user));
+    } else {
+      localStorage.setItem(CACHE_KEY, JSON.stringify(user));
+    }
     this.notify();
   }
 
@@ -330,6 +382,14 @@ class AuthWeb implements Auth {
     logger.setEnabled(enabled);
   }
 
+  setStorageAdapter(adapter: AuthStorageAdapter | undefined): void {
+    this._storageAdapter = adapter;
+    if (adapter) {
+      this.loadFromCache();
+      this.notify();
+    }
+  }
+
   name = "Auth";
   dispose() {}
   equals(other: any) {

package/src/AuthStorage.nitro.ts

@@ -0,0 +1,17 @@
+import type { HybridObject } from "react-native-nitro-modules";
+
+export interface AuthStorageAdapter
+  extends HybridObject<{ ios: "c++"; android: "c++" }> {
+  /**
+   * Called to save a value to the custom storage.
+   */
+  save(key: string, value: string): void;
+  /**
+   * Called to load a value from the custom storage.
+   */
+  load(key: string): string | undefined;
+  /**
+   * Called to remove a value from the custom storage.
+   */
+  remove(key: string): void;
+}

package/src/index.ts

@@ -1,4 +1,5 @@
 export * from "./Auth.nitro";
+export * from "./AuthStorage.nitro";
 export * from "./ui/social-button";
 export * from "./use-auth";
 export { AuthService } from "./service";

package/src/index.web.ts

@@ -1,4 +1,5 @@
 export * from "./Auth.nitro";
+export * from "./AuthStorage.nitro";
 export * from "./ui/social-button.web";
 export * from "./use-auth";
 export { AuthService } from "./service.web";