fix(frontend): preserve iframe theme across login

The iframe login used a fixed queryless redirect and called setTheme, which both dropped the iframe theme and persisted it into the normal site's shared theme storage.
This commit is contained in:
Rocky Kwok committed 2026-07-27 09:54:22 +08:00
1 parent 59f2f56124
commit 75aad4eef6
6 files changed
+187 -30

No files matched your search

+22 -13
View File
@@ -27,6 +27,10 @@ import {
persistAccessToken,
persistToolsToken,
} from '@/core/auth/yuser-login';
import {
isChatsIframeLogin,
resolvePostLoginRedirectPath,
} from '@/core/page-layout/chats-iframe-route';
import {
isSidebarEnabledFromSearch,
persistSidebarEnabled,
@@ -66,9 +70,7 @@ function summarizeStoredUserInfo(): Record<string, unknown> | null {
}
const DEFAULT_USERNAME = 'guest';
const IFRAME_LOGIN_SOURCE = 'iframe';
const DEFAULT_REDIRECT_PATH = '/page/workspace/chats/new';
const IFRAME_REDIRECT_PATH = '/page/workspace/chats_iframe/new';
type LoginStatus = 'loading' | 'success' | 'error';
@@ -89,19 +91,25 @@ export default function LoginPage() {
() => readEmbedParam(searchParams.toString()) === 'on',
[searchParams],
);
const redirectPath = useMemo(
() =>
searchParams.get('from') === IFRAME_LOGIN_SOURCE
? IFRAME_REDIRECT_PATH
: DEFAULT_REDIRECT_PATH,
const isIframeLogin = useMemo(
() => isChatsIframeLogin(searchParams.toString()),
[searchParams],
);
const resolvedUsername = useMemo(() => {
const raw = username?.trim();
return raw && raw.length > 0 ? raw : DEFAULT_USERNAME;
}, [username]);
// 外部跳转参数:theme 直接调 setTheme;ticket/userId/env 持久化到 localStorage
// 外部跳转参数:普通登录的 theme 调 setTheme;iframe 登录由 forcedTheme 接管。
// ticket/userId/env 仍按既有登录存储策略持久化。
const yUserId = searchParams.get('yUserId')?.trim() ?? '';
const redirectPath = useMemo(
() =>
resolvePostLoginRedirectPath(
searchParams.toString(),
yUserId ? GENERAL_QA_ROUTE : DEFAULT_REDIRECT_PATH,
),
[searchParams, yUserId],
);
const toolToken = searchParams.get('token')?.trim() ?? '';
// 内网登录透传的访问令牌:存入 userInfo.accessToken,供 XK/PG 中任务 iframe 拼参
const access_token = searchParams.get('access_token')?.trim() ?? '';
@@ -176,14 +184,14 @@ export default function LoginPage() {
// 主题切换:无界 props.theme 优先,其次 URL query theme。两者取值约定一致
// (dark-blue → 深色;red → 浅色;blue 暂无该主题,先用浅色代替),统一走 resolveThemeFromQuery。
// **嵌入模式跳过**:setTheme 会写 localStorage `strategy-theme`,跨标签污染正常系统页;
// 嵌入主题改由 forcedTheme(按标签页、不持久化)接管。
// **嵌入模式和 chats_iframe 登录跳过**:setTheme 会写 localStorage `strategy-theme`,
// 跨标签污染正常系统页;iframe 主题改由 forcedTheme(按标签页、不持久化)接管。
useEffect(() => {
if (isEmbed) return;
if (isEmbed || isIframeLogin) return;
const themeValue = wujieProps.theme || externalParams.theme;
if (!themeValue) return;
setTheme(resolveThemeFromQuery(themeValue));
}, [isEmbed, wujieProps.theme, externalParams.theme, setTheme]);
}, [isEmbed, isIframeLogin, wujieProps.theme, externalParams.theme, setTheme]);
// ticket / userId / env 持久化(与 login 请求并行做,互不阻塞)
useEffect(() => {
@@ -341,7 +349,7 @@ export default function LoginPage() {
}
}
router.replace(yUserId ? GENERAL_QA_ROUTE : redirectPath);
router.replace(redirectPath);
} catch (error) {
if (cancelled) {
return;
@@ -371,6 +379,7 @@ export default function LoginPage() {
taskId,
goPath,
redirectTo,
redirectPath,
isEmbed,
]);