diff --git a/src/components/mode-toggle.tsx b/src/components/mode-toggle.tsx index 6715664..a7eb990 100644 --- a/src/components/mode-toggle.tsx +++ b/src/components/mode-toggle.tsx @@ -12,61 +12,47 @@ import { DropdownMenuTrigger, } from "@/components/ui/dropdown-menu"; -type ThemeIconProps = { - theme?: string; -}; - -const ThemeIcon = ({ theme }: ThemeIconProps) => { +export function ModeToggle() { + const { setTheme, theme } = useTheme(); const [mounted, setMounted] = React.useState(false); React.useEffect(() => { setMounted(true); }, []); - if (!mounted) { - return null; - } - - switch (theme) { - case "light": - return ( - - ); - case "dark": - return ( - - ); - case "system": - return ; - default: - return ( - <> - - - - ); - } -}; - -export function ModeToggle() { - const { setTheme, theme } = useTheme(); - return ( - - setTheme("light")}> + setTheme("light")} + className={theme === "light" ? "bg-accent" : ""} + > + Light - setTheme("dark")}> + setTheme("dark")} + className={theme === "dark" ? "bg-accent" : ""} + > + Dark - setTheme("system")}> + setTheme("system")} + className={theme === "system" ? "bg-accent" : ""} + > + System