Function useUnitBase

src/react/apiBase.ts:46–176

(
  shape: Shape | {'@@unitShape': () => Shape},
  scope?: Scope,
)

}

export function useUnitBase<Shape extends {[key: string]: Unit<any>}>(
  shape: Shape | {'@@unitShape': () => Shape},
  scope?: Scope,
) {
  const isSingleUnit = is.unit(shape)
  let normShape: {[key: string]: Unit<any>} = {}
  if (isSingleUnit) {
    normShape = {unit: shape}
  } else if ('@@unitShape' in shape) {
    if (typeof shape['@@unitShape'] === 'function') {
      normShape = shape['@@unitShape']()
    } else {
      throwError('expect @@unitShape to be a function')
    }
  } else {
    normShape = shape
  }
  const isList = Array.isArray(normShape)
  const flagsRef = React.useRef({
    stale: true,
    justSubscribed: false,
    scope,
  })
  const [eventsShape, storeKeys, storeValues, eventKeys, eventValues] =
    React.useMemo(() => {
      flagsRef.current.stale = true
      const shape = Array.isArray(normShape) ? [] : ({} as any)
      const storeKeys: string[] = []
      const storeValues: Array<Store<any>> = []
      const eventKeys: string[] = []
      const eventValues: Array<Unit<any>> = []
      for (const key in normShape) {
        if (!Object.prototype.hasOwnProperty.call(normShape, key)) continue
        const unit = normShape[key]
        if (!is.unit(unit)) {
          const keyMessage = isSingleUnit ? 'argument' : `value in key "${key}"`
          throwError(`expect useUnit ${keyMessage} to be a unit`)
        }
        if (is.event(unit) || is.effect(unit)) {
          shape[key] = scope ? scopeBind(unit as Event<any>, {scope}) : unit
          eventKeys.push(key)
          eventValues.push(unit)
        } else {
          shape[key] = null
          storeKeys.push(key)
          storeValues.push(unit as Store<any>)
        }
      }
      return [shape, storeKeys, storeValues, eventKeys, eventValues]
    }, [
      flagsRef,
      scope,
      ...Object.keys(normShape),
      ...Object.values(normShape),
    ])
  const stateRef = React.useRef({
    value: eventsShape,
    storeKeys,