fix: 修复 AI 设计会话失效后的伪登录
问题:Main 刷新令牌返回 401 后已清空会话,但 Renderer 仍保留用户信息,AI 设计持续返回 AUTH_REQUIRED。 实现:认证初始化等待 Main 同步完成;同步或刷新失败清理登录态;Workspace 识别结构化认证错误并在重新登录后恢复加载;路由守卫响应登录状态变化。 验证:49 个聚焦测试通过,TypeScript、聚焦 Lint 与 Vite 生产构建通过。
This commit is contained in:
@@ -9,6 +9,7 @@ import {
|
||||
renameImageWorkspaceProject,
|
||||
sendImageWorkspaceMessage,
|
||||
} from '@/lib/image-workspace';
|
||||
import { useAuthStore } from '@/stores/auth';
|
||||
import {
|
||||
IMAGE_WORKSPACE_UNAVAILABLE_CODE,
|
||||
type DesignGenerationTask,
|
||||
@@ -17,7 +18,13 @@ import {
|
||||
type DesignWorkspaceSummary,
|
||||
} from '../../shared/image-workspace';
|
||||
|
||||
export type ImageWorkspaceLoadStatus = 'idle' | 'loading' | 'ready' | 'unavailable' | 'error';
|
||||
export type ImageWorkspaceLoadStatus =
|
||||
| 'idle'
|
||||
| 'loading'
|
||||
| 'ready'
|
||||
| 'unavailable'
|
||||
| 'error'
|
||||
| 'auth-required';
|
||||
|
||||
type ImageWorkspaceState = {
|
||||
status: ImageWorkspaceLoadStatus;
|
||||
@@ -44,6 +51,10 @@ function unavailable(error: unknown): boolean {
|
||||
&& (error.status === 501 || error.code === IMAGE_WORKSPACE_UNAVAILABLE_CODE);
|
||||
}
|
||||
|
||||
function authenticationRequired(error: unknown): boolean {
|
||||
return error instanceof ImageWorkspaceApiError && error.code === 'AUTH_REQUIRED';
|
||||
}
|
||||
|
||||
function messageOf(error: unknown): string {
|
||||
return error instanceof Error ? error.message : String(error);
|
||||
}
|
||||
@@ -69,6 +80,22 @@ function upsertSummary(
|
||||
}
|
||||
|
||||
export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) => {
|
||||
const handleRequestError = (error: unknown): string => {
|
||||
const message = messageOf(error);
|
||||
if (authenticationRequired(error)) {
|
||||
useAuthStore.getState().invalidateSession();
|
||||
set({
|
||||
status: 'auth-required',
|
||||
bootstrap: null,
|
||||
activeWorkspaceId: null,
|
||||
workspace: null,
|
||||
tasks: [],
|
||||
error: message,
|
||||
});
|
||||
}
|
||||
return message;
|
||||
};
|
||||
|
||||
const applyWorkspace = (workspace: DesignWorkspace): DesignWorkspace => {
|
||||
set((state) => ({
|
||||
status: 'ready',
|
||||
@@ -139,14 +166,17 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
if (activeWorkspaceId) await loadWorkspace(activeWorkspaceId);
|
||||
return bootstrap;
|
||||
} catch (error) {
|
||||
set({
|
||||
status: unavailable(error) ? 'unavailable' : 'error',
|
||||
bootstrap: null,
|
||||
activeWorkspaceId: null,
|
||||
workspace: null,
|
||||
tasks: [],
|
||||
error: messageOf(error),
|
||||
});
|
||||
const message = handleRequestError(error);
|
||||
if (!authenticationRequired(error)) {
|
||||
set({
|
||||
status: unavailable(error) ? 'unavailable' : 'error',
|
||||
bootstrap: null,
|
||||
activeWorkspaceId: null,
|
||||
workspace: null,
|
||||
tasks: [],
|
||||
error: message,
|
||||
});
|
||||
}
|
||||
return null;
|
||||
} finally {
|
||||
inFlightLoad = null;
|
||||
@@ -161,7 +191,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
set({ tasks: [] });
|
||||
return workspace;
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
throw error;
|
||||
}
|
||||
},
|
||||
@@ -170,7 +200,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
try {
|
||||
return applyWorkspace(await renameImageWorkspaceProject(workspaceId, title));
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
throw error;
|
||||
}
|
||||
},
|
||||
@@ -186,7 +216,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
try {
|
||||
await loadWorkspace(workspaceId);
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
}
|
||||
},
|
||||
|
||||
@@ -196,7 +226,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
try {
|
||||
return applyWorkspace(await fetchImageWorkspaceProject(workspaceId));
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
throw error;
|
||||
}
|
||||
},
|
||||
@@ -212,7 +242,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
set({ tasks });
|
||||
return tasks;
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
throw error;
|
||||
}
|
||||
},
|
||||
@@ -227,7 +257,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
message,
|
||||
));
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
return await recoverRevisionConflict(error);
|
||||
}
|
||||
},
|
||||
@@ -244,7 +274,7 @@ export const useImageWorkspaceStore = create<ImageWorkspaceState>((set, get) =>
|
||||
await get().refreshTasks().catch(() => []);
|
||||
return updated;
|
||||
} catch (error) {
|
||||
set({ error: messageOf(error) });
|
||||
set({ error: handleRequestError(error) });
|
||||
return await recoverRevisionConflict(error);
|
||||
}
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user