fix: evita logout al cambiar de tenant (race condition refresh token)
This commit is contained in:
@@ -590,18 +590,6 @@ export async function switchTenant(params: {
|
|||||||
throw new AppError(404, 'Empresa no encontrada o desactivada');
|
throw new AppError(404, 'Empresa no encontrada o desactivada');
|
||||||
}
|
}
|
||||||
|
|
||||||
// Persiste el target como "último tenant activo" — al re-loguear caerá aquí
|
|
||||||
// sin tener que volver a hacer switch.
|
|
||||||
const previousTenantId = user.lastTenantId;
|
|
||||||
await prisma.user.update({
|
|
||||||
where: { id: user.id },
|
|
||||||
data: { lastTenantId: targetTenant.id },
|
|
||||||
});
|
|
||||||
|
|
||||||
// Invalida el refresh token actual (puede no existir si el caller pasó el
|
|
||||||
// access token por error — deleteMany es idempotente).
|
|
||||||
await prisma.refreshToken.deleteMany({ where: { token: params.currentRefreshToken } });
|
|
||||||
|
|
||||||
const [platformRoles, tenants] = await Promise.all([
|
const [platformRoles, tenants] = await Promise.all([
|
||||||
getPlatformRoles(user.id),
|
getPlatformRoles(user.id),
|
||||||
getUserTenants(user.id),
|
getUserTenants(user.id),
|
||||||
@@ -619,13 +607,26 @@ export async function switchTenant(params: {
|
|||||||
const accessToken = generateAccessToken(tokenPayload);
|
const accessToken = generateAccessToken(tokenPayload);
|
||||||
const refreshToken = generateRefreshToken(tokenPayload);
|
const refreshToken = generateRefreshToken(tokenPayload);
|
||||||
|
|
||||||
await prisma.refreshToken.create({
|
// Persiste el target como "último tenant activo" y atomiza la rotacion del
|
||||||
|
// refresh token (delete + create) para evitar race conditions con requests
|
||||||
|
// concurrentes que intenten refrescar con el token anterior.
|
||||||
|
const previousTenantId = user.lastTenantId;
|
||||||
|
await prisma.$transaction([
|
||||||
|
prisma.user.update({
|
||||||
|
where: { id: user.id },
|
||||||
|
data: { lastTenantId: targetTenant.id },
|
||||||
|
}),
|
||||||
|
// Invalida el refresh token actual (puede no existir si el caller pasó el
|
||||||
|
// access token por error — deleteMany es idempotente).
|
||||||
|
prisma.refreshToken.deleteMany({ where: { token: params.currentRefreshToken } }),
|
||||||
|
prisma.refreshToken.create({
|
||||||
data: {
|
data: {
|
||||||
userId: user.id,
|
userId: user.id,
|
||||||
token: refreshToken,
|
token: refreshToken,
|
||||||
expiresAt: new Date(Date.now() + 7 * 24 * 60 * 60 * 1000),
|
expiresAt: new Date(Date.now() + 7 * 24 * 60 * 60 * 1000),
|
||||||
},
|
},
|
||||||
});
|
}),
|
||||||
|
]);
|
||||||
|
|
||||||
auditLog({
|
auditLog({
|
||||||
userId: user.id,
|
userId: user.id,
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ import { Header } from '@/components/layouts/header';
|
|||||||
import { Card, CardContent, CardHeader, CardTitle, Button, Input, Label, Select, SelectContent, SelectItem, SelectTrigger, SelectValue, Dialog, DialogContent, DialogHeader, DialogTitle, DialogDescription, DialogFooter } from '@horux/shared-ui';
|
import { Card, CardContent, CardHeader, CardTitle, Button, Input, Label, Select, SelectContent, SelectItem, SelectTrigger, SelectValue, Dialog, DialogContent, DialogHeader, DialogTitle, DialogDescription, DialogFooter } from '@horux/shared-ui';
|
||||||
import { getMyTenants, addMyTenant, type MyTenantDetailed } from '@/lib/api/tenants';
|
import { getMyTenants, addMyTenant, type MyTenantDetailed } from '@/lib/api/tenants';
|
||||||
import { switchTenant } from '@/lib/api/auth';
|
import { switchTenant } from '@/lib/api/auth';
|
||||||
|
import { cancelAllApiRequests } from '@/lib/api/client';
|
||||||
import { useAuthStore } from '@/stores/auth-store';
|
import { useAuthStore } from '@/stores/auth-store';
|
||||||
import { formatCurrency } from '@/lib/utils';
|
import { formatCurrency } from '@/lib/utils';
|
||||||
import { Building2, Plus, Crown, ArrowRight, Loader2, AlertCircle, CheckCircle2 } from 'lucide-react';
|
import { Building2, Plus, Crown, ArrowRight, Loader2, AlertCircle, CheckCircle2 } from 'lucide-react';
|
||||||
@@ -61,6 +62,9 @@ export default function MisEmpresasPage() {
|
|||||||
router.push('/dashboard');
|
router.push('/dashboard');
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
// Cancela requests pendientes para evitar que intenten refrescar con el
|
||||||
|
// token que switchTenant va a invalidar.
|
||||||
|
cancelAllApiRequests();
|
||||||
try {
|
try {
|
||||||
const res = await switchTenant(tenantId);
|
const res = await switchTenant(tenantId);
|
||||||
setTokens(res.accessToken, res.refreshToken);
|
setTokens(res.accessToken, res.refreshToken);
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import { useState, useEffect } from 'react';
|
|||||||
import { useQueryClient } from '@tanstack/react-query';
|
import { useQueryClient } from '@tanstack/react-query';
|
||||||
import { useAuthStore } from '@/stores/auth-store';
|
import { useAuthStore } from '@/stores/auth-store';
|
||||||
import { switchTenant } from '@/lib/api/auth';
|
import { switchTenant } from '@/lib/api/auth';
|
||||||
|
import { cancelAllApiRequests } from '@/lib/api/client';
|
||||||
import { Building2, ChevronDown, Check, Loader2, Crown } from 'lucide-react';
|
import { Building2, ChevronDown, Check, Loader2, Crown } from 'lucide-react';
|
||||||
import { cn } from '@horux/shared-ui';
|
import { cn } from '@horux/shared-ui';
|
||||||
import { isGlobalAdminRfc } from '@horux/shared';
|
import { isGlobalAdminRfc } from '@horux/shared';
|
||||||
@@ -44,6 +45,9 @@ export function MembershipSwitcher() {
|
|||||||
const handleSwitch = async (tenantId: string) => {
|
const handleSwitch = async (tenantId: string) => {
|
||||||
if (tenantId === user?.tenantId) { setOpen(false); return; }
|
if (tenantId === user?.tenantId) { setOpen(false); return; }
|
||||||
setSwitching(true);
|
setSwitching(true);
|
||||||
|
// Cancela requests pendientes para evitar que intenten refrescar con el
|
||||||
|
// token que switchTenant va a invalidar.
|
||||||
|
cancelAllApiRequests();
|
||||||
try {
|
try {
|
||||||
const res = await switchTenant(tenantId);
|
const res = await switchTenant(tenantId);
|
||||||
setTokens(res.accessToken, res.refreshToken);
|
setTokens(res.accessToken, res.refreshToken);
|
||||||
|
|||||||
@@ -7,6 +7,14 @@ export const apiClient = axios.create({
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Lock para refrescos: solo un /auth/refresh puede estar en vuelo a la vez.
|
||||||
|
// Cualquier otra peticion 401 espera el resultado del refresh en curso.
|
||||||
|
let refreshPromise: Promise<{ accessToken: string; refreshToken: string }> | null = null;
|
||||||
|
|
||||||
|
// Controllers de peticiones activas, para poder cancelarlas en operaciones
|
||||||
|
// que invalidan el refresh token (ej. cambio de tenant real).
|
||||||
|
const activeControllers = new Set<AbortController>();
|
||||||
|
|
||||||
apiClient.interceptors.request.use((config) => {
|
apiClient.interceptors.request.use((config) => {
|
||||||
if (typeof window !== 'undefined') {
|
if (typeof window !== 'undefined') {
|
||||||
const token = localStorage.getItem('accessToken');
|
const token = localStorage.getItem('accessToken');
|
||||||
@@ -26,13 +34,30 @@ apiClient.interceptors.request.use((config) => {
|
|||||||
// Ignore parse errors
|
// Ignore parse errors
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Rastrear controller para cancelacion masiva
|
||||||
|
const controller = new AbortController();
|
||||||
|
config.signal = controller.signal;
|
||||||
|
(config as any)._horuxController = controller;
|
||||||
|
activeControllers.add(controller);
|
||||||
}
|
}
|
||||||
return config;
|
return config;
|
||||||
});
|
});
|
||||||
|
|
||||||
|
function releaseController(config: any) {
|
||||||
|
const controller = config?._horuxController as AbortController | undefined;
|
||||||
|
if (controller) {
|
||||||
|
activeControllers.delete(controller);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
apiClient.interceptors.response.use(
|
apiClient.interceptors.response.use(
|
||||||
(response) => response,
|
(response) => {
|
||||||
|
releaseController(response.config);
|
||||||
|
return response;
|
||||||
|
},
|
||||||
async (error) => {
|
async (error) => {
|
||||||
|
releaseController(error.config);
|
||||||
const originalRequest = error.config;
|
const originalRequest = error.config;
|
||||||
|
|
||||||
// Rate limit hit. El backend envía { message } — lo preservamos para que los
|
// Rate limit hit. El backend envía { message } — lo preservamos para que los
|
||||||
@@ -67,9 +92,11 @@ apiClient.interceptors.response.use(
|
|||||||
if (error.response?.status === 401 && !originalRequest._retry) {
|
if (error.response?.status === 401 && !originalRequest._retry) {
|
||||||
originalRequest._retry = true;
|
originalRequest._retry = true;
|
||||||
|
|
||||||
try {
|
if (!refreshPromise) {
|
||||||
|
refreshPromise = (async () => {
|
||||||
const refreshToken = localStorage.getItem('refreshToken');
|
const refreshToken = localStorage.getItem('refreshToken');
|
||||||
if (refreshToken) {
|
if (!refreshToken) throw new Error('No refresh token');
|
||||||
|
|
||||||
const response = await axios.post(
|
const response = await axios.post(
|
||||||
`${process.env.NEXT_PUBLIC_API_URL || 'http://localhost:4000/api'}/auth/refresh`,
|
`${process.env.NEXT_PUBLIC_API_URL || 'http://localhost:4000/api'}/auth/refresh`,
|
||||||
{ refreshToken }
|
{ refreshToken }
|
||||||
@@ -78,17 +105,35 @@ apiClient.interceptors.response.use(
|
|||||||
const { accessToken, refreshToken: newRefreshToken } = response.data;
|
const { accessToken, refreshToken: newRefreshToken } = response.data;
|
||||||
localStorage.setItem('accessToken', accessToken);
|
localStorage.setItem('accessToken', accessToken);
|
||||||
localStorage.setItem('refreshToken', newRefreshToken);
|
localStorage.setItem('refreshToken', newRefreshToken);
|
||||||
|
return { accessToken, refreshToken: newRefreshToken };
|
||||||
|
})().finally(() => {
|
||||||
|
refreshPromise = null;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const { accessToken } = await refreshPromise;
|
||||||
originalRequest.headers.Authorization = `Bearer ${accessToken}`;
|
originalRequest.headers.Authorization = `Bearer ${accessToken}`;
|
||||||
return apiClient(originalRequest);
|
return apiClient(originalRequest);
|
||||||
}
|
|
||||||
} catch {
|
} catch {
|
||||||
localStorage.removeItem('accessToken');
|
localStorage.removeItem('accessToken');
|
||||||
localStorage.removeItem('refreshToken');
|
localStorage.removeItem('refreshToken');
|
||||||
|
localStorage.removeItem('horux-tenant-view');
|
||||||
window.location.href = '/login';
|
window.location.href = '/login';
|
||||||
|
return Promise.reject(error);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return Promise.reject(error);
|
return Promise.reject(error);
|
||||||
}
|
}
|
||||||
);
|
);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Cancela todas las peticiones activas del apiClient.
|
||||||
|
* Util antes de operaciones que invalidan el refresh token (ej. switch-tenant)
|
||||||
|
* para evitar race conditions entre requests viejas y el nuevo par de tokens.
|
||||||
|
*/
|
||||||
|
export function cancelAllApiRequests() {
|
||||||
|
activeControllers.forEach((controller) => controller.abort());
|
||||||
|
activeControllers.clear();
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user