patchOnRaf.ts

import { getHookEnv, HookEffects, runWithHooks } from '@typed/hooks'
import { patch, PatchEnv } from './Patch'
import { raf, RafEnv } from './raf'

export function* patchOnRaf<E, A, B, C>(
  fn: () => HookEffects<A, B>,
  initial: C,
): HookEffects<E & A & RafEnv & PatchEnv<C, B, E>, never> {
  const env = yield* getHookEnv()

  let previous = yield* patch(initial, yield* runWithHooks(fn(), env))

  while (true) {
    yield* raf()

    if (env.updated) {
      previous = yield* patch(previous, yield* runWithHooks(fn(), env))
    }
  }
}