Add support for 'feature_flags' claim

Author: Karan Kajla

src/provider.tsx

@@ -36,14 +36,15 @@ export function AuthKitProvider(props: AuthKitProviderProps) {
   const handleRefresh = React.useCallback(
     (response: OnRefreshResponse) => {
       const { user, accessToken, organizationId } = response;
-      const { role = null, permissions = [] } = getClaims(accessToken);
+      const { role = null, permissions = [], feature_flags = [] } = getClaims(accessToken);
       setState((prev) => {
         const next = {
           ...prev,
           user,
           organizationId: organizationId ?? null,
           role,
           permissions,
+          featureFlags: feature_flags,
         };
         return isEquivalentWorkOSSession(prev, next) ? prev : next;
       });
@@ -106,7 +107,10 @@ function isEquivalentWorkOSSession(
     a.user?.updatedAt === b.user?.updatedAt &&
     a.organizationId === b.organizationId &&
     a.role === b.role &&
-    a.permissions.every((perm, i) => perm === b.permissions[i])
+    a.permissions.length === b.permissions.length &&
+    a.permissions.every((perm, i) => perm === b.permissions[i]) &&
+    a.featureFlags.length === b.featureFlags.length &&
+    a.featureFlags.every((flag, i) => flag === b.featureFlags[i])
   );
 }
 

src/state.ts

@@ -6,6 +6,7 @@ export interface State {
   role: string | null;
   organizationId: string | null;
   permissions: string[];
+  featureFlags: string[];
 }
 
 export const initialState: State = {
@@ -14,4 +15,5 @@ export const initialState: State = {
   role: null,
   organizationId: null,
   permissions: [],
+  featureFlags: [],
 };