feat: onAuthStateChange (#82)

* feat: onAuthStateChange * fix: remove onAuthStateChange param to keep SSOT
logto-io · Nov 4, 2021 · e8dda56 · e8dda56
1 parent efc2d01
commit e8dda56
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 1 deletion.
diff --git a/packages/client/src/index.test.ts b/packages/client/src/index.test.ts
@@ -380,4 +380,52 @@ describe('LogtoClient', () => {
       expect(storage.removeItem).toBeCalled();
     });
   });
+
+  describe('onAuthStateChange', () => {
+    test('should be called on tokenSet recovery', async () => {
+      const onAuthStateChange = jest.fn();
+      const storage = new MemoryStorage();
+      storage.setItem(
+        `LOGTO_TOKEN_SET_CACHE::${discoverResponse.issuer}::${CLIENT_ID}::${DEFAULT_SCOPE}`,
+        {
+          ...fakeTokenResponse,
+          id_token: (await generateIdToken()).idToken,
+        }
+      );
+      await LogtoClient.create({
+        domain: DOMAIN,
+        clientId: CLIENT_ID,
+        storage,
+        onAuthStateChange,
+      });
+      expect(onAuthStateChange).toHaveBeenCalled();
+    });
+
+    test('should be called on handleCallback', async () => {
+      const onAuthStateChange = jest.fn();
+      const storage = new MemoryStorage();
+      const logto = await LogtoClient.create({
+        domain: DOMAIN,
+        clientId: CLIENT_ID,
+        storage,
+        onAuthStateChange,
+      });
+      await logto.loginWithRedirect(REDIRECT_URI);
+      await logto.handleCallback(REDIRECT_CALLBACK);
+      expect(onAuthStateChange).toHaveBeenCalled();
+    });
+
+    test('should be called on logout', async () => {
+      const onAuthStateChange = jest.fn();
+      const storage = new MemoryStorage();
+      const logto = await LogtoClient.create({
+        domain: DOMAIN,
+        clientId: CLIENT_ID,
+        storage,
+        onAuthStateChange,
+      });
+      logto.logout(REDIRECT_URI);
+      expect(onAuthStateChange).toHaveBeenCalled();
+    });
+  });
 });
diff --git a/packages/client/src/index.ts b/packages/client/src/index.ts
@@ -23,6 +23,7 @@ export interface ConfigParameters {
   clientId: string;
   scope?: string | string[];
   storage?: ClientStorage;
+  onAuthStateChange?: () => void;
 }
 
 export const appendSlashIfNeeded = (url: string): string => {
@@ -39,10 +40,12 @@ export default class LogtoClient {
   private readonly scope: string;
   private readonly sessionManager: SessionManager;
   private readonly storage: ClientStorage;
+  private readonly onAuthStateChange: Optional<() => void>;
   private tokenSet: Optional<TokenSet>;
   constructor(config: ConfigParameters, oidcConfiguration: OIDCConfiguration) {
-    const { clientId, scope, storage } = config;
+    const { clientId, scope, storage, onAuthStateChange } = config;
     this.clientId = clientId;
+    this.onAuthStateChange = onAuthStateChange;
     this.scope = generateScope(scope);
     this.oidcConfiguration = oidcConfiguration;
     this.storage = storage ?? new LocalStorage();
@@ -119,6 +122,9 @@ export default class LogtoClient {
     );
     this.storage.setItem(this.tokenSetCacheKey, tokenParameters);
     this.tokenSet = new TokenSet(tokenParameters);
+    if (this.onAuthStateChange) {
+      this.onAuthStateChange();
+    }
   }
 
   public getClaims() {
@@ -159,6 +165,10 @@ export default class LogtoClient {
 
   public logout(redirectUri: string) {
     this.sessionManager.clear();
+    if (this.onAuthStateChange) {
+      this.onAuthStateChange();
+    }
+
     if (!this.tokenSet) {
       return;
     }
@@ -181,6 +191,9 @@ export default class LogtoClient {
     const parameters = this.storage.getItem<TokenSetParameters>(this.tokenSetCacheKey);
     if (parameters) {
       this.tokenSet = new TokenSet(parameters);
+      if (this.onAuthStateChange) {
+        this.onAuthStateChange();
+      }
     }
   }
 }