@fruits-chain/react-native-xiaoshu/src/hooks/useStateUpdate.ts

🌈 React Native UI library

import isFunction from 'lodash/isFunction'
import { useState, useCallback } from 'react'

import { isObject } from '../helpers'

import useDestroyed from './useDestroyed'

type UpdateStateParam<T> = Partial<T> | ((s: T) => Partial<T>)

type UpdateState<T> = (p: UpdateStateParam<T>) => void

/**
 * useState 类似 this.setState 可以传入部分字段更新
 * @param state 状态
 */
const useStateUpdate = <T>(state: T | (() => T)): [T, UpdateState<T>] => {
  const [localState, setLocalState] = useState<T>(state)
  const getDestroyed = useDestroyed()
  const updateState = useCallback(
    (s: UpdateStateParam<T>) => {
      if (!getDestroyed()) {
        setLocalState(ls => {
          const value = isFunction(s) ? s(ls) : s

          if (isObject(ls)) {
            return {
              ...ls,
              ...value,
            }
          }

          return value as T
        })
      }
    },
    [getDestroyed],
  )

  return [localState, updateState]
}

export default useStateUpdate