57 lines
2.0 KiB
TypeScript
57 lines
2.0 KiB
TypeScript
import * as LazyRef from "@typed/lazy-ref"
|
|
import { Effect, pipe, Stream } from "effect"
|
|
import * as React from "react"
|
|
import { ReffuseExtension, type ReffuseNamespace } from "reffuse"
|
|
import { SetStateAction } from "reffuse/types"
|
|
|
|
|
|
export const LazyRefExtension = ReffuseExtension.make(() => ({
|
|
useSubscribeLazyRefs<
|
|
const Refs extends readonly LazyRef.LazyRef<any>[],
|
|
R,
|
|
>(
|
|
this: ReffuseNamespace.ReffuseNamespace<R>,
|
|
...refs: Refs
|
|
): [...{ [K in keyof Refs]: Effect.Effect.Success<Refs[K]> }] {
|
|
const [reactStateValue, setReactStateValue] = React.useState(this.useMemo(
|
|
() => Effect.all(refs as readonly LazyRef.LazyRef<any>[]),
|
|
[],
|
|
{ doNotReExecuteOnRuntimeOrContextChange: true },
|
|
) as [...{ [K in keyof Refs]: Effect.Effect.Success<Refs[K]> }])
|
|
|
|
this.useFork(() => pipe(
|
|
refs.map(ref => Stream.changesWith(ref.changes, (x, y) => x === y)),
|
|
streams => Stream.zipLatestAll(...streams),
|
|
Stream.runForEach(v =>
|
|
Effect.sync(() => setReactStateValue(v as [...{ [K in keyof Refs]: Effect.Effect.Success<Refs[K]> }]))
|
|
),
|
|
), refs)
|
|
|
|
return reactStateValue
|
|
},
|
|
|
|
useLazyRefState<A, E, R>(
|
|
this: ReffuseNamespace.ReffuseNamespace<R>,
|
|
ref: LazyRef.LazyRef<A, E, R>,
|
|
): [A, React.Dispatch<React.SetStateAction<A>>] {
|
|
const [reactStateValue, setReactStateValue] = React.useState(this.useMemo(
|
|
() => ref,
|
|
[],
|
|
{ doNotReExecuteOnRuntimeOrContextChange: true },
|
|
))
|
|
|
|
this.useFork(() => Stream.runForEach(
|
|
Stream.changesWith(ref.changes, (x, y) => x === y),
|
|
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]
|
|
},
|
|
}))
|