72 lines
1.8 KiB
TypeScript
72 lines
1.8 KiB
TypeScript
import { twMerge } from "tailwind-merge";
|
|
|
|
type BooleanString<T extends string> = T extends "true"
|
|
? true
|
|
: T extends "false"
|
|
? false
|
|
: T;
|
|
|
|
type RawVariantProps<V extends Record<string, Record<string, string>>> = {
|
|
[VariantKey in keyof V]?: BooleanString<keyof V[VariantKey] & string>;
|
|
};
|
|
|
|
export function vcn<V extends Record<string, Record<string, string>>>({
|
|
base,
|
|
variants,
|
|
defaults,
|
|
}: {
|
|
base?: string | undefined;
|
|
variants: V;
|
|
defaults: {
|
|
[VariantKey in keyof V]: BooleanString<keyof V[VariantKey] & string>;
|
|
};
|
|
}): [
|
|
(variantProps: RawVariantProps<V> & { className?: string }) => string,
|
|
(
|
|
anyProps: Record<string, any>,
|
|
options?: {
|
|
includeClassName?: boolean;
|
|
}
|
|
) => [RawVariantProps<V> & { className?: string }, Record<string, any>],
|
|
] {
|
|
return [
|
|
({ className, ...variantProps }) => {
|
|
return twMerge(
|
|
base,
|
|
...(
|
|
Object.entries(defaults) as [keyof V, keyof V[keyof V]][]
|
|
).map<string>(
|
|
([variantKey, defaultValue]) =>
|
|
variants[variantKey][
|
|
(variantProps as unknown as RawVariantProps<V>)[variantKey] ??
|
|
defaultValue
|
|
]
|
|
),
|
|
className
|
|
);
|
|
},
|
|
(anyProps, options = {}) => {
|
|
const variantKeys = Object.keys(variants) as (keyof V)[];
|
|
|
|
return Object.entries(anyProps).reduce(
|
|
([variantProps, otherProps], [key, value]) => {
|
|
if (
|
|
variantKeys.includes(key) ||
|
|
(options.includeClassName && key === "className")
|
|
) {
|
|
return [{ ...variantProps, [key]: value }, otherProps];
|
|
}
|
|
return [variantProps, { ...otherProps, [key]: value }];
|
|
},
|
|
[{}, {}]
|
|
);
|
|
},
|
|
];
|
|
}
|
|
|
|
export type VariantProps<F extends ReturnType<typeof vcn>[0]> = F extends (
|
|
props: infer P
|
|
) => string
|
|
? P
|
|
: never;
|