12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
import { createElement, useEffect, useRef, useState } from 'react'import type { PropsWithChildren } from 'react'import type { Atom, Scope } from './atom'import { createScopeContainer, getScopeContext } from './contexts'import type { ScopeContainer } from './contexts'import { COMMIT_ATOM, createStoreForExport } from './store'import type { VersionObject } from './store'import { useDebugState } from './useDebugState'
export const Provider = ({  children,  initialValues,  scope,  unstable_createStore,  unstable_enableVersionedWrite,}: PropsWithChildren<{  initialValues?: Iterable<readonly [Atom<unknown>, unknown]>  scope?: Scope  /**   * This is an unstable feature to use exported createStore.   */  unstable_createStore?: typeof createStoreForExport  /**   * This is an unstable experimental feature for React 18.   * When this is enabled, a) write function must be pure   * (read function must be pure regardless of this),   * b) React will show warning in DEV mode,   * c) then state branching works.   */  unstable_enableVersionedWrite?: boolean}>) => {  const [version, setVersion] = useState<VersionObject>()  useEffect(() => {    if (version) {      ;(scopeContainerRef.current as ScopeContainer).s[COMMIT_ATOM](        null,        version      )      delete version.p    }  }, [version])
  const scopeContainerRef = useRef<ScopeContainer>()  if (!scopeContainerRef.current) {    // lazy initialization    scopeContainerRef.current = createScopeContainer(      initialValues,      unstable_createStore    )    if (unstable_enableVersionedWrite) {      scopeContainerRef.current.w = (write) => {        setVersion((parentVersion) => {          const nextVersion = parentVersion ? { p: parentVersion } : {}          write(nextVersion)          return nextVersion        })      }    }  }
  // LIMITATION: useDebugState prevents to work versioned write properly  if (__DEV__ && !unstable_enableVersionedWrite) {    // eslint-disable-next-line react-hooks/rules-of-hooks    useDebugState(scopeContainerRef.current)  }
  const ScopeContainerContext = getScopeContext(scope)  return createElement(    ScopeContainerContext.Provider,    {      value: scopeContainerRef.current,    },    children  )}