blitzkit/packages/website/src/stores/embedState/utilities.ts

import type { TextProps } from '@radix-ui/themes';
import { EmbedState, type EmbedConfig, type ExtractEmbedConfigTypes } from '.';
import type { EmbedConfigItemType, EmbedItemType } from './constants';

export function useEmbedStateCurry<Config extends EmbedConfig>() {
  function useEmbedState<Key extends keyof Config>(key: Key) {
    return EmbedState.use((state) => {
      return (state as ExtractEmbedConfigTypes<Config>)[key];
    });
  }

  function useRichText<
    Key extends {
      [K in keyof Config]: Config[K] extends { type: EmbedItemType.RichText }
        ? K
        : never;
    }[keyof Config],
  >(key: Key) {
    const { color, ...state } = useEmbedState(
      key,
    ) as EmbedConfigItemType<EmbedItemType.RichText>['default'];

    return {
      style: { color },
      ...state,
    } satisfies TextProps;
  }

  return { useEmbedState, useRichText };
}