Add more stuff. Trying to fix prettier now

src/components/context/theme.tsx

@@ -0,0 +1,55 @@
+'use client';
+import * as React from 'react';
+import { ThemeProvider as NextThemesProvider } from 'next-themes';
+import { Moon, Sun } from 'lucide-react';
+import { useTheme } from 'next-themes';
+import { Button } from '@/components/ui/button';
+
+export const ThemeProvider = ({
+  children,
+  ...props
+}: React.ComponentProps<typeof NextThemesProvider>) => {
+  const [mounted, setMounted] = React.useState(false);
+
+  React.useEffect(() => {
+    setMounted(true);
+  }, []);
+
+  if (!mounted) return null;
+
+  return <NextThemesProvider {...props}>{children}</NextThemesProvider>;
+};
+
+export const ThemeToggle = () => {
+  const { setTheme, resolvedTheme } = useTheme();
+  const [mounted, setMounted] = React.useState(false);
+
+  React.useEffect(() => {
+    setMounted(true);
+  }, []);
+
+  if (!mounted) {
+    return (
+      <Button variant='outline' size='icon'>
+        <span className='h-[1.2rem] w-[1.2rem]' />
+      </Button>
+    );
+  }
+
+  const toggleTheme = () => {
+    if (resolvedTheme === 'dark') setTheme('light');
+    else setTheme('dark');
+  };
+
+  return (
+    <Button variant='outline' size='icon' onClick={toggleTheme}>
+      <Sun className='h-[1.2rem] w-[1.2rem] rotate-0 scale-100
+        transition-all dark:-rotate-90 dark:scale-0'
+      />
+      <Moon className='absolute h-[1.2rem] w-[1.2rem] rotate-90
+        scale-0 transition-all dark:rotate-0 dark:scale-100'
+      />
+      <span className='sr-only'>Toggle theme</span>
+    </Button>
+  );
+};