28 lines
1.0 KiB
TypeScript
28 lines
1.0 KiB
TypeScript
import * as LazyRef from "@typed/lazy-ref"
|
|
import { Effect, Stream } from "effect"
|
|
import * as React from "react"
|
|
import { ReffuseExtension, type ReffuseHelpers, SetStateAction } from "reffuse"
|
|
|
|
|
|
export const LazyRefExtension = ReffuseExtension.make(() => ({
|
|
useLazyRefState<A, E, R>(
|
|
this: ReffuseHelpers.ReffuseHelpers<R>,
|
|
ref: LazyRef.LazyRef<A, E, R>,
|
|
): [A, React.Dispatch<React.SetStateAction<A>>] {
|
|
const initialState = this.useMemo(() => ref, [], { doNotReExecuteOnRuntimeOrContextChange: true })
|
|
const [reactStateValue, setReactStateValue] = React.useState(initialState)
|
|
|
|
this.useFork(() => Stream.runForEach(ref.changes, v => Effect.sync(() =>
|
|
setReactStateValue(v)
|
|
)), [ref])
|
|
|
|
const setValue = this.useCallbackSync((setStateAction: React.SetStateAction<A>) =>
|
|
LazyRef.update(ref, prevState =>
|
|
SetStateAction.value(setStateAction, prevState)
|
|
),
|
|
[ref])
|
|
|
|
return [reactStateValue, setValue]
|
|
},
|
|
}))
|