Function performCoalescedRefresh

apps/sim/app/api/auth/oauth/utils.ts:336–442

({
  accountId,
  providerId,
  refreshToken,
  requestId,
  userId,
}: CoalescedRefreshOptions)

}

async function performCoalescedRefresh({
  accountId,
  providerId,
  refreshToken,
  requestId,
  userId,
}: CoalescedRefreshOptions): Promise<string | null> {
  const logContext = {
    ...(requestId ? { requestId } : {}),
    ...(userId ? { userId } : {}),
    providerId,
    accountId,
  }

  const deadCode = await getRecentTerminalError(accountId)
  if (deadCode) {
    logger.warn('Skipping refresh: credential recently failed', {
      ...logContext,
      errorCode: deadCode,
    })
    return null
  }

  const lockKey = `oauth:refresh:${accountId}`

  const refreshPromise = coalesceLocally(lockKey, () =>
    withLeaderLock<string>({
      key: lockKey,
      onLeader: async () => {
        try {
          const result = await refreshOAuthToken(providerId, refreshToken)

          if (!result.ok) {
            logger.error('Failed to refresh token', {
              ...logContext,
              errorCode: result.errorCode,
            })
            if (result.errorCode && isTerminalRefreshError(result.errorCode)) {
              await markCredentialDead(accountId, result.errorCode)
            }
            return null
          }

          const updateData: Record<string, unknown> = {
            accessToken: result.accessToken,
            accessTokenExpiresAt: new Date(Date.now() + result.expiresIn * 1000),
            updatedAt: new Date(),
          }
          if (result.refreshToken && result.refreshToken !== refreshToken) {
            updateData.refreshToken = result.refreshToken
          }
          if (isMicrosoftProvider(providerId)) {
            updateData.refreshTokenExpiresAt = getMicrosoftRefreshTokenExpiry()
          }

          await db.update(account).set(updateData).where(eq(account.id, accountId))

          logger.info('Successfully refreshed access token', logContext)