554 lines
15 KiB
TypeScript
554 lines
15 KiB
TypeScript
// @ts-nocheck
|
|
|
|
import { routeCacheAtom, setRouteCache } from '@/atoms/route-cache';
|
|
import { GPUStackVersionAtom, UpdateCheckAtom, userAtom } from '@/atoms/user';
|
|
import DarkMask from '@/components/dark-mask';
|
|
import IconFont from '@/components/icon-font';
|
|
import ShortCuts, {
|
|
modalConfig as ShortCutsConfig
|
|
} from '@/components/short-cuts';
|
|
import VersionInfo, { modalConfig } from '@/components/version-info';
|
|
import routeCachekey from '@/config/route-cachekey';
|
|
import { DEFAULT_ENTER_PAGE } from '@/config/settings';
|
|
import useBodyScroll from '@/hooks/use-body-scroll';
|
|
import useOverlayScroller from '@/hooks/use-overlay-scroller';
|
|
import useUserSettings from '@/hooks/use-user-settings';
|
|
import { logout } from '@/pages/login/apis';
|
|
import { useAccessMarkedRoutes } from '@@/plugin-access';
|
|
import { useModel } from '@@/plugin-model';
|
|
import { ProLayout } from '@ant-design/pro-components';
|
|
import {
|
|
Link,
|
|
Outlet,
|
|
dropByCacheKey,
|
|
history,
|
|
matchRoutes,
|
|
useAppData,
|
|
useIntl,
|
|
useLocation,
|
|
useNavigate,
|
|
type IRoute
|
|
} from '@umijs/max';
|
|
import { Button, ConfigProvider, Modal, Tooltip, theme } from 'antd';
|
|
import 'driver.js/dist/driver.css';
|
|
import { useAtom } from 'jotai';
|
|
import 'overlayscrollbars/overlayscrollbars.css';
|
|
import { useEffect, useMemo, useState } from 'react';
|
|
import { PageContainerInner } from '../pages/_components/page-box';
|
|
import Exception from './Exception';
|
|
import './Layout.css';
|
|
import { LogoIcon, SLogoIcon } from './Logo';
|
|
import ErrorBoundary from './error-boundary';
|
|
import { ExtraContent } from './extraRender';
|
|
import { getRightRenderContent } from './rightRender';
|
|
import { patchRoutes } from './runtime';
|
|
import SiderMenu from './sider-menu';
|
|
|
|
const NO_CONTAINER_PAGES = [
|
|
'chat',
|
|
'rerank',
|
|
'embedding',
|
|
'speech',
|
|
'image',
|
|
'text2images',
|
|
'clusterDetail',
|
|
'clusterCreate'
|
|
];
|
|
|
|
const loginPath = DEFAULT_ENTER_PAGE.login;
|
|
|
|
type InitialStateType = {
|
|
fetchUserInfo: () => Promise<Global.UserInfo>;
|
|
currentUser?: Global.UserInfo;
|
|
};
|
|
|
|
// Filter out the routes that need to be displayed, where filterFn indicates the levels that should not be shown
|
|
const filterRoutes = (
|
|
routes: IRoute[],
|
|
filterFn: (route: IRoute) => boolean
|
|
) => {
|
|
if (routes.length === 0) {
|
|
return [];
|
|
}
|
|
|
|
let newRoutes = [];
|
|
for (const route of routes) {
|
|
const newRoute = { ...route };
|
|
if (filterFn(route)) {
|
|
if (Array.isArray(newRoute.routes)) {
|
|
newRoutes.push(...filterRoutes(newRoute.routes, filterFn));
|
|
}
|
|
} else {
|
|
if (Array.isArray(newRoute.children)) {
|
|
newRoute.children = filterRoutes(newRoute.children, filterFn);
|
|
newRoute.routes = newRoute.children;
|
|
}
|
|
newRoutes.push(newRoute);
|
|
}
|
|
}
|
|
|
|
return newRoutes;
|
|
};
|
|
|
|
const mapRoutes = (routes: IRoute[], role: string) => {
|
|
if (routes.length === 0) {
|
|
return [];
|
|
}
|
|
return routes.map((route) => {
|
|
const newRoute = { ...route, role };
|
|
if (route.originPath) {
|
|
newRoute.path = route.originPath;
|
|
}
|
|
|
|
if (Array.isArray(route.routes)) {
|
|
newRoute.routes = mapRoutes(route.routes, role);
|
|
}
|
|
|
|
if (Array.isArray(route.children)) {
|
|
newRoute.children = mapRoutes(route.children, role);
|
|
}
|
|
|
|
return newRoute;
|
|
});
|
|
};
|
|
|
|
export default (props: any) => {
|
|
const { initialize: initialize } = useOverlayScroller({
|
|
defer: false
|
|
});
|
|
const [modal, contextHolder] = Modal.useModal();
|
|
const { themeData, setTheme, setUserSettings, userSettings, isDarkTheme } =
|
|
useUserSettings();
|
|
const { saveScrollHeight, restoreScrollHeight } = useBodyScroll();
|
|
const { initialize: initializeMenu } = useOverlayScroller();
|
|
const [userInfo] = useAtom(userAtom);
|
|
const [routeCache] = useAtom(routeCacheAtom);
|
|
const [version] = useAtom(GPUStackVersionAtom);
|
|
const [updateCheck] = useAtom(UpdateCheckAtom);
|
|
const location = useLocation();
|
|
const navigate = useNavigate();
|
|
const intl = useIntl();
|
|
const { clientRoutes, pluginManager } = useAppData();
|
|
// const [collapsed, setCollapsed] = useState(userSettings.collapsed || false);
|
|
const [collapseValue, setCollapseValue] = useState(false);
|
|
const [collapseKeys, setCollapseKeys] = useState<Set<string>>(new Set());
|
|
|
|
const initialInfo = (useModel && useModel('@@initialState')) || {
|
|
initialState: undefined,
|
|
loading: false,
|
|
setInitialState: null
|
|
};
|
|
|
|
const { initialState, loading, setInitialState } = initialInfo;
|
|
|
|
const userConfig = {
|
|
title: '',
|
|
locale: true
|
|
};
|
|
|
|
const formatMessage = (args) => {
|
|
return intl.formatMessage({ id: args.id });
|
|
};
|
|
|
|
const showVersion = () => {
|
|
saveScrollHeight();
|
|
modal.info({
|
|
...modalConfig,
|
|
width: 460,
|
|
content: <VersionInfo intl={intl} />,
|
|
onCancel: restoreScrollHeight
|
|
});
|
|
};
|
|
|
|
const showShortcuts = () => {
|
|
Modal.info({
|
|
...ShortCutsConfig,
|
|
content: <ShortCuts intl={intl} />
|
|
});
|
|
};
|
|
|
|
const initRouteCacheValue = (pathname) => {
|
|
if (routeCache.get(pathname) === undefined && routeCachekey[pathname]) {
|
|
setRouteCache(pathname, false);
|
|
}
|
|
};
|
|
|
|
const dropRouteCache = (pathname) => {
|
|
for (let key of routeCache.keys()) {
|
|
if (key !== pathname && !routeCache.get(key) && routeCachekey[key]) {
|
|
dropByCacheKey(key);
|
|
routeCache.delete(key);
|
|
}
|
|
}
|
|
};
|
|
|
|
const runtimeConfig = {
|
|
...initialInfo,
|
|
logout: async (userInfo) => {
|
|
await logout();
|
|
navigate(loginPath);
|
|
},
|
|
showVersion: () => {
|
|
return showVersion();
|
|
},
|
|
showShortcuts: () => {
|
|
return showShortcuts();
|
|
},
|
|
notFound: <span>404 not found</span>
|
|
};
|
|
|
|
const handleToggleCollapse = (e: any) => {
|
|
e.stopPropagation();
|
|
// setCollapsed(!collapsed);
|
|
setUserSettings({
|
|
...userSettings,
|
|
collapsed: !userSettings.collapsed
|
|
});
|
|
};
|
|
|
|
const newRoutes = filterRoutes(
|
|
clientRoutes.filter((route) => route.id === 'max-tabs'),
|
|
(route) => {
|
|
return (
|
|
(!!route.isLayout && route.id !== '@@/global-layout') ||
|
|
!!route.isWrapper
|
|
);
|
|
}
|
|
);
|
|
|
|
const role = initialState?.currentUser?.is_admin ? 'admin' : 'user';
|
|
const [route] = useAccessMarkedRoutes(mapRoutes(newRoutes, role));
|
|
|
|
patchRoutes({
|
|
routes: route.children,
|
|
initialState: initialInfo.initialState
|
|
});
|
|
|
|
const matchedRoute = useMemo(
|
|
() => matchRoutes(route?.children || [], location.pathname)?.pop?.()?.route,
|
|
[location.pathname]
|
|
);
|
|
|
|
const isNoContainerPage = useMemo(() => {
|
|
return NO_CONTAINER_PAGES.includes(matchedRoute?.name);
|
|
}, [matchedRoute]);
|
|
|
|
console.log('matchedRoute=========', matchedRoute, route);
|
|
|
|
const allRouteKeys = useMemo(() => {
|
|
const keys = new Set<string>();
|
|
const childrenRoutes = route?.children || [];
|
|
const traverseRoutes = (routes) => {
|
|
routes.forEach((r) => {
|
|
if (r.path) {
|
|
keys.add(r.path);
|
|
}
|
|
if (r.children) {
|
|
traverseRoutes(r.children);
|
|
}
|
|
});
|
|
};
|
|
traverseRoutes(childrenRoutes);
|
|
|
|
return keys;
|
|
}, [route?.children]);
|
|
|
|
const showUpgrade = useMemo(() => {
|
|
return (
|
|
initialState?.currentUser?.is_admin &&
|
|
updateCheck.latest_version &&
|
|
updateCheck.latest_version !== version?.version &&
|
|
updateCheck.latest_version?.indexOf('0.0.0') === -1 &&
|
|
updateCheck.latest_version?.indexOf('rc') === -1
|
|
);
|
|
}, [
|
|
updateCheck.latest_version,
|
|
version.version,
|
|
initialState?.currentUser?.is_admin
|
|
]);
|
|
|
|
useEffect(() => {
|
|
const body = document.querySelector('body');
|
|
if (body) {
|
|
const ins = initialize(body);
|
|
window.__GPUSTACK_BODY_SCROLLER__ = ins;
|
|
}
|
|
}, [initialize]);
|
|
|
|
const collapsed = useMemo(() => {
|
|
return userSettings.collapsed || false;
|
|
}, [userSettings.collapsed]);
|
|
|
|
const renderMenuHeader = (logo, title) => {
|
|
return (
|
|
<>
|
|
{logo}
|
|
<div className="collapse-wrap" onClick={handleToggleCollapse}>
|
|
<Button
|
|
style={{ marginRight: collapsed ? 0 : -14, border: 'none' }}
|
|
size="small"
|
|
type={collapsed ? 'default' : 'text'}
|
|
>
|
|
<>
|
|
<IconFont
|
|
type="icon-expand-left"
|
|
className="font-size-18 text-secondary"
|
|
style={{ display: collapsed ? 'block' : 'none' }}
|
|
/>
|
|
<IconFont
|
|
type="icon-expand-right"
|
|
className="font-size-18 text-secondary"
|
|
style={{ display: !collapsed ? 'block' : 'none' }}
|
|
/>
|
|
</>
|
|
</Button>
|
|
</div>
|
|
</>
|
|
);
|
|
};
|
|
|
|
const handleToggleGroup = (menuItemProps, e) => {
|
|
e.stopPropagation();
|
|
|
|
if (collapseKeys.has(menuItemProps.key)) {
|
|
collapseKeys.delete(menuItemProps.key);
|
|
} else {
|
|
collapseKeys.add(menuItemProps.key);
|
|
}
|
|
setCollapseKeys(new Set(collapseKeys));
|
|
};
|
|
|
|
const menuContentRender = (menuProps, defaultDom) => {
|
|
return <SiderMenu {...menuProps}></SiderMenu>;
|
|
};
|
|
|
|
const actionRender = (layoutProps) => {
|
|
const dom = getRightRenderContent({
|
|
runtimeConfig,
|
|
loading,
|
|
initialState,
|
|
setInitialState,
|
|
intl,
|
|
isDarkTheme: userSettings.isDarkTheme,
|
|
siderWidth: layoutProps?.siderWidth,
|
|
collapsed: layoutProps?.collapsed,
|
|
showUpgrade
|
|
});
|
|
|
|
return dom;
|
|
};
|
|
|
|
const menuItemRender = (menuItemProps, defaultDom) => {
|
|
console.log('defaultdom==========', menuItemProps, defaultDom);
|
|
if (menuItemProps.isUrl || menuItemProps.children) {
|
|
return defaultDom;
|
|
}
|
|
if (menuItemProps.path && location.pathname !== menuItemProps.path) {
|
|
return (
|
|
<Tooltip
|
|
title={collapsed ? menuItemProps.name : false}
|
|
placement="right"
|
|
>
|
|
<Link
|
|
to={menuItemProps.path.replace('/*', '')}
|
|
target={menuItemProps.target}
|
|
>
|
|
{defaultDom}
|
|
</Link>
|
|
</Tooltip>
|
|
);
|
|
}
|
|
return (
|
|
<Tooltip title={collapsed ? menuItemProps.name : false} placement="right">
|
|
{defaultDom}
|
|
</Tooltip>
|
|
);
|
|
};
|
|
|
|
const menuDataRender = (menuData) => {
|
|
const currentItem = menuData.find((s) => location.pathname === s.path);
|
|
const result = menuData.map((item) => {
|
|
const newItem = { ...item };
|
|
|
|
const selected =
|
|
location.pathname === newItem.path ||
|
|
location.pathname.indexOf(newItem.path) > -1;
|
|
|
|
if (newItem.icon) {
|
|
newItem.icon = selected ? (
|
|
<IconFont type={newItem.selectedIcon} />
|
|
) : (
|
|
<IconFont type={newItem.defaultIcon} />
|
|
);
|
|
}
|
|
if (newItem.children) {
|
|
newItem.children = menuDataRender(newItem.children);
|
|
}
|
|
return newItem;
|
|
});
|
|
|
|
return result;
|
|
};
|
|
|
|
const onPageChange = (route) => {
|
|
const { location } = history;
|
|
const { pathname } = location;
|
|
console.log('onPageChange', pathname, route);
|
|
|
|
initRouteCacheValue(pathname);
|
|
dropRouteCache(pathname);
|
|
|
|
// if user is not change password, redirect to change password page
|
|
if (location.pathname !== loginPath && userInfo?.require_password_change) {
|
|
history.push(loginPath);
|
|
return;
|
|
}
|
|
|
|
// if user is not logged in, redirect to login page
|
|
if (!initialState?.currentUser && location.pathname !== loginPath) {
|
|
history.push(loginPath);
|
|
} else if (location.pathname === '/') {
|
|
const pathname = initialState?.currentUser?.is_admin
|
|
? DEFAULT_ENTER_PAGE.adminForNormal
|
|
: DEFAULT_ENTER_PAGE.user;
|
|
history.push(pathname);
|
|
}
|
|
};
|
|
|
|
const onMenuHeaderClick = (e) => {
|
|
e.stopPropagation();
|
|
e.preventDefault();
|
|
const pagepath = initialState?.currentUser?.is_admin
|
|
? DEFAULT_ENTER_PAGE.adminForNormal
|
|
: DEFAULT_ENTER_PAGE.user;
|
|
|
|
navigate(pagepath);
|
|
};
|
|
|
|
const onCollapse = (value) => {
|
|
setUserSettings({
|
|
...userSettings,
|
|
collapsed: value
|
|
});
|
|
};
|
|
|
|
useEffect(() => {
|
|
// clear previous marks and measures
|
|
performance.clearMarks();
|
|
performance.clearMeasures();
|
|
|
|
// record start time
|
|
performance.mark('route-start');
|
|
|
|
requestAnimationFrame(() => {
|
|
// record end time
|
|
performance.mark('route-end');
|
|
|
|
// make sure `route-start` exists
|
|
if (performance.getEntriesByName('route-start').length > 0) {
|
|
performance.measure('route-change', 'route-start', 'route-end');
|
|
|
|
const measure = performance.getEntriesByName('route-change')[0];
|
|
console.log(
|
|
`[Performance] Route change to ${location.pathname} took ${measure.duration.toFixed(2)}ms`
|
|
);
|
|
|
|
// clear marks and measures
|
|
performance.clearMarks();
|
|
performance.clearMeasures();
|
|
} else {
|
|
console.warn('Missing performance mark: route-start');
|
|
}
|
|
});
|
|
}, [location.pathname]);
|
|
|
|
const onRenderCallback = (id, phase, actualDuration) => {
|
|
console.log(
|
|
`[Profiler] Route: ${id} - Phase: ${phase} - Render time: ${actualDuration.toFixed(2)}ms`
|
|
);
|
|
};
|
|
|
|
const currentTheme = useMemo(() => {
|
|
const data = {
|
|
algorithm: userSettings.isDarkTheme
|
|
? theme.darkAlgorithm
|
|
: theme.defaultAlgorithm,
|
|
...themeData
|
|
};
|
|
return data;
|
|
}, [userSettings.isDarkTheme, themeData]);
|
|
|
|
const outlet = runtimeConfig.childrenRender ? (
|
|
runtimeConfig.childrenRender(<Outlet />, props)
|
|
) : (
|
|
<Outlet />
|
|
);
|
|
|
|
return (
|
|
<ConfigProvider
|
|
componentSize="large"
|
|
key={userSettings.colorPrimary}
|
|
theme={{
|
|
algorithm: userSettings.isDarkTheme
|
|
? theme.darkAlgorithm
|
|
: theme.defaultAlgorithm,
|
|
...themeData
|
|
}}
|
|
>
|
|
<DarkMask></DarkMask>
|
|
<ProLayout
|
|
fixSiderbar
|
|
fixedHeader
|
|
breadcrumbRender={false}
|
|
route={route}
|
|
location={location}
|
|
title={userConfig.title}
|
|
navTheme={userSettings.theme}
|
|
layout="side"
|
|
contentStyle={{
|
|
paddingBlock: 0,
|
|
paddingInline: 0
|
|
}}
|
|
openKeys={false}
|
|
disableMobile={true}
|
|
header={{
|
|
title: <div style={{ fontSize: 36 }}> gpuStack </div>
|
|
}}
|
|
siderWidth={220}
|
|
onCollapse={onCollapse}
|
|
onMenuHeaderClick={onMenuHeaderClick}
|
|
menuHeaderRender={renderMenuHeader}
|
|
extra={[<ExtraContent key="extra-content" />]}
|
|
collapsed={userSettings.collapsed}
|
|
onPageChange={onPageChange}
|
|
formatMessage={formatMessage}
|
|
menu={{
|
|
locale: true,
|
|
type: 'group'
|
|
}}
|
|
splitMenus={true}
|
|
logo={collapsed ? SLogoIcon : LogoIcon}
|
|
menuContentRender={menuContentRender}
|
|
disableContentMargin
|
|
{...runtimeConfig}
|
|
ErrorBoundary={ErrorBoundary}
|
|
>
|
|
<Exception
|
|
route={matchedRoute}
|
|
noFound={runtimeConfig?.noFound}
|
|
notFound={runtimeConfig?.notFound}
|
|
unAccessible={runtimeConfig?.unAccessible}
|
|
noAccessible={runtimeConfig?.noAccessible}
|
|
>
|
|
{isNoContainerPage ? (
|
|
outlet
|
|
) : (
|
|
<PageContainerInner>{outlet}</PageContainerInner>
|
|
)}
|
|
</Exception>
|
|
</ProLayout>
|
|
{contextHolder}
|
|
</ConfigProvider>
|
|
);
|
|
};
|