Pass onRefreshFailure, onRefresh to the client from provider

Signed-off-by: Mitchell Hynes <[email protected]>

Author: ecumene

src/provider.tsx

@@ -25,14 +25,17 @@ export function AuthKitProvider(props: AuthKitProviderProps) {
     port,
     redirectUri,
     children,
+    onRefresh,
+    onRefreshFailure,
     onRedirectCallback,
     refreshBufferInterval,
   } = props;
   const [client, setClient] = React.useState<Client>(NOOP_CLIENT);
   const [state, setState] = React.useState(initialState);
 
-  const onRefresh = React.useCallback(
-    ({ user, accessToken, organizationId }: AuthenticationResponse) => {
+  const handleRefresh = React.useCallback(
+    (response: AuthenticationResponse) => {
+      const { user, accessToken, organizationId } = response;
       const { role = null, permissions = [] } = getClaims(accessToken);
       setState((prev) => {
         const next = {
@@ -44,8 +47,9 @@ export function AuthKitProvider(props: AuthKitProviderProps) {
         };
         return isEquivalentWorkOSSession(prev, next) ? prev : next;
       });
+      onRefresh?.(response);
     },
-    [client],
+    [client]
   );
 
   React.useEffect(() => {
@@ -58,7 +62,8 @@ export function AuthKitProvider(props: AuthKitProviderProps) {
           redirectUri,
           devMode,
           onRedirectCallback,
-          onRefresh,
+          onRefresh: handleRefresh,
+          onRefreshFailure,
           refreshBufferInterval,
         }).then(async (client) => {
           const user = client.getUser();
@@ -95,7 +100,7 @@ export function AuthKitProvider(props: AuthKitProviderProps) {
 // poor-man's "deep equality" check
 function isEquivalentWorkOSSession(
   a: typeof initialState,
-  b: typeof initialState,
+  b: typeof initialState
 ) {
   return (
     a.user?.updatedAt === b.user?.updatedAt &&