import React, { useEffect, useState } from 'react'; import ProLayout, { PageLoading } from '@ant-design/pro-layout'; import { enable as enableDarkMode, disable as disableDarkMode, auto as followSystemColorScheme, setFetchMethod, } from 'darkreader'; import defaultProps from './defaultProps'; import { Link, history } from 'umi'; import { LogoutOutlined } from '@ant-design/icons'; import config from '@/utils/config'; import { request } from '@/utils/http'; import './index.less'; import vhCheck from 'vh-check'; import { version, changeLog } from '../version'; import { useCtx, useTheme } from '@/utils/hooks'; import { message } from 'antd'; export default function (props: any) { const ctx = useCtx(); const theme = useTheme(); const [user, setUser] = useState(); const [loading, setLoading] = useState(true); const [systemInfo, setSystemInfo] = useState<{ isInitialized: boolean }>(); const logout = () => { request.post(`${config.apiPrefix}logout`).then(() => { localStorage.removeItem(config.authKey); history.push('/login'); }); }; const getSystemInfo = () => { request.get(`${config.apiPrefix}system`).then(({ code, data }) => { if (code === 200) { setSystemInfo(data); if (!data.isInitialized) { history.push('/initialization'); setLoading(false); } else { getUser(); } } else { message.error(data); } }); }; const getUser = (needLoading = true) => { needLoading && setLoading(true); request.get(`${config.apiPrefix}user`).then(({ code, data }) => { if (code === 200 && data.username) { setUser(data); localStorage.setItem('isLogin', 'true'); if (props.location.pathname === '/') { history.push('/crontab'); } } else { message.error(data); } needLoading && setLoading(false); }); }; const reloadUser = () => { getUser(false); }; const setTheme = () => { const media = window.matchMedia('(prefers-color-scheme: dark)'); const storageTheme = localStorage.getItem('qinglong_dark_theme'); const isDark = (media.matches && storageTheme !== 'light') || storageTheme === 'dark'; if (isDark) { document.body.setAttribute('data-dark', 'true'); } else { document.body.setAttribute('data-dark', 'false'); } }; useEffect(() => { vhCheck(); }, []); useEffect(() => { if (systemInfo && systemInfo.isInitialized && !user) { getUser(); } }, [props.location.pathname]); useEffect(() => { if (!systemInfo) { getSystemInfo(); } }, [systemInfo]); useEffect(() => { setTheme(); }, [theme.theme]); useEffect(() => { const _theme = localStorage.getItem('qinglong_dark_theme') || 'auto'; setFetchMethod(window.fetch); if (_theme === 'dark') { enableDarkMode({}); } else if (_theme === 'light') { disableDarkMode(); } else { followSystemColorScheme({}); } }, []); if (['/login', '/initialization'].includes(props.location.pathname)) { document.title = `${ (config.documentTitleMap as any)[props.location.pathname] } - 控制面板`; if (systemInfo) { return props.children; } } const isFirefox = navigator.userAgent.includes('Firefox'); const isSafari = navigator.userAgent.includes('Safari') && !navigator.userAgent.includes('Chrome'); const isQQBrowser = navigator.userAgent.includes('QQBrowser'); return loading ? ( ) : ( 控制面板 {version} } menuItemRender={(menuItemProps: any, defaultDom: any) => { if ( menuItemProps.isUrl || !menuItemProps.path || location.pathname === menuItemProps.path ) { return defaultDom; } return {defaultDom}; }} postMenuData={(menuData) => { return [ ...(menuData || []), { icon: , name: '退出登录', path: 'logout', onTitleClick: () => logout(), }, ]; }} pageTitleRender={(props, pageName, info) => { if (info && typeof info.pageName === 'string') { return `${info.pageName} - 控制面板`; } return '控制面板'; }} {...defaultProps} > {React.Children.map(props.children, (child) => { return React.cloneElement(child, { ...ctx, ...theme, user, reloadUser, reloadTheme: setTheme, }); })} ); }