File size: 2,144 Bytes
c30ea2a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | import * as React from "react";
import { Moon, Sun } from "lucide-react";
import { useTheme } from "next-themes";
import { Button } from "@/components/ui/button";
import {
DropdownMenu,
DropdownMenuContent,
DropdownMenuItem,
DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import { cn } from "@/lib/utils";
const ThemeToggle = ({ children }: { children?: React.ReactNode }) => {
const { theme, setTheme, systemTheme } = useTheme();
const [mounted, setMounted] = React.useState(false);
// 确保组件挂载后再渲染
React.useEffect(() => {
setMounted(true);
}, []);
// 在客户端渲染之前返回null
if (!mounted) {
return null;
}
// 获取当前实际主题
const currentTheme = theme === "system" ? systemTheme : theme;
return (
<DropdownMenu>
<DropdownMenuTrigger asChild>
{!children ? (
<Button
variant="outline"
size="icon"
className="relative overflow-hidden"
>
<Sun
className={cn(
"h-[1.2rem] w-[1.2rem] transition-all duration-500",
currentTheme === "dark"
? "-rotate-90 scale-0"
: "rotate-0 scale-100"
)}
/>
<Moon
className={cn(
"absolute h-[1.2rem] w-[1.2rem] transition-all duration-500",
currentTheme === "dark"
? "rotate-0 scale-100"
: "rotate-90 scale-0"
)}
/>
<span className="sr-only">Toggle theme</span>
</Button>
) : (
children
)}
</DropdownMenuTrigger>
<DropdownMenuContent align="end">
<DropdownMenuItem onClick={() => setTheme("light")}>
Light
</DropdownMenuItem>
<DropdownMenuItem onClick={() => setTheme("dark")}>
Dark
</DropdownMenuItem>
<DropdownMenuItem onClick={() => setTheme("system")}>
System
</DropdownMenuItem>
</DropdownMenuContent>
</DropdownMenu>
);
};
export default ThemeToggle;
|