useTarget.js 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  3. Object.defineProperty(exports, "__esModule", {
  4. value: true
  5. });
  6. exports.default = useTarget;
  7. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  8. var _useEvent = _interopRequireDefault(require("rc-util/lib/hooks/useEvent"));
  9. var _useLayoutEffect = _interopRequireDefault(require("rc-util/lib/hooks/useLayoutEffect"));
  10. var _react = require("react");
  11. var _util = require("../util");
  12. function isValidNumber(val) {
  13. return typeof val === 'number' && !Number.isNaN(val);
  14. }
  15. function useTarget(target, open, gap, scrollIntoViewOptions) {
  16. // ========================= Target =========================
  17. // We trade `undefined` as not get target by function yet.
  18. // `null` as empty target.
  19. var _useState = (0, _react.useState)(undefined),
  20. _useState2 = (0, _slicedToArray2.default)(_useState, 2),
  21. targetElement = _useState2[0],
  22. setTargetElement = _useState2[1];
  23. (0, _useLayoutEffect.default)(function () {
  24. var nextElement = typeof target === 'function' ? target() : target;
  25. setTargetElement(nextElement || null);
  26. });
  27. // ========================= Align ==========================
  28. var _useState3 = (0, _react.useState)(null),
  29. _useState4 = (0, _slicedToArray2.default)(_useState3, 2),
  30. posInfo = _useState4[0],
  31. setPosInfo = _useState4[1];
  32. var updatePos = (0, _useEvent.default)(function () {
  33. if (targetElement) {
  34. // Exist target element. We should scroll and get target position
  35. if (!(0, _util.isInViewPort)(targetElement) && open) {
  36. targetElement.scrollIntoView(scrollIntoViewOptions);
  37. }
  38. var _targetElement$getBou = targetElement.getBoundingClientRect(),
  39. left = _targetElement$getBou.left,
  40. top = _targetElement$getBou.top,
  41. width = _targetElement$getBou.width,
  42. height = _targetElement$getBou.height;
  43. var nextPosInfo = {
  44. left: left,
  45. top: top,
  46. width: width,
  47. height: height,
  48. radius: 0
  49. };
  50. setPosInfo(function (origin) {
  51. if (JSON.stringify(origin) !== JSON.stringify(nextPosInfo)) {
  52. return nextPosInfo;
  53. }
  54. return origin;
  55. });
  56. } else {
  57. // Not exist target which means we just show in center
  58. setPosInfo(null);
  59. }
  60. });
  61. var getGapOffset = function getGapOffset(index) {
  62. var _ref;
  63. return (_ref = Array.isArray(gap === null || gap === void 0 ? void 0 : gap.offset) ? gap === null || gap === void 0 ? void 0 : gap.offset[index] : gap === null || gap === void 0 ? void 0 : gap.offset) !== null && _ref !== void 0 ? _ref : 6;
  64. };
  65. (0, _useLayoutEffect.default)(function () {
  66. updatePos();
  67. // update when window resize
  68. window.addEventListener('resize', updatePos);
  69. return function () {
  70. window.removeEventListener('resize', updatePos);
  71. };
  72. }, [targetElement, open, updatePos]);
  73. // ======================== PosInfo =========================
  74. var mergedPosInfo = (0, _react.useMemo)(function () {
  75. if (!posInfo) {
  76. return posInfo;
  77. }
  78. var gapOffsetX = getGapOffset(0);
  79. var gapOffsetY = getGapOffset(1);
  80. var gapRadius = isValidNumber(gap === null || gap === void 0 ? void 0 : gap.radius) ? gap === null || gap === void 0 ? void 0 : gap.radius : 2;
  81. return {
  82. left: posInfo.left - gapOffsetX,
  83. top: posInfo.top - gapOffsetY,
  84. width: posInfo.width + gapOffsetX * 2,
  85. height: posInfo.height + gapOffsetY * 2,
  86. radius: gapRadius
  87. };
  88. }, [posInfo, gap]);
  89. return [mergedPosInfo, targetElement];
  90. }