|
| 1 | +import { CreateRenderer, Renderer, RenderResult, RenderHook } from '../types' |
| 2 | +import { ResultContainer, RenderHookOptions } from '../types/internal' |
| 3 | + |
| 4 | +import asyncUtils from './asyncUtils' |
| 5 | +import { cleanup, addCleanup, removeCleanup } from './cleanup' |
| 6 | + |
| 7 | +function resultContainer<TValue>(): ResultContainer<TValue> { |
| 8 | +const results: Array<{ value?: TValue; error?: Error }> = [] |
| 9 | +const resolvers: Array<() => void> = [] |
| 10 | + |
| 11 | +const result: RenderResult<TValue> = { |
| 12 | +get all() { |
| 13 | +return results.map(({ value, error }) => error ?? value) |
| 14 | +}, |
| 15 | +get current() { |
| 16 | +const { value, error } = results[results.length - 1] |
| 17 | +if (error) { |
| 18 | +throw error |
| 19 | +} |
| 20 | +return value as TValue |
| 21 | +}, |
| 22 | +get error() { |
| 23 | +const { error } = results[results.length - 1] |
| 24 | +return error |
| 25 | +} |
| 26 | +} |
| 27 | + |
| 28 | +const updateResult = (value?: TValue, error?: Error) => { |
| 29 | +results.push({ value, error }) |
| 30 | +resolvers.splice(0, resolvers.length).forEach((resolve) => resolve()) |
| 31 | +} |
| 32 | + |
| 33 | +return { |
| 34 | +result, |
| 35 | +addResolver: (resolver: () => void) => { |
| 36 | +resolvers.push(resolver) |
| 37 | +}, |
| 38 | +setValue: (value: TValue) => updateResult(value), |
| 39 | +setError: (error: Error) => updateResult(undefined, error) |
| 40 | +} |
| 41 | +} |
| 42 | + |
| 43 | +const createRenderHook = <TProps, TResult, TOptions extends {}, TRenderer extends Renderer<TProps>>( |
| 44 | +createRenderer: CreateRenderer<TProps, TResult, TOptions, TRenderer> |
| 45 | +) => ( |
| 46 | +callback: (props: TProps) => TResult, |
| 47 | +options: RenderHookOptions<TProps, TOptions> = {} as RenderHookOptions<TProps, TOptions> |
| 48 | +): RenderHook<TProps, TResult, TRenderer> => { |
| 49 | +const { result, setValue, setError, addResolver } = resultContainer<TResult>() |
| 50 | +const renderProps = { callback, setValue, setError } |
| 51 | +let hookProps = options.initialProps |
| 52 | + |
| 53 | +const { render, rerender, unmount, act, ...renderUtils } = createRenderer(renderProps, options) |
| 54 | + |
| 55 | +render(hookProps) |
| 56 | + |
| 57 | +function rerenderHook(newProps = hookProps) { |
| 58 | +hookProps = newProps |
| 59 | +rerender(hookProps) |
| 60 | +} |
| 61 | + |
| 62 | +function unmountHook() { |
| 63 | +removeCleanup(unmountHook) |
| 64 | +unmount() |
| 65 | +} |
| 66 | + |
| 67 | +addCleanup(unmountHook) |
| 68 | + |
| 69 | +return { |
| 70 | +result, |
| 71 | +rerender: rerenderHook, |
| 72 | +unmount: unmountHook, |
| 73 | +...asyncUtils(act, addResolver), |
| 74 | +...renderUtils |
| 75 | +} |
| 76 | +} |
| 77 | + |
| 78 | +export { createRenderHook, cleanup, addCleanup, removeCleanup } |
0 commit comments