diff --git a/lib/keyserver-conn/call-keyserver-endpoint-provider.react.js b/lib/keyserver-conn/call-keyserver-endpoint-provider.react.js
index bf45d02fa..2236b7f7b 100644
--- a/lib/keyserver-conn/call-keyserver-endpoint-provider.react.js
+++ b/lib/keyserver-conn/call-keyserver-endpoint-provider.react.js
@@ -1,444 +1,448 @@
 // @flow
 
 import invariant from 'invariant';
 import _memoize from 'lodash/memoize.js';
 import * as React from 'react';
 import { createSelector } from 'reselect';
 
 import { useKeyserverCallInfos } from './keyserver-call-infos.js';
 import {
   setNewSession,
   type SingleKeyserverActionFunc,
   type ActionFunc,
 } from './keyserver-conn-types.js';
 import {
   canResolveKeyserverSessionInvalidation,
   resolveKeyserverSessionInvalidation,
 } from './recovery-utils.js';
 import { logInActionSources } from '../types/account-types.js';
 import type { PlatformDetails } from '../types/device-types.js';
 import type { Endpoint, SocketAPIHandler } from '../types/endpoints.js';
 import type { Dispatch } from '../types/redux-types.js';
 import type { ClientSessionChange } from '../types/session-types.js';
 import type { CurrentUserInfo } from '../types/user-types.js';
 import type {
   CallSingleKeyserverEndpoint,
   CallSingleKeyserverEndpointOptions,
 } from '../utils/call-single-keyserver-endpoint.js';
 import callSingleKeyserverEndpoint from '../utils/call-single-keyserver-endpoint.js';
 import { promiseAll } from '../utils/promises.js';
 import { useSelector, useDispatch } from '../utils/redux-utils.js';
 
 type CreateCallSingleKeyserverEndpointSelector = (
   keyserverID: string,
 ) => ServerCallSelectorParams => CallSingleKeyserverEndpoint;
 
 type GetBoundSingleKeyserverActionFunc = <F>(
   keyserverID: string,
   actionFunc: SingleKeyserverActionFunc<F>,
 ) => F;
 
 type SingleKeyserverActionFuncSelectorParams = {
   +callSingleKeyserverEndpoint: CallSingleKeyserverEndpoint,
 };
 type CreateBoundSingleKeyserverActionFuncSelector = <F>(
   actionFunc: SingleKeyserverActionFunc<F>,
 ) => SingleKeyserverActionFuncSelectorParams => F;
 
 type GetBoundKeyserverActionFunc = <Args, Return>(
   actionFunc: ActionFunc<Args, Return>,
 ) => Args => Promise<Return>;
 
 type CallKeyserverEndpointContextType = {
   +createCallSingleKeyserverEndpointSelector: CreateCallSingleKeyserverEndpointSelector,
   +getBoundSingleKeyserverActionFunc: GetBoundSingleKeyserverActionFunc,
   +getBoundKeyserverActionFunc: GetBoundKeyserverActionFunc,
 };
 
 const CallKeyserverEndpointContext: React.Context<?CallKeyserverEndpointContextType> =
   React.createContext();
 
 type OngoingRecoveryAttempt = {
   +waitingCalls: Array<
     (callSingleKeyserverEndpoint: ?CallSingleKeyserverEndpoint) => mixed,
   >,
 };
 
 export type ServerCallSelectorParams = {
   +dispatch: Dispatch,
   +cookie: ?string,
   +urlPrefix: string,
   +sessionID: ?string,
   +currentUserInfo: ?CurrentUserInfo,
   +isSocketConnected: boolean,
+  +sessionRecoveryInProgress: boolean,
   +lastCommunicatedPlatformDetails: ?PlatformDetails,
 };
 type BindServerCallsParams = $ReadOnly<{
   ...ServerCallSelectorParams,
   +keyserverID: string,
 }>;
 
 type Props = {
   +children: React.Node,
 };
 function CallKeyserverEndpointProvider(props: Props): React.Node {
   // SECTION 1: bindCookieAndUtilsIntoCallServerEndpoint
 
   const ongoingRecoveryAttemptsRef = React.useRef<
     Map<string, OngoingRecoveryAttempt>,
   >(new Map());
 
   const bindCookieAndUtilsIntoCallSingleKeyserverEndpoint: (
     params: BindServerCallsParams,
   ) => CallSingleKeyserverEndpoint = React.useCallback(params => {
     const {
       dispatch,
       cookie,
       urlPrefix,
       sessionID,
       currentUserInfo,
       isSocketConnected,
       lastCommunicatedPlatformDetails,
       keyserverID,
     } = params;
     const loggedIn = !!(currentUserInfo && !currentUserInfo.anonymous && true);
     const boundSetNewSession = (
       sessionChange: ClientSessionChange,
       error: ?string,
     ) =>
       setNewSession(
         dispatch,
         sessionChange,
         {
           currentUserInfo,
           cookiesAndSessions: { [keyserverID]: { cookie, sessionID } },
         },
         error,
         undefined,
         keyserverID,
       );
     const canResolveInvalidation = canResolveKeyserverSessionInvalidation();
     // This function gets called before callSingleKeyserverEndpoint sends a
     // request, to make sure that we're not in the middle of trying to recover
     // an invalidated cookie
     const waitIfCookieInvalidated = () => {
       if (!canResolveInvalidation) {
         // If there is no way to resolve the session invalidation,
         // just let the caller callSingleKeyserverEndpoint instance continue
         return Promise.resolve(null);
       }
       const ongoingRecoveryAttempt =
         ongoingRecoveryAttemptsRef.current.get(keyserverID);
       if (!ongoingRecoveryAttempt) {
         // Our cookie seems to be valid
         return Promise.resolve(null);
       }
       // Wait to run until we get our new cookie
       return new Promise<?CallSingleKeyserverEndpoint>(r =>
         ongoingRecoveryAttempt.waitingCalls.push(r),
       );
     };
     // These functions are helpers for cookieInvalidationRecovery, defined below
     const attemptToResolveInvalidationHelper = async (
       sessionChange: ClientSessionChange,
     ) => {
       const newAnonymousCookie = sessionChange.cookie;
       const newSessionChange = await resolveKeyserverSessionInvalidation(
         dispatch,
         newAnonymousCookie,
         urlPrefix,
         logInActionSources.cookieInvalidationResolutionAttempt,
         keyserverID,
       );
 
       return newSessionChange
         ? bindCookieAndUtilsIntoCallSingleKeyserverEndpoint({
             ...params,
             cookie: newSessionChange.cookie,
             sessionID: newSessionChange.sessionID,
             currentUserInfo: newSessionChange.currentUserInfo,
           })
         : null;
     };
     const attemptToResolveInvalidation = (
       sessionChange: ClientSessionChange,
     ) => {
       return new Promise<?CallSingleKeyserverEndpoint>(
         // eslint-disable-next-line no-async-promise-executor
         async (resolve, reject) => {
           try {
             const newCallSingleKeyserverEndpoint =
               await attemptToResolveInvalidationHelper(sessionChange);
             const ongoingRecoveryAttempt =
               ongoingRecoveryAttemptsRef.current.get(keyserverID);
             ongoingRecoveryAttemptsRef.current.delete(keyserverID);
             const currentWaitingCalls =
               ongoingRecoveryAttempt?.waitingCalls ?? [];
 
             resolve(newCallSingleKeyserverEndpoint);
 
             for (const func of currentWaitingCalls) {
               func(newCallSingleKeyserverEndpoint);
             }
           } catch (e) {
             reject(e);
           }
         },
       );
     };
     // If this function is called, callSingleKeyserverEndpoint got a response
     // invalidating its cookie, and is wondering if it should just like...
     // give up? Or if there's a chance at redemption
     const cookieInvalidationRecovery = (sessionChange: ClientSessionChange) => {
       if (!canResolveInvalidation) {
         // If there is no way to resolve the session invalidation,
         // just let the caller callSingleKeyserverEndpoint instance continue
         return Promise.resolve(null);
       }
       if (!loggedIn) {
         // We don't want to attempt any use native credentials of a logged out
         // user to log-in after a cookieInvalidation while logged out
         return Promise.resolve(null);
       }
       const ongoingRecoveryAttempt =
         ongoingRecoveryAttemptsRef.current.get(keyserverID);
       if (ongoingRecoveryAttempt) {
         return new Promise<?CallSingleKeyserverEndpoint>(r =>
           ongoingRecoveryAttempt.waitingCalls.push(r),
         );
       }
       ongoingRecoveryAttemptsRef.current.set(keyserverID, { waitingCalls: [] });
       return attemptToResolveInvalidation(sessionChange);
     };
 
     return (
       endpoint: Endpoint,
       data: Object,
       options?: ?CallSingleKeyserverEndpointOptions,
     ) =>
       callSingleKeyserverEndpoint(
         cookie,
         boundSetNewSession,
         waitIfCookieInvalidated,
         cookieInvalidationRecovery,
         urlPrefix,
         sessionID,
         isSocketConnected,
         lastCommunicatedPlatformDetails,
         socketAPIHandler,
         endpoint,
         data,
         dispatch,
         options,
         loggedIn,
         keyserverID,
       );
   }, []);
 
   // SECTION 2: createCallSingleKeyserverEndpointSelector
 
   // For each keyserver, we have a set of params that configure our connection
   // to it. These params get bound into callSingleKeyserverEndpoint before it's
   // passed to a SingleKeyserverActionFunc. This helper function lets us create
   // a selector for a given keyserverID that will regenerate the bound
   // callSingleKeyserverEndpoint function only if one of the params changes.
   // This lets us skip some React render cycles.
   const createCallSingleKeyserverEndpointSelector = React.useCallback(
     (
       keyserverID: string,
     ): (ServerCallSelectorParams => CallSingleKeyserverEndpoint) =>
       createSelector(
         (params: ServerCallSelectorParams) => params.dispatch,
         (params: ServerCallSelectorParams) => params.cookie,
         (params: ServerCallSelectorParams) => params.urlPrefix,
         (params: ServerCallSelectorParams) => params.sessionID,
         (params: ServerCallSelectorParams) => params.currentUserInfo,
         (params: ServerCallSelectorParams) => params.isSocketConnected,
+        (params: ServerCallSelectorParams) => params.sessionRecoveryInProgress,
         (params: ServerCallSelectorParams) =>
           params.lastCommunicatedPlatformDetails,
         (
           dispatch: Dispatch,
           cookie: ?string,
           urlPrefix: string,
           sessionID: ?string,
           currentUserInfo: ?CurrentUserInfo,
           isSocketConnected: boolean,
+          sessionRecoveryInProgress: boolean,
           lastCommunicatedPlatformDetails: ?PlatformDetails,
         ) =>
           bindCookieAndUtilsIntoCallSingleKeyserverEndpoint({
             dispatch,
             cookie,
             urlPrefix,
             sessionID,
             currentUserInfo,
             isSocketConnected,
+            sessionRecoveryInProgress,
             lastCommunicatedPlatformDetails,
             keyserverID,
           }),
       ),
     [bindCookieAndUtilsIntoCallSingleKeyserverEndpoint],
   );
 
   // SECTION 3: getBoundSingleKeyserverActionFunc
 
   const dispatch = useDispatch();
   const currentUserInfo = useSelector(state => state.currentUserInfo);
 
   const keyserverInfos = useSelector(
     state => state.keyserverStore.keyserverInfos,
   );
   const keyserverCallInfos = useKeyserverCallInfos(keyserverInfos);
 
   const callSingleKeyserverEndpointSelectorCacheRef = React.useRef<
     Map<string, (ServerCallSelectorParams) => CallSingleKeyserverEndpoint>,
   >(new Map());
   const getCallSingleKeyserverEndpoint = React.useCallback(
     (keyserverID: string) => {
       let selector =
         callSingleKeyserverEndpointSelectorCacheRef.current.get(keyserverID);
       if (!selector) {
         selector = createCallSingleKeyserverEndpointSelector(keyserverID);
         callSingleKeyserverEndpointSelectorCacheRef.current.set(
           keyserverID,
           selector,
         );
       }
       const keyserverCallInfo = keyserverCallInfos[keyserverID];
       return selector({
         ...keyserverCallInfo,
         dispatch,
         currentUserInfo,
       });
     },
     [
       createCallSingleKeyserverEndpointSelector,
       dispatch,
       currentUserInfo,
       keyserverCallInfos,
     ],
   );
 
   const createBoundSingleKeyserverActionFuncSelector: CreateBoundSingleKeyserverActionFuncSelector =
     React.useCallback(
       actionFunc =>
         createSelector(
           (params: SingleKeyserverActionFuncSelectorParams) =>
             params.callSingleKeyserverEndpoint,
           actionFunc,
         ),
       [],
     );
 
   const createBoundSingleKeyserverActionFuncsCache: () => CreateBoundSingleKeyserverActionFuncSelector =
     React.useCallback(
       () => _memoize(createBoundSingleKeyserverActionFuncSelector),
       [createBoundSingleKeyserverActionFuncSelector],
     );
 
   const boundSingleKeyserverActionFuncSelectorCacheRef = React.useRef<
     Map<string, CreateBoundSingleKeyserverActionFuncSelector>,
   >(new Map());
   const getBoundSingleKeyserverActionFunc: GetBoundSingleKeyserverActionFunc =
     React.useCallback(
       <F>(keyserverID: string, actionFunc: SingleKeyserverActionFunc<F>): F => {
         let selector =
           boundSingleKeyserverActionFuncSelectorCacheRef.current.get(
             keyserverID,
           );
         if (!selector) {
           selector = createBoundSingleKeyserverActionFuncsCache();
           boundSingleKeyserverActionFuncSelectorCacheRef.current.set(
             keyserverID,
             selector,
           );
         }
         const callEndpoint = getCallSingleKeyserverEndpoint(keyserverID);
         return selector(actionFunc)({
           callSingleKeyserverEndpoint: callEndpoint,
         });
       },
       [
         createBoundSingleKeyserverActionFuncsCache,
         getCallSingleKeyserverEndpoint,
       ],
     );
 
   // SECTION 4: getBoundKeyserverActionFunc
 
   const callKeyserverEndpoint = React.useCallback(
     (
       endpoint: Endpoint,
       requests: { +[keyserverID: string]: ?{ +[string]: mixed } },
       options?: ?CallSingleKeyserverEndpointOptions,
     ) => {
       const makeCallToSingleKeyserver = (keyserverID: string) => {
         const boundCallSingleKeyserverEndpoint =
           getCallSingleKeyserverEndpoint(keyserverID);
         return boundCallSingleKeyserverEndpoint(
           endpoint,
           requests[keyserverID],
           options,
         );
       };
       const promises: { [string]: Promise<Object> } = {};
       for (const keyserverID in requests) {
         promises[keyserverID] = makeCallToSingleKeyserver(keyserverID);
       }
       return promiseAll(promises);
     },
     [getCallSingleKeyserverEndpoint],
   );
 
   const keyserverIDs = React.useMemo(
     () => Object.keys(keyserverCallInfos),
     [keyserverCallInfos],
   );
 
   const getBoundKeyserverActionFunc: GetBoundKeyserverActionFunc =
     React.useMemo(
       () =>
         _memoize(actionFunc => actionFunc(callKeyserverEndpoint, keyserverIDs)),
       [callKeyserverEndpoint, keyserverIDs],
     );
 
   const value = React.useMemo(
     () => ({
       createCallSingleKeyserverEndpointSelector,
       getBoundSingleKeyserverActionFunc,
       getBoundKeyserverActionFunc,
     }),
     [
       createCallSingleKeyserverEndpointSelector,
       getBoundSingleKeyserverActionFunc,
       getBoundKeyserverActionFunc,
     ],
   );
 
   return (
     <CallKeyserverEndpointContext.Provider value={value}>
       {props.children}
     </CallKeyserverEndpointContext.Provider>
   );
 }
 
 function useCallKeyserverEndpointContext(): CallKeyserverEndpointContextType {
   const callKeyserverEndpointContext = React.useContext(
     CallKeyserverEndpointContext,
   );
   invariant(
     callKeyserverEndpointContext,
     'callKeyserverEndpointContext should be set',
   );
   return callKeyserverEndpointContext;
 }
 
 let socketAPIHandler: ?SocketAPIHandler = null;
 function registerActiveSocket(passedSocketAPIHandler: ?SocketAPIHandler) {
   socketAPIHandler = passedSocketAPIHandler;
 }
 
 export {
   CallKeyserverEndpointProvider,
   useCallKeyserverEndpointContext,
   registerActiveSocket,
 };
