DocSpace-client/packages/shared/components/toggle-button/ToggleButton.theme.tsx

import React, { PropsWithChildren, forwardRef, useContext } from "react";
import { ThemeContext } from "styled-components";

import { ContainerToggleButtonTheme } from "./ToggleButton.styled";
import { ToggleButtonThemeProps } from "./ToggleButton.types";

const ToggleButtonTheme = forwardRef<
  HTMLDivElement,
  PropsWithChildren<ToggleButtonThemeProps>
>((props, ref) => {
  const defaultTheme = useContext(ThemeContext);

  const currentColorScheme = defaultTheme?.currentColorScheme;

  return (
    <ContainerToggleButtonTheme
      ref={ref}
      $currentColorScheme={currentColorScheme}
      {...props}
    />
  );
});

ToggleButtonTheme.displayName = "ToggleButtonTheme";

export default ToggleButtonTheme;