50 lines
1.1 KiB
TypeScript
50 lines
1.1 KiB
TypeScript
'use client';
|
|
|
|
import { Moon, Sun } from 'lucide-react';
|
|
import { useTheme } from 'next-themes';
|
|
import { Button } from '../ui/button';
|
|
import { useEffect, useState } from 'react';
|
|
|
|
export function ThemeToggle() {
|
|
const { theme, setTheme } = useTheme();
|
|
const [mounted, setMounted] = useState(false);
|
|
|
|
// 避免服务端渲染和客户端渲染不匹配
|
|
useEffect(() => {
|
|
setMounted(true);
|
|
}, []);
|
|
|
|
const toggleTheme = () => {
|
|
setTheme(theme === 'light' ? 'dark' : 'light');
|
|
};
|
|
|
|
// 在组件挂载前不渲染，避免闪烁
|
|
if (!mounted) {
|
|
return (
|
|
<Button
|
|
variant="ghost"
|
|
size="icon"
|
|
disabled
|
|
className="transition-colors"
|
|
>
|
|
<Sun className="w-5 h-5" />
|
|
</Button>
|
|
);
|
|
}
|
|
|
|
return (
|
|
<Button
|
|
variant="ghost"
|
|
size="icon"
|
|
onClick={toggleTheme}
|
|
title={theme === 'light' ? '切换到深色模式' : '切换到浅色模式'}
|
|
className="transition-colors"
|
|
>
|
|
{theme === 'light' ? (
|
|
<Moon className="w-5 h-5" />
|
|
) : (
|
|
<Sun className="w-5 h-5" />
|
|
)}
|
|
</Button>
|
|
);
|
|
} |