useWatch.js 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  3. Object.defineProperty(exports, "__esModule", {
  4. value: true
  5. });
  6. exports.default = useWatch;
  7. var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
  8. var _useLayoutEffect = _interopRequireDefault(require("rc-util/lib/hooks/useLayoutEffect"));
  9. var _util = require("../util");
  10. function useWatch(open, target, popup, onAlign, onScroll) {
  11. (0, _useLayoutEffect.default)(function () {
  12. if (open && target && popup) {
  13. var targetElement = target;
  14. var popupElement = popup;
  15. var targetScrollList = (0, _util.collectScroller)(targetElement);
  16. var popupScrollList = (0, _util.collectScroller)(popupElement);
  17. var win = (0, _util.getWin)(popupElement);
  18. var mergedList = new Set([win].concat((0, _toConsumableArray2.default)(targetScrollList), (0, _toConsumableArray2.default)(popupScrollList)));
  19. function notifyScroll() {
  20. onAlign();
  21. onScroll();
  22. }
  23. mergedList.forEach(function (scroller) {
  24. scroller.addEventListener('scroll', notifyScroll, {
  25. passive: true
  26. });
  27. });
  28. win.addEventListener('resize', notifyScroll, {
  29. passive: true
  30. });
  31. // First time always do align
  32. onAlign();
  33. return function () {
  34. mergedList.forEach(function (scroller) {
  35. scroller.removeEventListener('scroll', notifyScroll);
  36. win.removeEventListener('resize', notifyScroll);
  37. });
  38. };
  39. }
  40. }, [open, target, popup]);
  41. }