30 lines
721 B
TypeScript
30 lines
721 B
TypeScript
import { React } from '@revenge-mod/react'
|
|
import { findInTree } from './trees'
|
|
import type { ReactNode } from 'react'
|
|
import type { SearchFilter } from './trees'
|
|
|
|
export function useIsFirstRender() {
|
|
let firstRender = false
|
|
|
|
React.useEffect(() => {
|
|
firstRender = true
|
|
}, [])
|
|
|
|
return firstRender
|
|
}
|
|
|
|
export function useReRender() {
|
|
const [, forceUpdate] = React.useReducer(x => ~x, 0)
|
|
return forceUpdate
|
|
}
|
|
|
|
const ReactFiberWalkable = new Set(['props', 'children', 'child', 'sibling'])
|
|
|
|
export function findInReactFiber(
|
|
tree: Extract<ReactNode, { props: unknown }>,
|
|
filter: SearchFilter,
|
|
): any {
|
|
return findInTree(tree, filter, {
|
|
walkable: ReactFiberWalkable,
|
|
})
|
|
}
|