一个帮助合并 state 的自定义 hook` 大家看看有啥问题·

import { useCallback, useState } from 'react';

const getType = (value) => {
  const type = Object.prototype.toString.call(value);
  if (type === '[object Object]') {
    return 'isObject'
  } else if (type === '[object Array]') {
    return 'isArray'
  }
  return typeof value;
}

const useMerge = (initialState) => {
  const [state, setState] = useState(initialState);
  const stateType = getType(initialState);

  const handleMerge = useCallback((handleObject = {}) => {

    const { id, type, value } = handleObject;
    const merge = (state) => {
      let mergeData = {};
      if (stateType === 'isObject') {
        mergeData = {
          ...state,
          ...value
        }
      }
      if (stateType === 'isArray') {
        if (type === 'remove') {
          mergeData = state.filter((item) => {
            return item.id !== id
          })
        }

        if (type === 'add') {
          mergeData = [...state, value];
        }

        if (type === 'update') {
          mergeData = state.map((item) => {
            if (item.id === id) {
              return {
                ...item,
                ...value
              }
            }
            return item;
          })
        }
      }
      return mergeData;
    }

    setState((prevState) => merge(prevState))
  }, [])

  return [state, handleMerge];
}

export default useMerge;