useUnmount.js 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. var _typeof = require("@babel/runtime/helpers/typeof");
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = void 0;
  8. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  9. var React = _interopRequireWildcard(require("react"));
  10. var _useLayoutEffect = _interopRequireDefault(require("rc-util/lib/hooks/useLayoutEffect"));
  11. function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
  12. function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
  13. /**
  14. * Trigger only when component unmount
  15. */
  16. function useUnmount(triggerStart, triggerEnd) {
  17. var _React$useState = React.useState(false),
  18. _React$useState2 = (0, _slicedToArray2.default)(_React$useState, 2),
  19. firstMount = _React$useState2[0],
  20. setFirstMount = _React$useState2[1];
  21. (0, _useLayoutEffect.default)(function () {
  22. if (firstMount) {
  23. triggerStart();
  24. return function () {
  25. triggerEnd();
  26. };
  27. }
  28. }, [firstMount]);
  29. (0, _useLayoutEffect.default)(function () {
  30. setFirstMount(true);
  31. return function () {
  32. setFirstMount(false);
  33. };
  34. }, []);
  35. }
  36. var _default = exports.default = useUnmount;