useForm.js 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  3. var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = useForm;
  8. exports.toNamePathStr = toNamePathStr;
  9. var React = _interopRequireWildcard(require("react"));
  10. var _rcFieldForm = require("rc-field-form");
  11. var _findDOMNode = require("rc-util/lib/Dom/findDOMNode");
  12. var _scrollIntoViewIfNeeded = _interopRequireDefault(require("scroll-into-view-if-needed"));
  13. var _util = require("../util");
  14. var __rest = void 0 && (void 0).__rest || function (s, e) {
  15. var t = {};
  16. for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  17. if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
  18. if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  19. }
  20. return t;
  21. };
  22. function toNamePathStr(name) {
  23. const namePath = (0, _util.toArray)(name);
  24. return namePath.join('_');
  25. }
  26. function getFieldDOMNode(name, wrapForm) {
  27. const field = wrapForm.getFieldInstance(name);
  28. const fieldDom = (0, _findDOMNode.getDOM)(field);
  29. if (fieldDom) {
  30. return fieldDom;
  31. }
  32. const fieldId = (0, _util.getFieldId)((0, _util.toArray)(name), wrapForm.__INTERNAL__.name);
  33. if (fieldId) {
  34. return document.getElementById(fieldId);
  35. }
  36. }
  37. function useForm(form) {
  38. const [rcForm] = (0, _rcFieldForm.useForm)();
  39. const itemsRef = React.useRef({});
  40. const wrapForm = React.useMemo(() => form !== null && form !== void 0 ? form : Object.assign(Object.assign({}, rcForm), {
  41. __INTERNAL__: {
  42. itemRef: name => node => {
  43. const namePathStr = toNamePathStr(name);
  44. if (node) {
  45. itemsRef.current[namePathStr] = node;
  46. } else {
  47. delete itemsRef.current[namePathStr];
  48. }
  49. }
  50. },
  51. scrollToField: (name, options = {}) => {
  52. const {
  53. focus
  54. } = options,
  55. restOpt = __rest(options, ["focus"]);
  56. const node = getFieldDOMNode(name, wrapForm);
  57. if (node) {
  58. (0, _scrollIntoViewIfNeeded.default)(node, Object.assign({
  59. scrollMode: 'if-needed',
  60. block: 'nearest'
  61. }, restOpt));
  62. // Focus if scroll success
  63. if (focus) {
  64. wrapForm.focusField(name);
  65. }
  66. }
  67. },
  68. focusField: name => {
  69. var _a, _b;
  70. const itemRef = wrapForm.getFieldInstance(name);
  71. if (typeof (itemRef === null || itemRef === void 0 ? void 0 : itemRef.focus) === 'function') {
  72. itemRef.focus();
  73. } else {
  74. (_b = (_a = getFieldDOMNode(name, wrapForm)) === null || _a === void 0 ? void 0 : _a.focus) === null || _b === void 0 ? void 0 : _b.call(_a);
  75. }
  76. },
  77. getFieldInstance: name => {
  78. const namePathStr = toNamePathStr(name);
  79. return itemsRef.current[namePathStr];
  80. }
  81. }), [form, rcForm]);
  82. return [wrapForm];
  83. }