UNPKG

@bee-design/ui

Version:

Bee Design React UI Library.

64 lines (63 loc) 2.47 kB
var __read = (this && this.__read) || function (o, n) { var m = typeof Symbol === "function" && o[Symbol.iterator]; if (!m) return o; var i = m.call(o), r, ar = [], e; try { while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value); } catch (error) { e = { error: error }; } finally { try { if (r && !r.done && (m = i["return"])) m.call(i); } finally { if (e) throw e.error; } } return ar; }; import { useState, useContext, useEffect, useRef } from 'react'; import get from 'lodash/get'; import { isArray, isString } from '../../_util/is'; import { FormContext } from '../context'; import warn from '../../_util/warning'; var useWatch = function (field, form) { var formCtx = useContext(FormContext); var formInstance = form || formCtx.store; var _a = __read(useState(function () { var fieldValues = formInstance === null || formInstance === void 0 ? void 0 : formInstance.getFieldsValue([].concat(field)); if (isString(field)) { return get(fieldValues, field); } return fieldValues; }), 2), value = _a[0], setValue = _a[1]; // if field change, get the real value from fieldRef.current var fieldRef = useRef(field); fieldRef.current = field; var valueRef = useRef(JSON.stringify(value)); useEffect(function () { if (!formInstance) { warn(true, 'formInstance is not available'); return; } var registerWatcher = (formInstance === null || formInstance === void 0 ? void 0 : formInstance.getInnerMethods(true)).registerWatcher; var updateValue = function () { var field = fieldRef.current; var formValues = formInstance.getFieldsValue([].concat(field)); var newValue = formValues; if (!isArray(field)) { newValue = get(formValues, field); } var newValueString = JSON.stringify(newValue); if (valueRef.current !== newValueString) { setValue(newValue); valueRef.current = newValueString; } }; updateValue(); var cancelWatch = registerWatcher && registerWatcher(updateValue); return function () { cancelWatch && cancelWatch(); }; }, []); return value; }; export default useWatch;