import { Switch as SwitchPrimitive, useSwitchContext } from "@seed-design/react-switch";
import { switchStyle, type SwitchVariantProps } from "@seed-design/css/recipes/switch";
import { switchmark, type SwitchmarkVariantProps } from "@seed-design/css/recipes/switchmark";
import { createSlotRecipeContext } from "../../utils/createSlotRecipeContext";
import { Primitive, type PrimitiveProps } from "@seed-design/react-primitive";
import { createWithStateProps } from "../../utils/createWithStateProps";
import React from "react";
import clsx from "clsx";
import { splitMultipleVariantsProps } from "../../utils/splitMultipleVariantsProps";

const { withContext, ClassNamesProvider } = createSlotRecipeContext(switchStyle);
const {
  withContext: withControlContext,
  PropsProvider: ControlPropsProvider,
  withProvider: withControlProvider,
} = createSlotRecipeContext(switchmark);
const withStateProps = createWithStateProps([useSwitchContext]);

////////////////////////////////////////////////////////////////////////////////////

export interface SwitchRootProps
  extends SwitchVariantProps,
    Omit<SwitchmarkVariantProps, "size">,
    SwitchPrimitive.RootProps {}

export const SwitchRoot = React.forwardRef<HTMLLabelElement, SwitchRootProps>(
  ({ className, ...props }, ref) => {
    const [{ switch: switchVariantProps, switchmark: switchmarkVariantProps }, otherProps] =
      splitMultipleVariantsProps(props, { switchmark, switch: switchStyle });

    const classNames = switchStyle(switchVariantProps);

    return (
      <ControlPropsProvider value={switchmarkVariantProps}>
        <ClassNamesProvider value={classNames}>
          <SwitchPrimitive.Root
            ref={ref}
            className={clsx(classNames.root, className)}
            {...otherProps}
          />
        </ClassNamesProvider>
      </ControlPropsProvider>
    );
  },
);
SwitchRoot.displayName = "SwitchRoot";

////////////////////////////////////////////////////////////////////////////////////

export interface SwitchControlProps extends SwitchmarkVariantProps, SwitchPrimitive.ControlProps {}

export const SwitchControl = withControlProvider<HTMLDivElement, SwitchControlProps>(
  SwitchPrimitive.Control,
  "root",
);

////////////////////////////////////////////////////////////////////////////////////

export interface SwitchThumbProps extends SwitchPrimitive.ThumbProps {}

export const SwitchThumb = withControlContext<HTMLDivElement, SwitchThumbProps>(
  SwitchPrimitive.Thumb,
  "thumb",
);

////////////////////////////////////////////////////////////////////////////////////

export interface SwitchLabelProps extends PrimitiveProps, React.HTMLAttributes<HTMLSpanElement> {}

export const SwitchLabel = withContext<HTMLSpanElement, SwitchLabelProps>(
  withStateProps(Primitive.span),
  "label",
);

////////////////////////////////////////////////////////////////////////////////////

export interface SwitchHiddenInputProps extends SwitchPrimitive.HiddenInputProps {}

export const SwitchHiddenInput = SwitchPrimitive.HiddenInput;