diff --git a/lib/keyserver-conn/keyserver-call-infos.js b/lib/keyserver-conn/keyserver-call-infos.js
index 476ef132d..a3db2b17f 100644
--- a/lib/keyserver-conn/keyserver-call-infos.js
+++ b/lib/keyserver-conn/keyserver-call-infos.js
@@ -1,56 +1,61 @@
 // @flow
 
 import { createSelector } from 'reselect';
 
 import { useDerivedObject } from '../hooks/objects.js';
 import type { PlatformDetails } from '../types/device-types.js';
 import type { KeyserverInfo } from '../types/keyserver-types.js';
 
 export type KeyserverInfoPartial = $ReadOnly<{
   ...Partial<KeyserverInfo>,
   +urlPrefix: $PropertyType<KeyserverInfo, 'urlPrefix'>,
 }>;
 
 export type KeyserverCallInfo = {
   +cookie: ?string,
   +urlPrefix: string,
   +sessionID: ?string,
   +isSocketConnected: boolean,
+  +sessionRecoveryInProgress: boolean,
   +lastCommunicatedPlatformDetails: ?PlatformDetails,
 };
 
 const createKeyserverCallSelector: () => KeyserverInfoPartial => KeyserverCallInfo =
   () =>
     createSelector(
       (keyserverInfo: KeyserverInfoPartial) => keyserverInfo.cookie,
       (keyserverInfo: KeyserverInfoPartial) => keyserverInfo.urlPrefix,
       (keyserverInfo: KeyserverInfoPartial) => keyserverInfo.sessionID,
       (keyserverInfo: KeyserverInfoPartial) =>
         keyserverInfo.connection?.status === 'connected',
+      (keyserverInfo: KeyserverInfoPartial) =>
+        !!keyserverInfo.connection?.sessionRecoveryInProgress,
       (keyserverInfo: KeyserverInfoPartial) =>
         keyserverInfo.lastCommunicatedPlatformDetails,
       (
         cookie: ?string,
         urlPrefix: string,
         sessionID: ?string,
         isSocketConnected: boolean,
+        sessionRecoveryInProgress: boolean,
         lastCommunicatedPlatformDetails: ?PlatformDetails,
       ) => ({
         cookie,
         urlPrefix,
         sessionID,
         isSocketConnected,
+        sessionRecoveryInProgress,
         lastCommunicatedPlatformDetails,
       }),
     );
 
 function useKeyserverCallInfos(keyserverInfos: {
   +[keyserverID: string]: KeyserverInfoPartial,
 }): { +[keyserverID: string]: KeyserverCallInfo } {
   return useDerivedObject<KeyserverInfoPartial, KeyserverCallInfo>(
     keyserverInfos,
     createKeyserverCallSelector,
   );
 }
 
 export { useKeyserverCallInfos };
