Stitches Token Variant

tokenVariant.ts

export function tokenVariant<
  T extends keyof typeof theme,
  V extends keyof typeof theme[T],
  K extends (value: V) => ReturnType<K>
>(type: T, variantCss: K) {
  return Object.keys(theme[type]).reduce(
    (acc, key) => ({
      ...acc,
      [key]: variantCss(key as V),
    }),
    {}
  ) as Record<V, ReturnType<K>>;
}