fix: avatar upload during registration completes before redirect

Registration avatar upload raced with AuthRedirect — setting the Zustand
token triggered navigation before the upload could finish. Now the token
is stored in localStorage (for API auth) but not in Zustand until the
avatar upload completes, so the page stays mounted throughout.

Extracts initSession() from login/register for reuse.
This commit is contained in:
Jannis Braun
2026-03-15 19:39:56 +01:00
parent 2e6fa3cdc6
commit 56811b9333
2 changed files with 24 additions and 15 deletions
+10 -10
View File
@@ -15,6 +15,7 @@ interface AuthState {
user: User | null;
isLoading: boolean;
error: string | null;
initSession: (token: string, user: User) => void;
login: (username: string, password: string) => Promise<void>;
register: (username: string, password: string, displayName?: string, avatarColor?: string) => Promise<void>;
logout: () => void;
@@ -42,15 +43,18 @@ export const useAuthStore = create<AuthState>((set, get) => ({
isLoading: false,
error: null,
initSession: (token: string, user: User) => {
resetUserStores();
localStorage.setItem('backspace_token', token);
set({ token, user, isLoading: false });
useInstanceStore.getState().autoConnectAll().catch(() => {});
},
login: async (username: string, password: string) => {
set({ isLoading: true, error: null });
try {
const response = await api.auth.login({ username, password });
resetUserStores();
localStorage.setItem('backspace_token', response.token);
set({ token: response.token, user: response.user, isLoading: false });
// Auto-connect to remote instances (fire-and-forget)
useInstanceStore.getState().autoConnectAll().catch(() => {});
get().initSession(response.token, response.user);
} catch (err) {
set({ isLoading: false, error: err instanceof Error ? err.message : 'Login failed' });
throw err;
@@ -61,11 +65,7 @@ export const useAuthStore = create<AuthState>((set, get) => ({
set({ isLoading: true, error: null });
try {
const response = await api.auth.register({ username, password, displayName, avatarColor });
resetUserStores();
localStorage.setItem('backspace_token', response.token);
set({ token: response.token, user: response.user, isLoading: false });
// Auto-connect to remote instances (fire-and-forget)
useInstanceStore.getState().autoConnectAll().catch(() => {});
get().initSession(response.token, response.user);
} catch (err) {
set({ isLoading: false, error: err instanceof Error ? err.message : 'Registration failed' });
throw err;