import { useGlobal } from '@/lib/global' import { saveDarkModeToCookies } from '@/themes/theme' import { Moon, Sun } from '@/components/HeroIcons' import { useImperativeHandle } from 'react' /** * 深色模式按钮 */ const DarkModeButton = (props) => { const { cRef, className } = props const { isDarkMode, updateDarkMode } = useGlobal() /** * 对外暴露方法 */ useImperativeHandle(cRef, () => { return { handleChangeDarkMode: () => { handleChangeDarkMode() } } }) // 用户手动设置主题 const handleChangeDarkMode = () => { const newStatus = !isDarkMode saveDarkModeToCookies(newStatus) updateDarkMode(newStatus) const htmlElement = document.getElementsByTagName('html')[0] htmlElement.classList?.remove(newStatus ? 'light' : 'dark') htmlElement.classList?.add(newStatus ? 'dark' : 'light') } return