|
| 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