diff --git a/lib/selectors/server-calls.js b/lib/selectors/server-calls.js
index ebb72ab66..ba21ca727 100644
--- a/lib/selectors/server-calls.js
+++ b/lib/selectors/server-calls.js
@@ -1,63 +1,65 @@
 // @flow
 
 import _memoize from 'lodash/memoize.js';
 import { createSelector } from 'reselect';
 
 import {
   cookieSelector,
   sessionIDSelector,
   urlPrefixSelector,
   connectionSelector,
   lastCommunicatedPlatformDetailsSelector,
 } from './keyserver-selectors.js';
 import type { PlatformDetails } from '../types/device-types.js';
 import type { AppState } from '../types/redux-types.js';
 import type { ConnectionInfo } from '../types/socket-types.js';
 import { type CurrentUserInfo } from '../types/user-types.js';
 
 export type ServerCallState = {
   +cookie: ?string,
   +urlPrefix: ?string,
   +sessionID: ?string,
   +currentUserInfo: ?CurrentUserInfo,
   +isSocketConnected: ?boolean,
+  +sessionRecoveryInProgress: ?boolean,
   +lastCommunicatedPlatformDetails: ?PlatformDetails,
 };
 
 const baseServerCallStateSelector: (
   keyserverID: string,
 ) => (state: AppState) => ServerCallState = keyserverID =>
   createSelector(
     cookieSelector(keyserverID),
     urlPrefixSelector(keyserverID),
     sessionIDSelector(keyserverID),
     (state: AppState) => state.currentUserInfo,
     connectionSelector(keyserverID),
     lastCommunicatedPlatformDetailsSelector(keyserverID),
     (
       cookie: ?string,
       urlPrefix: ?string,
       sessionID: ?string,
       currentUserInfo: ?CurrentUserInfo,
       connectionInfo: ?ConnectionInfo,
       lastCommunicatedPlatformDetails: ?PlatformDetails,
     ) => ({
       cookie,
       urlPrefix,
       sessionID,
       currentUserInfo,
       isSocketConnected:
         connectionInfo?.status !== undefined
           ? connectionInfo?.status === 'connected'
           : undefined,
+      sessionRecoveryInProgress: connectionInfo?.sessionRecoveryInProgress,
       lastCommunicatedPlatformDetails,
     }),
   );
 
 const serverCallStateSelector: (
   keyserverID: string,
 ) => (state: AppState) => ServerCallState = _memoize(
   baseServerCallStateSelector,
 );
 
 export { serverCallStateSelector };
