import { useContext } from 'react'
import { shallow } from 'zustand/shallow'
import { FormStoreContext } from './FormStoreContext.js'
import type { FormValuesState } from './types.js'

export function useFormStore<T>(
  selector: (state: FormValuesState) => T,
  equalityFn = shallow
): T {
  const useStore = useContext(FormStoreContext)

  if (!useStore) {
    throw new Error('You forgot to wrap your component in <FormStoreProvider>.')
  }

  return useStore(selector, equalityFn)
}