More stuff

src/lib/hooks/context/use-theme.tsx

@@ -25,17 +25,12 @@ const ThemeProvider = ({
 
 type ThemeToggleProps = {
   size?: number;
-  buttonClassName?: ComponentProps<typeof Button>['className'];
-  buttonProps?: Omit<ComponentProps<typeof Button>, 'className' | 'onClick'>;
+  buttonProps?: Omit<ComponentProps<typeof Button>, 'onClick'>;
 };
 
 const ThemeToggle = ({
   size = 1,
-  buttonClassName,
-  buttonProps = {
-    variant: 'outline',
-    size: 'icon',
-  },
+  buttonProps,
 }: ThemeToggleProps) => {
 
   const { setTheme, resolvedTheme } = useTheme();
@@ -45,7 +40,7 @@ const ThemeToggle = ({
 
   if (!mounted) {
     return (
-      <Button className={buttonClassName} {...buttonProps}>
+      <Button {...buttonProps}>
         <span style={{ height: `${size}rem`, width: `${size}rem` }} />
       </Button>
     );
@@ -58,9 +53,11 @@ const ThemeToggle = ({
 
   return (
     <Button
-      className={cn('cursor-pointer', buttonClassName)}
-      onClick={toggleTheme}
+      variant='outline'
+      size='icon'
       {...buttonProps}
+      onClick={toggleTheme}
+      className={cn('cursor-pointer', buttonProps?.className)}
     >
       <Sun
         style={{ height: `${size}rem`, width: `${size}rem` }}