diff --git a/lib/utils/action-utils.js b/lib/utils/action-utils.js
index 4a49887c8..7285992a3 100644
--- a/lib/utils/action-utils.js
+++ b/lib/utils/action-utils.js
@@ -1,69 +1,73 @@
 // @flow
 
 import invariant from 'invariant';
 import * as React from 'react';
 
 import { useSelector, useDispatch } from './redux-utils.js';
 import { ashoatKeyserverID } from './validation-utils.js';
 import {
   type ServerCallSelectorParams,
   useCallKeyserverEndpointContext,
 } from '../keyserver-conn/call-keyserver-endpoint-provider.react.js';
 import type { SingleKeyserverActionFunc } from '../keyserver-conn/keyserver-conn-types.js';
 import { serverCallStateSelector } from '../selectors/server-calls.js';
 
 function useLegacyAshoatKeyserverCall<F>(
   serverCall: SingleKeyserverActionFunc<F>,
   paramOverride?: ?Partial<ServerCallSelectorParams>,
 ): F {
   const {
     createCallSingleKeyserverEndpointSelector,
     getBoundSingleKeyserverActionFunc,
   } = useCallKeyserverEndpointContext();
 
   const cachedNonOverridenBoundServerCall = React.useMemo(
     () => getBoundSingleKeyserverActionFunc(ashoatKeyserverID, serverCall),
     [getBoundSingleKeyserverActionFunc, serverCall],
   );
 
   const customSelector = React.useMemo(
     () => createCallSingleKeyserverEndpointSelector(ashoatKeyserverID),
     [createCallSingleKeyserverEndpointSelector],
   );
   const dispatch = useDispatch();
   const serverCallState = useSelector(
     serverCallStateSelector(ashoatKeyserverID),
   );
 
   return React.useMemo(() => {
     if (!paramOverride) {
       return cachedNonOverridenBoundServerCall;
     }
 
-    const { urlPrefix, isSocketConnected } = serverCallState;
+    const { urlPrefix, isSocketConnected, sessionRecoveryInProgress } =
+      serverCallState;
     invariant(
       !!urlPrefix &&
         isSocketConnected !== undefined &&
-        isSocketConnected !== null,
+        isSocketConnected !== null &&
+        sessionRecoveryInProgress !== undefined &&
+        sessionRecoveryInProgress !== null,
       'keyserver missing from keyserverStore',
     );
 
     const callSingleKeyserverEndpoint = customSelector({
       ...serverCallState,
       urlPrefix,
       isSocketConnected,
+      sessionRecoveryInProgress,
       dispatch,
       ...paramOverride,
     });
     return serverCall(callSingleKeyserverEndpoint);
   }, [
     cachedNonOverridenBoundServerCall,
     serverCall,
     serverCallState,
     dispatch,
     paramOverride,
     customSelector,
   ]);
 }
 
 export { useLegacyAshoatKeyserverCall };
