import React, { useEffect, useState, useRef } 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, MenuFoldOutlined, MenuUnfoldOutlined, UserOutlined, } from '@ant-design/icons'; import config from '@/utils/config'; import { request } from '@/utils/http'; import './index.less'; import vhCheck from 'vh-check'; import { version, changeLogLink, changeLog } from '../version'; import { useCtx, useTheme } from '@/utils/hooks'; import { message, Badge, Modal, Avatar, Dropdown, Menu, Image } from 'antd'; // @ts-ignore import SockJS from 'sockjs-client'; import * as Sentry from '@sentry/react'; import { Integrations } from '@sentry/tracing'; Sentry.init({ dsn: 'https://3406424fb1dc4813a62d39e844a9d0ac@o1098464.ingest.sentry.io/6122818', integrations: [new Integrations.BrowserTracing()], release: version, tracesSampleRate: 1.0, }); 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 ws = useRef(null); const [socketMessage, setSocketMessage] = useState(); const [collapsed, setCollapsed] = useState(false); const logout = () => { request.post(`${config.apiPrefix}user/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); } }) .catch((error) => { console.log(error); }); }; const getUser = (needLoading = true) => { needLoading && setLoading(true); request .get(`${config.apiPrefix}user`) .then(({ code, data }) => { if (code === 200 && data.username) { setUser(data); if (props.location.pathname === '/') { history.push('/crontab'); } } else { message.error(data); } needLoading && setLoading(false); }) .catch((error) => { console.log(error); }); }; 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({}); } }, []); useEffect(() => { if (!user) return; ws.current = new SockJS( `${location.origin}/api/ws?token=${localStorage.getItem(config.authKey)}`, ); ws.current.onmessage = (e: any) => { try { const data = JSON.parse(e.data); if (data.type === 'ping') { if (data && data.message === 'hanhh') { console.log('websocket连接成功', e); } else { console.log('websocket连接失败', e); } } setSocketMessage(data); } catch (error) { console.log('websocket连接失败', e); } }; const wsCurrent = ws.current; return () => { wsCurrent.close(); }; }, [user]); useEffect(() => { window.onload = () => { const timing = performance.timing; console.log(`白屏时间: ${timing.responseStart - timing.navigationStart}`); console.log( `请求完毕至DOM加载: ${timing.domInteractive - timing.responseEnd}`, ); console.log( `解释dom树耗时: ${timing.domComplete - timing.domInteractive}`, ); console.log( `从开始至load总耗时: ${timing.loadEventEnd - timing.navigationStart}`, ); Sentry.captureMessage( `白屏时间 ${timing.responseStart - timing.navigationStart}`, ); }; }, []); if (['/login', '/initialization'].includes(props.location.pathname)) { document.title = `${ (config.documentTitleMap as any)[props.location.pathname] } - 控制面板`; if ( systemInfo?.isInitialized && props.location.pathname === '/initialization' ) { history.push('/crontab'); } if (systemInfo) { return React.Children.map(props.children, (child) => { return React.cloneElement(child, { ...ctx, ...theme, user, reloadUser, reloadTheme: setTheme, ws: ws.current, }); }); } } const isFirefox = navigator.userAgent.includes('Firefox'); const isSafari = navigator.userAgent.includes('Safari') && !navigator.userAgent.includes('Chrome'); const isQQBrowser = navigator.userAgent.includes('QQBrowser'); const menu = ( } onClick={logout}> 退出登录 ); return loading ? ( ) : ( } title={ <> 控制面板 { e.stopPropagation(); }} > v{version} } menuItemRender={(menuItemProps: any, defaultDom: any) => { if ( menuItemProps.isUrl || !menuItemProps.path || location.pathname === menuItemProps.path ) { return defaultDom; } return {defaultDom}; }} pageTitleRender={(props, pageName, info) => { if (info && typeof info.pageName === 'string') { return `${info.pageName} - 控制面板`; } return '控制面板'; }} onCollapse={setCollapsed} collapsed={collapsed} rightContentRender={() => ctx.isPhone && ( } /> admin ) } collapsedButtonRender={(collapsed) => ( { e.preventDefault(); e.stopPropagation(); }} > {!collapsed && !ctx.isPhone && ( } /> admin )} setCollapsed(!collapsed)} > {collapsed ? : } )} {...defaultProps} > {React.Children.map(props.children, (child) => { return React.cloneElement(child, { ...ctx, ...theme, user, reloadUser, reloadTheme: setTheme, socketMessage, }); })} ); }