diff --git a/lib/utils/keyserver-call.js b/lib/utils/keyserver-call.js
index ca6372979..ab446f92e 100644
--- a/lib/utils/keyserver-call.js
+++ b/lib/utils/keyserver-call.js
@@ -1,120 +1,122 @@
 // @flow
 
 import _memoize from 'lodash/memoize.js';
 import * as React from 'react';
 
 import type { CallSingleKeyserverEndpointOptions } from './call-single-keyserver-endpoint.js';
 import { promiseAll } from './promises.js';
 import { useSelector, useDispatch } from './redux-utils.js';
 import { useCallKeyserverEndpointContext } from '../keyserver-conn/call-keyserver-endpoint-provider.react.js';
 import {
   useKeyserverCallInfos,
   type KeyserverInfoPartial,
 } from '../keyserver-conn/keyserver-call-infos.js';
 import type { ActionFunc } from '../keyserver-conn/keyserver-conn-types.js';
 import type { Endpoint } from '../types/endpoints.js';
 import type { Dispatch } from '../types/redux-types.js';
 import type { CurrentUserInfo } from '../types/user-types.js';
 
 export type KeyserverCallParamOverride = Partial<{
   +dispatch: Dispatch,
   +currentUserInfo: ?CurrentUserInfo,
   +keyserverInfos: { +[keyserverID: string]: KeyserverInfoPartial },
 }>;
 
 function useKeyserverCall<Args: mixed, Return>(
   keyserverCall: ActionFunc<Args, Return>,
   paramOverride?: ?KeyserverCallParamOverride,
 ): Args => Promise<Return> {
   const baseDispatch = useDispatch();
   const baseCurrentUserInfo = useSelector(state => state.currentUserInfo);
 
   const keyserverInfos = useSelector(
     state => state.keyserverStore.keyserverInfos,
   );
   const baseCombinedInfo = {
     dispatch: baseDispatch,
     currentUserInfo: baseCurrentUserInfo,
     keyserverInfos,
     ...paramOverride,
   };
 
   const {
     dispatch,
     currentUserInfo,
     keyserverInfos: keyserverInfoPartials,
   } = baseCombinedInfo;
   const keyserverCallInfos = useKeyserverCallInfos(keyserverInfoPartials);
 
   const {
     createCallSingleKeyserverEndpointSelector,
     getBoundKeyserverActionFunc,
   } = useCallKeyserverEndpointContext();
   const getCallSingleKeyserverEndpointSelector: typeof createCallSingleKeyserverEndpointSelector =
     React.useMemo(
       () => _memoize(createCallSingleKeyserverEndpointSelector),
       [createCallSingleKeyserverEndpointSelector],
     );
 
   const cachedNonOverridenBoundKeyserverCall = React.useMemo(
     () => getBoundKeyserverActionFunc(keyserverCall),
     [getBoundKeyserverActionFunc, keyserverCall],
   );
 
   return React.useMemo(() => {
     if (!paramOverride) {
       return cachedNonOverridenBoundKeyserverCall;
     }
 
     const callKeyserverEndpoint = (
       endpoint: Endpoint,
       requests: { +[keyserverID: string]: ?{ +[string]: mixed } },
       options?: ?CallSingleKeyserverEndpointOptions,
     ) => {
       const makeCallToSingleKeyserver = (keyserverID: string) => {
         const {
           cookie,
           urlPrefix,
           sessionID,
           isSocketConnected,
+          sessionRecoveryInProgress,
           lastCommunicatedPlatformDetails,
         } = keyserverCallInfos[keyserverID];
 
         const boundCallSingleKeyserverEndpoint =
           getCallSingleKeyserverEndpointSelector(keyserverID)({
             dispatch,
             currentUserInfo,
             cookie,
             urlPrefix,
             sessionID,
             isSocketConnected,
+            sessionRecoveryInProgress,
             lastCommunicatedPlatformDetails,
           });
 
         return boundCallSingleKeyserverEndpoint(
           endpoint,
           requests[keyserverID],
           options,
         );
       };
 
       const promises: { [string]: Promise<Object> } = {};
       for (const keyserverID in requests) {
         promises[keyserverID] = makeCallToSingleKeyserver(keyserverID);
       }
       return promiseAll(promises);
     };
     const keyserverIDs = Object.keys(keyserverCallInfos);
     return keyserverCall(callKeyserverEndpoint, keyserverIDs);
   }, [
     paramOverride,
     cachedNonOverridenBoundKeyserverCall,
     dispatch,
     currentUserInfo,
     keyserverCallInfos,
     getCallSingleKeyserverEndpointSelector,
     keyserverCall,
   ]);
 }
 
 export { useKeyserverCall };